from supSMSGecko import make_xml, symbols

disable_z_addrs = {
  'GMSJ01': (0x800EB020, 0x800EB02C),
  'GMSJ0A': (0x8027770C, 0x80277718),
  'GMSP01': (0x8028F8FC, 0x8028F908),
  'GMSE01': (0x80297A64, 0x80297A70),
}

def main(g, ver):
  S = symbols[ver]

  # disable Z button
  g.branch(*disable_z_addrs[ver])

  # Instant Level Select main function
  g.C2(0x220 + S['TApplication_gameLoop'], 'src/main.s', extra_as_input=[
    '.set $LevelSelect.area, 0x817F0028',
    '.set $LevelSelect.epFlag, 0x817F002A',
    '.set $LevelSelect.AreaLock, 0x817F002B',
  ]);

make_xml(main)