on master help sign in
Implement takeArr.
5b979f5Paweł Nowak
Data/Syntax/Combinator.hs+10−1
@@ -26,6 +26,9 @@ module Data.Syntax.Combinator
26 sepBy,
26 sepBy,
27 sepBy1,
27 sepBy1,
28
28
29+ -- * Arrowized combinators.
30+ takeArr,
31+
29 -- * Vectors.
32 -- * Vectors.
30 vecNSepBy,
33 vecNSepBy,
31 ivecNSepBy,
34 ivecNSepBy,
@@ -48,10 +51,11 @@ import Control.Category.Structures
48import Control.Lens
51import Control.Lens
49import Control.Lens.SemiIso
52import Control.Lens.SemiIso
50import Control.SIArrow
53import Control.SIArrow
54+import qualified Data.MonoTraversable as Seq
51import Data.Syntax
55import Data.Syntax
52import qualified Data.Vector as V
56import qualified Data.Vector as V
53import qualified Data.Vector.Unboxed as VU
57import qualified Data.Vector.Unboxed as VU
54−import Prelude hiding (id, (.))
58+import Prelude hiding (id, (.), take)
55
59
56-- | One or zero occurences of @f@.
60-- | One or zero occurences of @f@.
57optional :: SIArrow cat => cat () a -> cat () (Maybe a)
61optional :: SIArrow cat => cat () a -> cat () (Maybe a)
@@ -89,6 +93,11 @@ sepBy v s = sepBy1 v s /+/ sipure _Empty
89sepBy1 :: SIArrow cat => cat () a -> cat () () -> cat () [a]
93sepBy1 :: SIArrow cat => cat () a -> cat () () -> cat () [a]
90sepBy1 v s = _Cons /$/ v /*/ (s */ sepBy1 v s /+/ sipure _Empty)
94sepBy1 v s = _Cons /$/ v /*/ (s */ sepBy1 v s /+/ sipure _Empty)
91
95
96+-- | A string of given length.
97+takeArr :: Syntax syn => syn Int (Seq syn)
98+takeArr = sibind $ iso (\n -> constant n #>> take n)
99+ (\s -> constant (Seq.olength s) #>> take (Seq.olength s))
100+
92-- | Constant size vector with separators.
101-- | Constant size vector with separators.
93--
102--
94-- @vecNSepBy n e sep@ describes a size @n@ vector with elements @e@ separated by @sep@.
103-- @vecNSepBy n e sep@ describes a size @n@ vector with elements @e@ separated by @sep@.