Rattus-0.2: test/Rewrite.hs

{-# LANGUAGE TypeOperators #-}

module Main (module Main) where

import Rattus
import Rattus.Stream
import Prelude hiding ((<*>), map,zip,const)

{-# ANN module Rattus #-}

twice :: Str Int -> Str Int
twice = map (box (+1)) . map (box (+1))

scanAndMap :: Str Int -> Str Int
scanAndMap xs = map (box (+1)) (scan (box (+)) 0 xs)

sums :: Str Int -> Str Int
sums xs = scan (box (+)) 0 xs

twiceScan :: Str Int -> Str Int
twiceScan xs = scan (box (+)) 0  (scan (box (+)) 0 xs)

twiceScanMap :: Str Int -> Str Int
twiceScanMap xs = scan (box (+)) 0  (scanMap (box (+)) (box (+1)) 0 xs)

zipMap :: Str Int -> Str Int -> Str Int
zipMap xs ys = map (box (\ (x:*y) -> x + y)) (zip xs ys)

constMap :: Str Int
constMap = map (box (+1)) (const 5)



apply :: O (Int -> Int -> Int) -> O Int -> O Int -> O Int
apply f x y = f <*> x <*> y


apply' :: O (Int -> Int -> Int) -> Int -> O Int -> O Int
apply' f x y = f <** x <*> y


{-# ANN main NotRattus #-}
main = putStrLn "This is just to test the rewrite rules"