1module IRTS.CodegenC (codegenC) where
7import IRTS.Defunctionalise
9import IRTS.CodegenCommon
16import Data.List (intercalate)
20import System.Directory
21import System.FilePath ((</>), (<.>))
26codegenC :: CodeGenerator
27codegenC ci = do codegenC' (simpleDecls ci)
32 (map mkLib (compileLibs ci) ++
33 map incdir (importDirs ci))
38 when (interfaces ci) $
39 codegenH (exportDecls ci)
41 where mkLib l = "-l" ++ l
44codegenC' :: [(Name, SDecl)] ->
45 String -> -- output file name
46 OutputType -> -- generate executable if True, only .o if False
47 [FilePath] -> -- include files
48 [String] -> -- extra object files
49 [String] -> -- extra compiler flags (libraries)
50 [String] -> -- extra compiler flags (anything)
52 Bool -> -- interfaces too (so make a .o instead)
55codegenC' defs out exec incs objs libs flags exports iface dbg
57 let bc = map toBC defs
58 let h = concatMap toDecl (map fst bc)
59 let cc = concatMap (uncurry toC) bc
60 let hi = concatMap ifaceC (concatMap getExp exports)
62 mprog <- readFile (d </> "rts" </> "idris_main" <.> "c")
63 let cout = headers incs ++ debug dbg ++ h ++ cc ++
64 (if (exec == Executable) then mprog else hi)
66 MavenProject -> putStrLn ("FAILURE: output type not supported")
67 Raw -> writeSource out cout
69 (tmpn, tmph) <- tempfile
74 libFlags <- getLibFlags
75 incFlags <- getIncFlags
76 envFlags <- getEnvFlags
77 let args = [gccDbg dbg] ++
79 -- # Any flags defined here which alter the RTS API must also be added to config.mk
80 ["-DHAS_PTHREAD", "-DIDRIS_ENABLE_STATS",
81 "-I."] ++ objs ++ ["-x", "c"] ++ envFlags ++
82 (if (exec == Executable) then [] else ["-c"]) ++
84 (if not iface then concatMap words libFlags else []) ++
85 concatMap words incFlags ++
86 (if not iface then concatMap words libs else []) ++
87 concatMap words flags ++
89-- putStrLn (show args)
90 exit <- rawSystem comp args
91 when (exit /= ExitSuccess) $
92 putStrLn ("FAILURE: " ++ show comp ++ " " ++ show args)
94 getExp (Export _ _ exp) = exp
98 (\h -> "#include <" ++ h ++ ">\n")
99 (xs ++ ["idris_rts.h", "idris_bitstring.h", "idris_stdfgn.h"])
101debug TRACE = "#define IDRIS_TRACE\n\n"
104-- We're using signed integers now. Make sure we get consistent semantics
105-- out of them from gcc. See e.g. http://thiemonagel.de/2010/01/signed-integer-overflow/
106gccFlags i = if i then ["-fwrapv"]
107 else ["-fwrapv", "-fno-strict-overflow"]
113cname :: Name -> String
114cname n = "_idris_" ++ concatMap cchar (showCG n)
115 where cchar x | isAlpha x || isDigit x = [x]
116 | otherwise = "_" ++ show (fromEnum x) ++ "_"
118indent :: Int -> String
119indent n = replicate (n*4) ' '
122creg (L i) = "LOC(" ++ show i ++ ")"
123creg (T i) = "TOP(" ++ show i ++ ")"
126toDecl :: Name -> String
127toDecl f = "void " ++ cname f ++ "(VM*, VAL*);\n"
129toC :: Name -> [BC] -> String
131 = -- "/* " ++ show code ++ "*/\n\n" ++
132 "void " ++ cname f ++ "(VM* vm, VAL* oldbase) {\n" ++
133 indent 1 ++ "INITFRAME;\n" ++
134 concatMap (bcc 1) code ++ "}\n\n"
136showCStr :: String -> String
137showCStr s = '"' : foldr ((++) . showChar) "\"" s
139 showChar :: Char -> String
140 showChar '"' = "\\\""
141 showChar '\\' = "\\\\"
143 -- Note: we need the double quotes around the codes because otherwise
144 -- "\n3" would get encoded as "\x0a3", which is incorrect.
145 -- Instead, we opt for "\x0a""3" and let the C compiler deal with it.
146 | ord c < 0x10 = "\"\"\\x0" ++ showHex (ord c) "\"\""
147 | ord c < 0x20 = "\"\"\\x" ++ showHex (ord c) "\"\""
148 | ord c < 0x7f = [c] -- 0x7f = \DEL
149 | otherwise = showHexes (utf8bytes (ord c))
151 utf8bytes :: Int -> [Int]
152 utf8bytes x = let (h : bytes) = split [] x in
153 headHex h (length bytes) : map toHex bytes
156 split acc x = let xbits = x .&. 0x3f
157 xrest = shiftR x 6 in
158 split (xbits : acc) xrest
160 headHex h 1 = h + 0xc0
161 headHex h 2 = h + 0xe0
162 headHex h 3 = h + 0xf0
163 headHex h n = error "Can't happen: Invalid UTF8 character"
167 showHexes = foldr ((++) . showUTF8) ""
168 showUTF8 c = "\"\"\\x" ++ showHex c "\"\""
170bcc :: Int -> BC -> String
171bcc i (ASSIGN l r) = indent i ++ creg l ++ " = " ++ creg r ++ ";\n"
172bcc i (ASSIGNCONST l c)
173 = indent i ++ creg l ++ " = " ++ mkConst c ++ ";\n"
175 mkConst (I i) = "MKINT(" ++ show i ++ ")"
176 mkConst (BI i) | i < (2^30) = "MKINT(" ++ show i ++ ")"
177 | otherwise = "MKBIGC(vm,\"" ++ show i ++ "\")"
178 mkConst (Fl f) = "MKFLOAT(vm, " ++ show f ++ ")"
179 mkConst (Ch c) = "MKINT(" ++ show (fromEnum c) ++ ")"
180 mkConst (Str s) = "MKSTR(vm, " ++ showCStr s ++ ")"
181 mkConst (B8 x) = "idris_b8const(vm, " ++ show x ++ "U)"
182 mkConst (B16 x) = "idris_b16const(vm, " ++ show x ++ "U)"
183 mkConst (B32 x) = "idris_b32const(vm, " ++ show x ++ "UL)"
184 mkConst (B64 x) = "idris_b64const(vm, " ++ show x ++ "ULL)"
185 -- if it's a type constant, we won't use it, but equally it shouldn't
186 -- report an error. These might creep into generated for various reasons
187 -- (especially if erasure is disabled).
188 mkConst c | isTypeConst c = "MKINT(42424242)"
189 mkConst c = error $ "mkConst of (" ++ show c ++ ") not implemented"
191bcc i (UPDATE l r) = indent i ++ creg l ++ " = " ++ creg r ++ ";\n"
192bcc i (MKCON l loc tag []) | tag < 256
193 = indent i ++ creg l ++ " = NULL_CON(" ++ show tag ++ ");\n"
194bcc i (MKCON l loc tag args)
195 = indent i ++ alloc loc tag ++
196 indent i ++ setArgs 0 args ++ "\n" ++
197 indent i ++ creg l ++ " = " ++ creg Tmp ++ ";\n"
199-- "MKCON(vm, " ++ creg l ++ ", " ++ show tag ++ ", " ++
200-- show (length args) ++ concatMap showArg args ++ ");\n"
201 where showArg r = ", " ++ creg r
203 setArgs i (x : xs) = "SETARG(" ++ creg Tmp ++ ", " ++ show i ++ ", " ++ creg x ++
204 "); " ++ setArgs (i + 1) xs
206 = "allocCon(" ++ creg Tmp ++ ", vm, " ++ show tag ++ ", " ++
207 show (length args) ++ ", 0);\n"
209 = "updateCon(" ++ creg Tmp ++ ", " ++ creg old ++ ", " ++ show tag ++ ", " ++
210 show (length args) ++ ");\n"
212bcc i (PROJECT l loc a) = indent i ++ "PROJECT(vm, " ++ creg l ++ ", " ++ show loc ++
213 ", " ++ show a ++ ");\n"
214bcc i (PROJECTINTO r t idx)
215 = indent i ++ creg r ++ " = GETARG(" ++ creg t ++ ", " ++ show idx ++ ");\n"
216bcc i (CASE True r code def)
217 | length code < 4 = showCase i def code
219 showCode :: Int -> [BC] -> String
220 showCode i bc = "{\n" ++ concatMap (bcc (i + 1)) bc ++
223 showCase :: Int -> Maybe [BC] -> [(Int, [BC])] -> String
224 showCase i Nothing [(t, c)] = indent i ++ showCode i c
225 showCase i (Just def) [] = indent i ++ showCode i def
226 showCase i def ((t, c) : cs)
227 = indent i ++ "if (CTAG(" ++ creg r ++ ") == " ++ show t ++ ") " ++ showCode i c
228 ++ indent i ++ "else\n" ++ showCase i def cs
230bcc i (CASE safe r code def)
231 = indent i ++ "switch(" ++ ctag safe ++ "(" ++ creg r ++ ")) {\n" ++
232 concatMap (showCase i) code ++
239 showCase i (t, bc) = indent i ++ "case " ++ show t ++ ":\n"
240 ++ concatMap (bcc (i+1)) bc ++ indent (i + 1) ++ "break;\n"
241 showDef i Nothing = ""
242 showDef i (Just c) = indent i ++ "default:\n"
243 ++ concatMap (bcc (i+1)) c ++ indent (i + 1) ++ "break;\n"
244bcc i (CONSTCASE r code def)
246-- = indent i ++ "switch(GETINT(" ++ creg r ++ ")) {\n" ++
247-- concatMap (showCase i) code ++
250 = concatMap (iCase (creg r)) code ++
251 indent i ++ "{\n" ++ showDefS i def ++ indent i ++ "}\n"
253 = concatMap (strCase ("GETSTR(" ++ creg r ++ ")")) code ++
254 indent i ++ "{\n" ++ showDefS i def ++ indent i ++ "}\n"
256 = concatMap (biCase (creg r)) code ++
257 indent i ++ "{\n" ++ showDefS i def ++ indent i ++ "}\n"
258 | otherwise = error $ "Can't happen: Can't compile const case " ++ show code
260 intConsts ((I _, _ ) : _) = True
261 intConsts ((Ch _, _ ) : _) = True
264 bigintConsts ((BI _, _ ) : _) = True
265 bigintConsts _ = False
267 strConsts ((Str _, _ ) : _) = True
271 indent i ++ "if (strcmp(" ++ sv ++ ", " ++ show s ++ ") == 0) {\n" ++
272 concatMap (bcc (i+1)) bc ++ indent i ++ "} else\n"
273 biCase bv (BI b, bc) =
274 indent i ++ "if (bigEqConst(" ++ bv ++ ", " ++ show b ++ ")) {\n"
275 ++ concatMap (bcc (i+1)) bc ++ indent i ++ "} else\n"
277 indent i ++ "if (GETINT(" ++ v ++ ") == " ++ show b ++ ") {\n"
278 ++ concatMap (bcc (i+1)) bc ++ indent i ++ "} else\n"
280 indent i ++ "if (GETINT(" ++ v ++ ") == " ++ show (fromEnum b) ++ ") {\n"
281 ++ concatMap (bcc (i+1)) bc ++ indent i ++ "} else\n"
283 showCase i (t, bc) = indent i ++ "case " ++ show t ++ ":\n"
284 ++ concatMap (bcc (i+1)) bc ++
285 indent (i + 1) ++ "break;\n"
286 showDef i Nothing = ""
287 showDef i (Just c) = indent i ++ "default:\n"
288 ++ concatMap (bcc (i+1)) c ++
289 indent (i + 1) ++ "break;\n"
290 showDefS i Nothing = ""
291 showDefS i (Just c) = concatMap (bcc (i+1)) c
293bcc i (CALL n) = indent i ++ "CALL(" ++ cname n ++ ");\n"
294bcc i (TAILCALL n) = indent i ++ "TAILCALL(" ++ cname n ++ ");\n"
295bcc i (SLIDE n) = indent i ++ "SLIDE(vm, " ++ show n ++ ");\n"
296bcc i REBASE = indent i ++ "REBASE;\n"
297bcc i (RESERVE 0) = ""
298bcc i (RESERVE n) = indent i ++ "RESERVE(" ++ show n ++ ");\n"
300bcc i (ADDTOP n) = indent i ++ "ADDTOP(" ++ show n ++ ");\n"
301bcc i (TOPBASE n) = indent i ++ "TOPBASE(" ++ show n ++ ");\n"
302bcc i (BASETOP n) = indent i ++ "BASETOP(" ++ show n ++ ");\n"
303bcc i STOREOLD = indent i ++ "STOREOLD;\n"
304bcc i (OP l fn args) = indent i ++ doOp (creg l ++ " = ") fn args ++ ";\n"
305bcc i (FOREIGNCALL l rty (FStr fn) args)
307 c_irts (toFType rty) (creg l ++ " = ")
308 (fn ++ "(" ++ showSep "," (map fcall args) ++ ")") ++ ";\n"
309 where fcall (t, arg) = irts_c (toFType t) (creg arg)
310bcc i (NULL r) = indent i ++ creg r ++ " = NULL;\n" -- clear, so it'll be GCed
311bcc i (ERROR str) = indent i ++ "fprintf(stderr, " ++ show str ++ "); fprintf(stderr, \"\\n\"); exit(-1);\n"
312-- bcc i c = error (show c) -- indent i ++ "// not done yet\n"
314-- Deconstruct the Foreign type in the defunctionalised expression and build
315-- a foreign type description for c_irts and irts_c
317 | i == sUN "C_IntChar" = ATInt ITChar
318 | i == sUN "C_IntNative" = ATInt ITNative
319 | i == sUN "C_IntBits8" = ATInt (ITFixed IT8)
320 | i == sUN "C_IntBits16" = ATInt (ITFixed IT16)
321 | i == sUN "C_IntBits32" = ATInt (ITFixed IT32)
322 | i == sUN "C_IntBits64" = ATInt (ITFixed IT64)
323toAType t = error (show t ++ " not defined in toAType")
326 | c == sUN "C_Str" = FString
327 | c == sUN "C_Float" = FArith ATFloat
328 | c == sUN "C_Ptr" = FPtr
329 | c == sUN "C_MPtr" = FManagedPtr
330 | c == sUN "C_Unit" = FUnit
331toFType (FApp c [_,ity])
332 | c == sUN "C_IntT" = FArith (toAType ity)
334 | c == sUN "C_Any" = FAny
337c_irts (FArith (ATInt ITNative)) l x = l ++ "MKINT((i_int)(" ++ x ++ "))"
338c_irts (FArith (ATInt ITChar)) l x = c_irts (FArith (ATInt ITNative)) l x
339c_irts (FArith (ATInt (ITFixed ity))) l x
340 = l ++ "idris_b" ++ show (nativeTyWidth ity) ++ "const(vm, " ++ x ++ ")"
341c_irts FString l x = l ++ "MKSTR(vm, " ++ x ++ ")"
343c_irts FPtr l x = l ++ "MKPTR(vm, " ++ x ++ ")"
344c_irts FManagedPtr l x = l ++ "MKMPTR(vm, " ++ x ++ ")"
345c_irts (FArith ATFloat) l x = l ++ "MKFLOAT(vm, " ++ x ++ ")"
346c_irts FAny l x = l ++ x
348irts_c (FArith (ATInt ITNative)) x = "GETINT(" ++ x ++ ")"
349irts_c (FArith (ATInt ITChar)) x = irts_c (FArith (ATInt ITNative)) x
350irts_c (FArith (ATInt (ITFixed ity))) x
351 = "(" ++ x ++ "->info.bits" ++ show (nativeTyWidth ity) ++ ")"
352irts_c FString x = "GETSTR(" ++ x ++ ")"
354irts_c FPtr x = "GETPTR(" ++ x ++ ")"
355irts_c FManagedPtr x = "GETMPTR(" ++ x ++ ")"
356irts_c (FArith ATFloat) x = "GETFLOAT(" ++ x ++ ")"
359bitOp v op ty args = v ++ "idris_b" ++ show (nativeTyWidth ty) ++ op ++ "(vm, " ++ intercalate ", " (map creg args) ++ ")"
361bitCoerce v op input output arg
362 = v ++ "idris_b" ++ show (nativeTyWidth input) ++ op ++ show (nativeTyWidth output) ++ "(vm, " ++ creg arg ++ ")"
364signedTy :: NativeTy -> String
365signedTy t = "int" ++ show (nativeTyWidth t) ++ "_t"
367doOp v (LPlus (ATInt ITNative)) [l, r] = v ++ "ADD(" ++ creg l ++ ", " ++ creg r ++ ")"
368doOp v (LMinus (ATInt ITNative)) [l, r] = v ++ "INTOP(-," ++ creg l ++ ", " ++ creg r ++ ")"
369doOp v (LTimes (ATInt ITNative)) [l, r] = v ++ "MULT(" ++ creg l ++ ", " ++ creg r ++ ")"
370doOp v (LUDiv ITNative) [l, r] = v ++ "UINTOP(/," ++ creg l ++ ", " ++ creg r ++ ")"
371doOp v (LSDiv (ATInt ITNative)) [l, r] = v ++ "INTOP(/," ++ creg l ++ ", " ++ creg r ++ ")"
372doOp v (LURem ITNative) [l, r] = v ++ "UINTOP(%," ++ creg l ++ ", " ++ creg r ++ ")"
373doOp v (LSRem (ATInt ITNative)) [l, r] = v ++ "INTOP(%," ++ creg l ++ ", " ++ creg r ++ ")"
374doOp v (LAnd ITNative) [l, r] = v ++ "INTOP(&," ++ creg l ++ ", " ++ creg r ++ ")"
375doOp v (LOr ITNative) [l, r] = v ++ "INTOP(|," ++ creg l ++ ", " ++ creg r ++ ")"
376doOp v (LXOr ITNative) [l, r] = v ++ "INTOP(^," ++ creg l ++ ", " ++ creg r ++ ")"
377doOp v (LSHL ITNative) [l, r] = v ++ "INTOP(<<," ++ creg l ++ ", " ++ creg r ++ ")"
378doOp v (LLSHR ITNative) [l, r] = v ++ "UINTOP(>>," ++ creg l ++ ", " ++ creg r ++ ")"
379doOp v (LASHR ITNative) [l, r] = v ++ "INTOP(>>," ++ creg l ++ ", " ++ creg r ++ ")"
380doOp v (LCompl ITNative) [x] = v ++ "INTOP(~," ++ creg x ++ ")"
381doOp v (LEq (ATInt ITNative)) [l, r] = v ++ "INTOP(==," ++ creg l ++ ", " ++ creg r ++ ")"
382doOp v (LSLt (ATInt ITNative)) [l, r] = v ++ "INTOP(<," ++ creg l ++ ", " ++ creg r ++ ")"
383doOp v (LSLe (ATInt ITNative)) [l, r] = v ++ "INTOP(<=," ++ creg l ++ ", " ++ creg r ++ ")"
384doOp v (LSGt (ATInt ITNative)) [l, r] = v ++ "INTOP(>," ++ creg l ++ ", " ++ creg r ++ ")"
385doOp v (LSGe (ATInt ITNative)) [l, r] = v ++ "INTOP(>=," ++ creg l ++ ", " ++ creg r ++ ")"
386doOp v (LLt ITNative) [l, r] = v ++ "UINTOP(<," ++ creg l ++ ", " ++ creg r ++ ")"
387doOp v (LLe ITNative) [l, r] = v ++ "UINTOP(<=," ++ creg l ++ ", " ++ creg r ++ ")"
388doOp v (LGt ITNative) [l, r] = v ++ "UINTOP(>," ++ creg l ++ ", " ++ creg r ++ ")"
389doOp v (LGe ITNative) [l, r] = v ++ "UINTOP(>=," ++ creg l ++ ", " ++ creg r ++ ")"
391doOp v (LPlus (ATInt ITChar)) [l, r] = doOp v (LPlus (ATInt ITNative)) [l, r]
392doOp v (LMinus (ATInt ITChar)) [l, r] = doOp v (LMinus (ATInt ITNative)) [l, r]
393doOp v (LTimes (ATInt ITChar)) [l, r] = doOp v (LTimes (ATInt ITNative)) [l, r]
394doOp v (LUDiv ITChar) [l, r] = doOp v (LUDiv ITNative) [l, r]
395doOp v (LSDiv (ATInt ITChar)) [l, r] = doOp v (LSDiv (ATInt ITNative)) [l, r]
396doOp v (LURem ITChar) [l, r] = doOp v (LURem ITNative) [l, r]
397doOp v (LSRem (ATInt ITChar)) [l, r] = doOp v (LSRem (ATInt ITNative)) [l, r]
398doOp v (LAnd ITChar) [l, r] = doOp v (LAnd ITNative) [l, r]
399doOp v (LOr ITChar) [l, r] = doOp v (LOr ITNative) [l, r]
400doOp v (LXOr ITChar) [l, r] = doOp v (LXOr ITNative) [l, r]
401doOp v (LSHL ITChar) [l, r] = doOp v (LSHL ITNative) [l, r]
402doOp v (LLSHR ITChar) [l, r] = doOp v (LLSHR ITNative) [l, r]
403doOp v (LASHR ITChar) [l, r] = doOp v (LASHR ITNative) [l, r]
404doOp v (LCompl ITChar) [x] = doOp v (LCompl ITNative) [x]
405doOp v (LEq (ATInt ITChar)) [l, r] = doOp v (LEq (ATInt ITNative)) [l, r]
406doOp v (LSLt (ATInt ITChar)) [l, r] = doOp v (LSLt (ATInt ITNative)) [l, r]
407doOp v (LSLe (ATInt ITChar)) [l, r] = doOp v (LSLe (ATInt ITNative)) [l, r]
408doOp v (LSGt (ATInt ITChar)) [l, r] = doOp v (LSGt (ATInt ITNative)) [l, r]
409doOp v (LSGe (ATInt ITChar)) [l, r] = doOp v (LSGe (ATInt ITNative)) [l, r]
410doOp v (LLt ITChar) [l, r] = doOp v (LLt ITNative) [l, r]
411doOp v (LLe ITChar) [l, r] = doOp v (LLe ITNative) [l, r]
412doOp v (LGt ITChar) [l, r] = doOp v (LGt ITNative) [l, r]
413doOp v (LGe ITChar) [l, r] = doOp v (LGe ITNative) [l, r]
415doOp v (LPlus ATFloat) [l, r] = v ++ "FLOATOP(+," ++ creg l ++ ", " ++ creg r ++ ")"
416doOp v (LMinus ATFloat) [l, r] = v ++ "FLOATOP(-," ++ creg l ++ ", " ++ creg r ++ ")"
417doOp v (LTimes ATFloat) [l, r] = v ++ "FLOATOP(*," ++ creg l ++ ", " ++ creg r ++ ")"
418doOp v (LSDiv ATFloat) [l, r] = v ++ "FLOATOP(/," ++ creg l ++ ", " ++ creg r ++ ")"
419doOp v (LEq ATFloat) [l, r] = v ++ "FLOATBOP(==," ++ creg l ++ ", " ++ creg r ++ ")"
420doOp v (LSLt ATFloat) [l, r] = v ++ "FLOATBOP(<," ++ creg l ++ ", " ++ creg r ++ ")"
421doOp v (LSLe ATFloat) [l, r] = v ++ "FLOATBOP(<=," ++ creg l ++ ", " ++ creg r ++ ")"
422doOp v (LSGt ATFloat) [l, r] = v ++ "FLOATBOP(>," ++ creg l ++ ", " ++ creg r ++ ")"
423doOp v (LSGe ATFloat) [l, r] = v ++ "FLOATBOP(>=," ++ creg l ++ ", " ++ creg r ++ ")"
425doOp v (LIntFloat ITBig) [x] = v ++ "idris_castBigFloat(vm, " ++ creg x ++ ")"
426doOp v (LFloatInt ITBig) [x] = v ++ "idris_castFloatBig(vm, " ++ creg x ++ ")"
427doOp v (LPlus (ATInt ITBig)) [l, r] = v ++ "idris_bigPlus(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
428doOp v (LMinus (ATInt ITBig)) [l, r] = v ++ "idris_bigMinus(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
429doOp v (LTimes (ATInt ITBig)) [l, r] = v ++ "idris_bigTimes(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
430doOp v (LSDiv (ATInt ITBig)) [l, r] = v ++ "idris_bigDivide(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
431doOp v (LSRem (ATInt ITBig)) [l, r] = v ++ "idris_bigMod(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
432doOp v (LEq (ATInt ITBig)) [l, r] = v ++ "idris_bigEq(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
433doOp v (LSLt (ATInt ITBig)) [l, r] = v ++ "idris_bigLt(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
434doOp v (LSLe (ATInt ITBig)) [l, r] = v ++ "idris_bigLe(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
435doOp v (LSGt (ATInt ITBig)) [l, r] = v ++ "idris_bigGt(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
436doOp v (LSGe (ATInt ITBig)) [l, r] = v ++ "idris_bigGe(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
438doOp v (LIntFloat ITNative) [x] = v ++ "idris_castIntFloat(" ++ creg x ++ ")"
439doOp v (LFloatInt ITNative) [x] = v ++ "idris_castFloatInt(" ++ creg x ++ ")"
440doOp v (LSExt ITNative ITBig) [x] = v ++ "idris_castIntBig(vm, " ++ creg x ++ ")"
441doOp v (LTrunc ITBig ITNative) [x] = v ++ "idris_castBigInt(vm, " ++ creg x ++ ")"
442doOp v (LStrInt ITBig) [x] = v ++ "idris_castStrBig(vm, " ++ creg x ++ ")"
443doOp v (LIntStr ITBig) [x] = v ++ "idris_castBigStr(vm, " ++ creg x ++ ")"
444doOp v (LIntStr ITNative) [x] = v ++ "idris_castIntStr(vm, " ++ creg x ++ ")"
445doOp v (LStrInt ITNative) [x] = v ++ "idris_castStrInt(vm, " ++ creg x ++ ")"
446doOp v (LIntStr (ITFixed _)) [x] = v ++ "idris_castBitsStr(vm, " ++ creg x ++ ")"
447doOp v LFloatStr [x] = v ++ "idris_castFloatStr(vm, " ++ creg x ++ ")"
448doOp v LStrFloat [x] = v ++ "idris_castStrFloat(vm, " ++ creg x ++ ")"
450doOp v (LSLt (ATInt (ITFixed ty))) [x, y] = bitOp v "SLt" ty [x, y]
451doOp v (LSLe (ATInt (ITFixed ty))) [x, y] = bitOp v "SLte" ty [x, y]
452doOp v (LEq (ATInt (ITFixed ty))) [x, y] = bitOp v "Eq" ty [x, y]
453doOp v (LSGe (ATInt (ITFixed ty))) [x, y] = bitOp v "SGte" ty [x, y]
454doOp v (LSGt (ATInt (ITFixed ty))) [x, y] = bitOp v "SGt" ty [x, y]
455doOp v (LLt (ITFixed ty)) [x, y] = bitOp v "Lt" ty [x, y]
456doOp v (LLe (ITFixed ty)) [x, y] = bitOp v "Lte" ty [x, y]
457doOp v (LGe (ITFixed ty)) [x, y] = bitOp v "Gte" ty [x, y]
458doOp v (LGt (ITFixed ty)) [x, y] = bitOp v "Gt" ty [x, y]
460doOp v (LSHL (ITFixed ty)) [x, y] = bitOp v "Shl" ty [x, y]
461doOp v (LLSHR (ITFixed ty)) [x, y] = bitOp v "LShr" ty [x, y]
462doOp v (LASHR (ITFixed ty)) [x, y] = bitOp v "AShr" ty [x, y]
463doOp v (LAnd (ITFixed ty)) [x, y] = bitOp v "And" ty [x, y]
464doOp v (LOr (ITFixed ty)) [x, y] = bitOp v "Or" ty [x, y]
465doOp v (LXOr (ITFixed ty)) [x, y] = bitOp v "Xor" ty [x, y]
466doOp v (LCompl (ITFixed ty)) [x] = bitOp v "Compl" ty [x]
468doOp v (LPlus (ATInt (ITFixed ty))) [x, y] = bitOp v "Plus" ty [x, y]
469doOp v (LMinus (ATInt (ITFixed ty))) [x, y] = bitOp v "Minus" ty [x, y]
470doOp v (LTimes (ATInt (ITFixed ty))) [x, y] = bitOp v "Times" ty [x, y]
471doOp v (LUDiv (ITFixed ty)) [x, y] = bitOp v "UDiv" ty [x, y]
472doOp v (LSDiv (ATInt (ITFixed ty))) [x, y] = bitOp v "SDiv" ty [x, y]
473doOp v (LURem (ITFixed ty)) [x, y] = bitOp v "URem" ty [x, y]
474doOp v (LSRem (ATInt (ITFixed ty))) [x, y] = bitOp v "SRem" ty [x, y]
476doOp v (LSExt (ITFixed from) ITBig) [x]
477 = v ++ "MKBIGSI(vm, (" ++ signedTy from ++ ")" ++ creg x ++ "->info.bits" ++ show (nativeTyWidth from) ++ ")"
478doOp v (LSExt ITNative (ITFixed to)) [x]
479 = v ++ "idris_b" ++ show (nativeTyWidth to) ++ "const(vm, GETINT(" ++ creg x ++ "))"
480doOp v (LSExt ITChar (ITFixed to)) [x]
481 = doOp v (LSExt ITNative (ITFixed to)) [x]
482doOp v (LSExt (ITFixed from) ITNative) [x]
483 = v ++ "MKINT((i_int)((" ++ signedTy from ++ ")" ++ creg x ++ "->info.bits" ++ show (nativeTyWidth from) ++ "))"
484doOp v (LSExt (ITFixed from) ITChar) [x]
485 = doOp v (LSExt (ITFixed from) ITNative) [x]
486doOp v (LSExt (ITFixed from) (ITFixed to)) [x]
487 | nativeTyWidth from < nativeTyWidth to = bitCoerce v "S" from to x
488doOp v (LZExt ITNative (ITFixed to)) [x]
489 = v ++ "idris_b" ++ show (nativeTyWidth to) ++ "const(vm, (uintptr_t)GETINT(" ++ creg x ++ "))"
490doOp v (LZExt ITChar (ITFixed to)) [x]
491 = doOp v (LZExt ITNative (ITFixed to)) [x]
492doOp v (LZExt (ITFixed from) ITNative) [x]
493 = v ++ "MKINT((i_int)" ++ creg x ++ "->info.bits" ++ show (nativeTyWidth from) ++ ")"
494doOp v (LZExt (ITFixed from) ITChar) [x]
495 = doOp v (LZExt (ITFixed from) ITNative) [x]
496doOp v (LZExt (ITFixed from) ITBig) [x]
497 = v ++ "MKBIGUI(vm, " ++ creg x ++ "->info.bits" ++ show (nativeTyWidth from) ++ ")"
498doOp v (LZExt ITNative ITBig) [x]
499 = v ++ "MKBIGUI(vm, (uintptr_t)GETINT(" ++ creg x ++ "))"
500doOp v (LZExt (ITFixed from) (ITFixed to)) [x]
501 | nativeTyWidth from < nativeTyWidth to = bitCoerce v "Z" from to x
502doOp v (LTrunc ITNative (ITFixed to)) [x]
503 = v ++ "idris_b" ++ show (nativeTyWidth to) ++ "const(vm, GETINT(" ++ creg x ++ "))"
504doOp v (LTrunc ITChar (ITFixed to)) [x]
505 = doOp v (LTrunc ITNative (ITFixed to)) [x]
506doOp v (LTrunc (ITFixed from) ITNative) [x]
507 = v ++ "MKINT((i_int)" ++ creg x ++ "->info.bits" ++ show (nativeTyWidth from) ++ ")"
508doOp v (LTrunc (ITFixed from) ITChar) [x]
509 = doOp v (LTrunc (ITFixed from) ITNative) [x]
510doOp v (LTrunc ITBig (ITFixed to)) [x]
511 = v ++ "idris_b" ++ show (nativeTyWidth to) ++ "const(vm, ISINT(" ++ creg x ++ ") ? GETINT(" ++ creg x ++ ") : mpz_get_ui(GETMPZ(" ++ creg x ++ ")))"
512doOp v (LTrunc (ITFixed from) (ITFixed to)) [x]
513 | nativeTyWidth from > nativeTyWidth to = bitCoerce v "T" from to x
515doOp v LFExp [x] = v ++ flUnOp "exp" (creg x)
516doOp v LFLog [x] = v ++ flUnOp "log" (creg x)
517doOp v LFSin [x] = v ++ flUnOp "sin" (creg x)
518doOp v LFCos [x] = v ++ flUnOp "cos" (creg x)
519doOp v LFTan [x] = v ++ flUnOp "tan" (creg x)
520doOp v LFASin [x] = v ++ flUnOp "asin" (creg x)
521doOp v LFACos [x] = v ++ flUnOp "acos" (creg x)
522doOp v LFATan [x] = v ++ flUnOp "atan" (creg x)
523doOp v LFSqrt [x] = v ++ flUnOp "sqrt" (creg x)
524doOp v LFFloor [x] = v ++ flUnOp "floor" (creg x)
525doOp v LFCeil [x] = v ++ flUnOp "ceil" (creg x)
526doOp v LFNegate [x] = v ++ "MKFLOAT(vm, -GETFLOAT(" ++ (creg x) ++ "))"
528-- String functions which don't need to know we're UTF8
529doOp v LStrConcat [l,r] = v ++ "idris_concat(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
530doOp v LStrLt [l,r] = v ++ "idris_strlt(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
531doOp v LStrEq [l,r] = v ++ "idris_streq(vm, " ++ creg l ++ ", " ++ creg r ++ ")"
533doOp v LReadStr [_] = v ++ "idris_readStr(vm, stdin)"
534doOp v LWriteStr [_,s]
535 = v ++ "MKINT((i_int)(idris_writeStr(stdout"
540-- String functions which need to know we're UTF8
541doOp v LStrHead [x] = v ++ "idris_strHead(vm, " ++ creg x ++ ")"
542doOp v LStrTail [x] = v ++ "idris_strTail(vm, " ++ creg x ++ ")"
543doOp v LStrCons [x, y] = v ++ "idris_strCons(vm, " ++ creg x ++ "," ++ creg y ++ ")"
544doOp v LStrIndex [x, y] = v ++ "idris_strIndex(vm, " ++ creg x ++ "," ++ creg y ++ ")"
545doOp v LStrRev [x] = v ++ "idris_strRev(vm, " ++ creg x ++ ")"
546doOp v LStrLen [x] = v ++ "idris_strlen(vm, " ++ creg x ++ ")"
548doOp v LFork [x] = v ++ "MKPTR(vm, vmThread(vm, " ++ cname (sMN 0 "EVAL") ++ ", " ++ creg x ++ "))"
549doOp v LPar [x] = v ++ creg x -- "MKPTR(vm, vmThread(vm, " ++ cname (MN 0 "EVAL") ++ ", " ++ creg x ++ "))"
550doOp v (LChInt ITNative) args = v ++ creg (last args)
551doOp v (LChInt ITChar) args = doOp v (LChInt ITNative) args
552doOp v (LIntCh ITNative) args = v ++ creg (last args)
553doOp v (LIntCh ITChar) args = doOp v (LIntCh ITNative) args
555doOp v LSystemInfo [x] = v ++ "idris_systemInfo(vm, " ++ creg x ++ ")"
556doOp v LNoOp args = v ++ creg (last args)
558-- Pointer primitives (declared as %extern in Builtins.idr)
559doOp v (LExternal rf) [_,x]
560 | rf == sUN "prim__readFile"
561 = v ++ "idris_readStr(vm, GETPTR(" ++ creg x ++ "))"
562doOp v (LExternal wf) [_,x,s]
563 | wf == sUN "prim__writeFile"
564 = v ++ "MKINT((i_int)(idris_writeStr(GETPTR(" ++ creg x
567doOp v (LExternal vm) [] | vm == sUN "prim__vm" = v ++ "MKPTR(vm, vm)"
568doOp v (LExternal si) [] | si == sUN "prim__stdin" = v ++ "MKPTR(vm, stdin)"
569doOp v (LExternal so) [] | so == sUN "prim__stdout" = v ++ "MKPTR(vm, stdout)"
570doOp v (LExternal se) [] | se == sUN "prim__stderr" = v ++ "MKPTR(vm, stderr)"
572doOp v (LExternal nul) [] | nul == sUN "prim__null" = v ++ "MKPTR(vm, NULL)"
573doOp v (LExternal eqp) [x, y] | eqp == sUN "prim__eqPtr"
574 = v ++ "MKINT((i_int)(GETPTR(" ++ creg x ++ ") == GETPTR(" ++ creg y ++ ")))"
575doOp v (LExternal eqp) [x, y] | eqp == sUN "prim__eqManagedPtr"
576 = v ++ "MKINT((i_int)(GETMPTR(" ++ creg x ++ ") == GETMPTR(" ++ creg y ++ ")))"
577doOp v (LExternal rp) [p, i] | rp == sUN "prim__registerPtr"
578 = v ++ "MKMPTR(vm, GETPTR(" ++ creg p ++ "), GETINT(" ++ creg i ++ "))"
580doOp _ op args = error $ "doOp not implemented (" ++ show (op, args) ++ ")"
582flUnOp :: String -> String -> String
583flUnOp name val = "MKFLOAT(vm, " ++ name ++ "(GETFLOAT(" ++ val ++ ")))"
585-------------------- Interface file generation
587-- First, the wrappers in the C file
589ifaceC :: Export -> String
590ifaceC (ExportData n) = "typedef VAL " ++ cdesc n ++ ";\n"
591ifaceC (ExportFun n cn ret args)
592 = ctype ret ++ " " ++ cdesc cn ++
593 "(VM* vm" ++ showArgs (zip argNames args) ++ ") {\n"
594 ++ mkBody n (zip argNames args) ret ++ "}\n\n"
595 where showArgs [] = ""
596 showArgs ((n, t) : ts) = ", " ++ ctype t ++ " " ++ n ++
599 argNames = zipWith (++) (repeat "arg") (map show [0..])
601mkBody n as t = indent 1 ++ "INITFRAME;\n" ++
602 indent 1 ++ "RESERVE(" ++ show (max (length as) 3) ++ ");\n" ++
603 push 0 as ++ call n ++ retval t
605 push i ((n, t) : ts) = indent 1 ++ c_irts (toFType t)
606 ("TOP(" ++ show i ++ ") = ") n
607 ++ ";\n" ++ push (i + 1) ts
609 call _ = indent 1 ++ "STOREOLD;\n" ++
610 indent 1 ++ "BASETOP(0);\n" ++
611 indent 1 ++ "ADDTOP(" ++ show (length as) ++ ");\n" ++
612 indent 1 ++ "CALL(" ++ cname n ++ ");\n"
615 = indent 1 ++ "TOP(0) = NULL;\n" ++
616 indent 1 ++ "TOP(1) = NULL;\n" ++
617 indent 1 ++ "TOP(2) = RVAL;\n" ++
618 indent 1 ++ "STOREOLD;\n" ++
619 indent 1 ++ "BASETOP(0);\n" ++
620 indent 1 ++ "ADDTOP(3);\n" ++
621 indent 1 ++ "CALL(" ++ cname (sUN "call__IO") ++ ");\n" ++
623 retval t = indent 1 ++ "return " ++ irts_c (toFType t) "RVAL" ++ ";\n"
626 | c == sUN "C_Str" = "char*"
627 | c == sUN "C_Float" = "float"
628 | c == sUN "C_Ptr" = "void*"
629 | c == sUN "C_MPtr" = "void*"
630 | c == sUN "C_Unit" = "void"
631ctype (FApp c [_,ity])
632 | c == sUN "C_IntT" = carith ity
634 | c == sUN "C_Any" = "VAL"
636ctype FUnknown = "void*"
637ctype (FIO t) = ctype t
638ctype t = error "Can't happen: Not a valid interface type " ++ show t
641 | i == sUN "C_IntChar" = "char"
642 | i == sUN "C_IntNative" = "int"
643carith t = error "Can't happen: Not an exportable arithmetic type"
646cdesc s = error "Can't happen: Not a valid C name"
648-- Then, the header files
650codegenH :: [ExportIFace] -> IO ()
651codegenH es = mapM_ writeIFace es
653writeIFace :: ExportIFace -> IO ()
654writeIFace (Export ffic hdr exps)
655 | ffic == sNS (sUN "FFI_C") ["FFI_C"]
656 = do let hfile = "#ifndef " ++ hdr_guard hdr ++ "\n" ++
657 "#define " ++ hdr_guard hdr ++ "\n\n" ++
658 "#include <idris_rts.h>\n\n" ++
659 concatMap hdr_export exps ++ "\n" ++
662 | otherwise = return ()
664hdr_guard x = "__" ++ map hchar x
665 where hchar x | isAlphaNum x = toUpper x
668hdr_export :: Export -> String
669hdr_export (ExportData n) = "typedef VAL " ++ cdesc n ++ ";\n"
670hdr_export (ExportFun n cn ret args)
671 = ctype ret ++ " " ++ cdesc cn ++
672 "(VM* vm" ++ showArgs (zip argNames args) ++ ");\n"
673 where showArgs [] = ""
674 showArgs ((n, t) : ts) = ", " ++ ctype t ++ " " ++ n ++
677 argNames = zipWith (++) (repeat "arg") (map show [0..])
keysesc to close
- j k
- move down, up
- enter
- open
- esc
- back, or up a level
- 1 2 3 4
- readme, files, commits, refs
- /
- filter
- y
- copy the clone command
- c
- compare the chosen commit with another (or shift-click both)
- ?
- show or hide this list