module CodeGen.X86.Examples where
import Foreign
import CodeGen.X86
idCode = do
mov result arg1
ret
idFun :: Word64 -> Word64
idFun = compile idCode
fibCode = saveNonVolatile $ do
mov rdi arg1
inc rdi
xor_ rdx rdx
mov rax 1
doWhile NZ $ do
mov rcx rax
mov rax rdx
add rdx rcx
dec rdi
fibFun :: Word64 -> Word64
fibFun = compile fibCode
tracedFibCode = saveNonVolatile $ do
mov rdi arg1
inc rdi
xor_ rdx rdx
mov rax 1
doWhile NZ $ do
mov rcx rax
mov rax rdx
add rdx rcx
dec rdi
traceReg "d" rax
tracedFibFun :: Word64 -> Word64
tracedFibFun = compile tracedFibCode
callHsCode = do
callFun r11 (hsPtr fib)
ret
fib :: Word64 -> Word64
fib n = go n 0 1
where
go 0 a b = b `seq` a
go n a b = go (n1) b (a+b)
callHsFun :: Word64 -> Word64
callHsFun = compile callHsCode
callCCode name = saveNonVolatile $ do
leaData arg1 $ CString "Hello %s!\n"
leaData arg2 $ CString name
xor_ rax rax
callFun r11 printf
callCFun :: String -> IO ()
callCFun name = compile $ callCCode name
memTestFun :: Word64 -> IO Bool
memTestFun v = do
r <- mallocBytes 8
pokeByteOff r 0 (v :: Word64)
let code = saveNonVolatile $ do
mov rdi arg1
mov rax (addr rdi)
return $ compile code r == v