aboutsummaryrefslogtreecommitdiffstats
path: root/atk16_emu/cli.py
blob: 613af6c539b3421fb7ae0d76ed69479ecdae7c50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import sys
from dataclasses import dataclass
from .emu import Machine
from .debugger import Debugger

@dataclass
class Options:
  rom_image_path: str | None
  debugger_enabled: bool
  peripherals_enabled: bool

options = Options(
  rom_image_path=None,
  debugger_enabled=False,
  peripherals_enabled=True,
)

def print_help():
  print("Usage: emu_cli.py [-d] [-n] [-h|--help] <rom_path>")
  print("")
  print("Options:")
  print("  <rom_path>: path to the rom image")
  print("  -h: print help")
  print("  -d: enable debugger")
  print("  -n: no peripherals (useful for automated testing)")

def main():
  for arg in sys.argv[1:]:
    if arg == "-d":
      options.debugger_enabled = True
    elif arg == "-n":
      options.peripherals_enabled = False
    elif arg == "-h" or arg == "--help" or arg == "-?":
      print_help()
      sys.exit(0)
    elif arg[0] == "-":
      print(f"Unknown option: {arg}")
      print_help()
      sys.exit(1)
    else:
      options.rom_image_path = arg

  if options.rom_image_path is None:
    print_help()
    sys.exit(1)

  if not options.debugger_enabled:
    with open(options.rom_image_path, "rb") as f:
      rom_image = bytearray(f.read())

    machine = Machine(options.peripherals_enabled)
    machine.load_rom_image(rom_image)
    machine.reset()
    machine.run_until_halted()

    print("Machine halted.")
    print("===============")

    machine.print_state_summary()
    machine.print_hotspot_summary()

  else:
    debugger = Debugger(options.peripherals_enabled)
    debugger.load_rom_image(options.rom_image_path)
    debugger.activate()

if __name__ == "__main__":
  main()