6 import Data.Functor.Identity
7 import Control.Applicative
8 import Control.Monad => qualified join
9 import Control.Monad.State
10 import Control.Monad.Trans
11 import Control.Monad.Writer
17 :: In a b = In infix 0 a b
20 lit :: a -> v a | toString a
21 (+.) infixl 6 :: (v a) (v a) -> v a | + a
24 var :: (.(v .a) -> .(In .(v .a) (w b))) -> w b
27 arr :: *(arr a) -> *v *(arr a) | Array arr a & toString a
28 (++.) infixr 5 :: *(v *(arr a)) *(v *(arr a)) -> *v *(arr a) | Array arr a
29 (!.) infixl 8 :: .(v .(arr a)) .(v Int) -> .v a | Array arr a
30 (=.) infixl 8 :: (v .(arr a)) (v Int, v a) -> (v .(arr a)) | Array arr a
32 :: Printer a = P (WriterT [String] (StateT Int Identity) ())
33 runPrinter (P a) = concat $ snd $ evalState (runWriterT a) 0
34 instance uexpr Printer where
35 lit a = P $ tell [toString a]
36 +. (P l) (P r) = P $ tell ["("] >>| l >>| tell ["+"] >>| r >>| tell [")"]
38 instance uvar Printer Printer where
41 >>= \s->liftT (put (s+1))
42 >>| let (P i In P b) = def $ P $ tell $ varName s []
43 in tell ["let ":varName s [" = "]] >>| i >>| tell [" in\n"] >>| b
46 varName i c = ["v",toString i:c]
48 instance uarr Printer where
49 arr a = P $ tell ["{",join "," [toString a\\a<-:a],"}"]
50 ++. (P l) (P r) = P $ tell ["("] >>| l >>| tell ["+++"] >>| r >>| tell [")"]
51 !. (P a) (P i) = P $ a >>| tell [".["] >>| i >>| tell ["]"]
52 =. (P a) (P idx, P el) = P $ a >>| tell ["&"] >>| idx >>| tell ["="] >>| el
56 instance uexpr Eval where
58 +. (E l) (E r) = E $ l + r
60 instance uvar Eval Eval where
61 var def = let (i In b) = def i in b
63 instance uarr Eval where
65 ++. (E l) (E r) = E (appendArr l r)
66 !. (E a) (E i) = E (select a i)
67 =. (E a) (E idx, E el) = undef//E (update a idx el)
69 runBoth (Both l r) = (runPrinter l, runEval r)
71 //Start = both (var \v=arr {!1,2,3} In v ++. arr {!4,5,6})
72 //Start = runPrinter (var \v=arr {!1,2,3} In v ++. arr {!4,5,6})
75 // , runPrinter $ var \v=arr {!1,2,3} In v ++. v ?. lit 0 =. lit 42
77 // , runEval $ var \v=arr {!1,2,3} In v ++. v ?. lit 0 =. lit 42
79 , runPrinter $ var \v=arr {!1,2,3} In v ++. arr {!4,5,65}
81 , runPrinter $ var \v=arr {!1,2,3} In v ++. arr {!4,5,6}
83 , runEval $ var \v=arr {!1,2,3} In arr {!4,5,6} ++. v
85 , runPrinter $ var \v=arr {!1,2,3} In arr {!4,5,6} ++. v
88 //i :: v Int | uexpr v
91 //ai :: v {Int} | uexpr, uarr v
94 //aip :: v {Int} | uexpr, uarr v
95 //aip = arr {1,2,3} ++. arr {4,5,6}
97 //aii :: v Int | uexpr, uarr v
100 //somearray :: *{Int}
101 //somearray = {1,2,3}