src/Gargantext/Core/Methods/Distances.hs

   1 {-|
   2 Module      : Gargantext.Graph.Distances
   3 Description : Distance management tools
   4 Copyright   : (c) CNRS, 2017-Present
   5 License     : AGPL + CECILL v3
   6 Maintainer  : team@gargantext.org
   7 Stability   : experimental
   8 Portability : POSIX
   9
  10 -}
  11
  12 {-# LANGUAGE Strict            #-}
  13
  14 module Gargantext.Core.Methods.Distances
  15   where
  16
  17 import Data.Aeson
  18 import Data.Array.Accelerate (Matrix)
  19 import Data.Swagger
  20 import GHC.Generics (Generic)
  21 import Gargantext.Core.Methods.Distances.Accelerate.Conditional (measureConditional)
  22 import Gargantext.Core.Methods.Distances.Accelerate.Distributional (logDistributional)
  23 import Gargantext.Prelude (Ord, Eq, Int, Double)
  24 import Gargantext.Prelude (Show)
  25 import Prelude (Enum, Bounded, minBound, maxBound)
  26 import Test.QuickCheck (elements)
  27 import Test.QuickCheck.Arbitrary
  28
  29 ------------------------------------------------------------------------
  30 data Distance = Conditional | Distributional
  31   deriving (Show)
  32
  33 measure :: Distance -> Matrix Int -> Matrix Double
  34 measure Conditional    = measureConditional
  35 measure Distributional = logDistributional
  36
  37 ------------------------------------------------------------------------
  38 withMetric :: GraphMetric -> Distance
  39 withMetric Order1 = Conditional
  40 withMetric Order2 = Distributional
  41
  42 ------------------------------------------------------------------------
  43 data GraphMetric = Order1 | Order2
  44     deriving (Generic, Eq, Ord, Enum, Bounded, Show)
  45
  46 instance FromJSON  GraphMetric
  47 instance ToJSON    GraphMetric
  48 instance ToSchema  GraphMetric
  49 instance Arbitrary GraphMetric where
  50   arbitrary = elements [ minBound .. maxBound ]
  51
  52 ------------------------------------------------------------------------