Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fhdl/visit: fix TransformModule

This commit is contained in:
Sebastien Bourdeauducq 2015-03-14 17:45:11 +01:00
parent aef9275c99
commit c824379878
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
migen/fhdl/module.py
View File

@ -83,8 +83,8 @@ class _ModuleClockDomains(_ModuleProxy, _ModuleForwardAttr):
class Module:
def get_fragment(self):
assert(not self._get_fragment_called)
self._get_fragment_called = True
assert(not self.get_fragment_called)
self.get_fragment_called = True
self.finalize()
return self._fragment
@ -127,9 +127,9 @@ class Module:
elif name == "_clock_domains":
self._clock_domains = []
return self._clock_domains
elif name == "_get_fragment_called":
self._get_fragment_called = False
return self._get_fragment_called
elif name == "get_fragment_called":
self.get_fragment_called = False
return self.get_fragment_called
else:
raise AttributeError("'"+self.__class__.__name__+"' object has no attribute '"+name+"'")
@ -144,7 +144,7 @@ class Module:
def _collect_submodules(self):
r = []
for name, submodule in self._submodules:
if not submodule._get_fragment_called:
if not submodule.get_fragment_called:
r.append((name, submodule.get_fragment()))
return r

3
migen/fhdl/visit.py
View File

@ -202,6 +202,9 @@ class TransformModule:
def __init__(self, transform, module):
self.transform = transform
self.module = module
self.get_fragment_called = False
def get_fragment(self):
assert(not self.get_fragment_called)
self.get_fragment_called = True
return self.transform(self.module.get_fragment())