def run_sim(vcdfile='lcd.vcd', speed=0.1, fps=20, timeout=0.0, visible=1, image_file=''):
firmware = Firmware(path(__file__).dirname() / 'lcd.elf')
avr = Avr(firmware, f_cpu=16000000)
lcd = Lcd(avr)
ledrow = LedRow(avr, size=7)
# period=1000 -> vcd error
vcd = VcdFile(avr, period=10, filename=vcdfile)
def state_func(i):
return (ledrow.pinstate(i), ledrow.reset_dirty(i))
led_game = LedRowGame(state_func=state_func,
labels='D4 D5 D6 D7 RS E RW'.split()
)
ac = Ac(avr)
connect_pins_by_rule('''
avr.B0 <=> lcd.D4 -> vcd
avr.B1 <=> lcd.D5 -> vcd
avr.B2 <=> lcd.D6 -> vcd
avr.B3 <=> lcd.D7 -> vcd
avr.B4 ==> lcd.RS -> vcd
avr.B5 ==> lcd.E -> vcd
avr.B6 ==> lcd.RW -> vcd
vcd <- ac.OUT -> avr.D2
lcd.D4 -> led.0
lcd.D5 -> led.1
lcd.D6 -> led.2
lcd.D7 -> led.3
lcd.RS -> led.4
lcd.E -> led.5
lcd.RW -> led.6
''',
dict(
avr=avr,
led=ledrow,
lcd=lcd,
ac=ac
),
vcd=vcd,
)
dev = CompositeGame([
CompositeGame(
[LcdGame(
lambda x, y:lcd.get_char(x, y), (20, 2)),
led_game,
],
align=1),
InfoGame(avr),
])
scrshot_by_exit = [(dev, image_file)] if image_file else None
AvrSimMain(
avr, dev, vcd, speed=speed, fps=fps, visible=visible, timeout=timeout,
scrshot_by_exit=scrshot_by_exit).run_game()
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 arduino_sim(
elf='',
mcu='atmega328',
f_cpu=16000000,
vcdfile='arduino.vcd',
speed=0.5,
fps=20,
timeout=0.0,
visible=1,
image_file='',
rate=11025,
buttons_enable=1,
vcd_enable=0,
spk_enable=0,
# udp_enable=1,
avcc=5000,
vcc=5000,
code=None,
):
'''
MCU:
- atmega168 OK
- atmega328p OK
- atmega2560 NO
- atmega1280 NO
:param mcu:
:param avcc: AVcc in mV
:param vcc: Vcc in mV
'''
if code and pyavrutils:
cc = pyavrutils.Arduino()
cc.build(code)
elf = cc.output
if not elf:
elf = find_elf()
firmware = Firmware(elf)
avr = Avr(mcu=mcu, f_cpu=f_cpu, vcc=vcc / 1000.0, avcc=avcc / 1000.0)
avr.load_firmware(firmware)
# udpReader = UdpReader()
# if udp_enable:
# udp = Udp(avr)
# udp.connect()
# udpReader.start()
lcd = Lcd(avr)
vcd = VcdFile(avr, period=1000, filename=vcdfile) if vcd_enable else None
ledrow = LedRow(avr, size=14)
buttons = [Button(avr, pullup=0)
for x in range(14)] if buttons_enable else 14 * [None]
spk = Spk(avr, rate=rate, speed=speed) if spk_enable else None
connect_pins_by_rule(
'''
but0 .OUT ==> avr.D0 ==> dig.0 -> vcd
but1 .OUT ==> avr.D1 ==> dig.1 -> vcd
but2 .OUT ==> avr.D2 ==> dig.2 -> vcd
but3 .OUT ==> avr.D3 ==> dig.3 -> vcd
but4 .OUT ==> avr.D4 ==> dig.4 -> vcd
but5 .OUT ==> avr.D5 ==> dig.5 -> vcd
but6 .OUT ==> avr.D6 ==> dig.6 -> vcd
but7 .OUT ==> avr.D7 ==> dig.7 -> vcd
but8 .OUT ==> avr.B0 ==> dig.8 -> vcd
but9 .OUT ==> avr.B1 ==> dig.9 -> vcd
but10.OUT ==> avr.B2 ==> dig.10 -> vcd
but11.OUT ==> avr.B3 ==> dig.11 -> vcd
but12.OUT ==> avr.B4 ==> dig.12 -> vcd
but13.OUT ==> avr.B5 ==> dig.13 -> vcd
dig.13 --> spk.IN -> vcd
dig.5 <=> lcd.D4 -> vcd
dig.4 <=> lcd.D5 -> vcd
dig.3 <=> lcd.D6 -> vcd
dig.2 <=> lcd.D7 -> vcd
dig.12 ==> lcd.RS -> vcd
dig.11 ==> lcd.RW -> vcd
dig.10 ==> lcd.E -> vcd
''',
dict(avr=avr,
dig=ledrow,
lcd=lcd,
spk=spk,
**dict([('but' + str(i), b) for i, b in enumerate(buttons)])),
vcd=vcd,
)
#################
# GUI
#################
if spk_enable:
def spk_func(size):
return spk.read()
spk_game = SpkGame(spk_func, rate=rate)
def state_func(i):
return (ledrow.pinstate(i), ledrow.reset_dirty(i))
led_game = LedRowGame(
state_func=state_func,
labels=[str(x) for x in range(14)],
align=1,
)
but_guis = [
ButtonGame(
label=label,
shortcut=shortcut,
hook=dict(up=x.up, down=x.down) if x else {},
size=(50, 30),
) for x, label, shortcut in zip(
buttons,
'0 1 2 3 4 5 6 7 8 9 10 11 12 13'.split(),
'0 1 2 3 4 5 6 7 8 9 a b c d'.split(),
)
]
info = InfoGame(avr)
def reload_firmware():
firmware = Firmware(find_elf())
avr.load_firmware(firmware)
lcd.reset()
info.reload()
class MyFloat(object):
def __init__(self, value=0.0):
self.value = value
def __float__(self):
return float(self.value)
def inc(self):
self.value *= 10.0
def dec(self):
self.value /= 10.0
speed = MyFloat(speed)
# def udp_read():
# if not hasattr(udp_read, 'display'):
# udp_read.display = ''
# if not hasattr(udp_read, 'lastline'):
# udp_read.lastline = ''
# s = udpReader.read()
# if s:
# sys.stdout.write(s)
# udp_read.lastline += s
# udp_read.lastline = lastline(udp_read.lastline)
# udp_read.display = udp_read.lastline.replace(
# '\n', '\\n').replace('\r', '\\r')
# return udp_read.display
def uart_read():
return avr.uart.last_line.replace('\n', '\\n').replace('\r', '\\r')
dev = CompositeGame([
CompositeGame([
led_game,
CompositeGame(but_guis, align=1),
], align=0), info,
CompositeGame([
CompositeGame([
ButtonGame(label='reload',
shortcut='r',
hook=dict(down=reload_firmware)),
ButtonGame(
label='speed up', shortcut='+', hook=dict(down=speed.inc)),
ButtonGame(label='speed down',
shortcut='-',
hook=dict(down=speed.dec)),
TextGame((lambda: "speed set= %fx" % float(speed))),
],
align=1),
LcdGame(lambda x, y: lcd.get_char(x, y), (16, 2)),
TextGame((lambda: 'ser=' + uart_read())),
],
align=1)
])
scrshot_by_exit = [(dev, image_file)] if image_file else None
class ArduinoMain(AvrSimMain):
bufpos = 0
def cb_loop(self):
AvrSimMain.cb_loop(self)
# if len(udpReader.buffer) > self.bufpos:
# sys.stdout.write(''.join(udpReader.buffer[self.bufpos:]))
# self.bufpos = len(udpReader.buffer)
sim = ArduinoMain(avr,
dev,
vcd,
speed=speed,
fps=fps,
visible=visible,
timeout=timeout,
scrshot_by_exit=scrshot_by_exit)
sim.run_game()
time.sleep(1)
if spk_enable:
spk_game.terminate()
def arduino_sim(
elf='',
mcu='atmega328',
f_cpu=16000000,
vcdfile='arduino.vcd',
speed=0.5,
fps=20,
timeout=0.0,
visible=1,
image_file='',
rate=11025,
buttons_enable=1,
vcd_enable=0,
spk_enable=0,
# udp_enable=1,
avcc=5000,
vcc=5000,
code=None,
):
'''
MCU:
- atmega168 OK
- atmega328p OK
- atmega2560 NO
- atmega1280 NO
:param mcu:
:param avcc: AVcc in mV
:param vcc: Vcc in mV
'''
if code and pyavrutils:
cc = pyavrutils.Arduino()
cc.build(code)
elf = cc.output
if not elf:
elf = find_elf()
firmware = Firmware(elf)
avr = Avr(mcu=mcu, f_cpu=f_cpu, vcc=vcc / 1000.0, avcc=avcc / 1000.0)
avr.load_firmware(firmware)
# udpReader = UdpReader()
# if udp_enable:
# udp = Udp(avr)
# udp.connect()
# udpReader.start()
lcd = Lcd(avr)
vcd = VcdFile(avr, period=1000, filename=vcdfile) if vcd_enable else None
ledrow = LedRow(avr, size=14)
buttons = [Button(
avr, pullup=0) for x in range(14)] if buttons_enable else 14 * [None]
spk = Spk(avr, rate=rate, speed=speed) if spk_enable else None
connect_pins_by_rule('''
but0 .OUT ==> avr.D0 ==> dig.0 -> vcd
but1 .OUT ==> avr.D1 ==> dig.1 -> vcd
but2 .OUT ==> avr.D2 ==> dig.2 -> vcd
but3 .OUT ==> avr.D3 ==> dig.3 -> vcd
but4 .OUT ==> avr.D4 ==> dig.4 -> vcd
but5 .OUT ==> avr.D5 ==> dig.5 -> vcd
but6 .OUT ==> avr.D6 ==> dig.6 -> vcd
but7 .OUT ==> avr.D7 ==> dig.7 -> vcd
but8 .OUT ==> avr.B0 ==> dig.8 -> vcd
but9 .OUT ==> avr.B1 ==> dig.9 -> vcd
but10.OUT ==> avr.B2 ==> dig.10 -> vcd
but11.OUT ==> avr.B3 ==> dig.11 -> vcd
but12.OUT ==> avr.B4 ==> dig.12 -> vcd
but13.OUT ==> avr.B5 ==> dig.13 -> vcd
dig.13 --> spk.IN -> vcd
dig.5 <=> lcd.D4 -> vcd
dig.4 <=> lcd.D5 -> vcd
dig.3 <=> lcd.D6 -> vcd
dig.2 <=> lcd.D7 -> vcd
dig.12 ==> lcd.RS -> vcd
dig.11 ==> lcd.RW -> vcd
dig.10 ==> lcd.E -> vcd
''',
dict(
avr=avr,
dig=ledrow,
lcd=lcd,
spk=spk,
**dict([('but' + str(i), b) for i, b in enumerate(buttons)])
),
vcd=vcd,
)
#################
# GUI
#################
if spk_enable:
def spk_func(size):
return spk.read()
spk_game = SpkGame(spk_func, rate=rate)
def state_func(i):
return (ledrow.pinstate(i), ledrow.reset_dirty(i))
led_game = LedRowGame(state_func=state_func,
labels=[str(x) for x in range(14)],
align=1,)
but_guis = [ButtonGame(
label=label,
shortcut=shortcut,
hook=dict(up=x.up, down=x.down) if x else {},
size=(50, 30),
) for x, label, shortcut in zip(
buttons,
'0 1 2 3 4 5 6 7 8 9 10 11 12 13'.split(),
'0 1 2 3 4 5 6 7 8 9 a b c d'.split(),
)
]
info = InfoGame(avr)
def reload_firmware():
firmware = Firmware(find_elf())
avr.load_firmware(firmware)
lcd.reset()
info.reload()
class MyFloat(object):
def __init__(self, value=0.0):
self.value = value
def __float__(self):
return float(self.value)
def inc(self):
self.value *= 10.0
def dec(self):
self.value /= 10.0
speed = MyFloat(speed)
# def udp_read():
# if not hasattr(udp_read, 'display'):
# udp_read.display = ''
# if not hasattr(udp_read, 'lastline'):
# udp_read.lastline = ''
# s = udpReader.read()
# if s:
# sys.stdout.write(s)
# udp_read.lastline += s
# udp_read.lastline = lastline(udp_read.lastline)
# udp_read.display = udp_read.lastline.replace(
# '\n', '\\n').replace('\r', '\\r')
# return udp_read.display
def uart_read():
return avr.uart.last_line.replace('\n', '\\n').replace('\r', '\\r')
dev = CompositeGame([
CompositeGame([led_game,
CompositeGame(but_guis, align=1),
], align=0),
info,
CompositeGame([
CompositeGame([
ButtonGame(label='reload',
shortcut='r',
hook=dict(down=reload_firmware)),
ButtonGame(label='speed up',
shortcut='+',
hook=dict(down=speed.inc)),
ButtonGame(label='speed down',
shortcut='-',
hook=dict(down=speed.dec)),
TextGame((lambda:
"speed set= %fx" % float(speed))),
], align=1),
LcdGame(
lambda x, y:lcd.get_char(x, y), (16, 2)),
TextGame((lambda: 'ser=' + uart_read())),
], align=1)
])
scrshot_by_exit = [(dev, image_file)] if image_file else None
class ArduinoMain(AvrSimMain):
bufpos = 0
def cb_loop(self):
AvrSimMain.cb_loop(self)
# if len(udpReader.buffer) > self.bufpos:
# sys.stdout.write(''.join(udpReader.buffer[self.bufpos:]))
# self.bufpos = len(udpReader.buffer)
sim = ArduinoMain(
avr, dev, vcd, speed=speed, fps=fps, visible=visible, timeout=timeout,
scrshot_by_exit=scrshot_by_exit)
sim.run_game()
time.sleep(1)
if spk_enable:
spk_game.terminate()