def testpatch2():
with open(f'{EXTRACT_ROOT_PATH}/DATA/files/Stage/D000/D000_stg_l0.arc.LZ',
'rb') as f:
extracted_data = nlzss11.decompress(f.read())
stagearc = U8File.parse_u8(BytesIO(extracted_data))
roomarc = U8File.parse_u8(
BytesIO(stagearc.get_file_data(f'rarc/D000_r00.arc')))
room = parseBzs(roomarc.get_file_data('dat/room.bzs'))
objects = room['LAY ']['l0']['OBJS']
# find chest with id 68 and replace the content with hookshot
for obj in objects:
if obj['name'] == b'TBox\x00\x00\x00\x00':
if (obj['unk4'] & 0xFE00) >> 9 == 68:
obj['posy'] = obj['posy'] + 50
obj['unk4'] = (obj['unk4'] & 0xFE00) + 0x14
print('patched hookshot')
if (obj['unk4'] & 0xFE00) >> 9 == 67:
obj['posy'] = obj['posy'] + 50
obj['unk4'] = (obj['unk4'] & 0xFE00) + 0x31
print('patched gust bellows')
roomarc.set_file_data('dat/room.bzs', buildBzs(room))
stagearc.set_file_data('rarc/D000_r00.arc', roomarc.to_buffer())
# add gust bellows and hookshot oarcs so they properly work
with open('oarc/GetHookShot.arc', 'rb') as f:
arc = f.read()
stagearc.add_file_data('oarc/GetHookShot.arc', arc)
with open('oarc/GetVacuum.arc', 'rb') as f:
arc = f.read()
stagearc.add_file_data('oarc/GetVacuum.arc', arc)
with open(f'{MODIFIED_ROOT_PATH}/DATA/files/Stage/D000/D000_stg_l0.arc.LZ',
'wb') as f:
f.write(nlzss11.compress(stagearc.to_buffer()))
def patch_faron():
with get_stagepath('F100', 0).open('rb') as f:
extracted_data = nlzss11.decompress(f.read())
stagearc = U8File.parse_u8(BytesIO(extracted_data))
# patch layers, force layer 1
stagedef = parseBzs(stagearc.get_file_data('dat/stage.bzs'))
stagedef['LYSE'] = [
OrderedDict((('story_flag', -1), ('night', 0), ('layer', 1)))
]
stagearc.set_file_data('dat/stage.bzs', buildBzs(stagedef))
room0arc = U8File.parse_u8(
BytesIO(stagearc.get_file_data('rarc/F100_r00.arc')))
roomdef = parseBzs(room0arc.get_file_data('dat/room.bzs'))
# grab the trial from layer 3 and put in on layer 0
trial = next(
filter(lambda x: x['name'] == 'WarpObj',
roomdef['LAY ']['l3']['OBJ ']))
trial_butterflies = next(
filter(lambda x: x['name'] == 'InsctTg',
roomdef['LAY ']['l3']['STAG']))
# trial['posy'] += 100
# fix object ID of trial
trial['id'] = 0x02F2
trial_butterflies['id'] = 0xFEF3
roomdef['LAY ']['l3']['OBJ '].remove(trial)
roomdef['LAY ']['l0']['OBJ '].append(trial)
roomdef['LAY ']['l3']['STAG'].remove(trial_butterflies)
roomdef['LAY ']['l0']['STAG'].append(trial_butterflies)
roomdef['LAY ']['l0']['ARCN'].append('SirenEntrance')
roomdef['LAY ']['l0']['ARCN'].append('PLSwordStick')
roomdef['LAY ']['l0']['OBJN'].append('WarpObj')
room0arc.set_file_data('dat/room.bzs', buildBzs(roomdef))
roomdat = BytesIO()
room0arc.writeto(roomdat)
stagearc.set_file_data('rarc/F100_r00.arc', roomdat.getbuffer())
# add the trial arc(s)
with open('oarc/SirenEntrance.arc', 'rb') as f:
arc = f.read()
stagearc.add_file_data('oarc/SirenEntrance.arc', arc)
with open('oarc/PLHarpPlay.arc', 'rb') as f:
arc = f.read()
stagearc.add_file_data('oarc/PLHarpPlay.arc', arc)
with open('oarc/PLSwordStick.arc', 'rb') as f:
arc = f.read()
stagearc.add_file_data('oarc/PLSwordStick.arc', arc)
stagedat = BytesIO()
stagearc.writeto(stagedat)
with get_stagepath('F100', 0, rootpath=MODIFIED_ROOT_PATH).open('wb') as f:
f.write(nlzss11.compress(stagedat.getbuffer()))
def extract_stage_rooms(name: str) -> OrderedDict:
with open(
f'{EXTRACT_ROOT_PATH}/DATA/files/Stage/{name}/{name}_stg_l0.arc.LZ',
'rb') as f:
extracted_data = nlzss11.decompress(f.read())
stagearc = U8File.parse_u8(BytesIO(extracted_data))
stage = parseBzs(stagearc.get_file_data('dat/stage.bzs'))
rooms = OrderedDict()
for i in range(len(stage['RMPL'])):
roomarc = U8File.parse_u8(
BytesIO(stagearc.get_file_data(f'rarc/{name}_r{i:02}.arc')))
rooms[f'r{i:02}'] = parseBzs(roomarc.get_file_data('dat/room.bzs'))
return stage, rooms
def extract_obj_pack():
data = (Path(__file__).parent / EXTRACT_ROOT_PATH / 'DATA' / 'files' /
'Object' / 'ObjectPack.arc.LZ').read_bytes()
data = U8File.parse_u8(BytesIO(nlzss11.decompress(data)))
oarc_cache_path = Path(__file__).parent / 'oarc'
for oarc in filter(lambda x: x.endswith('.arc'), data.get_all_paths()):
oarc_data = data.get_file_data(oarc)
(oarc_cache_path / oarc.split('/')[-1]).write_bytes(oarc_data)
def get_names():
with open(EXTRACT_ROOT_PATH + '/DATA/files/Stage/F000/F000_stg_l4.arc.LZ',
'rb') as f:
data = nlzss11.decompress(f.read())
data = BytesIO(data)
data = U8File.parse_u8(data)
# room=data.get_data('rarc/D000_r00.arc:dat/room.bzs')
for arc in len(data.get_all_paths):
if arc.endswith('.arc'):
print(arc)
def upgrade_test():
# patch stage
with get_stagepath('D000', 0).open('rb') as f:
extracted_data = nlzss11.decompress(f.read())
stagearc = U8File.parse_u8(BytesIO(extracted_data))
stagedef = parseBzs(stagearc.get_file_data('dat/stage.bzs'))
room0arc = U8File.parse_u8(
BytesIO(stagearc.get_file_data('rarc/D000_r00.arc')))
roomdef = parseBzs(room0arc.get_file_data('dat/room.bzs'))
# get chest
chest = next(
filter(lambda x: x['name'] == 'TBox', roomdef['LAY ']['l0']['OBJS']))
chest['anglez'] = (chest['anglez'] & ~0x1FF) | 53 # Beetle
room0arc.set_file_data('dat/room.bzs', buildBzs(roomdef))
# add both beetle models
with open('oarc/GetBeetleA.arc', 'rb') as h:
stagearc.add_file_data('oarc/GetBeetleA.arc', h.read())
with open('oarc/GetBeetleB.arc', 'rb') as h:
stagearc.add_file_data('oarc/GetBeetleB.arc', h.read())
stagearc.set_file_data('rarc/D000_r00.arc', room0arc.to_buffer())
# write back
with get_stagepath('D000', 0, rootpath=MODIFIED_ROOT_PATH).open('wb') as f:
f.write(nlzss11.compress(stagearc.to_buffer()))
# patch get item event
with open(
Path(__file__).parent / EXTRACT_ROOT_PATH / 'DATA' / 'files' /
'EU' / 'Object' / 'en_GB' / '0-Common.arc', 'rb') as f:
evntarc = U8File.parse_u8(BytesIO(f.read()))
itemmsbf = parseMSB(evntarc.get_file_data('0-Common/003-ItemGet.msbf'))
evnt = itemmsbf['FLW3']['flow'][
422] # event triggered after beetle text box
evnt['type'] = 'type3'
evnt['subType'] = 0
evnt['param1'] = 0
evnt['param3'] = 9
evnt['param2'] = 75 # Hook Beetle
evntarc.set_file_data('0-Common/003-ItemGet.msbf', buildMSB(itemmsbf))
with open(
Path(__file__).parent / MODIFIED_ROOT_PATH / 'DATA' / 'files' /
'EU' / 'Object' / 'en_GB' / '0-Common.arc', 'wb') as f:
f.write(evntarc.to_buffer())
def extract_objects():
try:
os.mkdir('oarc')
except:
pass
for (file, layer), objs in extracts.items():
with get_stagepath(file, layer).open('rb') as f:
data = nlzss11.decompress(f.read())
data = BytesIO(data)
data = U8File.parse_u8(data)
for objname in objs:
outdata = data.get_file_data(objname)
with open(objname, 'wb') as out:
out.write(outdata)
def testpatch():
# open the skyloft cave file
with open(EXTRACT_ROOT_PATH + '/DATA/files/Stage/D000/D000_stg_l0.arc.LZ',
'rb') as f:
# extract in memory
data = nlzss11.decompress(f.read())
data = BytesIO(data)
data = U8File.parse_u8(data)
# add hookshot and gust bellows
with open('oarc/GetHookShot.arc', 'rb') as h:
data.add_file_data('oarc/GetHookShot.arc', h.read())
with open('oarc/GetVacuum.arc', 'rb') as h:
data.add_file_data('oarc/GetVacuum.arc', h.read())
# open room
# room=parse_bzs(data.get_data('rarc/D000_r00.arc:dat/room.bzs'))
# get objects
# objects=room.children['LAY '].layers[0].children['OBJS'].objects
# find chest with id 68 and replace the content with hookshot
# for obj in objects:
# if obj['name']==b'TBox\x00\x00\x00\x00':
# if (obj['talk_behaviour']&0xF700)>>9 == 64:
# obj['talk_behaviour']==(obj['talk_behaviour']&0xF700)+0x14
# print('patched hookshot')
# for obj in objects:
# if obj['name']==b'TBox\x00\x00\x00\x00':
# if (obj['talk_behaviour']&0xF700)>>9 == 67:
# obj['talk_behaviour']==(obj['talk_behaviour']&0xF700)+0x31
# print('patched gust bellows')
# write room back
# data.update_file('rarc/D000_r00.arc:dat/room.bzs',build_bzs(room))
# write stage to memory
# with open('D000_stg_l0.arc', 'wb') as o:
# data.writeto(o)
return data