forked from M-Labs/artiq
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from llvm import core as lc
|
|
from llvm import target as lt
|
|
|
|
lt.initialize_all()
|
|
|
|
_syscalls = [
|
|
("rpc", "i+:i"),
|
|
("gpio_set", "ii:v"),
|
|
("rtio_set", "iii:v"),
|
|
("rtio_sync", "i:v"),
|
|
("dds_program", "ii:v"),
|
|
]
|
|
|
|
def _str_to_functype(s):
|
|
_chr_to_type = {
|
|
"v": lc.Type.void,
|
|
"i": lc.Type.int
|
|
}
|
|
assert(s[-2] == ":")
|
|
type_ret = _chr_to_type[s[-1]]()
|
|
|
|
var_arg_fixcount = None
|
|
type_args = []
|
|
for n, c in enumerate(s[:-2]):
|
|
if c == "+":
|
|
type_args.append(lc.Type.int())
|
|
var_arg_fixcount = n
|
|
else:
|
|
type_args.append(_chr_to_type[c]())
|
|
return var_arg_fixcount, lc.Type.function(type_ret, type_args, var_arg=var_arg_fixcount is not None)
|
|
|
|
class LinkInterface:
|
|
def set_module(self, module):
|
|
self.var_arg_fixcount = dict()
|
|
for func_name, func_type_str in _syscalls:
|
|
var_arg_fixcount, func_type = _str_to_functype(func_type_str)
|
|
if var_arg_fixcount is not None:
|
|
self.var_arg_fixcount[func_name] = var_arg_fixcount
|
|
module.add_function(func_type, "__syscall_"+func_name)
|
|
|
|
self.module = module
|
|
|
|
def emit_syscall(self, builder, syscall_name, args):
|
|
if syscall_name in self.var_arg_fixcount:
|
|
fixcount = self.var_arg_fixcount[syscall_name]
|
|
args = args[:fixcount] \
|
|
+ [lc.Constant.int(lc.Type.int(), len(args) - fixcount)] \
|
|
+ args[fixcount:]
|
|
return builder.call(self.module.get_function_named("__syscall_"+syscall_name), args)
|
|
|
|
class Environment(LinkInterface):
|
|
def __init__(self, ref_period):
|
|
self.ref_period = ref_period
|
|
self.initial_time = 8000
|
|
|
|
def emit_object(self):
|
|
tm = lt.TargetMachine.new(triple="or1k", cpu="generic")
|
|
return tm.emit_object(self.module)
|