src/Gargantext/Viz/Graph/Utils.hs

   1 {-|
   2 Module      : Gargantext.Viz.Graph.Utils
   3 Description :
   4 Copyright   : (c) CNRS, 2017-Present
   5 License     : AGPL + CECILL v3
   6 Maintainer  : team@gargantext.org
   7 Stability   : experimental
   8 Portability : POSIX
   9
  10 These functions are used for Vector.Matrix only.
  11
  12 -}
  13
  14 {-# LANGUAGE BangPatterns      #-}
  15 {-# LANGUAGE NoImplicitPrelude #-}
  16 {-# LANGUAGE FlexibleContexts  #-}
  17 {-# LANGUAGE Strict            #-}
  18
  19 module Gargantext.Viz.Graph.Utils
  20   where
  21
  22 import Data.Matrix hiding (identity)
  23
  24 import Data.Vector (Vector)
  25 import qualified Data.Vector as V
  26
  27 import qualified Data.List   as L
  28 import Gargantext.Prelude
  29
  30 ------------------------------------------------------------------------
  31 -- | Some utils to build the matrix from cooccurrence results
  32
  33 -- | For tests only, to be removed
  34 -- m1 :: Matrix Double
  35 -- m1 = fromList 300 300 [1..]
  36 ------------------------------------------------------------------------
  37 ------------------------------------------------------------------------
  38 data Axis = Col | Row
  39 ------------------------------------------------------------------------
  40 -- | Matrix functions
  41 type AxisId = Int
  42
  43 -- Data.Vector.Additions
  44 dropAt :: Int -> Vector a -> Vector a
  45 dropAt n v = debut <> (V.tail fin)
  46   where
  47     debut = V.take n v
  48     fin   = V.drop n v
  49
  50 total :: Num a => Matrix a -> a
  51 total m = V.sum $ V.map (\c -> V.sum (getCol c m)) (V.enumFromTo 1 (nOf Col m))
  52
  53 nOf :: Axis -> Matrix a -> Int
  54 nOf Row = nrows
  55 nOf Col = ncols
  56
  57 axis :: Axis -> AxisId -> Matrix a -> Vector a
  58 axis Col = getCol
  59 axis Row = getRow
  60
  61
  62 toListsWithIndex :: Matrix a ->  [((Int, Int), a)]
  63 toListsWithIndex m = concat' $ zip [1..] $ map (\c -> zip [1..] c) $ toLists m
  64   where
  65     concat' :: [(Int, [(Int, a)])] -> [((Int, Int), a)]
  66     concat' xs = L.concat $ map (\(x, ys) -> map (\(y, a) -> ((x,y), a)) ys ) xs
  67
  68