179 lines
6.4 KiB
Python
179 lines
6.4 KiB
Python
from wire_components import wire
|
|
|
|
|
|
""" LOGIC GATE COMPONENTS """
|
|
|
|
|
|
class logic_gate():
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w"):
|
|
self.a = wire(name=prefix+"_"+a.name.replace(prefix+"_", ""), value=a.value)
|
|
self.b = wire(name=prefix+"_"+b.name.replace(prefix+"_", ""), value=b.value)
|
|
self.prefix = prefix
|
|
|
|
def get_component_types(self):
|
|
return list([self])
|
|
|
|
# FLAT C GENERATION #
|
|
@staticmethod
|
|
def get_includes_c():
|
|
return f"#include <stdio.h>\n#include <stdint.h>\n\n"
|
|
|
|
def get_prototype_c(self):
|
|
self.a.name = self.a.name.replace(self.prefix+"_", "")
|
|
self.b.name = self.b.name.replace(self.prefix+"_", "")
|
|
return f"uint8_t {self.gate_type}(uint8_t {self.a.name}, uint8_t {self.b.name})" + "{" + "\n"
|
|
|
|
def get_declaration_c_flat(self):
|
|
return f"{self.a.get_declaration_c()}{self.b.get_declaration_c()}{self.output.get_declaration_c()}"
|
|
|
|
def get_init_c_flat(self):
|
|
return f"{self.a.name} {self.operator} {self.b.name}"
|
|
|
|
def get_function_c(self):
|
|
self.a.prefix = self.a.name.replace(self.prefix+"_", "")
|
|
self.b.prefix = self.b.name.replace(self.prefix+"_", "")
|
|
return f"{self.a.get_wire_value_c()} {self.operator} {self.b.get_wire_value_c()}"
|
|
|
|
# Generating flat C code representation of separate logic gate
|
|
# (i.e. not as a component of bigger circuit)
|
|
def get_c_code(self, file_object):
|
|
file_object.write(self.get_includes_c())
|
|
file_object.write(self.get_prototype_c())
|
|
file_object.write(" return "+(self.get_function_c())+";\n}")
|
|
file_object.close()
|
|
|
|
# HIERARCHICAL C GENERATION #
|
|
def get_function_block_c(self):
|
|
self.a.name = "a"
|
|
self.b.name = "b"
|
|
return f"{self.get_prototype_c()}" + \
|
|
f" return "+(self.get_function_c())+";\n}\n\n"
|
|
|
|
def get_invocation_c(self, a: wire, b: wire):
|
|
return f"{self.gate_type}({a.name}, {b.name});"
|
|
|
|
def get_gate_output_c(self, a: wire, b: wire, offset: int = 0):
|
|
return f"({self.gate_type}({a.name}, {b.name}) & 0x01) << {offset}"
|
|
|
|
|
|
# Single-input
|
|
class not_gate(logic_gate):
|
|
def __init__(self, a: wire, prefix: str = "w", outid: int = 0):
|
|
self.gate_type = "not_gate"
|
|
self.operator = "~"
|
|
self.a = wire(prefix+"_"+a.name.replace(prefix+"_", ""), a.value)
|
|
self.prefix = prefix
|
|
|
|
if self.a.value == 1:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
|
|
# FLAT C GENERATION #
|
|
def get_prototype_c(self):
|
|
return f"uint8_t {self.gate_type}(uint8_t a)" + "{" + "\n"
|
|
|
|
def get_declaration_c(self):
|
|
return f"{self.a.get_declaration_c()}{self.output.get_declaration_c()}"
|
|
|
|
def get_init_c_flat(self):
|
|
return f"{self.operator}{self.a.name}"
|
|
|
|
def get_function_c(self):
|
|
self.a.prefix = self.a.name.replace(self.prefix+"_", "")
|
|
return f"{self.operator}{self.a.get_wire_value_c()} & 0x01 << 0"
|
|
|
|
# HIERARCHICAL C GENERATION #
|
|
def get_function_block_c(self):
|
|
self.a.name = "a"
|
|
return f"{self.get_prototype_c()}" + \
|
|
f" return "+(self.get_function_c())+";\n}\n\n"
|
|
|
|
def get_invocation_c(self, a: wire):
|
|
return f"{self.gate_type}({a.name});"
|
|
|
|
def get_gate_output_c(self, a: wire, offset: int = 0):
|
|
return f"({self.gate_type}({a.name}) & 0x01) << {offset}"
|
|
|
|
|
|
# Two-input
|
|
class and_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "and_gate"
|
|
self.operator = "&"
|
|
|
|
if a.value == 1 and b.value == 1:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
|
|
|
|
class nand_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "nand_gate"
|
|
self.operator = "&"
|
|
|
|
if (self.a.value == 1 and self.b.value == 1):
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
|
|
def get_function_c(self):
|
|
return "~("+(super().get_function_c())+f") & 0x01 << 0"
|
|
|
|
|
|
class or_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "or_gate"
|
|
self.operator = "|"
|
|
|
|
if self.a.value == 1 or self.b.value == 1:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
|
|
|
|
class nor_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "nor_gate"
|
|
self.operator = "|"
|
|
|
|
if self.a.value == 1 or self.b.value == 1:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
|
|
def get_function_c(self):
|
|
return "~("+(super().get_function_c())+") & 0x01 << 0"
|
|
|
|
|
|
class xor_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "xor_gate"
|
|
self.operator = "^"
|
|
|
|
if (a.value == 1 and b.value == 0) or (a.value == 0 and b.value == 1):
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
|
|
|
|
class xnor_gate(logic_gate):
|
|
def __init__(self, a: wire, b: wire, prefix: str = "w", outid: int = 0):
|
|
super().__init__(a, b, prefix)
|
|
self.gate_type = "xnor_gate"
|
|
self.operator = "^"
|
|
|
|
if (self.a.value == 1 and self.b.value == 0) or (self.a.value == 0 and self.b.value == 1):
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=0)
|
|
else:
|
|
self.output = wire(name=prefix+"_y"+str(outid), value=1)
|
|
|
|
def get_function_c(self):
|
|
return "~("+(super().get_function_c())+") & 0x01 << 0"
|