mirror of
https://github.com/blawar/ooot.git
synced 2024-07-02 09:03:36 +00:00
167 lines
3.7 KiB
Python
167 lines
3.7 KiB
Python
#!/usr/bin/python3
|
|
|
|
import os.path
|
|
from pathlib import Path
|
|
import sys
|
|
from oot import *
|
|
|
|
magic = '// GENERATED'
|
|
includeHeader = '#include "asset_common.h"'
|
|
removeHeaders = ['#include "z64animation.h"', '#include "z64bgcheck.h"', '#include "asset_common.h"']
|
|
noLuts = ['gameplay_keep', 'gameplay_dangeon', 'gameplay_field']
|
|
|
|
def writeFile(path, buffer):
|
|
with open(path, 'r') as f:
|
|
if f.read() == buffer:
|
|
return
|
|
|
|
print('writing %s' % path)
|
|
with open(path, 'w') as f:
|
|
f.write(buffer)
|
|
|
|
def clean(buffer):
|
|
lines = []
|
|
for line in buffer.split('\n'):
|
|
line = line.replace('\r', '').strip()
|
|
|
|
if len(line) == 0 or line[0] == '#':
|
|
continue
|
|
|
|
if line.startswith('static void*') or line.startswith(magic):
|
|
break
|
|
|
|
lines.append(line)
|
|
|
|
return '\n'.join(lines)
|
|
|
|
def cleanC(buffer):
|
|
lines = []
|
|
for line in buffer.split('\n'):
|
|
line = line.replace('\r', '').strip()
|
|
|
|
if line.startswith(magic):
|
|
break
|
|
|
|
lines.append(line)
|
|
|
|
return '\n'.join(lines)
|
|
|
|
dir = assetPath()
|
|
files = []
|
|
search = []
|
|
|
|
if len(sys.argv) > 1:
|
|
for x in sys.argv[1:]:
|
|
search.append(x.replace('\\', '/'))
|
|
|
|
for path in Path(dir).rglob('*.h'):
|
|
x = str(path).replace('\\', '/')
|
|
|
|
if len(search):
|
|
for s in search:
|
|
if s in x:
|
|
files.append(path)
|
|
break
|
|
else:
|
|
files.append(path)
|
|
|
|
for path in files:
|
|
x = str(path).replace('\\', '/')
|
|
skip = False
|
|
|
|
if '/text/' in x:
|
|
continue
|
|
|
|
if '/audio/' in x:
|
|
continue
|
|
|
|
if fixSlashPath(x).startswith(assetPath('misc')):
|
|
if x.split('/')[-2] == 'misc': # skip files directly in misc
|
|
continue
|
|
|
|
if '/overlays/' in x:
|
|
skip = True
|
|
|
|
name = path.name[0:-2]
|
|
c_path = str(path)[0:-2] + '.cpp'
|
|
|
|
lutVarName = name + '_lut'
|
|
|
|
with open(path, 'r') as f:
|
|
buffer = clean(f.read())
|
|
|
|
map = []
|
|
mapFull = []
|
|
lutMap = []
|
|
|
|
if not skip:
|
|
isAllBytes = True
|
|
for line in buffer.split('\n'):
|
|
if not line.startswith('extern '):
|
|
continue
|
|
bits = line.split(' ')
|
|
var = bits[-1]
|
|
|
|
if bits[-2] != 'u8':
|
|
isAllBytes = False
|
|
if var[-3:] != '[];':
|
|
#raise IOError('incorrect format: ' + var)
|
|
map.append(var[0:-1])
|
|
else:
|
|
map.append(var[0:-3])
|
|
mapFull.append(var[0:-1])
|
|
|
|
if isAllBytes:
|
|
lutMap.append(var[0:-3])
|
|
#print(var[0:-3])
|
|
|
|
result = '#pragma once\n'
|
|
result += buffer + '\n\n%s\n\n' % magic
|
|
|
|
if not skip:
|
|
if mapFull[0][-2:] == '[]':
|
|
startPtr = '(u8*)%s' % map[0]
|
|
else:
|
|
startPtr = '(u8*)&%s' % map[0]
|
|
|
|
if mapFull[-1][-2:] == '[]':
|
|
endPtr = '(u8*)%s' % map[-1]
|
|
else:
|
|
endPtr = '(u8*)&%s' % map[-1]
|
|
|
|
endPtr = '(u8*)&_%sSegmentRomLastSymbol' % name
|
|
|
|
if name not in noLuts and len(lutMap) > 0:
|
|
result += 'static void* %s[] = {\n%s\n};\n\n' % (lutVarName, ',\n'.join(lutMap))
|
|
|
|
result += 'extern u8 _%sSegmentRomLastSymbol;\n' % name
|
|
if not skip:
|
|
#result += '#define _%sSegmentStart (%s)\n' % (name, startPtr)
|
|
#result += '#define _%sSegmentEnd (%s)\n' % (name, endPtr)
|
|
result += '#define _%sSegmentRomStart (%s)\n' % (name, startPtr)
|
|
result += '#define _%sSegmentRomEnd (%s)\n' % (name, endPtr)
|
|
|
|
if skip:
|
|
with open(c_path, 'r') as f:
|
|
resultC = buffer = f.read()
|
|
else:
|
|
with open(c_path, 'r') as f:
|
|
buffer = cleanC(f.read()).strip()
|
|
resultC = buffer + '\n\n%s\n\n' % magic
|
|
resultC += 'u8 _%sSegmentRomLastSymbol = 0;\n' % (name)
|
|
|
|
#print(resultC)
|
|
#break
|
|
|
|
for i in removeHeaders:
|
|
resultC = resultC.replace(i + '\n', '')
|
|
|
|
if includeHeader not in resultC:
|
|
resultC = resultC.replace('#include "ultra64.h"', '#include "ultra64.h"\n' + includeHeader)
|
|
|
|
if '_scene.c' in c_path:
|
|
resultC = resultC.replace('#include "z64cutscene.h"', '')
|
|
resultC = resultC.replace('#include "ultra64.h"', '#include "ultra64.h"\n' + '#include "z64cutscene.h"')
|
|
|
|
writeFile(path, result)
|
|
writeFile(c_path, resultC) |