class Controller:
def __init__(self, sock, mcu_name, freq=16000000, version=1):
self.mcu_name = mcu_name
self.mcu = Avr(mcu=mcu_name, f_cpu=freq)
self.socket = sock
self.fw_path = None
self.version = version
self.bind_callback_for_digit_pins(atmega_328_digit_pin_table)
self.vcd = None
self.connect_vcd()
def upload_firmware(self, fw_path):
self.fw_path = fw_path
fw = Firmware(fw_path)
self.mcu.load_firmware(fw)
def connect_vcd(self):
self.vcd = VcdFile(self.mcu, filename='out' + str(self.version))
connect_pins_by_rule('''
avr.D0 ==> vcd
avr.D1 ==> vcd
avr.D2 ==> vcd
avr.D3 ==> vcd
avr.D4 ==> vcd
avr.D5 ==> vcd
avr.D6 ==> vcd
avr.D7 ==> vcd
avr.B0 ==> vcd
avr.B1 ==> vcd
avr.B2 ==> vcd
avr.B3 ==> vcd
avr.B4 ==> vcd
avr.B5 ==> vcd
''',
dict(avr=self.mcu),
vcd=self.vcd)
def bind_callback_for_digit_pins(self, ports):
def port_callback(irq, new_val):
msg = "change " + self.mcu_name + ' ' + irq.name[
0] + " " + irq.name[1] + " " + str(new_val)
self.socket.send_msg(msg)
callback = Mock(side_effect=port_callback)
for port in ports:
p = self.mcu.irq.ioport_register_notify(callback,
(port[0], int(port[1:])))
p.get_irq().name = port
def new_pin_val(self, port_pin, new_val):
# type: ((str, int), int) -> None
irq = self.mcu.irq.getioport(port_pin)
avr_raise_irq(irq, new_val)
def run(self):
self.vcd.start()
self.mcu.run()
def pause(self):
self.vcd.stop()
self.mcu.pause()
def terminate(self):
if self.vcd:
self.vcd.terminate()
self.mcu.terminate()
