45 lines
843 B
Python
45 lines
843 B
Python
from migen import *
|
|
from migen.fhdl import verilog
|
|
|
|
from functools import reduce
|
|
from operator import or_
|
|
|
|
|
|
def gen_list(n):
|
|
s = [Signal() for i in range(n)]
|
|
return s
|
|
|
|
|
|
def gen_2list(n):
|
|
s = [Signal(2) for i in range(n)]
|
|
return s
|
|
|
|
|
|
class Foo:
|
|
def __init__(self):
|
|
la = gen_list(3)
|
|
lb = gen_2list(2)
|
|
self.sigs = la + lb
|
|
|
|
|
|
class Bar:
|
|
def __init__(self):
|
|
self.sigs = gen_list(2)
|
|
|
|
|
|
class Example(Module):
|
|
def __init__(self):
|
|
a = [Bar() for x in range(3)]
|
|
b = [Foo() for x in range(3)]
|
|
c = b
|
|
b = [Bar() for x in range(2)]
|
|
|
|
output = Signal()
|
|
allsigs = []
|
|
for lst in [a, b, c]:
|
|
for obj in lst:
|
|
allsigs.extend(obj.sigs)
|
|
self.comb += output.eq(reduce(or_, allsigs))
|
|
|
|
print(verilog.convert(Example()))
|