module Data.CRF.Chain1.Constrained.Dataset.Internal
( Ob (..)
, Lb (..)
, X (..)
, mkX
, unX
, unR
, Xs
, Y (..)
, mkY
, unY
, Ys
, AVec (unAVec)
, fromList
, fromSet
) where
import Control.Applicative ((<$>), (<*>))
import Data.Vector.Generic.Base
import Data.Vector.Generic.Mutable
import Data.Binary (Binary, get, put, putWord8, getWord8)
import Data.Vector.Binary ()
import Data.Ix (Ix)
import qualified Data.Set as S
import qualified Data.Vector as V
import qualified Data.Vector.Unboxed as U
newtype Ob = Ob { unOb :: Int }
deriving ( Show, Read, Eq, Ord, Binary
, Vector U.Vector, MVector U.MVector, U.Unbox )
newtype Lb = Lb { unLb :: Int }
deriving ( Show, Read, Eq, Ord, Binary
, Vector U.Vector, MVector U.MVector, U.Unbox
, Num, Ix )
newtype AVec a = AVec { unAVec :: U.Vector a }
deriving (Show, Read, Eq, Ord, Binary)
fromList :: (Ord a, U.Unbox a) => [a] -> AVec a
fromList = fromSet . S.fromList
fromSet :: (Ord a, U.Unbox a) => S.Set a -> AVec a
fromSet = AVec . U.fromList . S.toList
data X
= X { _unX :: AVec Ob }
| R { _unX :: AVec Ob
, _unR :: AVec Lb }
deriving (Show, Read, Eq, Ord)
instance Binary X where
put X{..} = putWord8 0 >> put _unX
put R{..} = putWord8 1 >> put _unX >> put _unR
get = getWord8 >>= \i -> case i of
0 -> X <$> get
_ -> R <$> get <*> get
mkX :: [Ob] -> [Lb] -> X
mkX x [] = X (fromList x)
mkX x r = R (fromList x) (fromList r)
unX :: X -> [Ob]
unX = U.toList . unAVec . _unX
unR :: AVec Lb -> X -> [Lb]
unR r0 X{..} = U.toList . unAVec $ r0
unR _ R{..} = U.toList . unAVec $ _unR
type Xs = V.Vector X
newtype Y = Y { _unY :: AVec (Lb, Double) }
deriving (Show, Read, Eq, Ord, Binary)
mkY :: [(Lb, Double)] -> Y
mkY = Y . fromList
unY :: Y -> [(Lb, Double)]
unY = U.toList . unAVec . _unY
type Ys = V.Vector Y