@@ -1,37 +1,43 @@
1+{-# LANGUAGE PatternGuards #-}
1module IRTS.CodegenJava (codegenJava) where
2module IRTS.CodegenJava (codegenJava) where
3import Core.TT
4import Core.TT
4import IRTS.BCImp
5import IRTS.BCImp
5import IRTS.CodegenCommon
6import IRTS.CodegenCommon
7+import IRTS.Java.ASTBuilding
8+import IRTS.Java.JTypes
9+import IRTS.Java.Mangling
6import IRTS.Lang
10import IRTS.Lang
7import IRTS.Simplified
11import IRTS.Simplified
8import Paths_idris
12import Paths_idris
9import Util.System
13import Util.System
11−import Control.Applicative
15+import Control.Applicative hiding (Const)
12import Control.Arrow
16import Control.Arrow
13import Control.Monad
17import Control.Monad
14−import qualified Control.Monad.Trans as T
18+import qualified Control.Monad.Trans as T
15import Control.Monad.Trans.State
19import Control.Monad.Trans.State
17−import Data.Maybe (fromJust)
18−import Data.List (isPrefixOf, isSuffixOf, intercalate, foldl')
19import Data.Int
20import Data.Int
21+import Data.List (foldl', intercalate, isPrefixOf,
23+import Data.Maybe (fromJust)
20−import qualified Data.Text as T
24+import qualified Data.Text as T
21−import qualified Data.Text.IO as TIO
25+import qualified Data.Text.IO as TIO
26+import qualified Data.Vector.Unboxed as V
22import Language.Java.Parser
27import Language.Java.Parser
23import Language.Java.Pretty
28import Language.Java.Pretty
24−import Language.Java.Syntax hiding (Name)
29+import Language.Java.Syntax hiding (Name)
25−import qualified Language.Java.Syntax as J
30+import qualified Language.Java.Syntax as J
26import System.Directory
31import System.Directory
27import System.Exit
32import System.Exit
28import System.FilePath
33import System.FilePath
29import System.IO
34import System.IO
30import System.Process
35import System.Process
32−data CodeGenerationEnv = CodeGenerationEnv { globalVariablePositions :: [(Name, Integer)] }
37+data CodeGenerationEnv = CodeGenerationEnv { globalVariablePositions :: [(Name, Int)] }
34type CodeGeneration = StateT (CodeGenerationEnv) (Either String)
39type CodeGeneration = StateT (CodeGenerationEnv) (Either String)
40+type BlockResultTransformation = [BlockStmt] -> Exp -> CodeGeneration [BlockStmt]
36codegenJava :: [(Name, SExp)] -> -- initialization of globals
42codegenJava :: [(Name, SExp)] -> -- initialization of globals
37 [(Name, SDecl)] ->
43 [(Name, SDecl)] ->
@@ -44,11 +50,10 @@ codegenJava globalInit defs out hdrs libs exec = do
44 withTempdir (takeBaseName out) $ \ tmpDir -> do
50 withTempdir (takeBaseName out) $ \ tmpDir -> do
45 let srcdir = tmpDir </> "src" </> "main" </> "java"
51 let srcdir = tmpDir </> "src" </> "main" </> "java"
46 createDirectoryIfMissing True srcdir
52 createDirectoryIfMissing True srcdir
47− let (Ident clsName) =
48− either error id (evalStateT (mkClassName out) (mkCodeGenEnv globalInit))
53+ let (Ident clsName) = either error id (mkClassName out)
49 let outjava = srcdir </> clsName <.> "java"
54 let outjava = srcdir </> clsName <.> "java"
50 let jout = either error
55 let jout = either error
51− (flatIndent . prettyPrint)
56+ (prettyPrint)-- flatIndent . prettyPrint)
52 (evalStateT (mkCompilationUnit globalInit defs hdrs out) (mkCodeGenEnv globalInit))
57 (evalStateT (mkCompilationUnit globalInit defs hdrs out) (mkCodeGenEnv globalInit))
53 writeFile outjava jout
58 writeFile outjava jout
54 if (exec == Raw)
59 if (exec == Raw)
@@ -58,8 +63,8 @@ codegenJava globalInit defs out hdrs libs exec = do
58 execPomTemplate <- TIO.readFile execPom
63 execPomTemplate <- TIO.readFile execPom
59 let execPom = T.replace (T.pack "$MAIN-CLASS$")
64 let execPom = T.replace (T.pack "$MAIN-CLASS$")
60 (T.pack clsName)
65 (T.pack clsName)
61− (T.replace (T.pack "$ARTIFACT-NAME$")
66+ (T.replace (T.pack "$ARTIFACT-NAME$")
62− (T.pack $ takeBaseName out)
67+ (T.pack $ takeBaseName out)
63 (T.replace (T.pack "$DEPENDENCIES$")
68 (T.replace (T.pack "$DEPENDENCIES$")
64 (mkPomDependencies libs)
69 (mkPomDependencies libs)
65 execPomTemplate
70 execPomTemplate
@@ -72,11 +77,11 @@ codegenJava globalInit defs out hdrs libs exec = do
72 when (exit /= ExitSuccess) $ error ("FAILURE: " ++ mvnCmd ++ " compile\n" ++ err ++ mvout)
77 when (exit /= ExitSuccess) $ error ("FAILURE: " ++ mvnCmd ++ " compile\n" ++ err ++ mvout)
73 if (exec == Object)
78 if (exec == Object)
75− classFiles <-
80+ classFiles <-
76 map (\ clsFile -> tmpDir </> "target" </> "classes" </> clsFile)
81 map (\ clsFile -> tmpDir </> "target" </> "classes" </> clsFile)
77− . filter ((".class" ==) . takeExtension)
82+ . filter ((".class" ==) . takeExtension)
78 <$> getDirectoryContents (tmpDir </> "target" </> "classes")
83 <$> getDirectoryContents (tmpDir </> "target" </> "classes")
79− mapM_ (\ clsFile -> copyFile clsFile (takeDirectory out </> takeFileName clsFile))
84+ mapM_ (\ clsFile -> copyFile clsFile (takeDirectory out </> takeFileName clsFile))
80 classFiles
85 classFiles
82 (exit, mvout, err) <- readProcessWithExitCode mvnCmd (args ++ ["package"]) ""
87 (exit, mvout, err) <- readProcessWithExitCode mvnCmd (args ++ ["package"]) ""
@@ -94,8 +99,11 @@ codegenJava globalInit defs out hdrs libs exec = do
94 readProcess mvnCmd (args ++ ["clean"]) ""
99 readProcess mvnCmd (args ++ ["clean"]) ""
95 removeFile (tmpDir </> "pom.xml")
100 removeFile (tmpDir </> "pom.xml")
102+-----------------------------------------------------------------------
103+-- Jar and Pom infrastructure
97jarHeader :: String
105jarHeader :: String
98−jarHeader =
106+jarHeader =
99 "#!/bin/sh\n"
107 "#!/bin/sh\n"
100 ++ "MYSELF=`which \"$0\" 2>/dev/null`\n"
108 ++ "MYSELF=`which \"$0\" 2>/dev/null`\n"
101 ++ "[ $? -gt 0 -a -f \"$0\" ] && MYSELF=\"./$0\"\n"
109 ++ "[ $? -gt 0 -a -f \"$0\" ] && MYSELF=\"./$0\"\n"
@@ -110,7 +118,7 @@ mkPomDependencies :: [String] -> T.Text
110mkPomDependencies deps =
118mkPomDependencies deps =
111 T.concat $ map (T.concat . map (T.append (T.pack " ")) . mkDependency . T.pack) deps
119 T.concat $ map (T.concat . map (T.append (T.pack " ")) . mkDependency . T.pack) deps
113− mkDependency s =
121+ mkDependency s =
114 case T.splitOn (T.pack ":") s of
122 case T.splitOn (T.pack ":") s of
115 [g, a, v] ->
123 [g, a, v] ->
116 [ T.pack $ "<dependency>\n"
124 [ T.pack $ "<dependency>\n"
@@ -124,136 +132,127 @@ mkPomDependencies deps =
124 mkArtifactId a = T.append (T.pack $ "<artifactId>") (T.append a $ T.pack "</artifactId>\n")
132 mkArtifactId a = T.append (T.pack $ "<artifactId>") (T.append a $ T.pack "</artifactId>\n")
125 mkVersion v = T.append (T.pack $ "<version>") (T.append v $ T.pack "</version>\n")
133 mkVersion v = T.append (T.pack $ "<version>") (T.append v $ T.pack "</version>\n")
135+-----------------------------------------------------------------------
136+-- Code generation environment
127mkCodeGenEnv :: [(Name, SExp)] -> CodeGenerationEnv
138mkCodeGenEnv :: [(Name, SExp)] -> CodeGenerationEnv
128−mkCodeGenEnv globalInit =
139+mkCodeGenEnv globalInit =
129 CodeGenerationEnv $ zipWith (\ (name, _) pos -> (name, pos)) globalInit [0..]
140 CodeGenerationEnv $ zipWith (\ (name, _) pos -> (name, pos)) globalInit [0..]
142+flatIndent :: String -> String
143+flatIndent (' ' : ' ' : xs) = flatIndent xs
144+flatIndent (x:xs) = x:flatIndent xs
147+-----------------------------------------------------------------------
131mkCompilationUnit :: [(Name, SExp)] -> [(Name, SDecl)] -> [String] -> FilePath -> CodeGeneration CompilationUnit
150mkCompilationUnit :: [(Name, SExp)] -> [(Name, SDecl)] -> [String] -> FilePath -> CodeGeneration CompilationUnit
132−mkCompilationUnit globalInit defs hdrs out =
151+mkCompilationUnit globalInit defs hdrs out = do
152+ clsName <- mkClassName out
133 CompilationUnit Nothing ( [ ImportDecl False idrisRts True
153 CompilationUnit Nothing ( [ ImportDecl False idrisRts True
134 , ImportDecl True idrisForeign True
154 , ImportDecl True idrisForeign True
155+ , ImportDecl False idrisForeignWrapper False
156+ , ImportDecl False idrisPrimFn False
135 , ImportDecl False bigInteger False
157 , ImportDecl False bigInteger False
136 , ImportDecl False stringBuffer False
158 , ImportDecl False stringBuffer False
137 , ImportDecl False runtimeException False
159 , ImportDecl False runtimeException False
138 , ImportDecl False scanner False
160 , ImportDecl False scanner False
139 , ImportDecl False arrays False
161 , ImportDecl False arrays False
162+ , ImportDecl False callable False
140 ] ++ otherHdrs
163 ] ++ otherHdrs
142− <$> mkTypeDecl globalInit defs out
165+ <$> mkTypeDecl clsName globalInit defs
144 idrisRts = J.Name $ map Ident ["org", "idris", "rts"]
167 idrisRts = J.Name $ map Ident ["org", "idris", "rts"]
145 idrisForeign = J.Name $ map Ident ["org", "idris", "rts", "ForeignPrimitives"]
168 idrisForeign = J.Name $ map Ident ["org", "idris", "rts", "ForeignPrimitives"]
169+ idrisForeignWrapper = J.Name $ map Ident ["org", "idris", "rts", "ForeignWrapper"]
170+ idrisPrimFn = J.Name $ map Ident ["org", "idris", "rts", "PrimFn"]
146 bigInteger = J.Name $ map Ident ["java", "math", "BigInteger"]
171 bigInteger = J.Name $ map Ident ["java", "math", "BigInteger"]
147 stringBuffer = J.Name $ map Ident ["java", "lang", "StringBuffer"]
172 stringBuffer = J.Name $ map Ident ["java", "lang", "StringBuffer"]
148 runtimeException = J.Name $ map Ident ["java", "lang", "RuntimeException"]
173 runtimeException = J.Name $ map Ident ["java", "lang", "RuntimeException"]
149 scanner = J.Name $ map Ident ["java", "util", "Scanner"]
174 scanner = J.Name $ map Ident ["java", "util", "Scanner"]
150 arrays = J.Name $ map Ident ["java", "util", "Arrays"]
175 arrays = J.Name $ map Ident ["java", "util", "Arrays"]
176+ callable = J.Name $ map Ident ["java", "util", "concurrent", "Callable"]
151 otherHdrs = map ( (\ name -> ImportDecl False name False)
177 otherHdrs = map ( (\ name -> ImportDecl False name False)
152− . J.Name
178+ . J.Name
153− . map (Ident . T.unpack)
179+ . map (Ident . T.unpack)
154− . T.splitOn (T.pack ".")
180+ . T.splitOn (T.pack ".")
155− . T.pack)
181+ . T.pack)
156 $ filter (not . isSuffixOf ".h") hdrs
182 $ filter (not . isSuffixOf ".h") hdrs
158−flatIndent :: String -> String
184+-----------------------------------------------------------------------
159−flatIndent (' ' : ' ' : xs) = flatIndent xs
185+-- Main class
160−flatIndent (x:xs) = x:flatIndent xs
186+
163−prefixCallNamespaces :: Ident -> SExp -> SExp
164−prefixCallNamespaces (Ident name) (SApp tail (NS n ns) args) = SApp tail (NS n (name:ns)) args
165−prefixCallNamespaces name (SLet var e1 e2) = SLet var (prefixCallNamespaces name e1) (prefixCallNamespaces name e2)
166−prefixCallNamespaces name (SUpdate var e) = SUpdate var (prefixCallNamespaces name e)
167−prefixCallNamespaces name (SCase var alts) = SCase var (map (prefixCallNamespacesCase name) alts)
168−prefixCallNamespaces name (SChkCase var alts) = SChkCase var (map (prefixCallNamespacesCase name) alts)
169−prefixCallNamespaces _ exp = exp
171−prefixCallNamespacesCase :: Ident -> SAlt -> SAlt
172−prefixCallNamespacesCase name (SConCase x y n ns e) = SConCase x y n ns (prefixCallNamespaces name e)
173−prefixCallNamespacesCase name (SConstCase c e) = SConstCase c (prefixCallNamespaces name e)
174−prefixCallNamespacesCase name (SDefaultCase e) = SDefaultCase (prefixCallNamespaces name e)
176−prefixCallNamespacesDecl :: Ident -> SDecl -> SDecl
177−prefixCallNamespacesDecl name (SFun fname args i e) = SFun fname args i (prefixCallNamespaces name e)
179−mkTypeDecl :: [(Name, SExp)] -> [(Name, SDecl)] -> FilePath -> CodeGeneration [TypeDecl]
187+mkTypeDecl :: Ident -> [(Name, SExp)] -> [(Name, SDecl)] -> CodeGeneration [TypeDecl]
180−mkTypeDecl globalInit defs out =
188+mkTypeDecl name globalInit defs =
181− (\ name body -> [ClassTypeDecl $ ClassDecl [ Public
189+ (\ body -> [ClassTypeDecl $ ClassDecl [ Public
182− , Annotation $ SingleElementAnnotation
190+ , Annotation $ SingleElementAnnotation
183− (J.Name [Ident "SuppressWarnings"])
191+ (jName "SuppressWarnings")
184− (EVVal . InitExp . Lit $ String "unchecked")
192+ (EVVal . InitExp $ jString "unchecked")
188− Nothing
196+ Nothing
192− <*> ( mkClassName out
193− >>= (\ ident -> mkClassBody globalInit (map (second (prefixCallNamespacesDecl ident)) defs))
199+ <$> mkClassBody globalInit (map (second (prefixCallNamespaces name)) defs)
196−mkClassName :: FilePath -> CodeGeneration Ident
198− T.lift $ left (\ err -> "Parser error in \"" ++ path ++ "\": " ++ (show err))
199− (parser ident . takeBaseName $ takeFileName path)
201mkClassBody :: [(Name, SExp)] -> [(Name, SDecl)] -> CodeGeneration ClassBody
201mkClassBody :: [(Name, SExp)] -> [(Name, SDecl)] -> CodeGeneration ClassBody
202−mkClassBody globalInit defs =
202+mkClassBody globalInit defs =
203 (\ globals defs -> ClassBody . (globals++) . addMainMethod . mergeInnerClasses $ defs)
203 (\ globals defs -> ClassBody . (globals++) . addMainMethod . mergeInnerClasses $ defs)
204 <$> mkGlobalContext globalInit
204 <$> mkGlobalContext globalInit
205 <*> mapM mkDecl defs
205 <*> mapM mkDecl defs
207mkGlobalContext :: [(Name, SExp)] -> CodeGeneration [Decl]
207mkGlobalContext :: [(Name, SExp)] -> CodeGeneration [Decl]
208mkGlobalContext [] = return []
208mkGlobalContext [] = return []
209−mkGlobalContext initExps =
209+mkGlobalContext initExps =
210− (\ exps -> [ MemberDecl $ FieldDecl [Private, Static, Final]
210+ (\ exps -> [ MemberDecl $ FieldDecl [Private, Static, Final]
211− objectArrayType
211+ (array objectType)
212− [VarDecl (VarId $ Ident "globalContext")
212+ [ VarDecl (VarId $ globalContextID). Just . InitExp
213− (Just . InitArray $ ArrayInit exps)
213+ $ ArrayCreate objectType [jInt $ length initExps] 0
215+ , InitDecl True (Block $ concat exps)
220+ mkUpdate (Glob name) exp (\ b e -> return $ addToBlock b e)
217− <$> mapM (\ (_, exp) -> InitExp <$> mkExp exp) initExps
222+ initExps
221addMainMethod :: [Decl] -> [Decl]
224addMainMethod :: [Decl] -> [Decl]
222addMainMethod decls
225addMainMethod decls
223 | findMain decls = mkMainMethod : decls
226 | findMain decls = mkMainMethod : decls
224 | otherwise = decls
227 | otherwise = decls
226− findMain ((MemberDecl (MemberClassDecl (ClassDecl _ (Ident "Main") _ _ _ (ClassBody body)))):_) =
229+ findMain ((MemberDecl (MemberClassDecl (ClassDecl _ name _ _ _ (ClassBody body)))):_)
227− findMainMethod body
230+ | name == mangle' (UN "Main") = findMainMethod body
228 findMain (_:decls) = findMain decls
231 findMain (_:decls) = findMain decls
229 findMain [] = False
232 findMain [] = False
234+ innerMainMethod = (either error id $ mangle (UN "main"))
231− findMainMethod ((MemberDecl (MethodDecl _ _ _ (Ident "main") [] _ _)):_) = True
235+ findMainMethod ((MemberDecl (MethodDecl _ _ _ name [] _ _)):_)
236+ | name == mangle' (UN "main") = True
232 findMainMethod (_:decls) = findMainMethod decls
237 findMainMethod (_:decls) = findMainMethod decls
233 findMainMethod [] = False
238 findMainMethod [] = False
235mkMainMethod :: Decl
240mkMainMethod :: Decl
236−mkMainMethod =
241+mkMainMethod =
237− MemberDecl $ MethodDecl [Public, Static]
243+ [Public, Static]
239− Nothing
244+ Nothing
240− (Ident "main")
245+ "main"
241− [FormalParam [] stringArrayType False (VarId $ Ident "args")]
246+ [FormalParam [] (array stringType) False (VarId $ Ident "args")]
243− (MethodBody . Just $ Block [ BlockStmt . ExpStmt . MethodInv $
247+ $ Block [ BlockStmt . ExpStmt
244− MethodCall (J.Name [Ident "idris_initArgs"])
245− [ MethodInv $ TypeMethodCall
246− (J.Name [Ident "Thread"])
248− (Ident "currentThread")
248+ $ call "idris_initArgs" [ (threadType ~> "currentThread") []
249− []
249+ , jConst "args"
250− , ExpName $ J.Name [Ident "args"]
252− , BlockStmt . ExpStmt . MethodInv $
251+ , BlockStmt . ExpStmt $ call (mangle' (MN 0 "runMain")) []
253− MethodCall (J.Name [Ident "runMain_0"])
254+-----------------------------------------------------------------------
255+-- Inner classes (idris namespaces)
258mergeInnerClasses :: [Decl] -> [Decl]
257mergeInnerClasses :: [Decl] -> [Decl]
259mergeInnerClasses = foldl' mergeInner []
258mergeInnerClasses = foldl' mergeInner []
@@ -274,1305 +273,343 @@ mergeInnerClasses = foldl' mergeInner []
274 mergeInner [] decl' = [decl']
273 mergeInner [] decl' = [decl']
277−mkIdentifier :: Name -> CodeGeneration Ident
278−mkIdentifier (NS name _) = mkIdentifier name
279−mkIdentifier (MN i name) = (\ (Ident x) -> Ident $ x ++ ('_' : show i))
280− <$> mkIdentifier (UN name)
281−mkIdentifier (UN name) =
282− T.lift $ left (\ err -> "Parser error in \"" ++ name ++ "\": " ++ (show err))
287− $ cleanWs False name)
290− | isNumber x = '_' : (x:xs)
291− | otherwise = (x:xs)
293− cleanNonLetter (x:xs)
294− | x == '#' = "_Hash" ++ cleanNonLetter xs
295− | x == '@' = "_At" ++ cleanNonLetter xs
296− | x == '$' = "_Dollar" ++ cleanNonLetter xs
297− | x == '!' = "_Bang" ++ cleanNonLetter xs
298− | x == '.' = "_Dot" ++ cleanNonLetter xs
299− | x == '\'' = "_Prime" ++ cleanNonLetter xs
300− | x == '*' = "_Times" ++ cleanNonLetter xs
301− | x == '+' = "_Plus" ++ cleanNonLetter xs
302− | x == '/' = "_Divide" ++ cleanNonLetter xs
303− | x == '-' = "_Minus" ++ cleanNonLetter xs
304− | x == '%' = "_Mod" ++ cleanNonLetter xs
305− | x == '<' = "_LessThan" ++ cleanNonLetter xs
306− | x == '=' = "_Equals" ++ cleanNonLetter xs
307− | x == '>' = "_MoreThan" ++ cleanNonLetter xs
308− | x == '[' = "_LSBrace" ++ cleanNonLetter xs
309− | x == ']' = "_RSBrace" ++ cleanNonLetter xs
310− | x == '(' = "_LBrace" ++ cleanNonLetter xs
311− | x == ')' = "_RBrace" ++ cleanNonLetter xs
312− | x == '_' = "__" ++ cleanNonLetter xs
313− | not (isAlphaNum x) = "_" ++ (show $ ord x) ++ xs
314− | otherwise = x:cleanNonLetter xs
315− cleanNonLetter [] = []
316− cleanWs capitalize (x:xs)
317− | isSpace x = cleanWs True xs
318− | capitalize = (toUpper x) : (cleanWs False xs)
319− | otherwise = x : (cleanWs False xs)
323− cleanReserved "param" = "_param"
324− cleanReserved "globalContext" = "_globalContext"
325− cleanReserved "context" = "_context"
326− cleanReserved "newcontext" = "_newcontext"
328− cleanReserved "void" = "_void"
329− cleanReserved "null" = "_null"
330− cleanReserved "int" = "_int"
331− cleanReserved "long" = "_long"
332− cleanReserved "char" = "_char"
333− cleanReserved "byte" = "_byte"
334− cleanReserved "double" = "_double"
335− cleanReserved "float" = "_float"
336− cleanReserved "boolean" = "_boolean"
337− cleanReserved "Object" = "_Object"
338− cleanReserved "String" = "_String"
339− cleanReserved "StringBuilder" = "_StringBuilder"
340− cleanReserved "StringBuffer" = "_StringBuffer"
341− cleanReserved "Scanner" = "_Scanner"
342− cleanReserved "Integer" = "_Integer"
343− cleanReserved "Double" = "_Double"
344− cleanReserved "Byte" = "_Byte"
345− cleanReserved "Character" = "_Character"
346− cleanReserved "BigInteger" = "_BigInteger"
347− cleanReserved "Boolean" = "_Boolean"
348− cleanReserved "Closure" = "_Closure"
349− cleanReserved "IdrisObject" = "_IdrisObject"
350− cleanReserved "TailCallClosure" = "_TailCallClosure"
351− cleanReserved "System" = "_System"
352− cleanReserved "Math" = "_Math"
353− cleanReserved "Arrays" = "_Arrays"
354− cleanReserved "RuntimeException" = "_RuntimeException"
355− cleanReserved "Comparable" = "_Comparable"
357− cleanReserved "class" = "_class"
358− cleanReserved "enum" = "_enum"
359− cleanReserved "interface" = "_interface"
360− cleanReserved "extends" = "_extends"
361− cleanReserved "implements" = "_implements"
362− cleanReserved "public" = "_public"
363− cleanReserved "private" = "_private"
364− cleanReserved "protected" = "_protected"
365− cleanReserved "static" = "_static"
366− cleanReserved "final" = "_final"
367− cleanReserved "abstract" = "_abstract"
368− cleanReserved "strict" = "_strict"
369− cleanReserved "volatile" = "_volatile"
370− cleanReserved "transient" = "_transient"
371− cleanReserved "native" = "_native"
372− cleanReserved "const" = "_const"
374− cleanReserved "import" = "_import"
375− cleanReserved "package" = "_package"
377− cleanReserved "throw" = "_throw"
378− cleanReserved "throws" = "_throws"
379− cleanReserved "try" = "_try"
380− cleanReserved "catch" = "_catch"
382− cleanReserved "synchronized" = "_synchronized"
384− cleanReserved "if" = "_if"
385− cleanReserved "else" = "_else"
386− cleanReserved "switch" = "_switch"
387− cleanReserved "case" = "_case"
388− cleanReserved "default" = "_default"
390− cleanReserved "while" = "_while"
391− cleanReserved "for" = "_for"
392− cleanReserved "do" = "_do"
393− cleanReserved "break" = "_break"
394− cleanReserved "continue" = "_continue"
395− cleanReserved "goto" = "_goto"
397− cleanReserved "this" = "_this"
398− cleanReserved "super" = "_super"
399− cleanReserved "new" = "_new"
400− cleanReserved "return" = "_return"
402− cleanReserved "idris_initArgs" = "_idris_initArgs"
403− cleanReserved "idris_numArgs" = "_idris_numArgs"
404− cleanReserved "idris_getArg" = "_idris_getArg"
405− cleanReserved "getenv" = "_getenv"
406− cleanReserved "exit" = "_exit"
407− cleanReserved "usleep" = "_usleep"
408− cleanReserved "idris_sendMessage" = "_idris_sendMessage"
409− cleanReserved "idris_checkMessage" = "_idris_checkMessage"
410− cleanReserved "idris_recvMessage" = "_idris_recvMessage"
411− cleanReserved "putStr" = "_putStr"
412− cleanReserved "putchar" = "_putchar"
413− cleanReserved "getchar" = "_getchar"
414− cleanReserved "fileOpen" = "_fileOpen"
415− cleanReserved "fileClose" = "_fileClose"
416− cleanReserved "fputStr" = "_fputStr"
417− cleanReserved "fileEOF" = "_fileEOF"
418− cleanReserved "isNull" = "_isNull"
419− cleanReserved "idris_K" = "_idris_K"
420− cleanReserved "idris_flipK" = "_idris_flipK"
421− cleanReserved "idris_assignStack" = "_idris_assignStack"
422− cleanReserved "free" = "_free"
423− cleanReserved "malloc" = "_malloc"
424− cleanReserved "idris_memset" = "_idris_memset"
425− cleanReserved "idris_peek" = "_idris_peek"
426− cleanReserved "idris_poke" = "_idris_poke"
427− cleanReserved "idris_memmove" = "_idris_memmove"
431−mkName :: Name -> CodeGeneration J.Name
432−mkName (NS name nss) = (\ n ns -> J.Name (n:ns))
433− <$> mkIdentifier name
434− <*> mapM (mkIdentifier . UN) nss
435−mkName n = J.Name . (:[]) <$> mkIdentifier n
437−voidType :: ClassType
438−voidType = ClassType [(Ident "Void", [])]
440−objectType :: ClassType
441−objectType = ClassType [(Ident "Object", [])]
443−objectArrayType :: Language.Java.Syntax.Type
444−objectArrayType = RefType . ArrayType . RefType . ClassRefType $ objectType
446−idrisClosureType :: ClassType
447−idrisClosureType = ClassType [(Ident "Closure", [])]
449−idrisTailCallClosureType :: ClassType
450−idrisTailCallClosureType = ClassType [(Ident "TailCallClosure", [])]
452−idrisObjectType :: ClassType
453−idrisObjectType = ClassType [(Ident "IdrisObject", [])]
455−contextArray :: LVar -> Exp
456−contextArray (Loc _) = ExpName $ J.Name [Ident "context"]
457−contextArray (Glob _) = ExpName $ J.Name [Ident "globalContext"]
459−charType :: ClassType
460−charType = ClassType [(Ident "Character", [])]
462−byteType :: ClassType
463−byteType = ClassType [(Ident "Byte", [])]
465−shortType :: ClassType
466−shortType = ClassType [(Ident "Short", [])]
468−integerType :: ClassType
469−integerType = ClassType [(Ident "Integer", [])]
471−longType :: ClassType
472−longType = ClassType [(Ident "Long", [])]
474−nextIntTy :: IntTy -> IntTy
475−nextIntTy (ITFixed IT8) = (ITFixed IT16)
476−nextIntTy (ITFixed IT16) = (ITFixed IT32)
477−nextIntTy (ITFixed IT32) = (ITFixed IT64)
478−nextIntTy (ITFixed IT64) = (ITFixed IT64)
479−nextIntTy ITNative = (ITFixed IT64)
481−intTyToIdent :: IntTy -> Ident
482−intTyToIdent (ITFixed IT8) = Ident "Byte"
483−intTyToIdent (ITFixed IT16) = Ident "Short"
484−intTyToIdent (ITFixed IT32) = Ident "Integer"
485−intTyToIdent (ITFixed IT64) = Ident "Long"
486−intTyToIdent ITNative = Ident "Integer"
487−intTyToIdent ITBig = Ident "BigInteger"
489−intTyToClass :: IntTy -> ClassType
490−intTyToClass ty = ClassType [(intTyToIdent ty, [])]
492−intTyToMethod :: IntTy -> String
493−intTyToMethod (ITFixed IT8) = "byteValue"
494−intTyToMethod (ITFixed IT16) = "shortValue"
495−intTyToMethod (ITFixed IT32) = "intValue"
496−intTyToMethod (ITFixed IT64) = "longValue"
497−intTyToMethod ITNative = "intValue"
499−intTyToPrimTy :: IntTy -> PrimType
500−intTyToPrimTy (ITFixed IT8) = ByteT
501−intTyToPrimTy (ITFixed IT16) = ShortT
502−intTyToPrimTy (ITFixed IT32) = IntT
503−intTyToPrimTy (ITFixed IT64) = LongT
504−intTyToPrimTy ITNative = IntT
506−bigIntegerType :: ClassType
507−bigIntegerType = ClassType [(Ident "BigInteger", [])]
509−doubleType :: ClassType
510−doubleType = ClassType [(Ident "Double", [])]
512−stringType :: ClassType
513−stringType = ClassType [(Ident "String", [])]
515−stringArrayType :: Language.Java.Syntax.Type
516−stringArrayType = RefType . ArrayType . RefType . ClassRefType $ stringType
518−exceptionType :: ClassType
519−exceptionType = ClassType [(Ident "Throwable", [])]
521−runtimeExceptionType :: ClassType
522−runtimeExceptionType = ClassType [(Ident "RuntimeException", [])]
524−comparableType :: ClassType
525−comparableType = ClassType [(Ident "Comparable", [])]
527mkDecl :: (Name, SDecl) -> CodeGeneration Decl
277mkDecl :: (Name, SDecl) -> CodeGeneration Decl
528mkDecl ((NS n (ns:nss)), decl) =
278mkDecl ((NS n (ns:nss)), decl) =
529− (\ name body -> MemberDecl $ MemberClassDecl $ ClassDecl [Public, Static] name [] Nothing [] body)
280+ MemberDecl $ MemberClassDecl $ ClassDecl [Public, Static] name [] Nothing [] body)
530− <$> mkIdentifier (UN ns)
281+ <$> mangle (UN ns)
531 <*> mkClassBody [] [(NS n nss, decl)]
282 <*> mkClassBody [] [(NS n nss, decl)]
532mkDecl (_, SFun name params stackSize body) =
283mkDecl (_, SFun name params stackSize body) =
533− (\ name params paramNames methodBody ->
284+ (\ (Ident name) params paramNames methodBody ->
534− MemberDecl $ MethodDecl [Public, Static]
286+ [Public, Static]
536− (Just . RefType $ ClassRefType objectType)
287+ (Just objectType)
539− []
290+ ( Block $ ( declareFinalObjectArray
540− (MethodBody . Just $ Block
291+ localContextID
541− [ LocalVars [Final]
292+ (Just (arrayInitExps $ extendWithNull paramNames stackSize))
543− [ VarDecl (VarDeclArray . VarId $ Ident "context")
544− (Just . InitArray . ArrayInit $
546− ++ replicate stackSize (InitExp . Lit $ Null))
548− , BlockStmt . Return $ Just methodBody
293+ ) : methodBody
552− <$> mkIdentifier name
296+ <$> mangle name
553 <*> mapM mkFormalParam params
297 <*> mapM mkFormalParam params
554− <*> mapM (\ p -> (InitExp . ExpName) <$> mkName p) params
298+ <*> mapM (\ p -> (ExpName) <$> mangleFull p) params
555− <*> mkExp body
299+ <*> mkExp body (\ block res -> return $ block ++ [jReturn res])
558−mkClosure :: [Name] -> Int -> SExp -> CodeGeneration Exp
559−mkClosure params stackSize body =
560− (\ paramArray body ->
564− (Just $ ClassBody [body])
566− <$> mkStackInit params stackSize
567− <*> mkClosureCall body
569mkFormalParam :: Name -> CodeGeneration FormalParam
301mkFormalParam :: Name -> CodeGeneration FormalParam
570mkFormalParam name =
302mkFormalParam name =
571− (\ name -> FormalParam [Final] (RefType . ClassRefType $ objectType) False (VarId name))
303+ (\ name -> FormalParam [Final] objectType False (VarId name))
572− <$> mkIdentifier name
304+ <$> mangle name
574−mkClosureCall :: SExp -> CodeGeneration Decl
306+-----------------------------------------------------------------------
575−mkClosureCall body =
307+-- Expressions
576− (\ body -> MemberDecl $ MethodDecl [Public] [] (Just . RefType $ ClassRefType objectType) (Ident "call") [] [] body)
308+
577− <$> mkMethodBody body
309+-- | Compile a simple expression and use the given continuation to postprocess
578−
310+-- the resulting value.
579−mkMethodBody :: SExp -> CodeGeneration MethodBody
311+mkExp :: SExp -> BlockResultTransformation -> CodeGeneration [BlockStmt]
580−mkMethodBody exp =
312+-- Variables
581− (\ exp -> MethodBody . Just . Block $ [BlockStmt . Return . Just $ exp])
313+mkExp (SV var) cont = Nothing <>@! var >>= cont []
584−mkStackInit :: [Name] -> Int -> CodeGeneration Exp
316+mkExp (SApp True name args) cont =
585−mkStackInit params stackSize =
317+ ((\ name args -> call name args) <$> mangleFull name <*> mapM (Nothing <>@!) args) >>= cont []
586− (\ localVars -> ArrayCreateInit objectArrayType 0 . ArrayInit $
318+ --mkTailCallClosure name args >>= cont []
587− (map (InitExp . ExpName) localVars)
319+mkExp (SApp False name args) cont =
588− ++ (replicate (stackSize) (InitExp $ Lit Null)))
320+ ((\ name args -> call name args) <$> mangleFull name <*> mapM (Nothing <>@!) args) >>= cont []
589− <$> mapM mkName params
321+ --((\ closure -> (closure ~> "call") []) <$> mkTailCallClosure name args) >>= cont []
591−mkK :: Exp -> Exp -> Exp
323+-- Bindings
592−mkK result drop =
324+mkExp (SLet var newExp inExp) cont = mkLet var newExp inExp cont
593− MethodInv $ MethodCall (J.Name [Ident "idris_K"]) [ result, drop ]
325+mkExp (SUpdate var newExp) cont = mkUpdate var newExp cont
595−mkFlipK :: Exp -> Exp -> Exp
327+-- Objects
596−mkFlipK drop result =
328+mkExp (SCon conId _ args) cont = (mkIdrisObject conId args) >>= cont []
597− MethodInv $ MethodCall (J.Name [Ident "idris_flipK"]) [ drop, result ]
329+
598−
330+-- Case expressions
599−mkLet :: LVar -> SExp -> SExp -> CodeGeneration Exp
331+mkExp (SCase var alts) cont = mkCase True var alts cont
600−mkLet (Loc pos) oldExp newExp =
332+mkExp (SChkCase var alts) cont = mkCase False var alts cont
601− (\ oldExp newExp ->
333+
602− mkFlipK ( Assign ( ArrayLhs $ ArrayIndex (ExpName $ J.Name [Ident "context"])
334+-- Projections
603− (Lit $ Int (toInteger pos)))
335+mkExp (SProj var i) cont = (mkProjection var i) >>= cont []
605− newExp
337+-- Constants
611−mkLet (Glob _) _ _ = T.lift $ Left "Cannot let bind to global variable"
613−reverseNameSpace :: J.Name -> J.Name
614−reverseNameSpace (J.Name ids) =
615− J.Name ((tail ids) ++ [head ids])
617−mkCase :: Bool -> LVar -> [SAlt] -> CodeGeneration Exp
618−mkCase checked var ((SConCase parentStackPos consIndex _ params branchExpression):cases) =
626−mkCase checked var (c@(SConstCase constant branchExpression):cases) =
627− (\ constant branchExpression alternative var->
628− Cond ( MethodInv $ PrimaryMethodCall (constant)
636− <$> mkExp (SConst constant)
338+mkExp (SConst c) cont = cont [] $ mkConstant c
637− <*> mkExp branchExpression
339+
638− <*> mkCase checked var cases
340+-- Foreign function calls
639− <*> mkVarAccess Nothing var
341+mkExp (SForeign lang resTy text params) cont =
640−mkCase checked var (SDefaultCase exp:cases) = mkExp exp
342+ (mkForeign lang resTy text params) >>= cont []
641−mkCase checked _ [] = mkExp (SError "Non-exhaustive pattern")
343+
642−
344+-- Primitive functions
643−mkConsCase :: Bool -> LVar -> Int -> Int -> [Name] -> SExp -> SExp -> CodeGeneration Exp
345+mkExp (SOp LFork [arg]) cont = (mkThread arg) >>= cont []
644−mkConsCase checked
346+mkExp (SOp LPar [arg]) cont = (Nothing <>@! arg) >>= cont []
645− toDeconstruct
347+mkExp (SOp LNoOp args) cont = (Nothing <>@! (last args)) >>= cont []
646− parentStackStart
348+mkExp (SOp op args) cont = (mkPrimitiveFunction op args) >>= cont []
648− params
350+-- Empty expressions
649− branchExpression
351+mkExp (SNothing) cont = cont [] $ Lit Null
651− (\ caseBinding alternative var varCasted->
353+-- Errors
652− Cond (BinOp (InstanceOf (var)
354+mkExp (SError err) cont = mkException err
653− (ClassRefType idrisObjectType)
355+
654− )
356+-----------------------------------------------------------------------
655− CAnd
357+-- Variable access
657− ( MethodInv $ PrimaryMethodCall (varCasted)
659− (Ident "getConstructorId")
663− (Lit $ Int (toInteger consIndex))
669− <$> mkCaseBinding checked toDeconstruct parentStackStart params branchExpression
670− <*> mkExp alternative
671− <*> mkVarAccess (Nothing) toDeconstruct
672− <*> mkVarAccess (Just idrisObjectType) toDeconstruct
675−mkCaseBinding :: Bool -> LVar -> Int -> [Name] -> SExp -> CodeGeneration Exp
676−mkCaseBinding True var parentStackStart params branchExpression =
677− (\ branchExpression deconstruction ->
678− mkFlipK (MethodInv $ MethodCall (J.Name [Ident "idris_assignStack"])
679− ( (ExpName $ J.Name [Ident "context"])
680− : (Lit $ Int (toInteger parentStackStart))
685− <$> mkExp branchExpression
686− <*> mkCaseBindingDeconstruction var params
687−mkCaseBinding False var parentStackStart params branchExpression =
688− (\ bindingMethod deconstruction ->
689− MethodInv $ PrimaryMethodCall
690− ( MethodInv $ PrimaryMethodCall (InstanceCreation []
691− (ClassType [(Ident "Object", [])])
693− (Just $ ClassBody [MemberDecl $ bindingMethod])
697− ( contextArray (Loc undefined) : deconstruction )
703− <$> mkCaseBindingMethod parentStackStart params branchExpression
704− <*> mkCaseBindingDeconstruction var params
706−mkCaseBindingDeconstruction :: LVar -> [Name] -> CodeGeneration [Exp]
359+(<>@!) :: Maybe J.Type -> LVar -> CodeGeneration Exp
707−mkCaseBindingDeconstruction var members =
360+(<>@!) Nothing var =
708− mapM (mkProjection var) ([0..(length members - 1)])
361+ (ArrayAccess . ((contextArray var) @!)) <$> varPos var
710−mkCaseBindingMethod :: Int -> [Name] -> SExp -> CodeGeneration MemberDecl
711−mkCaseBindingMethod parentStackStart params branchExpression =
712− (\ formalParams caseBindingStack branchExpression ->
713− MethodDecl [Final, Public]
715− (Just . RefType $ ClassRefType idrisClosureType)
717− (mkContextParam:formalParams)
719− (MethodBody . Just . Block $
721− [BlockStmt . Return $ Just branchExpression]))
722− <$> mapM mkFormalParam params
723− <*> mkBindingStack False parentStackStart params
724− <*> mkBindingClosure branchExpression
726−mkContextParam :: FormalParam
728− FormalParam [Final] (objectArrayType) False (VarId (Ident "context"))
730−mkBindingClosure :: SExp -> CodeGeneration Exp
731−mkBindingClosure oldExp =
735− [ ExpName $ J.Name [Ident "new_context"] ]
736− (Just $ ClassBody [oldCall])
738− <$> mkClosureCall oldExp
741−mkBindingStack :: Bool -> Int -> [Name] -> CodeGeneration [BlockStmt]
742−mkBindingStack checked parentStackStart params =
746− [ VarDecl (VarDeclArray . VarId $ Ident "new_context")
747− (Just . InitExp $ mkContextCopy checked parentStackStart params)
749− : ( map (\ (param, pos) ->
750− BlockStmt . ExpStmt $
751− Assign (ArrayLhs $ ArrayIndex (ExpName $ J.Name [Ident "new_context"])
752− (Lit $ Int (toInteger pos)))
754− (ExpName param)) $ zip paramNames [parentStackStart..]
757− <$> mapM mkName params
759−mkContextCopy :: Bool -> Int -> [Name] -> Exp
760−mkContextCopy True parentStackStart params =
761− MethodInv $ PrimaryMethodCall (ExpName $ J.Name [Ident "context"])
765−mkContextCopy False parentStackStart params =
766− MethodInv $ TypeMethodCall (J.Name [Ident "Arrays"])
769− [ ExpName $ J.Name [Ident "context"]
771− $ TypeMethodCall (J.Name [Ident "Math"])
774− [ FieldAccess $ PrimaryFieldAccess (ExpName $ J.Name [Ident "context"])
776− , Lit . Int $ toInteger (parentStackStart + length params)
780−mkProjection :: LVar -> Int -> CodeGeneration Exp
781−mkProjection var memberNr =
782− (\ var -> ArrayAccess $ ArrayIndex ( MethodInv $ PrimaryMethodCall
788− (Lit $ Int (toInteger memberNr))
790− <$> mkVarAccess (Just idrisObjectType) var
792−type ClassName = String
794−mkPrimitive :: ClassName -> Literal -> Exp
795−mkPrimitive className value =
796− MethodInv $ TypeMethodCall (J.Name [Ident className])
801−mkClass :: ClassType -> Exp
803− ClassLit . Just . RefType .ClassRefType $ classType
805−mkBinOpExp :: ClassType -> Op -> [LVar] -> CodeGeneration Exp
806−mkBinOpExp castTo op (var:vars) = do
362+(<>@!) (Just castTo) var =
807− start <- mkVarAccess (Just castTo) var
363+ (castTo <>) <$> (Nothing <>@! var)
808− foldM (\ exp var -> BinOp exp op <$> mkVarAccess (Just castTo) var) start vars
364+
809−
365+varPos :: LVar -> CodeGeneration Int
810−mkBinOpExpTrans :: (Exp -> Exp) -> (Exp -> Exp) -> ClassType -> Op -> [LVar] -> CodeGeneration Exp
811−mkBinOpExpTrans opTransformation resultTransformation castTo op (var:vars) = do
812− start <- (mkVarAccess (Just castTo) var)
813− foldM (\ exp var -> resultTransformation
814− . BinOp (opTransformation exp) op
816− <$> mkVarAccess (Just castTo) var)
820−mkBinOpExpConv :: String -> PrimType -> ClassType -> Op -> [LVar] -> CodeGeneration Exp
821−mkBinOpExpConv fromMethodName toType fromType@(ClassType [(cls@(Ident _), [])]) op args =
822− mkBinOpExpTrans (\ exp -> MethodInv $ TypeMethodCall (J.Name [cls])
824− (Ident fromMethodName)
827− (\ exp -> MethodInv $ TypeMethodCall (J.Name [cls])
830− [Cast (PrimType $ toType) exp]
836−mkLogicalBinOpExp :: ClassType -> Op -> [LVar] -> CodeGeneration Exp
837−mkLogicalBinOpExp castTo op (var:vars) = do
838− start <- mkVarAccess (Just castTo) var
839− foldM (\ exp var -> mkBoolToNumber castTo . BinOp exp op <$> mkVarAccess (Just castTo) var)
843−mkMethodOpChain1 :: (Exp -> Exp) -> ClassType -> String -> [LVar] -> CodeGeneration Exp
844−mkMethodOpChain1 = mkMethodOpChain id
846−mkMethodOpChain :: (Exp -> Exp) -> (Exp -> Exp) -> ClassType -> String -> [LVar] -> CodeGeneration Exp
847−mkMethodOpChain initialTransformation resultTransformation castTo method (arg:args) = do
848− start <- initialTransformation <$> mkVarAccess (Just $ castTo) arg
849− foldM (\ exp arg' ->
850− resultTransformation
852− . PrimaryMethodCall exp [] (Ident method)
854− <$> mkVarAccess (Just $ castTo) arg'
859−mkBoolToNumber :: ClassType -> Exp -> Exp
860−mkBoolToNumber (ClassType [(Ident name, [])]) boolExp =
861− Cond boolExp (mkPrimitive name (Int 1)) (mkPrimitive name (Int 0))
863−mkZeroExt :: String -> Int -> ClassType -> ClassType -> LVar -> CodeGeneration Exp
864−mkZeroExt toMethod bits fromType toType@(ClassType [(toTypeName, [])]) var = do
866− MethodInv $ TypeMethodCall (J.Name [toTypeName])
869− [ Cond ( BinOp (var)
873− ( BinOp (Lit $ Int (2^bits))
880− <$> mkVarAccess (Just $ fromType) var
881− <*> mkSignedExt' toMethod fromType var
883−mkSignedExt :: String -> ClassType -> ClassType -> LVar -> CodeGeneration Exp
884−mkSignedExt toMethod fromType (ClassType [(toTypeName, [])]) var =
885− (\ sext -> MethodInv $ TypeMethodCall (J.Name [toTypeName])
890− <$> mkSignedExt' toMethod fromType var
892−mkSignedExt' :: String -> ClassType -> LVar -> CodeGeneration Exp
893−mkSignedExt' toMethod fromType var =
894− (\ var -> MethodInv $ PrimaryMethodCall (var)
899− <$> mkVarAccess (Just $ fromType) var
908−mkPartialOrder :: SPartialOrder -> Exp -> Exp
909−mkPartialOrder SLt x = (BinOp (Lit $ Int (-1)) Equal x)
910−mkPartialOrder SLe x =
911− BinOp (BinOp (Lit $ Int (-1)) Equal x)
913− (BinOp (Lit $ Int 0) Equal x)
914−mkPartialOrder SEq x = BinOp (Lit $ Int 0) Equal x
915−mkPartialOrder SGe x =
916− BinOp (BinOp (Lit $ Int 1) Equal x)
918− (BinOp (Lit $ Int 0) Equal x)
919−mkPartialOrder SGt x = (BinOp (Lit $ Int 1) Equal x)
921−varPos :: LVar -> CodeGeneration Integer
922−varPos (Loc i) = return (toInteger i)
366+varPos (Loc i) = return i
923varPos (Glob name) = do
367varPos (Glob name) = do
924 positions <- globalVariablePositions <$> get
368 positions <- globalVariablePositions <$> get
925 case lookup name positions of
369 case lookup name positions of
926 (Just pos) -> return pos
370 (Just pos) -> return pos
927 Nothing -> T.lift . Left $ "Invalid global variable id: " ++ show name
371 Nothing -> T.lift . Left $ "Invalid global variable id: " ++ show name
929−mkVarAccess :: Maybe ClassType -> LVar -> CodeGeneration Exp
373+-----------------------------------------------------------------------
930−mkVarAccess Nothing var =
374+-- Application (wrap method calls in tail call closures)
931− (\ pos -> ArrayAccess $ ArrayIndex (contextArray var) (Lit $ Int pos))
375+
932− <$> varPos var
376+mkTailCallClosure :: Name -> [LVar] -> CodeGeneration Exp
933−mkVarAccess (Just castTo) var =
377+mkTailCallClosure name args =
934− Cast (RefType . ClassRefType $ castTo) <$> (mkVarAccess Nothing var)
378+ (\ methClosure ->
935−
379+ InstanceCreation [] (toClassType idrisTailCallClosureType) [methClosure] Nothing
936−mkPrimitiveCast :: ClassType -> ClassType -> LVar -> CodeGeneration Exp
937−mkPrimitiveCast fromType (ClassType [(toType, [])]) var =
939− MethodInv $ TypeMethodCall (J.Name [toType])
944− <$> mkVarAccess (Just fromType) var
946−mkToString :: ClassType -> LVar -> CodeGeneration Exp
947−mkToString castTo var =
948− (\ var -> MethodInv $ PrimaryMethodCall (var)
953− <$> mkVarAccess (Just castTo) var
954−data Std = In | Out | Err
956−instance Show Std where
961−mkSystemStd :: Std -> Exp
962−mkSystemStd std = FieldAccess $ PrimaryFieldAccess (ExpName $ J.Name [Ident "System"]) (Ident $ show std)
964−mkSystemOutPrint :: Exp -> Exp
965−mkSystemOutPrint value =
966− MethodInv $ PrimaryMethodCall (mkSystemStd Out)
971−mkMathFun :: String -> LVar -> CodeGeneration Exp
972−mkMathFun funName var =
973− (\ var -> MethodInv $ TypeMethodCall (J.Name [Ident "Double"])
976− [ MethodInv $ TypeMethodCall (J.Name [Ident "Math"])
982− <$> mkVarAccess (Just doubleType) var
984−mkStringAtIndex :: LVar -> Exp -> CodeGeneration Exp
985−mkStringAtIndex var indexExp =
986− (\ var -> MethodInv $ TypeMethodCall (J.Name [Ident "Integer"])
989− [ MethodInv $ PrimaryMethodCall (var)
995− <$> mkVarAccess (Just stringType) var
997−mkForeignType :: FType -> Maybe ClassType
998−mkForeignType (FArith (ATInt ity)) = return (intTyToClass ity)
999−mkForeignType (FArith ATFloat) = return doubleType
1000−mkForeignType FChar = return integerType
1001−mkForeignType FString = return stringType
1002−mkForeignType FPtr = return objectType
1003−mkForeignType FAny = return objectType
1004−mkForeignType FUnit = Nothing
1006−mkForeignVarAccess :: FType -> LVar -> CodeGeneration Exp
1007−mkForeignVarAccess (FArith (ATInt ty)) var =
1008− (\ var -> MethodInv $ PrimaryMethodCall var
1010− (Ident (intTyToMethod ty))
1013− <$> mkVarAccess (Just $ intTyToClass ty) var
1014−mkForeignVarAccess FChar var = Cast (PrimType CharT) <$> mkForeignVarAccess (FArith (ATInt (ITFixed IT32))) var
1015−mkForeignVarAccess (FArith ATFloat) var =
1016− (\ var -> MethodInv $ PrimaryMethodCall (var)
1018− (Ident "doubleValue")
1021− <$> mkVarAccess (Just doubleType) var
1022−mkForeignVarAccess otherType var = mkVarAccess (mkForeignType otherType) var
1024−mkFromForeignType :: FType -> Exp -> Exp
1025−mkFromForeignType (FArith (ATInt ty)) from =
1026− MethodInv $ TypeMethodCall (J.Name [intTyToIdent ty])
1030−mkFromForeignType FChar from = mkFromForeignType (FArith (ATInt (ITFixed IT32))) from
1031−mkFromForeignType (FArith ATFloat) from =
1032− MethodInv $ TypeMethodCall (J.Name [Ident "Double"])
1036−mkFromForeignType _ from = from
1038−mkForeignInvoke :: FType -> String -> [(FType, LVar)] -> CodeGeneration Exp
1039−mkForeignInvoke fType method args =
1040− (\ foreignInvokeMeth ->
1041− MethodInv $ PrimaryMethodCall (InstanceCreation []
1044− (Just $ ClassBody [ MemberDecl $ foreignInvokeMeth ])
1047− (Ident "foreignInvoke")
1050− <$> mkForeignInvokeMethod fType method args
381+ <$> mkMethodCallClosure name args
1052−
383+mkMethodCallClosure :: Name -> [LVar] -> CodeGeneration Exp
1053−mkForeignInvokeMethod :: FType -> String -> [(FType, LVar)] -> CodeGeneration MemberDecl
384+mkMethodCallClosure name args =
1054−mkForeignInvokeMethod fType method args =
385+ (\ name args -> closure [localContext] (call name args))
1055− (\ tryBlock ->
386+ <$> mangleFull name
1056− MethodDecl [Public, Final]
387+ <*> mapM (Nothing <>@!) args
1058− (Just . RefType $ ClassRefType objectType)
389+-----------------------------------------------------------------------
1059− (Ident "foreignInvoke")
390+-- Updates (change context array) and Let bindings (Update, execute)
1061− []
392+mkUpdate :: LVar -> SExp -> BlockResultTransformation -> CodeGeneration [BlockStmt]
1062− (MethodBody . Just $ Block
393+mkUpdate var exp cont = do
1063− [ BlockStmt
394+ pos <- varPos var
1064− $ Try tryBlock
395+ mkExp exp (\ blk rhs -> cont blk $ ((contextArray var) @! pos) @:= rhs )
1065− [ Catch (FormalParam []
396+
1066− (RefType $ ClassRefType exceptionType)
397+mkLet :: LVar -> SExp -> SExp -> BlockResultTransformation -> CodeGeneration [BlockStmt]
1067− False
398+mkLet var@(Loc pos) newExp inExp cont = do
1068− (VarId $ Ident "ex")
399+ inBlk <- mkExp inExp cont
1069− )
400+ upd <- mkUpdate var newExp (\ blk exp -> return $ addToBlock blk exp)
1070− (Block [ BlockStmt
401+ return $ upd ++ inBlk
1071− . Throw
402+mkLet (Glob _) _ _ _ = T.lift $ Left "Cannot let bind to global variable"
404+-----------------------------------------------------------------------
407+mkIdrisObject :: Int -> [LVar] -> CodeGeneration Exp
408+mkIdrisObject conId args =
1072− $ InstanceCreation []
410+ InstanceCreation [] (toClassType idrisObjectType) ((jInt conId):args) Nothing
1073− runtimeExceptionType
1074− [ExpName $ J.Name [Ident "ex"]]
1083− <$> mkForeignInvokeTryBlock fType method args
412+ <$> mapM (Nothing <>@!) args
414+-----------------------------------------------------------------------
1086−mkForeignInvokeTryBlock :: FType -> String -> [(FType, LVar)] -> CodeGeneration Block
417+mkCase :: Bool -> LVar -> [SAlt] -> BlockResultTransformation -> CodeGeneration [BlockStmt]
1087−mkForeignInvokeTryBlock FUnit method args =
1088− (\ method args -> Block [ BlockStmt . ExpStmt . MethodInv $ MethodCall method args
1089− , BlockStmt $ Return (Just $ Lit Null)
1092− <$> ( T.lift $ left (\ err -> "Error parsing name \"" ++ method ++ "\" :" ++ (show err))
1093− (parser name method)
1095− <*> mapM (uncurry mkForeignVarAccess) args
1096−mkForeignInvokeTryBlock fType method args =
1097− (\ method args -> Block [ BlockStmt . Return
1099− . mkFromForeignType fType
1101− $ MethodCall method args
1104− <$> ( T.lift $ left (\ err -> "Error parsing name \"" ++ method ++ "\" :" ++ (show err))
1105− (parser name method)
1107− <*> mapM (uncurry mkForeignVarAccess) args
1109−mkMethodClosure :: Name -> [LVar] -> CodeGeneration Exp
1110−mkMethodClosure name args =
1112− InstanceCreation []
1114− [ (ExpName $ J.Name [Ident "context"]) ]
1118− $ MethodDecl [Public, Final]
1120− (Just . RefType $ ClassRefType objectType)
1126− $ Block [ BlockStmt
1130− $ MethodCall (reverseNameSpace name)
1138− <*> mapM (mkExp . SV) args
1140−mkThread :: LVar -> CodeGeneration Exp
419+mkCase checked var cases cont
1141−mkThread arg =
420+ | isConstCase cases = do
1142− (\ eval ->
421+ defaultStmts <- mkDefaultMatch cases cont
1143− MethodInv
422+ ifte <- mkConstMatch var defaultStmts cases cont
1144− $ PrimaryMethodCall (InstanceCreation []
423+ return [BlockStmt ifte]
1145− (ClassType [(Ident "Thread", [])])
424+mkCase checked var cases cont =
1146− [ eval ]
425+ (\ switchExp blocks defaultStmts ->
1147− ( Just
426+ [BlockStmt $ Switch switchExp (blocks ++ [SwitchBlock Default defaultStmts])]
1148− $ ClassBody [ MemberDecl
1149− $ MethodDecl [Public, Final]
1151− (Just . RefType $ ClassRefType objectType)
1157− $ Block [ BlockStmt
1160− $ MethodCall (J.Name [Ident "start"])
1175− <$> mkThreadBinding arg
428+ <$> mkGetConstructorId checked var
1176−
429+ <*> mkConsMatch var cases cont
1177−mkThreadBinding :: LVar -> CodeGeneration Exp
430+ <*> mkDefaultMatch cases cont
1178−mkThreadBinding var =
431+
1179− (\ bindingMethod var ->
432+isConstCase :: [SAlt] -> Bool
1180− MethodInv $ PrimaryMethodCall ( InstanceCreation []
433+isConstCase ((SConstCase _ _):_) = True
1181− objectType
434+isConstCase ((SDefaultCase _):cases) = isConstCase cases
1182− []
435+isConstCase _ = False
1183− (Just $ ClassBody [MemberDecl $ bindingMethod])
436+
1184− )
437+mkDefaultMatch :: [SAlt] -> BlockResultTransformation -> CodeGeneration [BlockStmt]
1185− []
438+mkDefaultMatch (x@(SDefaultCase branchExpression):_) cont = mkExp branchExpression cont
1186− (Ident "apply")
439+mkDefaultMatch (x:xs) cont = mkDefaultMatch xs cont
1187− [ var ]
440+mkDefaultMatch [] cont = mkException "Non-exhaustive pattern"
442+mkMatchConstExp :: LVar -> Const -> CodeGeneration Exp
443+mkMatchConstExp var c
445+ (\ var -> (primFnType ~> opName (LEq undefined)) [var, jc] ~==~ jInt 1)
446+ <$> (Just cty <>@! var)
448+ (\ var -> (arraysType ~> "equals") [var, jc])
449+ <$> (Just cty <>@! var)
451+ (\ var -> ((primFnType ~> opName (LStrEq)) [var, jc] ~==~ jInt 1))
452+ <$> (Just cty <>@! var)
454+ (\ var -> (var ~> "equals") [jc])
455+ <$> (Just cty <>@! var)
460+mkConstMatch :: LVar -> [BlockStmt] -> [SAlt] -> BlockResultTransformation -> CodeGeneration Stmt
461+mkConstMatch var defaultStmts ((SConstCase constant branchExpression):cases) cont =
462+ (\ exp block others -> IfThenElse exp (StmtBlock $ Block block) others)
463+ <$> mkMatchConstExp var constant
464+ <*> mkExp branchExpression cont
465+ <*> mkConstMatch var defaultStmts cases cont
466+mkConstMatch var defaultStmts (c:cases) cont = mkConstMatch var defaultStmts cases cont
467+mkConstMatch var defaultStmts [] cont = return $ StmtBlock (Block defaultStmts)
469+mkGetConstructorId :: Bool -> LVar -> CodeGeneration Exp
470+mkGetConstructorId True var =
471+ (\ var -> ((idrisObjectType <> var) ~> "getConstructorId") [])
472+ <$> (Nothing <>@! var)
473+mkGetConstructorId False var =
475+ Cond (InstanceOf var (toRefType idrisObjectType)) match (jInt (-1))
1189− <$> mkThreadBindingMethod
477+ <$> (Nothing <>@! var)
1190− <*> mkVarAccess Nothing var
478+ <*> mkGetConstructorId True var
1192−
480+mkConsMatch :: LVar -> [SAlt] -> BlockResultTransformation -> CodeGeneration [SwitchBlock]
1193−mkThreadBindingMethod :: CodeGeneration MemberDecl
481+mkConsMatch var ((SConCase parentStackPos consIndex _ params branchExpression):cases) cont =
1194−mkThreadBindingMethod =
482+ (\ block others -> (SwitchBlock (SwitchCase $ jInt consIndex) block):others)
1195− (\ compute ->
483+ <$> mkCaseBinding var parentStackPos params branchExpression cont
484+ <*> mkConsMatch var cases cont
485+mkConsMatch var (_:cases) cont = mkConsMatch var cases cont
486+mkConsMatch _ [] _ = return []
488+mkCaseBinding :: LVar -> Int -> [Name] -> SExp -> BlockResultTransformation -> CodeGeneration [BlockStmt]
489+mkCaseBinding var stackStart [] branchExpression cont = mkExp branchExpression cont
490+mkCaseBinding var stackStart (params) branchExpression cont = do
491+ deconstruction <- mkCaseBindingDeconstruction var stackStart params
492+ branchBlk <- mkExp branchExpression cont
493+ return $ deconstruction ++ branchBlk
495+mkCaseBindingDeconstruction :: LVar -> Int -> [Name] -> CodeGeneration [BlockStmt]
496+mkCaseBindingDeconstruction var stackStart members =
497+ mapM (\ pos -> BlockStmt . ExpStmt .
498+ ((contextArray var @! (stackStart + pos)) @:=) <$> mkProjection var pos
499+ ) ([0..(length members - 1)])
501+-----------------------------------------------------------------------
502+-- Projection (retrieve the n-th field of an object)
504+mkProjection :: LVar -> Int -> CodeGeneration Exp
505+mkProjection var memberNr =
506+ (\ var -> ArrayAccess $ ((var ~> "getData") []) @! memberNr)
507+ <$> (Just idrisObjectType <>@! var)
509+-----------------------------------------------------------------------
512+mkConstantArray :: (V.Unbox a) => J.Type -> (a -> Const) -> V.Vector a -> Exp
513+mkConstantArray cty elemToConst elems =
517+ (ArrayInit . map (InitExp . mkConstant . elemToConst) $ V.toList elems)
519+mkConstant :: Const -> Exp
520+mkConstant c@(I x) = constType c <> (Lit . Word $ toInteger x)
521+mkConstant c@(BI x) = bigInteger (show x)
522+mkConstant c@(Fl x) = constType c <> (Lit . Double $ x)
523+mkConstant c@(Ch x) = constType c <> (Lit . Char $ x)
524+mkConstant c@(Str x) = constType c <> (Lit . String $ x)
525+mkConstant c@(B8 x) = constType c <> (Lit . Word $ toInteger x)
526+mkConstant c@(B16 x) = constType c <> (Lit . Word $ toInteger x)
527+mkConstant c@(B32 x) = constType c <> (Lit . Word $ toInteger x)
528+mkConstant c@(B64 x) = (bigInteger (show c) ~> "longValue") []
529+mkConstant c@(B8V x) = mkConstantArray (constType c) B8 x
530+mkConstant c@(B16V x) = mkConstantArray (constType c) B16 x
531+mkConstant c@(B32V x) = mkConstantArray (constType c) B32 x
532+mkConstant c@(B64V x) = mkConstantArray (constType c) B64 x
533+mkConstant c@(AType x) = ClassLit (Just $ box (constType c))
534+mkConstant c@(ChType ) = ClassLit (Just $ box integerType)
535+mkConstant c@(StrType ) = ClassLit (Just $ stringType)
536+mkConstant c@(PtrType ) = ClassLit (Just $ objectType)
537+mkConstant c@(VoidType ) = ClassLit (Just $ voidType)
538+mkConstant c@(Forgot ) = ClassLit (Just $ objectType)
540+-----------------------------------------------------------------------
541+-- Foreign function calls
543+mkForeignWrappedMethod :: FCallType -> FType -> String -> [(FType, LVar)] -> CodeGeneration Decl
544+mkForeignWrappedMethod callType resTy text params
545+ | callType <- FStatic =
547+ wrapperMethod $ wrapReturn resTy (call method args)
549+ <$> liftParsed (parser name text)
550+ <*> foreignVarAccess params
551+ | callType <- FObject =
552+ (\ method (tgt:args) ->
553+ wrapperMethod $ wrapReturn resTy ((tgt ~> (show $ pretty method)) args)
555+ <$> liftParsed (parser ident text)
556+ <*> foreignVarAccess params
557+ | callType <- FConstructor =
559+ wrapperMethod $ wrapReturn resTy (InstanceCreation [] clsTy args Nothing)
561+ <$> liftParsed (parser classType text)
562+ <*> foreignVarAccess params
1196− MethodDecl [Final, Public]
565+ MemberDecl . MethodDecl
1197− []
566+ [Protected, Final] [] (Just objectType) (Ident $ "wrappedInvoke") [] [toRefType exceptionType]
1198− (Just . RefType $ ClassRefType idrisClosureType)
1200− [ FormalParam [Final] (RefType . ClassRefType $ objectType) False (VarId $ Ident "param") ]
1202− (MethodBody . Just $ Block
567+ . MethodBody . Just . Block
1203− [ mkThreadBindingStack
568+
1204− , BlockStmt . Return $ Just compute
569+
1205− ]
570+ wrapReturn FUnit exp = [ BlockStmt $ ExpStmt exp, jReturn (Lit Null) ]
1206− )
571+ wrapReturn _ exp = [ jReturn exp ]
573+ foreignVarAccess args =
574+ mapM (\ (fty, var) -> (foreignType fty <>@! var)) args
576+mkForeign :: FLang -> FType -> String -> [(FType, LVar)] -> CodeGeneration Exp
577+mkForeign (LANG_JAVA callType) resTy text params =
581+ (toClassType foreignWrapperType)
583+ (Just $ ClassBody [wrappedMethod])
1208− <$> mkBindingClosure (SUpdate (Loc 0) (SApp False (MN 0 "EVAL") [Loc 0]))
586+ <$> mkForeignWrappedMethod callType resTy text params
587+mkForeign (LANG_C) resTy text params = mkForeign (LANG_JAVA FStatic) resTy text params
589+-----------------------------------------------------------------------
590+-- Primitive functions
1211−mkThreadBindingStack :: BlockStmt
592+mkPrimitiveFunction :: PrimFn -> [LVar] -> CodeGeneration Exp
1212−mkThreadBindingStack =
593+mkPrimitiveFunction op args =
1213− LocalVars [Final]
594+ (\ args -> (primFnType ~> opName op) args)
1214− objectArrayType
595+ <$> sequence (zipWith (\ a t -> (Just t) <>@! a) args (sourceTypes op))
1215− [ VarDecl (VarDeclArray . VarId $ Ident "new_context")
596+
1216− (Just . InitArray $ ArrayInit [InitExp . ExpName $ J.Name [Ident "param"]])
597+mkThread :: LVar -> CodeGeneration Exp
599+ (\ closure -> (closure ~> "fork") []) <$> mkTailCallClosure (MN 0 "EVAL") [arg]
601+-----------------------------------------------------------------------
604+mkException :: String -> CodeGeneration [BlockStmt]
1218−
610+ (toClassType runtimeExceptionType)
1220−mkExp :: SExp -> CodeGeneration Exp
1221−mkExp (SV var) = mkVarAccess Nothing var
1222−mkExp (SApp False name args) =
1224− MethodInv $ PrimaryMethodCall ( InstanceCreation []
1225− idrisTailCallClosureType
1233− <$> mkMethodClosure name args
1234−mkExp (SApp True name args) =
1236− ( InstanceCreation []
1237− idrisTailCallClosureType
1242− <$> mkMethodClosure name args
1243−mkExp (SLet var new old) =
1245−mkExp (SUpdate var exp) =
1246− (\ rhs varPos -> Assign (ArrayLhs $ ArrayIndex (contextArray var) (Lit $ Int varPos))
1252−mkExp (SCon conId name args) =
1253− (\ args -> InstanceCreation []
1255− ((Lit $ Int (toInteger conId)):args)
1257− <$> mapM (mkExp .SV) args
1258−mkExp (SCase var alts) = mkCase False var alts
1259−mkExp (SChkCase var alts) = mkCase True var alts
1260−mkExp (SProj var i) = mkProjection var i
1261−mkExp (SConst (I x)) =
1262− let x' :: Int32; x' = fromInteger (toInteger x) in
1263− return $ mkPrimitive "Integer" (Int (fromInteger (toInteger x')))
1264−mkExp (SConst (BI x)) =
1265− return $ InstanceCreation []
1266− (ClassType [(Ident "BigInteger", [])])
1267− [Lit $ String (show x)]
1269−mkExp (SConst (Fl x)) = return $ mkPrimitive "Double" (Double x)
1270−mkExp (SConst (Ch x)) = return $ mkPrimitive "Integer" (Char x)
1271−mkExp (SConst (Str x)) = return $ Lit $ String x
1272−mkExp (SConst (AType (ATInt ITNative))) = return $ mkClass integerType
1273−mkExp (SConst (AType (ATInt ITBig))) = return $ mkClass bigIntegerType
1274−mkExp (SConst (AType ATFloat)) = return $ mkClass doubleType
1275−mkExp (SConst ChType) = return $ mkClass charType
1276−mkExp (SConst StrType) = return $ mkClass stringType
1277−mkExp (SConst (B8 x)) = return $ mkPrimitive "Byte" (String (show x))
1278−mkExp (SConst (B16 x)) = return $ mkPrimitive "Short" (String (show x))
1279−mkExp (SConst (B32 x)) = return $ mkPrimitive "Integer" (Int (toInteger x))
1280−mkExp (SConst (B64 x)) = return $ mkPrimitive "Long" (String (show x))
1281−mkExp (SConst (AType (ATInt (ITFixed IT8)))) = return $ mkClass byteType
1282−mkExp (SConst (AType (ATInt (ITFixed IT16)))) = return $ mkClass shortType
1283−mkExp (SConst (AType (ATInt (ITFixed IT32)))) = return $ mkClass integerType
1284−mkExp (SConst (AType (ATInt (ITFixed IT64)))) = return $ mkClass longType
1285−mkExp (SConst (PtrType)) = return $ mkClass objectType
1286−mkExp (SConst (VoidType)) = return $ mkClass voidType
1287−mkExp (SConst (Forgot)) = return $ mkClass objectType
1288−mkExp (SForeign _ fType meth args) = mkForeignInvoke fType meth args
1289−mkExp (SOp (LPlus (ATInt ITNative)) args) = mkExp (SOp (LPlus (ATInt (ITFixed IT32))) args)
1290−mkExp (SOp (LMinus (ATInt ITNative)) args) = mkExp (SOp (LMinus (ATInt (ITFixed IT32))) args)
1291−mkExp (SOp (LTimes (ATInt ITNative)) args) = mkExp (SOp (LTimes (ATInt (ITFixed IT32))) args)
1292−mkExp (SOp (LSDiv (ATInt ITNative)) args) = mkExp (SOp (LSDiv (ATInt (ITFixed IT32))) args)
1293−mkExp (SOp (LSRem (ATInt ITNative)) args) = mkExp (SOp (LSRem (ATInt (ITFixed IT32))) args)
1294−mkExp (SOp (LAnd ITNative) args) = mkExp (SOp (LAnd (ITFixed IT32)) args)
1295−mkExp (SOp (LOr ITNative) args) = mkExp (SOp (LOr (ITFixed IT32)) args)
1296−mkExp (SOp (LXOr ITNative) args) = mkExp (SOp (LXOr (ITFixed IT32)) args)
1297−mkExp (SOp (LCompl ITNative) args) = mkExp (SOp (LCompl (ITFixed IT32)) args)
1298−mkExp (SOp (LSHL ITNative) args) = mkExp (SOp (LSHL (ITFixed IT32)) args)
1299−mkExp (SOp (LASHR ITNative) args) = mkExp (SOp (LASHR (ITFixed IT32)) args)
1300−mkExp (SOp (LEq (ATInt ITNative)) args) = mkExp (SOp (LEq (ATInt (ITFixed IT32))) args)
1301−mkExp (SOp (LLt (ATInt ITNative)) args) = mkExp (SOp (LLt (ATInt (ITFixed IT32))) args)
1302−mkExp (SOp (LLe (ATInt ITNative)) args) = mkExp (SOp (LLe (ATInt (ITFixed IT32))) args)
1303−mkExp (SOp (LGt (ATInt ITNative)) args) = mkExp (SOp (LGt (ATInt (ITFixed IT32))) args)
1304−mkExp (SOp (LGe (ATInt ITNative)) args) = mkExp (SOp (LGe (ATInt (ITFixed IT32))) args)
1305−mkExp (SOp (LPlus ATFloat) args) = mkBinOpExp doubleType Add args
1306−mkExp (SOp (LMinus ATFloat) args) = mkBinOpExp doubleType Sub args
1307−mkExp (SOp (LTimes ATFloat) args) = mkBinOpExp doubleType Mult args
1308−mkExp (SOp (LSDiv ATFloat) args) = mkBinOpExp doubleType Div args
1309−mkExp (SOp (LEq ATFloat) args) =
1310− mkMethodOpChain1 (mkBoolToNumber doubleType) doubleType "equals" args
1311−mkExp (SOp (LLt ATFloat) args) = mkLogicalBinOpExp integerType LThan args
1312−mkExp (SOp (LLe ATFloat) args) = mkLogicalBinOpExp integerType LThanE args
1313−mkExp (SOp (LGt ATFloat) args) = mkLogicalBinOpExp integerType GThan args
1314−mkExp (SOp (LGe ATFloat) args) = mkLogicalBinOpExp integerType GThanE args
1315−mkExp (SOp (LPlus (ATInt ITBig)) args) = mkMethodOpChain1 id bigIntegerType "add" args
1316−mkExp (SOp (LMinus (ATInt ITBig)) args) = mkMethodOpChain1 id bigIntegerType "subtract" args
1317−mkExp (SOp (LTimes (ATInt ITBig)) args) = mkMethodOpChain1 id bigIntegerType "multiply" args
1318−mkExp (SOp (LSDiv (ATInt ITBig)) args) = mkMethodOpChain1 id bigIntegerType "divide" args
1319−mkExp (SOp (LSRem (ATInt ITBig)) args) = mkMethodOpChain1 id bigIntegerType "mod" args
1320−mkExp (SOp (LEq (ATInt ITBig)) args) =
1321− mkMethodOpChain1 (mkBoolToNumber bigIntegerType) bigIntegerType "equals" args
1322−mkExp (SOp (LLt (ATInt ITBig)) args) =
1323− mkMethodOpChain1 ( mkBoolToNumber bigIntegerType
1324− . mkPartialOrder SLt
1329−mkExp (SOp (LLe (ATInt ITBig)) args) =
1330− mkMethodOpChain1 ( mkBoolToNumber bigIntegerType
1331− . mkPartialOrder SLe
1336−mkExp (SOp (LGt (ATInt ITBig)) args) =
1337− mkMethodOpChain1 ( mkBoolToNumber bigIntegerType
1338− . mkPartialOrder SGt
1343−mkExp (SOp (LGe (ATInt ITBig)) args) =
1344− mkMethodOpChain1 ( mkBoolToNumber bigIntegerType
1345− . mkPartialOrder SGe
1350−mkExp (SOp LStrConcat args) =
1351− mkMethodOpChain (\ exp -> InstanceCreation []
1352− (ClassType [(Ident "StringBuilder", [])])
1356− (\ exp -> MethodInv $ PrimaryMethodCall exp [] (Ident "toString") [])
1360−mkExp (SOp LStrLt args@[_, _]) =
1361− mkMethodOpChain1 ( mkBoolToNumber integerType
1362− . mkPartialOrder SLt
1367−mkExp (SOp LStrEq args@[_, _]) =
1368− mkMethodOpChain1 ( mkBoolToNumber integerType)
1372−mkExp (SOp LStrLen [arg]) =
1373− (\ var -> MethodInv $ PrimaryMethodCall var [] (Ident "length") [])
1374− <$> mkVarAccess (Just stringType) arg
1375−mkExp (SOp (LIntFloat ity) [arg]) =
1376− mkPrimitiveCast (intTyToClass ity) doubleType arg
1377−mkExp (SOp (LFloatInt ity) [arg]) =
1378− mkSignedExt (intTyToMethod ity) doubleType (intTyToClass ity) arg
1379−mkExp (SOp (LIntStr ity) [arg]) =
1380− mkToString (intTyToClass ity) arg
1381−mkExp (SOp (LStrInt ity) [arg]) =
1382− mkPrimitiveCast stringType (intTyToClass ity) arg
1383−mkExp (SOp LFloatStr [arg]) =
1384− mkToString doubleType arg
1385−mkExp (SOp LStrFloat [arg]) =
1386− mkPrimitiveCast doubleType stringType arg
1387−mkExp (SOp (LSExt ITNative ITBig) [arg]) =
1388− mkPrimitiveCast integerType bigIntegerType arg
1389−mkExp (SOp (LTrunc ITBig ITNative) [arg]) =
1390− mkPrimitiveCast bigIntegerType integerType arg
1391−mkExp (SOp (LChInt ITNative) [arg]) =
1392− mkVarAccess (Just integerType) arg
1393−mkExp (SOp (LIntCh ITNative) [arg]) =
1394− mkVarAccess (Just integerType) arg
1395−mkExp (SOp LPrintNum [arg]) =
1396− mkSystemOutPrint <$> (mkVarAccess Nothing arg)
1397−mkExp (SOp LPrintStr [arg]) =
1398− mkSystemOutPrint <$> (mkVarAccess (Just stringType) arg)
1399−mkExp (SOp LReadStr [arg]) = mkExp (SForeign LANG_C FString "idris_readStr" [(FPtr, arg)])
1400−mkExp (SOp (LLt (ATInt ty)) args) = mkLogicalBinOpExp (intTyToClass ty) LThan args
1401−mkExp (SOp (LLe (ATInt ty)) args) = mkLogicalBinOpExp (intTyToClass ty) LThanE args
1402−mkExp (SOp (LEq (ATInt ty)) args) =
1403− mkMethodOpChain1 (mkBoolToNumber (intTyToClass ty)) (intTyToClass ty) "equals" args
1404−mkExp (SOp (LGt (ATInt ty)) args) = mkLogicalBinOpExp (intTyToClass ty) GThan args
1405−mkExp (SOp (LGe (ATInt ty)) args) = mkLogicalBinOpExp (intTyToClass ty) GThanE args
1406−mkExp (SOp (LPlus (ATInt ty)) args) = mkBinOpExp (intTyToClass ty) Add args
1407−mkExp (SOp (LMinus (ATInt ty)) args) = mkBinOpExp (intTyToClass ty) Sub args
1408−mkExp (SOp (LTimes (ATInt ty)) args) = mkBinOpExp (intTyToClass ty) Mult args
1409−mkExp (SOp (LUDiv (ITFixed IT64)) (arg:args)) = do
1410− (arg:args) <- mapM (mkVarAccess (Just longType)) (arg:args)
1411− return $ foldl (\ exp arg ->
1412− MethodInv $ PrimaryMethodCall
1413− ( MethodInv $ PrimaryMethodCall
1414− ( MethodInv $ TypeMethodCall (J.Name [Ident "BigInteger"])
1421− [ MethodInv $ TypeMethodCall (J.Name [Ident "BigInteger"])
1428− (Ident "longValue")
1433−mkExp (SOp (LUDiv ty) args) =
1434− mkBinOpExpConv (intTyToMethod $ nextIntTy ty)
1435− (intTyToPrimTy $ nextIntTy ty)
1439−mkExp (SOp (LSDiv (ATInt ty)) args) = mkBinOpExp (intTyToClass ty) Div args
1440−mkExp (SOp (LURem (ITFixed IT64)) (arg:args)) = do
1441− (arg:args) <- mapM (mkVarAccess (Just longType)) (arg:args)
1442− return $ foldl (\ exp arg ->
1443− MethodInv $ PrimaryMethodCall
1444− ( MethodInv $ PrimaryMethodCall
1445− ( MethodInv $ TypeMethodCall (J.Name [Ident "BigInteger"])
1451− (Ident "remainder")
1452− [ MethodInv $ TypeMethodCall (J.Name [Ident "BigInteger"])
1459− (Ident "longValue")
1464−mkExp (SOp (LURem ty) args) =
1465− mkBinOpExpConv (intTyToMethod $ nextIntTy ty)
1466− (intTyToPrimTy $ nextIntTy ty)
1470−mkExp (SOp (LSRem (ATInt ty)) args) = mkBinOpExp (intTyToClass ty) Rem args
1471−mkExp (SOp (LSHL ty) args) = mkBinOpExp (intTyToClass ty) LShift args
1472−mkExp (SOp (LLSHR ty) args) = mkBinOpExp (intTyToClass ty) RRShift args
1473−mkExp (SOp (LASHR ty) args) = mkBinOpExp (intTyToClass ty) RShift args
1474−mkExp (SOp (LAnd ty) args) = mkBinOpExp (intTyToClass ty) And args
1475−mkExp (SOp (LOr ty) args) = mkBinOpExp (intTyToClass ty) Or args
1476−mkExp (SOp (LXOr ty) args) = mkBinOpExp (intTyToClass ty) Xor args
1477−mkExp (SOp (LCompl ty) [var]) = PreBitCompl <$> mkVarAccess (Just $ intTyToClass ty) var
1478−mkExp (SOp (LZExt from to) [var])
1479− | intTyWidth from < intTyWidth to
1480− = mkZeroExt (intTyToMethod to) (intTyWidth from) (intTyToClass from) (intTyToClass to) var
1481−mkExp (SOp (LSExt from to) [var])
1482− | intTyWidth from < intTyWidth to
1483− = mkSignedExt (intTyToMethod to) (intTyToClass from) (intTyToClass to) var
1484−mkExp (SOp (LTrunc from to) [var])
1485− | intTyWidth from > intTyWidth to
1486− = mkSignedExt (intTyToMethod to) (intTyToClass from) (intTyToClass to) var
1487−mkExp (SOp LFExp [arg]) = mkMathFun "exp" arg
1488−mkExp (SOp LFLog [arg]) = mkMathFun "log" arg
1489−mkExp (SOp LFSin [arg]) = mkMathFun "sin" arg
1490−mkExp (SOp LFCos [arg]) = mkMathFun "cos" arg
1491−mkExp (SOp LFTan [arg]) = mkMathFun "tan" arg
1492−mkExp (SOp LFASin [arg]) = mkMathFun "asin" arg
1493−mkExp (SOp LFACos [arg]) = mkMathFun "acos" arg
1494−mkExp (SOp LFATan [arg]) = mkMathFun "atan" arg
1495−mkExp (SOp LFSqrt [arg]) = mkMathFun "sqrt" arg
1496−mkExp (SOp LFFloor [arg]) = mkMathFun "floor" arg
1497−mkExp (SOp LFCeil [arg]) = mkMathFun "ceil" arg
1498−mkExp (SOp LStrHead [arg]) = mkStringAtIndex arg (Lit $ Int 0)
1499−mkExp (SOp LStrTail [arg]) =
1500− (\ var -> MethodInv $ PrimaryMethodCall (var)
1502− (Ident "substring")
1505− <$> mkVarAccess (Just stringType) arg
1506−mkExp (SOp LStrCons [c, cs]) =
1507− (\ cVar csVar -> MethodInv $
1508− PrimaryMethodCall ( MethodInv $ PrimaryMethodCall (InstanceCreation []
1509− (ClassType [(Ident "StringBuilder", [])])
1516− Cast (PrimType CharT)
1517− (MethodInv $ PrimaryMethodCall
1529− <$> mkVarAccess (Just integerType) c
1530− <*> mkVarAccess (Just stringType) cs
1531−mkExp (SOp LStrIndex [str, i]) = mkVarAccess (Just integerType) i >>= mkStringAtIndex str
1532−mkExp (SOp LStrRev [str]) =
1533− (\ var -> MethodInv $
1534− PrimaryMethodCall ( MethodInv $ PrimaryMethodCall (InstanceCreation []
1535− (ClassType [(Ident "StringBuffer", [])])
1547− <$> mkVarAccess (Just stringType) str
1548−mkExp (SOp LStdIn []) = return $ mkSystemStd In
1549−mkExp (SOp LStdOut []) = return $ mkSystemStd Out
1550−mkExp (SOp LStdErr []) = return $ mkSystemStd Err
1551−mkExp (SOp LFork [arg]) = mkThread arg
1552−mkExp (SOp LPar [arg]) = mkExp (SV arg)
1553−mkExp (SOp LVMPtr []) =
1554− return $ MethodInv $ TypeMethodCall (J.Name [Ident "Thread"]) [] (Ident "currentThread") []
1555−mkExp (SOp LNoOp args) = mkExp . SV $ last args
1556−mkExp (SNothing) = return $ Lit Null
1557−mkExp (SError err) =
1558− return . MethodInv $
1559− PrimaryMethodCall (InstanceCreation []
1560− runtimeExceptionType
1563− [ MemberDecl $ MethodDecl [Public, Final]
1565− (Just . RefType $ ClassRefType objectType)
1566− (Ident "throwSelf")
1569− ( MethodBody . Just $
1570− Block [ BlockStmt (Throw This) ]
1576− (Ident "throwSelf")
1578−mkExp other = error (show other)