def __init__(self, func):
senslist = []
super(_AlwaysComb, self).__init__(func, senslist)
s = inspect.getsource(func)
s = _dedent(s)
tree = ast.parse(s)
# print ast.dump(tree)
v = _AttrRefTransformer(self)
v.visit(tree)
v = _SigNameVisitor(self.symdict)
v.visit(tree)
self.inputs = v.results['input']
self.outputs = v.results['output']
inouts = v.results['inout'] | self.inputs.intersection(self.outputs)
if inouts:
raise AlwaysCombError(_error.SignalAsInout % inouts)
if v.results['embedded_func']:
raise AlwaysCombError(_error.EmbeddedFunction)
for n in self.inputs:
s = self.symdict[n]
if isinstance(s, _Signal):
senslist.append(s)
elif _isListOfSigs(s):
senslist.extend(s)
self.senslist = tuple(senslist)
if len(self.senslist) == 0:
raise AlwaysCombError(_error.EmptySensitivityList)
def always_comb(func):
if not isinstance(func, FunctionType):
raise AlwaysCombError(_error.ArgType)
if _isGenFunc(func):
raise AlwaysCombError(_error.ArgType)
if func.__code__.co_argcount > 0:
raise AlwaysCombError(_error.NrOfArgs)
c = _AlwaysComb(func)
return c
def visit_FunctionDef(self, node):
if self.toplevel:
self.toplevel = 0 # skip embedded functions
for n in node.body:
self.visit(n)
else:
raise AlwaysCombError(_error.EmbeddedFunction)
def __init__(self, func, symdict):
self.func = func
self.symdict = symdict
s = inspect.getsource(func)
s = _dedent(s)
tree = ast.parse(s)
# print ast.dump(tree)
v = _AttrRefTransformer(self)
v.visit(tree)
v = _SigNameVisitor(self.symdict)
v.visit(tree)
self.inputs = v.inputs
self.outputs = v.outputs
senslist = []
for n in self.inputs:
s = self.symdict[n]
if isinstance(s, _Signal):
senslist.append(s)
else: # list of sigs
senslist.extend(s)
self.senslist = tuple(senslist)
self.gen = self.genfunc()
if len(self.senslist) == 0:
raise AlwaysCombError(_error.EmptySensitivityList)
if len(self.senslist) == 1:
W = _SignalWaiter
else:
W = _SignalTupleWaiter
self.waiter = W(self.gen)
def __init__(self, func, callinfo):
senslist = []
super(_AlwaysComb, self).__init__(func, senslist, callinfo=callinfo)
if self.embedded_func:
raise AlwaysCombError(_error.EmbeddedFunction)
for n in self.inputs:
s = self.symdict[n]
if isinstance(s, _Signal):
senslist.append(s)
elif _isListOfSigs(s):
senslist.extend(s)
self.senslist = tuple(senslist)
if len(self.senslist) == 0:
raise AlwaysCombError(_error.EmptySensitivityList)
def visit_Module(self, node):
inputs = self.inputs
outputs = self.outputs
for n in node.body:
self.visit(n)
for n in inputs:
if n in outputs:
raise AlwaysCombError(_error.SignalAsInout % n)
def always_comb(func):
if not isinstance(func, FunctionType):
raise AlwaysCombError(_error.ArgType)
if _isGenFunc(func):
raise AlwaysCombError(_error.ArgType)
if func.__code__.co_argcount > 0:
raise AlwaysCombError(_error.NrOfArgs)
varnames = func.__code__.co_varnames
symdict = {}
for n, v in func.__globals__.items():
if n not in varnames:
symdict[n] = v
# handle free variables
if func.__code__.co_freevars:
for n, c in zip(func.__code__.co_freevars, func.__closure__):
try:
obj = _cell_deref(c)
symdict[n] = obj
except NameError:
raise NameError(n)
c = _AlwaysComb(func, symdict)
return c
def visit_Name(self, node):
id = node.id
if id not in self.symdict:
return
s = self.symdict[id]
if isinstance(s, _Signal) or _isListOfSigs(s):
if self.context == INPUT:
self.inputs.add(id)
elif self.context == OUTPUT:
self.outputs.add(id)
elif self.context == INOUT:
raise AlwaysCombError(_error.SignalAsInout % id)
else:
raise AssertionError("bug in always_comb")
def __init__(self, func):
def senslistexpand(senslist, reg):
# print(repr(reg))
if isinstance(reg, _Signal):
senslist.append(reg)
elif isinstance(reg, StructType):
# print('senslistexpand StructType', repr(reg))
refs = vars(reg)
# print(refs)
for k in refs:
# print(refs[k])
if isinstance(refs[k], _Signal):
senslist.append(refs[k])
elif isinstance(refs[k], (list, Array)):
if len(refs[k]) != 0:
senslistexpand(senslist, refs[k])
elif isinstance(refs[k], StructType):
senslistexpand(senslist, refs[k])
else:
# print('senslistexpand passing {}?'.format(k))
pass
elif isinstance(reg, (list, Array)):
if isinstance(reg[0], (list, Array)):
for r in reg:
senslistexpand(senslist, r)
else:
if isinstance(reg[0], StructType):
for rr in reg:
senslistexpand(senslist, rr)
elif isinstance(reg[0], _Signal):
senslist.extend(reg)
# elif isinstance(reg, Array):
# senslist.extend(reg._array)
# elif isinstance(reg, list):
# senslist.extend(reg)
# self.func = func
# self.symdict = symdict
senslist = []
super(_AlwaysComb, self).__init__(func, senslist)
# print('_AlwaysComb', senslist)
s = inspect.getsource(func)
s = _dedent(s)
tree = ast.parse(s)
# print(ast.dump(tree))
# for item in ast.dump(tree).split():
# print(item)
v = _AttrRefTransformer(self)
v.visit(tree)
# print(ast.dump(tree))
# for item in ast.dump(tree).split():
# print(item)
v = _SigNameVisitor(self.symdict)
v.visit(tree)
# print(ast.dump(tree))
# for item in ast.dump(tree).split():
# print(item)
self.inputs = v.results['input']
self.outputs = v.results['output']
# print(v.results)
# print('inputs:', self.inputs)
# print('outputs:', self.outputs)
# inouts = v.results['inout'] | self.inputs.intersection(self.outputs)
# if inouts:
# raise AlwaysCombError(_error.SignalAsInout % inouts)
if v.results['embedded_func']:
raise AlwaysCombError(_error.EmbeddedFunction)
# print(self.inputs)
# print('2', self.senslist)
for n in self.inputs:
s = self.symdict[n]
# print(n,s)
# print(n, s, isinstance(s, StructType), isinstance(s, (list, Array)))
senslistexpand(senslist, s)
# if isinstance(s, _Signal):
# senslist.append(s)
# elif isinstance(s, list):
# # print(repr(s))
# # list or Array of sigs
# senslistexpand(senslist, s)
# elif isinstance(s, Array):
# # Array of sigs
# if not isinstance(s.element, intbv):
# senslistexpand(senslist, s)
# elif isinstance(s, StructType):
# senslistexpand(senslist, s)
# elif _isListOfSigs(s):
# senslist.extend(s)
# else :
# # print('_always_comb', n)
# pass
# print('3', self.senslist)
self.senslist = tuple(senslist)
# print('4', self.senslist)
self.gen = self.genfunc()
# for item in self.senslist:
# print(item._name, repr(item))
# print()
if len(self.senslist) == 0:
raise AlwaysCombError(_error.EmptySensitivityList)