315 lines
7.5 KiB
Python
315 lines
7.5 KiB
Python
from migen.fhdl.structure import *
|
|
from migen.fhdl.specials import Memory
|
|
from migen.bus import csr
|
|
from migen.bank import description, csrgen
|
|
from migen.bank.description import *
|
|
from migen.genlib.misc import optree
|
|
|
|
|
|
class RegParams:
|
|
def __init__(self, name, base, width, nb):
|
|
self.name = name
|
|
self.base = base
|
|
self.width = width
|
|
self.nb = nb
|
|
|
|
self.size = nb*width
|
|
self.words = int(2**bits_for(self.width-1)/8)
|
|
|
|
def list_regs(objects):
|
|
r = []
|
|
for object in objects:
|
|
if "_reg" in object:
|
|
r.append(objects[object])
|
|
return r
|
|
|
|
class Term:
|
|
#
|
|
# Definition
|
|
#
|
|
def __init__(self, width):
|
|
self.width = width
|
|
self.interface = None
|
|
|
|
self.i = Signal(width)
|
|
self.t = Signal(width)
|
|
self.m = Signal(width)
|
|
self.o = Signal()
|
|
|
|
self.reg_p = RegParams("term_reg", 0, width, 2)
|
|
self.reg = None
|
|
|
|
def get_registers_comb(self):
|
|
comb = [self.t.eq(self.reg.storage[0*self.width:1*self.width])]
|
|
comb += [self.m.eq(self.reg.storage[1*self.width:2*self.width])]
|
|
return comb
|
|
|
|
def get_fragment(self):
|
|
comb = [self.o.eq((self.m & self.i) == self.t)]
|
|
comb += self.get_registers_comb()
|
|
return Fragment(comb)
|
|
|
|
#
|
|
# Driver
|
|
#
|
|
def set(self, dat, mask=None):
|
|
if mask is None:
|
|
mask = (2**self.width)-1
|
|
self.interface.write_n(self.reg_p.base, mask, self.width)
|
|
self.interface.write_n(self.reg_p.base + self.reg_p.words, dat, self.width)
|
|
|
|
class RangeDetector:
|
|
#
|
|
# Definition
|
|
#
|
|
def __init__(self, width):
|
|
self.width = width
|
|
self.pipe = pipe
|
|
self.interface = None
|
|
|
|
self.reg_p = RegParams("range_reg", 0, width, 2)
|
|
self.reg = None
|
|
|
|
self.i = Signal(width)
|
|
self.low = Signal(width)
|
|
self.high = Signal(width)
|
|
self.o = Signal()
|
|
|
|
def get_registers_comb(self):
|
|
comb = [self.low.eq(self.reg.storage[0*self.width:1*self.width])]
|
|
comb += [self.low.eq(self.reg.storage[1*self.width:2*self.width])]
|
|
return comb
|
|
|
|
def get_fragment(self):
|
|
comb = [self.o.eq((self.i >= self.low) & (self.i <= self.high))]
|
|
comb += self.get_registers_comb()
|
|
return Fragment(comb)
|
|
#
|
|
# Driver
|
|
#
|
|
def set_low(self, dat):
|
|
self.interface.write_n(self.reg_p.base, dat ,self.width)
|
|
|
|
def set_high(self, dat):
|
|
self.interface.write_n(self.reg_p.base + self.reg_p.words, dat ,self.width)
|
|
|
|
class EdgeDetector:
|
|
#
|
|
# Definition
|
|
#
|
|
def __init__(self, width, mode="RFB"):
|
|
self.width = width
|
|
self.mode = mode
|
|
self.interface = None
|
|
|
|
self.reg_p = RegParams("edge_reg", 0, width, len(self.mode))
|
|
self.reg = None
|
|
|
|
self.i = Signal(self.width)
|
|
self.i_d = Signal(self.width)
|
|
if "R" in self.mode:
|
|
self.r_mask = Signal(self.width)
|
|
self.ro = Signal()
|
|
if "F" in self.mode:
|
|
self.f_mask = Signal(self.width)
|
|
self.fo = Signal()
|
|
if "B" in self.mode:
|
|
self.b_mask = Signal(self.width)
|
|
self.bo = Signal()
|
|
self.o = Signal()
|
|
|
|
def get_registers_comb(self):
|
|
comb = []
|
|
i = 0
|
|
if "R" in self.mode:
|
|
comb += [self.r_mask.eq(self.reg.storage[i*self.width:(i+1)*self.width])]
|
|
i += 1
|
|
if "F" in self.mode:
|
|
comb += [self.f_mask.eq(self.reg.storage[i*self.width:(i+1)*self.width])]
|
|
i += 1
|
|
if "B" in self.mode:
|
|
comb += [self.b_mask.eq(self.reg.storage[i*self.width:(i+1)*self.width])]
|
|
i += 1
|
|
return comb
|
|
|
|
def get_fragment(self):
|
|
comb = []
|
|
sync = [self.i_d.eq(self.i)]
|
|
|
|
# Rising Edge
|
|
if "R" in self.mode:
|
|
comb += [self.ro.eq(self.r_mask & self.i & (~self.i_d))]
|
|
else:
|
|
comb += [self.ro.eq(0)]
|
|
|
|
# Falling Edge
|
|
if "F" in self.mode:
|
|
comb += [self.fo.eq(self.f_mask & (~self.i) & self.i_d)]
|
|
else:
|
|
comb += [self.fo.eq(0)]
|
|
|
|
# Both
|
|
if "B" in self.mode:
|
|
comb += [self.bo.eq((self.b_mask & self.i) != self.i_d)]
|
|
else:
|
|
comb += [self.bo.eq(0)]
|
|
|
|
# Output
|
|
comb += [self.o.eq(self.ro | self.fo | self.bo)]
|
|
|
|
# Registers
|
|
comb += self.get_registers_comb()
|
|
|
|
return Fragment(comb, sync)
|
|
|
|
#
|
|
# Driver
|
|
#
|
|
def get_offset(self, type):
|
|
if type == "R":
|
|
r = 0
|
|
r = r + self.words if "F" in self.mode else r
|
|
r = r + self.words if "B" in self.mode else r
|
|
return r
|
|
elif type == "F":
|
|
r = 0
|
|
r = r + self.words if "B" in self.mode else r
|
|
return r
|
|
elif type == "B":
|
|
r = 0
|
|
return r
|
|
return 0
|
|
|
|
def set_r(self, dat):
|
|
self.interface.write_n(self.reg_p.base + self.get_offset("R"), dat ,self.width)
|
|
|
|
def set_f(self, dat):
|
|
self.interface.write_n(self.reg_p.base + self.get_offset("F"), dat ,self.width)
|
|
|
|
def set_b(self, dat):
|
|
self.interface.write_n(self.reg_p.base + self.get_offset("B"), dat ,self.width)
|
|
|
|
class Sum:
|
|
#
|
|
# Definition
|
|
#
|
|
def __init__(self, width=4):
|
|
self.width = width
|
|
self.interface = None
|
|
|
|
self.i = Signal(self.width)
|
|
self.o = Signal()
|
|
|
|
self.reg_p = RegParams("sum_reg", 0, 8, 4)
|
|
self.reg = None
|
|
|
|
self.prog_stb = Signal()
|
|
self.prog_adr = Signal(width)
|
|
self.prog_dat = Signal()
|
|
|
|
self._mem = Memory(1, 2**self.width)
|
|
self._lut_port = self._mem.get_port()
|
|
self._prog_port = self._mem.get_port(write_capable=True)
|
|
|
|
def get_registers_comb(self):
|
|
comb = [
|
|
self.prog_adr.eq(self.reg.storage[0:16]),
|
|
self.prog_dat.eq(self.reg.storage[16]),
|
|
self.prog_stb.eq(self.reg.storage[17])
|
|
]
|
|
return comb
|
|
|
|
def get_fragment(self):
|
|
comb = [
|
|
self._lut_port.adr.eq(self.i),
|
|
|
|
self._prog_port.adr.eq(self.prog_adr),
|
|
self._prog_port.we.eq(self.prog_stb),
|
|
self._prog_port.dat_w.eq(self.prog_dat),
|
|
|
|
self.o.eq(self._lut_port.dat_r),
|
|
]
|
|
comb += self.get_registers_comb()
|
|
return Fragment(comb, specials={self._mem})
|
|
|
|
#
|
|
# Driver
|
|
#
|
|
def set(self, truth_table):
|
|
for i in range(len(truth_table)):
|
|
val = truth_table[i]
|
|
we = 1<<17
|
|
dat = val<<16
|
|
addr = i
|
|
self.interface.write_n(self.reg_p.base, we + dat + addr, self.reg_p.size)
|
|
self.interface.write_n(self.reg_p.base, dat + addr, self.reg_p.size)
|
|
|
|
class Trigger:
|
|
#
|
|
# Definition
|
|
#
|
|
def __init__(self, width, ports, address=0x0000, interface=None):
|
|
self.width = width
|
|
self.ports = ports
|
|
|
|
self.sum = Sum(len(ports))
|
|
self.trig = Signal(self.width)
|
|
self.hit = Signal()
|
|
|
|
# insert port number in port reg name
|
|
for i in range(len(self.ports)):
|
|
self.ports[i].reg_p.name += "_%d"%i
|
|
|
|
# generate ports csr registers fields
|
|
for port in self.ports:
|
|
rf = CSRStorage(port.reg_p.size, reset=0, name=port.reg_p.name)
|
|
setattr(self, port.reg_p.name, rf)
|
|
|
|
# generate sum csr registers fields
|
|
self.sum_reg = CSRStorage(self.sum.reg_p.size, reset=0, name=self.sum.reg_p.name)
|
|
|
|
# generate registers
|
|
self.regs = list_regs(self.__dict__)
|
|
self.bank = csrgen.Bank(self.regs, address=address)
|
|
|
|
# update base addr & interface
|
|
self.set_address(address)
|
|
self.set_interface(interface)
|
|
self.set_registers()
|
|
|
|
def set_address(self, address):
|
|
self.address = address
|
|
self.bank = csrgen.Bank(self.regs, address=self.address)
|
|
for port in self.ports:
|
|
port.reg_p.base = self.bank.get_base(port.reg_p.name)
|
|
self.sum.reg_p.base = self.bank.get_base(self.sum.reg_p.name)
|
|
|
|
def set_interface(self, interface):
|
|
self.interface = interface
|
|
for port in self.ports:
|
|
port.interface = self.interface
|
|
self.sum.interface = self.interface
|
|
|
|
def set_registers(self):
|
|
self.sum.reg=self.sum_reg
|
|
for port in self.ports:
|
|
port.reg=getattr(self, port.reg_p.name)
|
|
|
|
def get_fragment(self):
|
|
# connect trig to input of each trig element
|
|
comb = [port.i.eq(self.trig) for port in self.ports]
|
|
|
|
# connect output of trig elements to sum
|
|
comb += [self.sum.i[j].eq(self.ports[j].o) for j in range(len(self.ports))]
|
|
|
|
# connect sum ouput to hit
|
|
comb += [self.hit.eq(self.sum.o)]
|
|
|
|
# add ports & sum to frag
|
|
frag = self.bank.get_fragment()
|
|
frag += self.sum.get_fragment()
|
|
for port in self.ports:
|
|
frag += port.get_fragment()
|
|
|
|
return frag + Fragment(comb)
|