module Data.Repa.Flow.Auto.ZipWith
( zipWith3_i
, zipWith4_i
, zipWith5_i
, zipWith6_i
, zipWith7_i)
where
import Data.Repa.Flow.Auto.Base
import Data.Repa.Array.Auto
import Data.Repa.Array.Material.Auto (Name(..))
import qualified Data.Repa.Flow.Chunked as C hiding (next)
#include "repa-flow.h"
zipWith3_i
:: ( Flow a, Flow b, Flow c
, Build b, Build c, Build d)
=> (a -> b -> c -> d)
-> Sources a -> Sources b -> Sources c
-> IO (Sources d)
zipWith3_i f sa sb sc
= C.szipWith_ii A (\_ a (b, c) -> f a b c) sa
=<< C.szipWith_ii A (\_ b c -> (b, c)) sb sc
zipWith4_i
:: ( Flow a, Flow b, Flow c, Flow d
, Build a, Build b, Build c, Build d, Build e)
=> (a -> b -> c -> d -> e)
-> Sources a -> Sources b -> Sources c -> Sources d
-> IO (Sources e)
zipWith4_i f sa sb sc sd
= do sab <- C.szipWith_ii A (\_ a b -> (a, b)) sa sb
scd <- C.szipWith_ii A (\_ c d -> (c, d)) sc sd
result <- C.szipWith_ii A (\_ (a, b) (c, d) -> f a b c d) sab scd
return result
zipWith5_i
:: ( Flow a, Flow b, Flow c, Flow d, Flow e
, Build a, Build b, Build c, Build d, Build e
, Build f)
=> (a -> b -> c -> d -> e -> f)
-> Sources a -> Sources b -> Sources c -> Sources d -> Sources e
-> IO (Sources f)
zipWith5_i f sa sb sc sd se
= do sab <- C.szipWith_ii A (\_ a b -> (a, b)) sa sb
scd <- C.szipWith_ii A (\_ c d -> (c, d)) sc sd
result <- zipWith3_i (\(a, b) (c, d) e -> f a b c d e)
sab scd se
return result
zipWith6_i
:: ( Flow a, Flow b, Flow c, Flow d, Flow e, Flow f
, Build a, Build b, Build c, Build d, Build e, Build f
, Build g)
=> (a -> b -> c -> d -> e -> f -> g)
-> Sources a -> Sources b -> Sources c -> Sources d -> Sources e -> Sources f
-> IO (Sources g)
zipWith6_i ff sa sb sc sd se sf
= do sab <- C.szipWith_ii A (\_ a b -> (a, b)) sa sb
scd <- C.szipWith_ii A (\_ c d -> (c, d)) sc sd
result <- zipWith4_i (\(a, b) (c, d) e f -> ff a b c d e f)
sab scd se sf
return result
zipWith7_i
:: ( Flow a, Flow b, Flow c, Flow d, Flow e, Flow f, Flow g
, Build a, Build b, Build c, Build d, Build e, Build f, Build g
, Build h)
=> (a -> b -> c -> d -> e -> f -> g -> h)
-> Sources a -> Sources b -> Sources c -> Sources d -> Sources e -> Sources f -> Sources g
-> IO (Sources h)
zipWith7_i ff sa sb sc sd se sf sg
= do sab <- C.szipWith_ii A (\_ a b -> (a, b)) sa sb
scd <- C.szipWith_ii A (\_ c d -> (c, d)) sc sd
result <- zipWith5_i (\(a, b) (c, d) e f g -> ff a b c d e f g)
sab scd se sf sg
return result