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

migen/fhdl/autofragment: factorize

This commit is contained in:
Sebastien Bourdeauducq 2013-03-09 23:23:24 +01:00
parent d0d2df3c4b
commit d0676e2dd1
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
migen/fhdl/autofragment.py
View File

@ -1,5 +1,14 @@
import collections
from migen.fhdl.structure import * from migen.fhdl.structure import *
from migen.fhdl.specials import Special from migen.fhdl.specials import Special
from migen.fhdl.tools import flat_iteration
def flat_list(e):
if isinstance(e, collections.Iterable):
return flat_iteration(e)
else:
return [e]
class _FModuleProxy: class _FModuleProxy:
def __init__(self, fm): def __init__(self, fm):
@ -7,10 +16,7 @@ class _FModuleProxy:
class _FModuleComb(_FModuleProxy): class _FModuleComb(_FModuleProxy):
def __iadd__(self, other): def __iadd__(self, other):
if isinstance(other, (list, tuple)): self._fm._fragment.comb += flat_list(other)
self._fm._fragment.comb += other
else:
self._fm._fragment.comb.append(other)
return self return self
def _cd_append(d, key, statements): def _cd_append(d, key, statements):
@ -19,10 +25,7 @@ def _cd_append(d, key, statements):
except KeyError: except KeyError:
l = [] l = []
d[key] = l d[key] = l
if isinstance(statements, (list, tuple)): l += flat_list(statements)
l += statements
else:
l.append(statements)
class _FModuleSyncCD: class _FModuleSyncCD:
def __init__(self, fm, cd): def __init__(self, fm, cd):
@ -47,18 +50,12 @@ class _FModuleSync(_FModuleProxy):
class _FModuleSpecials(_FModuleProxy): class _FModuleSpecials(_FModuleProxy):
def __iadd__(self, other): def __iadd__(self, other):
if isinstance(other, (set, list, tuple)): self._fm._fragment.specials |= set(flat_list(other))
self._fm._fragment.specials |= set(other)
else:
self._fm._fragment.specials.add(other)
return self return self
class _FModuleSubmodules(_FModuleProxy): class _FModuleSubmodules(_FModuleProxy):
def __iadd__(self, other): def __iadd__(self, other):
if isinstance(other, (list, tuple)): self._fm._submodules += flat_list(other)
self._fm._submodules += other
else:
self._fm._submodules.append(other)
return self return self
class FModule: class FModule: