def simulate(self): elf = self.cc.output # run firmware = Firmware(elf) avr = Avr(mcu=self.cc.mcu, f_cpu=self.cc.f_cpu) avr.load_firmware(firmware) udpReader = UdpReader() udp = Udp(avr) udp.connect() udpReader.start() simvcd = None if self.vcd: simvcd = VcdFile(avr, period=1000, filename=self.vcd) 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=avr, ), vcd=simvcd, ) simvcd.start() avr.move_time_marker(self.timespan) while avr.time_passed() < self.timespan * 0.99: time.sleep(0.05) if simvcd: simvcd.terminate() udpReader.terminate() log.debug('cycles=%s' % avr.cycle) log.debug('mcu time=%s' % avr.time_passed()) # time.sleep(1) self.serial = udpReader.read()
def test(): avr = Avr(mcu='atmega48', f_cpu=8000000) Ac(avr) Button(avr) Inverter(avr) LedRow(avr) Sgm7(avr) Lcd(avr) Spk(avr) Udp(avr)
def startAvr(self): self.avr = Avr(mcu='atmega88',f_cpu=8000000) self.udp = Udp(self.avr) self.udp.connect() self.uartBridge = UartBridge() firmware = Firmware(ELFNAME) self.avr.load_firmware(firmware); self.uartBridge.start(); self.avr.run()
class TestBase(unittest.TestCase): def setUp(self): self.first = True self.startAvr(); pass def tearDown(self): self.uartBridge.terminate() self.avr.terminate() pass def messageSanityCheck(self, bytes, commandId, expectedSize, first): ''' Check that the response has correct size, message Id, checksum @param bytes: list of bytes @param commandId: message identifier @param expectedSize: expected size of the response @param first: is is a first response @raise exception: unittest.TestCase.assertEqual @return: None ''' if (expectedSize != None): self.assertEqual(len(bytes), expectedSize, "Message length "+str(len(bytes))+", expected size "+str(expectedSize)); self.assertEqual(bytes[0], 0x7f, "1st sync byte is "+str(bytes[0])); self.assertEqual(bytes[1], 0xef, "2nd sync byte is "+str(bytes[1])); if (first): self.assertEqual(bytes[2], commandId, "Message id "+str(bytes[2])); else: self.assertEqual(bytes[2], commandId | 0x80, "Message id "+str(bytes[2])); self.assertEqual(bytes[3], expectedSize-4, "Payload size is "+str(bytes[3])); bytesWithoutChecksum = bytes[:-1] csExpected = calculateChecksum(bytesWithoutChecksum) csActual = bytes[len(bytes)-1] self.assertEqual(csActual, csExpected, "Checksum is "+hex(csActual)+" instead "+hex(csExpected)); def startAvr(self): self.avr = Avr(mcu='atmega88',f_cpu=8000000) self.udp = Udp(self.avr) self.udp.connect() self.uartBridge = UartBridge() firmware = Firmware(ELFNAME) self.avr.load_firmware(firmware); self.uartBridge.start(); self.avr.run() #self.time_passed(); def time_passed(self, timeout=0.99): while self.avr.time_passed() < 0.1 * timeout: time.sleep(0.05) def sendCommand(self, command, payload): syncBytes = [0x7f, 0xef]; command = syncBytes + command + [len(payload)+1] + payload; cs = calculateChecksum(command) command = command + [cs]; s = str(bytearray(command)) self.uartBridge.send_str(s); self.time_passed(0.05*len(s))