Files
Diddy-Kong-Racing/tools/python/calc_func_checksums.py
T
FazanaandGitHub e47fc4b0ef Couple matches (#270)
* match func_80077268

* Update borders.c

* Update borders.h

* game.c labels

* rename border

* match func_80068BF4 and func_80069484

* Update camera.c

* camera.c labels

* more camera.c documentation

oh god this is gonna be another one of THOSE PR's again isn't it?

* Update camera.c

* destruction

* nonmatching func_80002DF8

* make clean now removes the linker

* Update unknown_003260.c

* match func_800A0EB4

* match render_background

* fix warnings

* style fixes

* label indicator arrows in the UI

* requested changes

* struct renames

* match func_800A7B68

* Update README.md

* match func_800A5A64

* Fix unforgivable typo

sorry im not a fake fan I sware on me nan

* Update README.md

* more detailed documentation of game_ui

* match func_800A01A0

* prelim changes

* oh no

* nonmatching func_80024D54

* labbel

* fix build

* fix context warnings

* match func_800A7520

* fixes

* requested changes
2022-10-30 19:58:59 +00:00

86 lines
3.0 KiB
Python

import re
import os
import sys
from file_util import FileUtil
VERSION = sys.argv[1]
BUILD_DIR = 'build/' + VERSION
MAP_FILEPATH = BUILD_DIR + '/dkr.map'
ROM_FILEPATH = BUILD_DIR + '/dkr.z64'
FUNCTIONS_TO_CALC = [
# function checksum variable func size variable
('func_80019808', 'gFunc80019808Checksum', 'gFunc80019808Length'),
('render_scene', 'gTractionTableChecksum', 'gFunc80024D54Length'),
('obj_loop_goldenballoon', 'gObjLoopGoldenBalloonChecksum', 'gObjLoopGoldenBalloonLength'),
('func_80068158', 'gFunc80068158Checksum', 'gFunc80068158Length')
]
def getMatches(string, regex):
out = []
matches = re.finditer(regex, string, re.MULTILINE)
for matchNum, match in enumerate(matches, start=1):
start = match.start()
end = match.end()
out.append((string[start:end], match.groups(), (start, end)))
return out
rom = FileUtil.get_bytes_from_file(ROM_FILEPATH)
mapFile = {}
RAM_TO_ROM = None
def write_word_to_rom(romOffset, value):
global rom
rom[romOffset + 0] = (value >> 24) & 0xFF
rom[romOffset + 1] = (value >> 16) & 0xFF
rom[romOffset + 2] = (value >> 8) & 0xFF
rom[romOffset + 3] = value & 0xFF
def calculate_checksum_for_function(funcLabel, varLabel, funcSizeLabel):
if funcLabel not in mapFile:
raise Exception('Label "' + funcLabel + '" does not exist in the map file!')
if varLabel not in mapFile:
raise Exception('Label "' + varLabel + '" does not exist in the map file!')
if funcSizeLabel not in mapFile:
raise Exception('Label "' + funcSizeLabel + '" does not exist in the map file!')
funcEntry = mapFile[funcLabel]
funcAddress = funcEntry['value']
funcLength = funcEntry['length']
funcRomOffset = funcAddress - RAM_TO_ROM
count = 0
for i in range(0, funcLength):
count += rom[funcRomOffset + i]
# Save function checksum.
varEntry = mapFile[varLabel]
varAddress = varEntry['value']
varRomOffset = varAddress - RAM_TO_ROM
write_word_to_rom(varRomOffset, count)
# Save function size.
sizeEntry = mapFile[funcSizeLabel]
sizeAddress = sizeEntry['value']
sizeRomOffset = sizeAddress - RAM_TO_ROM
write_word_to_rom(sizeRomOffset, funcLength)
def calculate_matches():
global mapFile, RAM_TO_ROM
REGEX_MAP_GET_LABEL = r"[ ]*?0x[0-9A-Fa-f]{8}([0-9A-Fa-f]{8})[ ]*?([_A-Za-z0-9]+)"
mapText = FileUtil.get_text_from_file(MAP_FILEPATH)
mapMatches = getMatches(mapText, REGEX_MAP_GET_LABEL)
for i in range(0, len(mapMatches) - 1):
match = mapMatches[i]
labelValue = match[1]
value = int(labelValue[0], 16)
length = int(mapMatches[i + 1][1][0], 16) - value
mapFile[labelValue[1]] = { "value": value, "length": length }
RAM_TO_ROM = mapFile['__RAM_TO_ROM']['value']
def main():
calculate_matches()
for entry in FUNCTIONS_TO_CALC:
calculate_checksum_for_function(entry[0], entry[1], entry[2])
FileUtil.write_bytes_to_file(ROM_FILEPATH, rom)
main()