mirror of
https://github.com/zeldaret/oot.git
synced 2024-11-14 05:19:36 +00:00
f4a72303cb
* remove trailing whitespaces * minor docs tweaks * some more trailing whitespaces * few more tweaks
126 lines
3.2 KiB
Python
126 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import struct
|
|
import json
|
|
|
|
table = None
|
|
|
|
class AddrRange:
|
|
def __init__(self, start, end):
|
|
self.start = start
|
|
self.end = end
|
|
def __str__(self):
|
|
return "{0:08X}:{1:08X}".format(self.start, self.end)
|
|
|
|
class FileResult:
|
|
def __init__(self, name, vrom, vram, offset):
|
|
self.name = name
|
|
self.vrom = vrom
|
|
self.vram = vram
|
|
self.offset = offset
|
|
|
|
def __str__(self):
|
|
return "{0}: Rom {1} VRam {2} Offset {3:06X}".format(self.name, self.vrom, self.vram, self.offset)
|
|
|
|
def PVA(a):
|
|
return a & 0x1FFFFFF
|
|
def UVA(a):
|
|
#a *= 0x10
|
|
a += 0x80000000
|
|
return a
|
|
|
|
def CreateAddrLookup(dict, recs, tracer):
|
|
recs = sorted(recs, key= lambda x: x[0])
|
|
length = len(recs)
|
|
print(length)
|
|
recs.append((recs[length-1][1], None, None))
|
|
recs = sorted(recs, key=lambda x: x[0], reverse=True)
|
|
|
|
for i in range(length):
|
|
if recs[i][0] != recs[i+1][1]:
|
|
if tracer != "ra" or recs[i][0] != recs[i+1][1] + 0xFFF & -0x1000:
|
|
print(tracer)
|
|
print(recs[i])
|
|
print(recs[i+1])
|
|
for item in recs:
|
|
dict[item[0]] = item[2]
|
|
|
|
def CreateTable():
|
|
|
|
vrecs = []
|
|
rrecs = []
|
|
vrecs.append((PVA(0x80157D90), PVA(0x80800000), None))
|
|
|
|
dict = {
|
|
"va" : {},
|
|
"ra" : {},
|
|
"ft" : {}
|
|
}
|
|
|
|
with open("filetable.txt", "r") as file:
|
|
for line in file:
|
|
li = line.rstrip().split('\t')
|
|
rrec = (int(li[2],16), int(li[3],16), li[7])
|
|
vrec = (PVA(int(li[4],16)), PVA(int(li[5],16)), li[7])
|
|
rrecs.append(rrec)
|
|
vrecs.append(vrec)
|
|
dict["ft"][li[7]] = (rrec[0], rrec[1])
|
|
file = None
|
|
|
|
with open("filetable2.txt", "r") as file:
|
|
for line in file:
|
|
li = line.rstrip().split('\t')
|
|
rrec = (int(li[2],16), int(li[3],16), li[5])
|
|
rrecs.append(rrec)
|
|
dict["ft"][li[5]] = (rrec[0], rrec[1])
|
|
|
|
CreateAddrLookup(dict["va"], vrecs, "va")
|
|
CreateAddrLookup(dict["ra"], rrecs, "ra")
|
|
|
|
with open("filetable.json", "w") as table:
|
|
json.dump(dict, table)
|
|
|
|
def AddressLookup(lookupTable, addr):
|
|
start = 0;
|
|
end = 0;
|
|
key = None;
|
|
|
|
for k, v in lookupTable.items():
|
|
if addr >= k:
|
|
start = k
|
|
key = v
|
|
break
|
|
end = k
|
|
return (key, start, end)
|
|
|
|
def GetFromVRam(addr):
|
|
key, vramStart, vramEnd = AddressLookup(table["va"], PVA(addr))
|
|
|
|
if key is None:
|
|
return None
|
|
|
|
ftl = table["ft"][key]
|
|
vrom = AddrRange(ftl[0], ftl[1])
|
|
vram = AddrRange(UVA(vramStart), UVA(vramEnd))
|
|
|
|
offset = addr - vram.start
|
|
|
|
return FileResult(key, vrom, vram, offset)
|
|
|
|
def GetFromRom(addr):
|
|
key, romStart, romEnd = AddressLookup(table["ra"], addr)
|
|
|
|
if key is None:
|
|
return None
|
|
|
|
vrom = AddrRange(romStart, romEnd)
|
|
offset = addr - vrom.start
|
|
|
|
return FileResult(key, vrom, None, offset)
|
|
|
|
#CreateTable()
|
|
with open(os.path.join(os.path.dirname(__file__), 'filetable.json'), "r") as table:
|
|
table = json.load(table)
|
|
table["va"] = {int(k):v for k,v in table["va"].items()}
|
|
table["ra"] = {int(k):v for k,v in table["ra"].items()}
|