forked from M-Labs/artiq
1
0
Fork 0
artiq/artiq/compiler/builtins.py

186 lines
4.0 KiB
Python
Raw Normal View History

"""
2015-06-24 16:46:15 +08:00
The :mod:`builtins` module contains the builtin Python
and ARTIQ types, such as int or float.
"""
from collections import OrderedDict
from . import types
# Types
class TNone(types.TMono):
def __init__(self):
super().__init__("NoneType")
class TBool(types.TMono):
def __init__(self):
super().__init__("bool")
@staticmethod
def zero():
return False
@staticmethod
def one():
return True
class TInt(types.TMono):
def __init__(self, width=None):
if width is None:
width = types.TVar()
super().__init__("int", {"width": width})
@staticmethod
def zero():
return 0
@staticmethod
def one():
return 1
class TFloat(types.TMono):
def __init__(self):
super().__init__("float")
@staticmethod
def zero():
return 0.0
@staticmethod
def one():
return 1.0
2015-07-21 19:27:48 +08:00
class TStr(types.TMono):
def __init__(self):
super().__init__("str")
class TList(types.TMono):
def __init__(self, elt=None):
if elt is None:
elt = types.TVar()
super().__init__("list", {"elt": elt})
2015-06-26 23:53:20 +08:00
class TRange(types.TMono):
def __init__(self, elt=None):
if elt is None:
elt = types.TVar()
super().__init__("range", {"elt": elt})
self.attributes = OrderedDict([
("start", elt),
("stop", elt),
("step", elt),
])
2015-06-26 23:53:20 +08:00
2015-06-29 03:48:15 +08:00
class TException(types.TMono):
def __init__(self, name="Exception"):
super().__init__(name)
2015-07-17 21:05:02 +08:00
class TIndexError(TException):
def __init__(self):
super().__init__("IndexError")
2015-07-17 21:05:02 +08:00
class TValueError(TException):
2015-06-29 03:48:15 +08:00
def __init__(self):
super().__init__("ValueError")
2015-06-29 03:48:15 +08:00
2015-06-26 16:16:08 +08:00
def fn_bool():
2015-06-29 03:40:57 +08:00
return types.TConstructor("bool")
2015-06-26 16:16:08 +08:00
def fn_int():
2015-06-29 03:40:57 +08:00
return types.TConstructor("int")
2015-06-26 16:16:08 +08:00
def fn_float():
2015-06-29 03:40:57 +08:00
return types.TConstructor("float")
2015-06-26 16:16:08 +08:00
2015-07-21 19:27:48 +08:00
def fn_str():
return types.TConstructor("str")
2015-06-26 16:16:08 +08:00
def fn_list():
2015-06-29 03:40:57 +08:00
return types.TConstructor("list")
2015-06-26 16:16:08 +08:00
2015-06-29 03:48:15 +08:00
def fn_Exception():
return types.TExceptionConstructor("Exception")
def fn_IndexError():
return types.TExceptionConstructor("IndexError")
def fn_ValueError():
return types.TExceptionConstructor("ValueError")
2015-06-26 23:53:20 +08:00
def fn_range():
2015-06-29 03:40:57 +08:00
return types.TBuiltinFunction("range")
2015-06-26 23:53:20 +08:00
def fn_len():
2015-06-29 03:40:57 +08:00
return types.TBuiltinFunction("len")
def fn_round():
2015-06-29 03:40:57 +08:00
return types.TBuiltinFunction("round")
2015-07-22 03:32:10 +08:00
def fn_print():
return types.TBuiltinFunction("print")
def fn_syscall():
2015-06-29 03:40:57 +08:00
return types.TBuiltinFunction("syscall")
# Accessors
2015-06-14 17:07:13 +08:00
def is_none(typ):
return types.is_mono(typ, "NoneType")
def is_bool(typ):
return types.is_mono(typ, "bool")
2015-06-13 18:45:09 +08:00
def is_int(typ, width=None):
2015-06-29 03:48:15 +08:00
if width is not None:
2015-06-13 18:45:09 +08:00
return types.is_mono(typ, "int", {"width": width})
else:
return types.is_mono(typ, "int")
2015-06-14 17:07:13 +08:00
def get_int_width(typ):
if is_int(typ):
2015-06-15 13:40:37 +08:00
return types.get_value(typ.find()["width"])
2015-06-14 17:07:13 +08:00
def is_float(typ):
return types.is_mono(typ, "float")
2015-07-21 19:27:48 +08:00
def is_str(typ):
return types.is_mono(typ, "str")
def is_numeric(typ):
2015-06-14 17:07:13 +08:00
typ = typ.find()
return isinstance(typ, types.TMono) and \
typ.name in ('int', 'float')
2015-06-14 17:07:13 +08:00
def is_list(typ, elt=None):
2015-06-29 03:48:15 +08:00
if elt is not None:
2015-06-14 17:07:13 +08:00
return types.is_mono(typ, "list", {"elt": elt})
else:
return types.is_mono(typ, "list")
2015-06-26 23:53:20 +08:00
def is_range(typ, elt=None):
2015-06-29 03:48:15 +08:00
if elt is not None:
2015-06-26 23:53:20 +08:00
return types.is_mono(typ, "range", {"elt": elt})
else:
return types.is_mono(typ, "range")
2015-07-17 21:05:02 +08:00
def is_exception(typ):
return isinstance(typ.find(), TException)
2015-06-26 23:53:20 +08:00
def is_iterable(typ):
typ = typ.find()
return isinstance(typ, types.TMono) and \
typ.name in ('list', 'range')
def get_iterable_elt(typ):
if is_iterable(typ):
2015-07-17 21:05:02 +08:00
return typ.find()["elt"].find()
2015-06-26 23:53:20 +08:00
2015-06-14 17:07:13 +08:00
def is_collection(typ):
typ = typ.find()
return isinstance(typ, types.TTuple) or \
types.is_mono(typ, "list")
2015-06-24 16:46:15 +08:00
2015-07-21 19:27:48 +08:00
def is_allocated(typ):
2015-07-04 09:16:37 +08:00
return typ.fold(False, lambda accum, typ:
2015-07-21 19:27:48 +08:00
is_list(typ) or is_str(typ) or types.is_function(typ))