module ADP.Fusion.Term.Strng.Point where
import Data.Proxy
import Data.Strict.Tuple
import Debug.Trace
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.Point
import ADP.Fusion.Term.Strng.Type
instance
( TmkCtx1 m ls (Strng v x) (PointL i)
) => MkStream m (ls :!: Strng v x) (PointL i) where
mkStream (ls :!: strng@(Strng _ minL maxL xs)) sv us is
= S.map (\(ss,ee,ii) -> ElmStrng ee ii ss)
. addTermStream1 strng sv us is
$ mkStream ls (termStaticVar strng sv is) us (termStreamIndex strng sv is)
instance
( TstCtx m ts s x0 i0 is (PointL I)
) => TermStream m (TermSymbol ts (Strng v x)) s (is:.PointL I) where
termStream (ts:|Strng f minL maxL v) (cs:.IStatic d) (us:.PointL u) (is:.PointL i)
= S.map (\(TState s ii ee) ->
let RiPlI k = getIndex (getIdx s) (Proxy :: PRI is (PointL I))
in TState s (ii:.:RiPlI i) (ee:.f k (ik) v))
. termStream ts cs us is
termStream (ts:|Strng f minL maxL v) (cs:.IVariable d) (us:.PointL u) (is:.PointL i)
= S.flatten mk step . termStream ts cs us is
where mk (tstate@(TState s ii ee)) =
let RiPlI k = getIndex (getIdx s) (Proxy :: PRI is (PointL I))
in return (tstate, ikdminL)
step (tstate@(TState s ii ee), z)
| z >= 0 && (lk <= maxL) = return $ S.Yield (TState s (ii:.:RiPlI l) (ee:.f k (lk+1) v)) (tstate, z1)
| otherwise = return $ S.Done
where RiPlI k = getIndex (getIdx s) (Proxy :: PRI is (PointL I))
l = i z d
instance
( TstCtx m ts s x0 i0 is (PointL O)
) => TermStream m (TermSymbol ts (Strng v x)) s (is:.PointL O) where
termStream (ts:|Strng f minL maxL v) (cs:.OStatic d) (us:.PointL u) (is:.PointL i)
= S.map (\(TState s ii ee) ->
let RiPlO k o = getIndex (getIdx s) (Proxy :: PRI is (PointL O))
in TState s (ii:.:RiPlO (id+1) o) (ee:.f k (ik) v))
. termStream ts cs us is
termStream (ts:|Strng f minL maxL v) (cs:.ORightOf d) (us:.PointL u) (is:.PointL i)
= S.flatten mk step . termStream ts cs us is
where mk (tstate@(TState s ii ee)) =
let RiPlO k _ = getIndex (getIdx s) (Proxy :: PRI is (PointL O))
in return (tstate, ikdminL)
step (tstate@(TState s ii ee), z)
| z >= 0 && (lk <= maxL) = return $ S.Yield (TState s (ii:.:RiPlO l o) (ee:.f k (lk+1) v)) (tstate, z1)
| otherwise = return $ S.Done
where RiPlO k o = getIndex (getIdx s) (Proxy :: PRI is (PointL O))
l = i z d
instance TermStaticVar (Strng v x) (PointL I) where
termStaticVar (Strng _ minL maxL _) (IStatic d) _ = IVariable $ d + maxL minL
termStaticVar _ (IVariable d) _ = IVariable d
termStreamIndex (Strng _ minL _ _) (IStatic d) (PointL j) = PointL $ j minL
instance TermStaticVar (Strng v x) (PointL O) where
termStaticVar (Strng _ minL maxL _) (OStatic d) _ = ORightOf $ d + maxL minL
termStaticVar _ (ORightOf d) _ = ORightOf 0
termStreamIndex _ _ j = j