Merge branch 'feature_complement2_abs_sign' into develop

Author: shtaxxx

veriloggen/core/vtypes.py

@@ -283,6 +283,9 @@ def __pos__(self):
     def __invert__(self):
         raise TypeError('Not allowed operation.')
 
+    def __abs__(self):
+        raise TypeError('Not allowed operation.')
+
     def __getitem__(self, r):
         raise TypeError('Not allowed operation.')
 
@@ -361,6 +364,9 @@ def __pos__(self):
     def __invert__(self):
         return Unot(self)
 
+    def __abs__(self):
+        return Abs(self)
+
     def __getitem__(self, r):
         if isinstance(r, slice):
             size = self._len()
@@ -1745,6 +1751,24 @@ def Mux(condition, true_value, false_value):
     return Cond(condition, true_value, false_value)
 
 
+def Complement2(var):
+    if isinstance(var, (int, bool, float)):
+        return abs(var)
+
+    return Unot(var) + Int(1)
+
+
+def Abs(var):
+    return Mux(Sign(var), Complement2(var), var)
+
+
+def Sign(var):
+    if isinstance(var, (int, bool, float)):
+        return var < 0
+
+    return var < Int(0, signed=True)
+
+
 class Sensitive(VeriloggenNode):
 
     def __init__(self, name):

veriloggen/stream/stypes.py

@@ -346,6 +346,9 @@ def __neg__(self):
     def __pos__(self):
         return Uplus(self)
 
+    def __abs__(self):
+        return Abs(self)
+
     def __getitem__(self, r):
         if isinstance(r, slice):
             size = self.bit_length()
@@ -1134,7 +1137,12 @@ def eval(self):
 class Unot(_UnaryLogicalOperator):
 
     def eval(self):
-        return ~ self.right.eval()
+        right = self.right.eval()
+        try:
+            v = ~right
+        except:
+            v = Ulnot(right)
+        return v
 
 
 class Uand(_UnaryLogicalOperator):
@@ -1322,6 +1330,9 @@ def _implement(self, m, seq, svalid=None, senable=None):
 
         m.Assign(data(rdata))
 
+    def eval(self):
+        return self
+
 
 class _SpecialOperator(_Operator):
     latency = 1
@@ -1668,6 +1679,81 @@ def _implement(self, m, seq, svalid=None, senable=None):
         m.Instance(inst, self.name('lut'), ports=ports)
 
 
+class Complement2(_SpecialOperator):
+
+    def __init__(self, var):
+        _SpecialOperator.__init__(self, var)
+        self.op = vtypes.Complement2
+
+    def _set_attributes(self):
+        self.width = self.var.bit_length()
+        self.point = self.var.get_point()
+        self.signed = self.var.get_signed()
+
+    @property
+    def var(self):
+        return self.args[0]
+
+    @var.setter
+    def var(self, var):
+        self.args[0] = var
+
+    def eval(self):
+        var = self.var.eval()
+        ret = Complement2(var)
+        return ret
+
+
+class Abs(_SpecialOperator):
+
+    def __init__(self, var):
+        _SpecialOperator.__init__(self, var)
+        self.op = vtypes.Abs
+
+    def _set_attributes(self):
+        self.width = self.var.bit_length()
+        self.point = self.var.get_point()
+        self.signed = self.var.get_signed()
+
+    @property
+    def var(self):
+        return self.args[0]
+
+    @var.setter
+    def var(self, var):
+        self.args[0] = var
+
+    def eval(self):
+        var = self.var.eval()
+        ret = abs(var)
+        return ret
+
+
+class Sign(_SpecialOperator):
+
+    def __init__(self, var):
+        _SpecialOperator.__init__(self, var)
+        self.op = vtypes.Sign
+
+    def _set_attributes(self):
+        self.width = self.var.bit_length()
+        self.point = self.var.get_point()
+        self.signed = self.var.get_signed()
+
+    @property
+    def var(self):
+        return self.args[0]
+
+    @var.setter
+    def var(self, var):
+        self.args[0] = var
+
+    def eval(self):
+        var = self.var.eval()
+        ret = Sign(var)
+        return ret
+
+
 class _Delay(_UnaryOperator):
 
     def __init__(self, right):