|
| 1 | +import sys |
| 2 | +import os |
| 3 | +import math |
| 4 | + |
| 5 | +from veriloggen import * |
| 6 | + |
| 7 | +try: |
| 8 | + log2 = math.log2 |
| 9 | +except: |
| 10 | + def log2(v): |
| 11 | + return math.log(v, 2) |
| 12 | + |
| 13 | +def mkLed(numports=4): |
| 14 | + if log2(numports) % 1.0 != 0.0: |
| 15 | + raise ValueError('numports must be power of 2') |
| 16 | + |
| 17 | + m = Module('blinkled') |
| 18 | + clk = m.Input('CLK') |
| 19 | + rst = m.Input('RST') |
| 20 | + |
| 21 | + idata = [ m.Input('idata' + str(i), 32) for i in range(numports) ] |
| 22 | + ivalid = [ m.Input('ivalid' + str(i)) for i in range(numports) ] |
| 23 | + odata = m.OutputReg('odata', 32, initval=0) |
| 24 | + ovalid = m.OutputReg('ovalid', initval=0) |
| 25 | + |
| 26 | + par = lib.Parallel(m, 'par') |
| 27 | + pdata = idata |
| 28 | + pvalid = ivalid |
| 29 | + ndata = [] |
| 30 | + nvalid = [] |
| 31 | + |
| 32 | + for s in range( int(log2(numports)) ): |
| 33 | + for i in range( numports >> (s+1) ): |
| 34 | + td = m.TmpReg(32, initval=0) |
| 35 | + tv = m.TmpReg(initval=0) |
| 36 | + ndata.append(td) |
| 37 | + nvalid.append(tv) |
| 38 | + cond = AndList(pvalid[i*2], pvalid[i*2+1]) |
| 39 | + par.add( tv(cond) ) |
| 40 | + par.add( td(pdata[i*2] + pdata[i*2+1]), cond=cond ) |
| 41 | + pdata = ndata |
| 42 | + pvalid = nvalid |
| 43 | + |
| 44 | + par.add( odata(pdata[-1]) ) |
| 45 | + par.add( ovalid(pvalid[-1]) ) |
| 46 | + |
| 47 | + par.make_always(clk, rst) |
| 48 | + |
| 49 | + return m |
| 50 | + |
| 51 | +def mkTest(): |
| 52 | + m = Module('test') |
| 53 | + |
| 54 | + # target instance |
| 55 | + led = mkLed() |
| 56 | + |
| 57 | + # copy paras and ports |
| 58 | + params = m.copy_params(led) |
| 59 | + ports = m.copy_sim_ports(led) |
| 60 | + |
| 61 | + clk = ports['CLK'] |
| 62 | + rst = ports['RST'] |
| 63 | + idata = [ p for k, p in sorted(ports.items(), key=lambda x:x[0]) if k.startswith('idata') ] |
| 64 | + ivalid = [ p for k, p in sorted(ports.items(), key=lambda x:x[0]) if k.startswith('ivalid') ] |
| 65 | + |
| 66 | + uut = m.Instance(led, 'uut', |
| 67 | + params=m.connect_params(led), |
| 68 | + ports=m.connect_ports(led)) |
| 69 | + |
| 70 | + reset_stmt = [] |
| 71 | + for d in idata: |
| 72 | + reset_stmt.append( d(0) ) |
| 73 | + for v in ivalid: |
| 74 | + reset_stmt.append( v(0) ) |
| 75 | + |
| 76 | + lib.simulation.setup_waveform(m, uut) |
| 77 | + lib.simulation.setup_clock(m, clk, hperiod=5) |
| 78 | + init = lib.simulation.setup_reset(m, rst, reset_stmt, period=100) |
| 79 | + |
| 80 | + nclk = lib.simulation.next_clock |
| 81 | + |
| 82 | + init.add( |
| 83 | + Delay(1000), |
| 84 | + nclk(clk), |
| 85 | + |
| 86 | + [ d(0) for d in idata ], |
| 87 | + [ v(0) for v in ivalid ], |
| 88 | + nclk(clk), |
| 89 | + |
| 90 | + [ d(i+1) for i, d in enumerate(idata) ], |
| 91 | + [ v(1) for v in ivalid ], |
| 92 | + nclk(clk), |
| 93 | + |
| 94 | + [ d(1) for d in idata ], |
| 95 | + [ v(0) for v in ivalid ], |
| 96 | + nclk(clk), |
| 97 | + |
| 98 | + [ d(i+10) for i, d in enumerate(idata) ], |
| 99 | + [ v(1) for v in ivalid ], |
| 100 | + nclk(clk), |
| 101 | + |
| 102 | + [ v(0) for v in ivalid ], |
| 103 | + [ nclk(clk) for _ in range(10) ], |
| 104 | + |
| 105 | + Systask('finish'), |
| 106 | + ) |
| 107 | + |
| 108 | + return m |
| 109 | + |
| 110 | +if __name__ == '__main__': |
| 111 | + test = mkTest() |
| 112 | + verilog = test.to_verilog('tmp.v') |
| 113 | + print(verilog) |
0 commit comments