164 lines
3.9 KiB
Python
164 lines
3.9 KiB
Python
import os
|
|
|
|
usf_types = ['u8', 'u16', 'u32', 'u64', 's8', 's16', 's32', 's64', 'f32']
|
|
vec3_types = ['Vec3s', 'Vec3f']
|
|
typedef_pointers = ['BehaviorScript', 'ObjectAnimPointer']
|
|
|
|
exclude_structs = [
|
|
'SPTask',
|
|
'VblankHandler',
|
|
'GraphNodeRoot',
|
|
'MarioAnimDmaRelatedThing',
|
|
'UnusedArea28',
|
|
]
|
|
|
|
def get_path(p):
|
|
return os.path.dirname(os.path.realpath(__file__)) + '/../' + p
|
|
|
|
def translate_type_to_lvt(ptype):
|
|
if ptype == 'char':
|
|
ptype = 'u8'
|
|
|
|
if 'const ' in ptype:
|
|
ptype = ptype.replace('const ', '').strip()
|
|
|
|
if ('char' in ptype and '[' in ptype):
|
|
return 'LVT_STRING'
|
|
|
|
if ptype == 'char*':
|
|
return 'LVT_STRING_P'
|
|
|
|
if '[' in ptype or '{' in ptype:
|
|
return 'LOT_???'
|
|
|
|
if 'enum ' in ptype:
|
|
return 'LVT_S32'
|
|
|
|
if ptype == 'bool':
|
|
return 'LVT_BOOL'
|
|
|
|
if ptype in usf_types:
|
|
return 'LVT_' + ptype.upper()
|
|
|
|
if ptype in vec3_types:
|
|
return 'LVT_COBJECT'
|
|
|
|
if ptype == 'float':
|
|
return 'LVT_F32'
|
|
|
|
if ptype == 'LuaFunction':
|
|
return 'LVT_LUAFUNCTION'
|
|
|
|
if 'struct' in ptype:
|
|
if ptype.count('*') > 1:
|
|
return 'LVT_???'
|
|
if '*' in ptype:
|
|
return 'LVT_COBJECT_P'
|
|
return 'LVT_COBJECT'
|
|
|
|
if ptype.count('*') == 1 and '(' not in ptype and '[' not in ptype:
|
|
ptype = ptype.replace('const', '').replace('*', '').strip()
|
|
if ptype in usf_types or ptype in typedef_pointers:
|
|
return 'LVT_%s_P' % ptype.upper()
|
|
|
|
return 'LVT_???'
|
|
|
|
def translate_type_to_lot(ptype):
|
|
if ptype == 'const char*':
|
|
return 'LOT_NONE'
|
|
|
|
if ptype == 'char*' or ('char' in ptype and '[' in ptype):
|
|
return 'LOT_NONE'
|
|
|
|
if 'const ' in ptype:
|
|
ptype = ptype.replace('const ', '')
|
|
|
|
if ptype == 'char':
|
|
ptype = 'u8'
|
|
|
|
if '[' in ptype or '{' in ptype:
|
|
return 'LOT_???'
|
|
|
|
if 'enum ' in ptype:
|
|
return 'LOT_NONE'
|
|
|
|
if ptype == 'bool':
|
|
return 'LOT_NONE'
|
|
|
|
if ptype in usf_types:
|
|
return 'LOT_NONE'
|
|
|
|
if ptype in vec3_types:
|
|
return 'LOT_' + ptype.upper()
|
|
|
|
if ptype == 'float':
|
|
return 'LOT_NONE'
|
|
|
|
if ptype == 'LuaFunction':
|
|
return 'LOT_NONE'
|
|
|
|
if 'struct' in ptype:
|
|
if ptype.count('*') > 1:
|
|
return 'LOT_???'
|
|
|
|
struct_id = ptype.split(' ')[1].replace('*', '')
|
|
|
|
if struct_id in exclude_structs:
|
|
return 'LOT_???'
|
|
|
|
return 'LOT_' + struct_id.upper()
|
|
|
|
if ptype.count('*') == 1 and '???' not in translate_type_to_lvt(ptype):
|
|
return 'LOT_POINTER'
|
|
|
|
return 'LOT_???'
|
|
|
|
def translate_type_to_lua(ptype):
|
|
if ptype.startswith('struct '):
|
|
ptype = ptype.split(' ')[1].replace('*', '')
|
|
return ptype, 'structs.md#%s' % ptype
|
|
|
|
if ptype == 'const char*':
|
|
return '`string`', None
|
|
|
|
if ptype == 'char*' or ('char' in ptype and '[' in ptype):
|
|
return '`string`', None
|
|
|
|
if 'Vec3' in ptype:
|
|
return ptype, 'structs.md#%s' % ptype
|
|
|
|
if ptype.startswith('enum '):
|
|
return ptype, 'constants.md#%s' % ptype.replace(' ', '-')
|
|
|
|
if ptype in usf_types:
|
|
if ptype.startswith('f'):
|
|
return '`number`', None
|
|
else:
|
|
return '`integer`', None
|
|
|
|
if 'void' == ptype:
|
|
return None, None
|
|
|
|
if ptype == 'LuaFunction':
|
|
return '`Lua Function` ()', None
|
|
|
|
if ptype.count('*') == 1 and '???' not in translate_type_to_lvt(ptype):
|
|
ptype = ptype.replace('const', '').replace('*', '').strip()
|
|
s = '`Pointer` <%s>' % translate_type_to_lua(ptype)[0]
|
|
return s, None
|
|
|
|
if not ptype.startswith('`'):
|
|
ptype = '`' + ptype + '`'
|
|
|
|
return ptype, None
|
|
|
|
def gen_comment_header(f):
|
|
comment_h = "// " + f + " //"
|
|
comment_l = "/" * len(comment_h)
|
|
s = ""
|
|
s += " " + comment_l + "\n"
|
|
s += " " + comment_h + "\n"
|
|
s += "" + comment_l + "\n"
|
|
s += "\n"
|
|
return s
|