litex/migen/fhdl/structure.py

import math

def BitsFor(n):
	if n == 0:
		return 1
	else:
		return int(math.ceil(math.log(n+1, 2)))

class BV:
	def __init__(self, width=1, signed=False):
		self.width = width
		self.signed = signed
	
	def __str__(self):
		r = str(self.width) + "'"
		if self.signed:
			r += "s"
		r += "d"
		return r

class Value:
	def __invert__(self):
		return Operator("~", [self])

	def __add__(self, other):
		return Operator("+", [self, other])
	def __radd__(self, other):
		return Operator("+", [other, self])
	def __sub__(self, other):
		return Operator("-", [self, other])
	def __rsub__(self, other):
		return Operator("-", [other, self])
	def __mul__(self, other):
		return Operator("*", [self, other])
	def __rmul__(self, other):
		return Operator("*", [other, self])
	def __lshift__(self, other):
		return Operator("<<", [self, other])
	def __rlshift__(self, other):
		return Operator("<<", [other, self])
	def __rshift__(self, other):
		return Operator(">>", [self, other])
	def __rrshift__(self, other):
		return Operator(">>", [other, self])
	def __and__(self, other):
		return Operator("&", [self, other])
	def __rand__(self, other):
		return Operator("&", [other, self])
	def __xor__(self, other):
		return Operator("^", [self, other])
	def __rxor__(self, other):
		return Operator("^", [other, self])
	def __or__(self, other):
		return Operator("|", [self, other])
	def __ror__(self, other):
		return Operator("|", [other, self])
	
	def __lt__(self, other):
		return Operator("<", [self, other])
	def __le__(self, other):
		return Operator("<=", [self, other])
	def __eq__(self, other):
		return Operator("==", [self, other])
	def __ne__(self, other):
		return Operator("!=", [self, other])
	def __gt__(self, other):
		return Operator(">", [self, other])
	def __ge__(self, other):
		return Operator(">=", [self, other])
	
	
	def __getitem__(self, key):
		if isinstance(key, int):
			return Slice(self, key, key+1)
		elif isinstance(key, slice):
			start = key.start or 0
			stop = key.stop or self.bv.width
			if stop > self.bv.width:
				stop = self.bv.width
			if key.step != None:
				raise KeyError
			return Slice(self, start, stop)
		else:
			raise KeyError

class Operator(Value):
	def __init__(self, op, operands):
		self.op = op
		self.operands = list(map(_cst, operands))

class Slice(Value):
	def __init__(self, value, start, stop):
		self.value = value
		self.start = start
		self.stop = stop

class Cat(Value):
	def __init__(self, *args):
		self.l = list(map(_cst, args))

class Constant(Value):
	def __init__(self, n, bv=None):
		self.bv = bv or BV(BitsFor(n))
		self.n = n

def _cst(x):
	if isinstance(x, int):
		return Constant(x)
	else:
		return x

class Signal(Value):
	def __init__(self, bv=BV(), name="anonymous", variable=False, reset=0):
		self.bv = bv
		self.variable = variable
		self.name = name
		self.reset = Constant(reset, bv)

	def __hash__(self):
		return id(self)

def Declare(parent, name, bv=BV(), variable=False, reset=0):
	# try to find a meaningful prefix
	if parent.__module__ == "__main__":
		prefix = parent.__class__.__name__
	else:
		modules = parent.__module__.split('.')
		prefix = modules[len(modules)-1]
	setattr(parent, name, Signal(bv, prefix + "_" + name, variable, reset))

# statements

class Assign:
	def __init__(self, l, r):
		self.l = l
		self.r = _cst(r)

class StatementList:
	def __init__(self, l=[]):
		self.l = l

def _sl(x):
	if isinstance(x, list):
		return StatementList(x)
	else:
		return x

class If:
	def __init__(self, cond, t, f=StatementList()):
		self.cond = cond
		self.t = _sl(t)
		self.f = _sl(f)

class Case:
	def __init__(self, test, cases=[], default=StatementList()):
		self.test = test
		self.cases = [(c[0], _sl(c[1])) for c in cases]
		self.default = _sl(default)

#

class Instance:
	def __init__(self, of, outs=[], ins=[], parameters=[], clkport="", rstport="", name=""):
		self.of = of
		if name:
			self.name = name
		else:
			self.name = of
		def process_io(x):
			if isinstance(x[1], Signal):
				return x # override
			elif isinstance(x[1], BV):
				return (x[0], Signal(x[1], self.name + "_" + x[0]))
			else:
				raise TypeError
		self.outs = dict(map(process_io, outs))
		self.ins = dict(map(process_io, ins))
		self.parameters = parameters
		self.clkport = clkport
		self.rstport = rstport

	def __hash__(self):
		return id(self)

class Fragment:
	def __init__(self, comb=StatementList(), sync=StatementList(), instances=[]):
		self.comb = _sl(comb)
		self.sync = _sl(sync)
		self.instances = instances
	
	def __add__(self, other):
		return Fragment(self.comb.l + other.comb.l, self.sync.l + other.sync.l, self.instances + other.instances)
	def __iadd__(self, other):
		self.comb.l += other.comb.l
		self.sync.l += other.sync.l
		self.instances += other.instances
		return self