def read(self, br: BinaryReader):
br.seek(0, io.SEEK_SET)
self.magic = br.read(4)
if self.magic != b"smdl":
raise ValueError("SMDHeader does not start with magic value")
br.read_uint32() # 0
self.file_length = br.read_uint32()
self.version = br.read_uint16()
self.unk1 = br.read_uint8()
self.unk2 = br.read_uint8()
# 8 bytes of 0
br.read_uint32()
br.read_uint32()
self.year = br.read_uint16()
self.month = br.read_uint8()
self.day = br.read_uint8()
self.hour = br.read_uint8()
self.minute = br.read_uint8()
self.second = br.read_uint8()
self.centisecond = br.read_uint8()
self.file_name = br.read_string(16, encoding=None, pad=b"\xFF")
br.read_uint32() # 0x1
br.read_uint32() # 0x1
br.read_uint32() # 0xFFFFFFFF
br.read_uint32() # 0xFFFFFFFF
def read_stream(self, stream: BinaryIO):
if isinstance(stream, BinaryReader):
rdr = stream
else:
rdr = BinaryReader(stream)
with open("test.bin", "wb+") as f:
f.write(rdr.read())
rdr.seek(0)
palette_lenght = rdr.read_uint32()
for color_i in range(palette_lenght):
self.palette[color_i] = ndspy.color.unpack255(rdr.read_uint16())
if color_i:
self.palette[color_i, 3] = 255
n_tiles = rdr.read_uint32()
tiles = np.frombuffer(rdr.read(n_tiles * 0x40), np.uint8).reshape((n_tiles, 8, 8))
map_w = rdr.read_uint16()
map_h = rdr.read_uint16()
img_w = map_w * 8
img_h = map_h * 8
self.image = np.zeros((img_h, img_w), np.uint8)
for map_y in range(map_h):
for map_x in range(map_w):
img_y = map_y * 8
img_x = map_x * 8
self.image[img_y:img_y + 8, img_x:img_x + 8] = tiles[rdr.read_uint16()]
def read(self, br: BinaryReader):
br.seek(0)
self.magic = br.read(4)
if self.magic != b"swdl":
raise ValueError("SWDHeader does not start with magic value")
br.read_uint32() # 0
self.file_length = br.read_uint32()
self.version = br.read_uint16()
if self.version != 0x415:
raise ValueError("SMDHeader does not have correct version")
self.is_sample_bank = br.read_bool()
self.group = br.read_uint8()
# 8 bytes of 0
br.read_uint32()
br.read_uint32()
self.year = br.read_uint16()
self.month = br.read_uint8()
self.day = br.read_uint8()
self.hour = br.read_uint8()
self.minute = br.read_uint8()
self.second = br.read_uint8()
self.centisecond = br.read_uint8()
self.file_name = br.read_string(16, encoding=None, pad=b"\xAA")
self.unk10 = br.read_uint32()
# 8 bytes of 0
br.read_uint32()
br.read_uint32()
self.unk13 = br.read_uint32()
self.pcmdlen = br.read_uint32()
self.unk14 = br.read_uint32()
self.wavi_slot_count = br.read_uint16()
self.prgi_slot_count = br.read_uint16()
self.unk17 = br.read_uint16()
self.wavi_len = br.read_uint32()
def read_stream(self, stream):
if not isinstance(stream, BinaryReader):
rdr = BinaryReader(stream)
else:
rdr = stream
self.original_header = rdr.read(0x100)
rdr.seek(0)
self.chunk_id = rdr.read(4)
if self.chunk_id != b"sadl":
raise ValueError("SADL does not start with magic value")
rdr.seek(0x31)
self.loop_flag = rdr.read_uint8()
self.channels = rdr.read_uint8()
coding = rdr.read_uint8()
if coding & 0x06 == 4:
self.sample_rate = 32728
elif coding & 0x06 == 2:
self.sample_rate = 16364
self.coding = coding & 0xf0
rdr.seek(0x40)
self.file_size = rdr.read_uint32()
if self.coding == Coding.INT_IMA:
self.num_samples = int(
(self.file_size - 0x100) / self.channels * 2)
elif self.coding == Coding.NDS_PROCYON:
self.num_samples = int(
(self.file_size - 0x100) / self.channels / 16 * 30)
else:
raise NotImplementedError()
rdr.seek(0x54)
if self.loop_flag != 0:
if self.coding == Coding.INT_IMA:
self.loop_offset = int(
(rdr.read_uint32() - 0x100) / self.channels * 2)
elif self.coding == Coding.NDS_PROCYON:
self.loop_offset = int(
(rdr.read_uint32() - 0x100) / self.channels / 16 * 30)
rdr.seek(0x100)
buffer = rdr.read(self.file_size - 0x100)
buffer = np.frombuffer(buffer, dtype=np.uint8)
buffer = buffer.reshape(
((self.file_size - 0x100) // 0x10 // self.channels, self.channels,
0x10))
buffer = buffer.swapaxes(0, 1)
self.buffer = buffer.reshape(
(buffer.shape[0], buffer.shape[1] * buffer.shape[2]))
self.offset = [0] * self.channels
self.ima_decoders = []
self.procyon_decoders = []
for i in range(self.channels):
self.ima_decoders.append(ima_adpcm.ImaAdpcm())
self.procyon_decoders.append(procyon.Procyon())
self.blocks_done = 0
def read_stream(self, stream: BinaryIO):
if isinstance(stream, BinaryReader):
rdr = stream
else:
rdr = BinaryReader(stream)
n_entries = rdr.read_uint16()
header_length = rdr.read_uint16()
entry_length = rdr.read_uint16()
rdr.seek(header_length)
self._entries = []
for i in range(n_entries):
self._entries.append(rdr.read(entry_length))
def read(self, br: BinaryReader):
br.seek(0x40, io.SEEK_SET)
self.label = br.read(4)
if self.label != b"song":
raise ValueError("SongChunk does not start with magic value")
self.unk1 = br.read_uint32()
br.read_uint32() # 0xFF10
br.read_uint32() # 0xFFFFFFB0
br.read_uint16() # 0x1
self.tpqn = br.read_uint16()
br.read_uint16() # 0xFF10
self.num_tracks = br.read_uint8()
self.num_channels = br.read_uint8()
br.read_uint32() # 0x0F000000
br.read_uint32() # 0xFFFFFFFF
br.read_uint32() # 0x40000000
br.read_uint32() # 0x00404000
br.read_uint16() # 0x0200
br.read_uint16() # 0x0800
br.read_uint32() # 0xFFFFFF00
br.read(16) # 16 0xFF Padding
def read_stream(self, stream):
if not isinstance(stream, BinaryReader):
rdr = BinaryReader(stream)
else:
rdr = stream
self.chunk_id = rdr.read(4)
self.chunk_size = rdr.read_uint32()
self.format = rdr.read(4)
if self.chunk_id != b"RIFF" or self.format != b"WAVE":
raise NotImplementedError()
self.fmt.read(rdr)
if self.fmt.audio_format != WaveFormat.WAVE_FORMAT_PCM or self.fmt.bits_per_sample == 0x08:
raise NotImplementedError()
rdr.seek(0x14 + self.fmt.chunk_size)
data_id = rdr.read(4)
while data_id != b"data":
offset = rdr.read_uint32()
rdr.seek(rdr.tell() + offset)
data_id = rdr.read(4)
rdr.seek(-4, os.SEEK_CUR)
self.data.read(rdr, self.fmt)
def load(self, rdr):
if not isinstance(rdr, BinaryReader):
rdr = BinaryReader(rdr)
rdr: BinaryReader
self.original = rdr.read()
rdr.seek(0)
self.number = rdr.read_uint16()
rdr.read_uint16() # 112
self.title = rdr.read_string(encoding=self.encoding)
rdr.seek(0x34)
self.tutorial_id = rdr.read_uint8()
for i in range(3):
self.picarat_decay[i] = rdr.read_uint8()
self._flags = rdr.read_uint8()
self.location_id = rdr.read_uint8()
self.type = rdr.read_uint8()
self.bg_btm_id = rdr.read_uint8()
rdr.read_uint16()
self.bg_top_id = rdr.read_uint8()
self.reward_id = rdr.read_uint8()
puzzle_text_offset = 0x70 + rdr.read_uint32()
puzzle_correct_answer_offset = 0x70 + rdr.read_uint32()
puzzle_incorrect_answer_offset = 0x70 + rdr.read_uint32()
puzzle_hint1_offset = 0x70 + rdr.read_uint32()
puzzle_hint2_offset = 0x70 + rdr.read_uint32()
puzzle_hint3_offset = 0x70 + rdr.read_uint32()
rdr.seek(puzzle_text_offset)
self.text = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
rdr.seek(puzzle_correct_answer_offset)
self.correct_answer = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
rdr.seek(puzzle_incorrect_answer_offset)
self.incorrect_answer = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
rdr.seek(puzzle_hint1_offset)
self.hint1 = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
rdr.seek(puzzle_hint2_offset)
self.hint2 = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
rdr.seek(puzzle_hint3_offset)
self.hint3 = subs.replace_substitutions(rdr.read_string(encoding=self.encoding), True)
def read_stream(self, stream: BinaryIO):
if isinstance(stream, BinaryReader):
rdr = stream
else:
rdr = BinaryReader(stream)
n_images = rdr.read_uint16()
self.colordepth = 4 if rdr.read_uint16() == 3 else 8
# import the images
self.images = []
for img_i in range(n_images):
img_w = rdr.read_uint16()
img_h = rdr.read_uint16()
n_parts = rdr.read_uint16()
img = np.zeros((img_h, img_w), np.uint8)
rdr.seek(2, SEEK_CUR)
for part_i in range(n_parts):
part_x = rdr.read_uint16()
part_y = rdr.read_uint16()
part_w = 2**(3 + rdr.read_uint16())
part_h = 2**(3 + rdr.read_uint16())
part: np.ndarray
if self.colordepth == 8:
part = np.frombuffer(rdr.read(part_h * part_w), np.uint8)
part = part.reshape((part_h, part_w))
else:
bufpart = np.frombuffer(rdr.read(part_h * part_w // 2),
np.uint8)
part = np.zeros((part_w * part_h), np.uint8)
part[0::2] = bufpart & 0xf
part[1::2] = bufpart >> 4
part = part.reshape((part_h, part_w))
if (part_x + part_w) > img_w:
part_w = img_w - part_x
if (part_y + part_h) > img_h:
part_h = img_h - part_y
img[part_y:part_y + part_h,
part_x:part_x + part_w] = part[:part_h, :part_w]
self.images.append(img)
self.palette = np.zeros((256, 4), np.uint8)
palette_length = rdr.read_uint32()
for color_i in range(palette_length):
self.palette[color_i] = ndspy.color.unpack255(rdr.read_uint16())
if color_i:
self.palette[color_i, 3] = 255
rdr.seek(0x1E, SEEK_CUR)
n_animations = rdr.read_uint32()
animation_names = rdr.read_string_array(n_animations, 0x1e)
animation_frame_sets = []
for i in range(n_animations):
n_frames = rdr.read_uint32()
frame_indexes = rdr.read_uint32_array(n_frames)
frame_durations = rdr.read_uint32_array(n_frames)
image_indexes = rdr.read_uint32_array(n_frames)
animation_frame_sets.append([
AnimationFrame(index=frame_indexes[i],
duration=frame_durations[i],
image_index=image_indexes[i])
for i in range(n_frames)
])
self.animations = [
Animation(name=animation_names[i], frames=animation_frame_sets[i])
for i in range(n_animations)
]
if rdr.read_uint16() != 0x1234:
return # We hit end of stream
variable_labels = rdr.read_string_array(16, 16)
variable_data = [[] for _ in range(16)]
for _ in range(8):
for var_i in range(16):
variable_data[var_i].append(rdr.read_int16())
self.variables = {
variable_labels[i]: variable_data[i]
for i in range(16)
}
for anim in self.animations:
anim.child_image_x = rdr.read_uint16()
for anim in self.animations:
anim.child_image_y = rdr.read_uint16()
for anim in self.animations:
anim.child_image_animation_index = rdr.read_uint8()
self.child_image = rdr.read_string(128)
