module Data.Array.Repa.Operators.Select
(select)
where
import Data.Array.Repa.Index
import Data.Array.Repa.Internals.Elt
import Data.Array.Repa.Internals.Base
import Data.Array.Repa.Internals.Select
import qualified Data.Vector.Unboxed as V
import System.IO.Unsafe
select :: Elt a
=> (Int -> Bool)
-> (Int -> a)
-> Int
-> Array DIM1 a
select match produce len
= unsafePerformIO
$ do (sh, vec) <- selectIO
return $ sh `seq` vec `seq`
Array sh [Region RangeAll (GenManifest vec)]
where
selectIO
= do vecs <- selectChunkedP match produce len
vecs' <- mapM V.unsafeFreeze vecs
let result = V.concat vecs'
return (Z :. V.length result, result)