From 8061203a4facbbfa1f605ceea102d8efece2c275 Mon Sep 17 00:00:00 2001
From: PikalaxALT <pikalaxalt@gmail.com>
Date: Thu, 9 Sep 2021 17:56:18 -0400
Subject: Use NAIX in map headers

---
 map_header_resolve_fields.py | 112 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 map_header_resolve_fields.py

(limited to 'map_header_resolve_fields.py')

diff --git a/map_header_resolve_fields.py b/map_header_resolve_fields.py
new file mode 100644
index 00000000..1d230bac
--- /dev/null
+++ b/map_header_resolve_fields.py
@@ -0,0 +1,112 @@
+import collections
+import struct
+import typing
+import re
+
+romfname = 'baserom.nds'
+arm9offs = 0x00004000
+arm9load = 0x02000000
+mapheado = 0x020EEDBC
+mapheads = 0x00003468
+
+
+class MapHeader(typing.NamedTuple):
+    area_data_bank: int
+    move_model_bank: int
+    matrix_id: int
+    scripts_bank: int
+    level_scripts_bank: int
+    msg_bank: int
+    day_music_id: int
+    night_music_id: int
+    wild_encounter_bank: int
+    events_bank: int
+    mapsec: int
+    weather_type: int
+    camera_type: int
+    unk16: int
+    battle_bg: int
+    is_bike_allowed: bool
+    is_running_allowed: bool
+    is_escape_rope_allowed: bool
+    is_fly_allowed: bool
+
+    @classmethod
+    def from_buffer(cls, buffer: bytes) -> typing.Generator['MapHeader', None, typing.Any]:
+        for tup in struct.iter_unpack('<BBHHHHHHHHHBBBB', buffer):
+            *tup, flags = tup
+            tup += (flags & 15, flags & 16 != 0, flags & 32 != 0, flags & 64 != 0, flags & 128 != 0)
+            yield cls._make(tup)
+
+
+def read_sndseq_h():
+    with open('include/constants/sndseq.h') as fp:
+        for line in fp:
+            if line.startswith('#define SEQ_'):
+                yield line.split()[1]
+
+
+def read_mapsec_h():
+    with open('include/constants/map_sections.h') as fp:
+        for line in fp:
+            if line.startswith('#define MAPSEC_'):
+                yield line.split()[1]
+
+
+def read_naix(filename):
+    with open(filename) as fp:
+        for line in fp:
+            if (m := re.match(r'\s*(NARC_\w+)\s+=\s+\d+,', line)) is not None:
+                yield m[1]
+
+
+class MyList(list):
+    def __getitem__(self, item):
+        try:
+            return super().__getitem__(item)
+        except IndexError:
+            return str(item)
+
+
+def main():
+    sndseqs = MyList(read_sndseq_h())
+    mapsecs = MyList(read_mapsec_h())
+    msgnaix = MyList(read_naix('files/msgdata/msg.naix'))
+    scrseqnaix = MyList(read_naix('files/fielddata/script/scr_seq_release.naix'))
+    areanaix = MyList(read_naix('files/fielddata/areadata/area_data.naix'))
+    mmlistnaix = MyList(read_naix('files/fielddata/mm_list/move_model_list.naix'))
+    matlistnaix = MyList(read_naix('files/fielddata/mapmatrix/map_matrix.naix'))
+    d_encdatanaix = MyList(read_naix('files/fielddata/encountdata/d_enc_data.naix'))
+    p_encdatanaix = MyList(read_naix('files/fielddata/encountdata/p_enc_data.naix'))
+    eventnaix = MyList(read_naix('files/fielddata/eventdata/zone_event_release.naix'))
+    with open(romfname, 'rb') as rom:
+        rom.seek(mapheado - arm9load + arm9offs)
+        for header in MapHeader.from_buffer(rom.read(mapheads)):  # type: MapHeader
+            print('{' + ', '.join((
+                areanaix[header.area_data_bank],
+                mmlistnaix[header.move_model_bank],
+                matlistnaix[header.matrix_id],
+                scrseqnaix[header.scripts_bank],
+                scrseqnaix[header.level_scripts_bank],
+                msgnaix[header.msg_bank],
+                sndseqs[header.day_music_id - 1000],
+                sndseqs[header.night_music_id - 1000],
+                '0xFFFF' if header.wild_encounter_bank == 0xFFFF else 'ENCDATA({}, {})'.format(
+                    d_encdatanaix[header.wild_encounter_bank],
+                    p_encdatanaix[header.wild_encounter_bank],
+                ),
+                eventnaix[header.events_bank],
+                mapsecs[header.mapsec],
+                str(header.weather_type),
+                str(header.camera_type),
+                str(header.unk16),
+                str(header.battle_bg),
+                ['FALSE', 'TRUE'][header.is_bike_allowed],
+                ['FALSE', 'TRUE'][header.is_running_allowed],
+                ['FALSE', 'TRUE'][header.is_escape_rope_allowed],
+                ['FALSE', 'TRUE'][header.is_fly_allowed],
+            )) + '},')
+
+
+if __name__ == '__main__':
+    main()
-- 
cgit v1.2.3