mirror of
https://github.com/idris-lang/Idris2.git
synced 2024-12-17 08:11:45 +03:00
31 lines
854 B
Idris
31 lines
854 B
Idris
streamCount : Nat -> Stream Nat
|
|
streamCount n = n :: streamCount (S n)
|
|
|
|
badCount : Nat -> Stream Nat
|
|
badCount n = n :: map S (badCount n)
|
|
|
|
data SP : Type -> Type -> Type where
|
|
Get : (a -> SP a b) -> SP a b
|
|
Put : b -> Inf (SP a b) -> SP a b
|
|
|
|
copy : SP a a
|
|
copy = Get (\x => Put x copy)
|
|
|
|
process : SP a b -> Stream a -> Stream b
|
|
process (Get f) (x :: xs) = process (f x) xs
|
|
process (Put b sp) xs = b :: process sp xs
|
|
|
|
badProcess : SP a b -> Stream a -> Stream b
|
|
badProcess (Get f) (x :: xs) = badProcess (f x) xs
|
|
badProcess (Put b sp) xs = badProcess sp xs
|
|
|
|
doubleInt : SP Nat Integer
|
|
doubleInt = Get (\x => Put (the Integer (cast x))
|
|
(Put (the Integer (cast x) * 2) doubleInt))
|
|
|
|
countStream : Nat -> Stream Nat
|
|
countStream x = x :: countStream (x + 1)
|
|
|
|
main : IO ()
|
|
main = printLn (take 10 (process doubleInt (countStream 1)))
|