2 Module : Gargantext.Viz.Phylo.Tools
3 Description : Phylomemy Tools to build/manage it
4 Copyright : (c) CNRS, 2017-Present
5 License : AGPL + CECILL v3
6 Maintainer : team@gargantext.org
7 Stability : experimental
13 {-# LANGUAGE NoImplicitPrelude #-}
14 {-# LANGUAGE FlexibleContexts #-}
15 {-# LANGUAGE OverloadedStrings #-}
17 module Gargantext.Viz.Phylo.View.Metrics
20 import Control.Lens hiding (makeLenses, both, Level)
22 import Data.List (notElem,last,head,union,concat,null,nub,(++),init,tail,elemIndex,groupBy,(!!),sortOn,sort,(\\))
23 import Data.Map (Map,elems,adjust,unionWith,intersectionWith,fromList,mapKeys,insert)
24 import Data.Maybe (isNothing)
26 import Data.Text (Text,unwords)
27 import Data.Tuple (fst, snd)
28 import Data.Vector (Vector)
30 import Gargantext.Prelude hiding (head)
31 import Gargantext.Viz.Phylo
32 import Gargantext.Viz.Phylo.Tools
34 import qualified Data.List as List
35 import qualified Data.Map as Map
36 import qualified Data.Set as Set
37 import qualified Data.Vector as Vector
40 -- | To add a new meta Metric to a PhyloBranch
41 addBranchMeta :: PhyloBranchId -> Text -> Double -> PhyloView -> PhyloView
42 addBranchMeta id lbl val v = over (phylo_viewBranches
44 (\b -> if getBranchId b == id
45 then b & phylo_branchMeta %~ insert lbl val
49 -- | To get the age (in year) of all the branches of a PhyloView
50 branchAge :: PhyloView -> PhyloView
51 branchAge v = foldl (\v' b -> let bId = (fst . head) b
52 prds = sortOn fst $ map snd b
53 in addBranchMeta bId "age" ((abs . fromIntegral)
54 $ ((snd . last) prds) - ((fst . head) prds)) v') v
55 $ groupBy ((==) `on` fst)
57 $ map (\n -> (getNodeBranchId n, (fst . fst) $ getNodeId n))
58 $ getNodesInBranches v
61 -- | To process a list of Metrics to a PhyloView
62 processMetrics :: [Metric] -> Phylo -> PhyloView -> PhyloView
63 processMetrics ms p v = foldl (\v' m -> case m of
64 BranchAge -> branchAge v'
65 _ -> panic "[ERR][Viz.Phylo.Example.processMetrics] metric not found") v ms