module ADP.Fusion.PointL.Term.MultiChr where
import Data.Proxy
import Data.Strict.Tuple
import Debug.Trace
import GHC.Exts
import qualified Data.Vector.Fusion.Stream.Monadic as S
import qualified Data.Vector.Generic as VG
import Data.PrimitiveArray
import ADP.Fusion.Core
import ADP.Fusion.Core.Term.MultiChr
import ADP.Fusion.PointL.Core
type instance LeftPosTy (IStatic d) (MultiChr c v x) (PointL I) = IStatic d
type instance LeftPosTy (OStatic d) (MultiChr c v x) (PointL O) = OStatic (d + c)
instance
forall pos posLeft m ls c v x i
. ( TermStream m (Z:.pos) (TermSymbol M (MultiChr c v x)) (Elm (Term1 (Elm ls (PointL i))) (Z :. PointL i)) (Z:.PointL i)
, posLeft ~ LeftPosTy pos (MultiChr c v x) (PointL i)
, TermStaticVar pos (MultiChr c v x) (PointL i)
, MkStream m posLeft ls (PointL i)
)
⇒ MkStream m pos (ls :!: MultiChr c v x) (PointL i) where
mkStream pos (ls :!: MultiChr xs) grd us is
= S.map (\(ss,ee,ii) -> ElmMultiChr ee ii ss)
. addTermStream1 pos (MultiChr @v @x @c xs) us is
$ mkStream (Proxy ∷ Proxy posLeft) ls (termStaticCheck pos (MultiChr @v @x @c xs) us is grd) us (termStreamIndex pos (MultiChr @v @x @c xs) is)
{-# Inline mkStream #-}
instance
( TermStreamContext m ps ts s x0 i0 is (PointL I)
, KnownNat c
) => TermStream m (ps:.IStatic d) (TermSymbol ts (MultiChr c v x)) s (is:.PointL I) where
termStream Proxy (ts:|MultiChr xs) (us:..LtPointL u) (is:.PointL i)
= let !c = fromIntegral $ natVal (Proxy ∷ Proxy c) in
S.map (\(TState s ii ee) -> TState s (ii:.:RiPlI i) (ee:. VG.unsafeSlice (i-c) c xs))
. termStream (Proxy ∷ Proxy ps) ts us is
{-# Inline termStream #-}
instance
( TermStreamContext m ps ts s x0 i0 is (PointL O)
, KnownNat c
) => TermStream m (ps:.OStatic d) (TermSymbol ts (MultiChr c v x)) s (is:.PointL O) where
termStream Proxy (ts:|MultiChr xs) (us:..LtPointL u) (is:.PointL i)
= S.map (\(TState s ii ee) ->
let RiPlO k o = getIndex (getIdx s) (Proxy :: PRI is (PointL O))
c = fromIntegral $ natVal (Proxy ∷ Proxy c)
in TState s (ii:.: RiPlO (k+c) o) (ee:.VG.unsafeSlice k c xs))
. termStream (Proxy ∷ Proxy ps) ts us is
{-# Inline termStream #-}
instance (KnownNat c) ⇒ TermStaticVar (IStatic d) (MultiChr c v x) (PointL I) where
termStreamIndex Proxy (MultiChr x) (PointL j) = PointL $ j-(fromIntegral $ natVal (Proxy ∷ Proxy c))
termStaticCheck Proxy (MultiChr x) _ (PointL j) grd = grd
{-# Inline [0] termStreamIndex #-}
{-# Inline [0] termStaticCheck #-}
instance TermStaticVar (OStatic d) (MultiChr c v x) (PointL O) where
termStreamIndex Proxy (MultiChr x) (PointL j) = PointL $ j
termStaticCheck Proxy (MultiChr x) _ (PointL j) grd = grd
{-# Inline [0] termStreamIndex #-}
{-# Inline [0] termStaticCheck #-}