2255a7e5dd
git-svn-id: http://llvm-py.googlecode.com/svn/trunk@4 8d1e9007-1d4e-0410-b67e-1979fd6579aa
43 lines
1,010 B
Python
43 lines
1,010 B
Python
#!/usr/bin/env python
|
|
|
|
from llvm.core import *
|
|
|
|
## create a module
|
|
module = Module.new("my_module")
|
|
|
|
## create a function type taking two doubles and returning a (32-bit) integer
|
|
ty_double = Type.double()
|
|
ty_int = Type.int()
|
|
ty_func = Type.function( ty_int, [ ty_double, ty_double ] )
|
|
|
|
## create a function of this type
|
|
func = Function.new( module, ty_func, "foobar" )
|
|
|
|
# name function args
|
|
func.args[0].name = "arg1"
|
|
func.args[1].name = "arg2"
|
|
|
|
## implement the function
|
|
|
|
# add a basic block
|
|
entry = func.append_basic_block("entry")
|
|
|
|
# create an llvm::IRBuilder
|
|
builder = Builder.new()
|
|
builder.position_at_end(entry)
|
|
|
|
# add two args into tmp1
|
|
tmp1 = builder.add(func.args[0], func.args[1], "tmp1")
|
|
|
|
# sub `1' from that
|
|
one = Constant.real( ty_double, 1.0 )
|
|
tmp2 = builder.sub(tmp1, one, "tmp2")
|
|
|
|
# convert to integer
|
|
tmp3 = builder.fptoui(tmp2, ty_int, "tmp3")
|
|
|
|
# return it
|
|
builder.ret(tmp3)
|
|
|
|
# dump the module to see the llvm "assembly" code
|
|
print module
|