69e129805e
* More math improvements * more math functions * stack check * added error message when trying to modify read-only table; fix gSmluaConstants printed to console
103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
import sys
|
|
|
|
|
|
VEC3X_TO_VEC3Y = """
|
|
/* |description|
|
|
Converts a 3D {{desc}} vector `a` into a 3D {{desc_2}} vector and stores the result in `dest`
|
|
|descriptionEnd| */
|
|
INLINE OPTIMIZE_O3 Vec3{{suffix_2}}p vec3{{suffix}}_to_vec3{{suffix_2}}(Vec3{{suffix_2}} dest, Vec3{{suffix}} a) {
|
|
dest[0] = a[0]{{rounding_0}};
|
|
dest[1] = a[1]{{rounding_1}};
|
|
dest[2] = a[2]{{rounding_2}};
|
|
return dest;
|
|
}
|
|
"""
|
|
|
|
ROUNDING_FORMULA = " + ((a[%d] > 0) ? 0.5f : -0.5f)"
|
|
|
|
def vec3_write_conversion_functions(generated: str, curr_template: dict, templates: list, size: int) -> str:
|
|
for template in templates:
|
|
if template["suffix"] == curr_template["suffix"]:
|
|
continue
|
|
|
|
generated += VEC3X_TO_VEC3Y \
|
|
.replace("{{desc}}", curr_template["desc"]) \
|
|
.replace("{{suffix}}", curr_template["suffix"]) \
|
|
.replace("{{desc_2}}", template["desc"]) \
|
|
.replace("{{suffix_2}}", template["suffix"])
|
|
|
|
for i in range(size):
|
|
rounding_i = "{{rounding_%d}}" % (i)
|
|
generated = generated.replace(rounding_i, ROUNDING_FORMULA % (i) if curr_template["rounding"] else "")
|
|
|
|
return generated
|
|
|
|
|
|
TEMPLATES = {
|
|
"src/engine/math_util_vec3.tmpl": {
|
|
"size": 3,
|
|
"templates": [
|
|
{
|
|
"desc": "floating-point",
|
|
"type": "f32",
|
|
"suffix": "f",
|
|
"rounding": True
|
|
},
|
|
{
|
|
"desc": "integer",
|
|
"type": "s32",
|
|
"suffix": "i",
|
|
"rounding": False
|
|
},
|
|
{
|
|
"desc": "short integer",
|
|
"type": "s16",
|
|
"suffix": "s",
|
|
"rounding": False
|
|
}
|
|
],
|
|
"post-template": {
|
|
"function": vec3_write_conversion_functions,
|
|
"args": {
|
|
"size": 3
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
def main():
|
|
verbose = len(sys.argv) > 1 and (sys.argv[1] == "-v" or sys.argv[1] == "--verbose")
|
|
for template_file, x in TEMPLATES.items():
|
|
with open(template_file) as f:
|
|
tmpl = f.read()
|
|
templates = x["templates"]
|
|
post_template = x.get("post-template", None)
|
|
|
|
if verbose:
|
|
print("Generating the following files from template: %s" % (template_file))
|
|
|
|
for template in templates:
|
|
generated = (
|
|
"/* THIS FILE IS AUTO-GENERATED */\n" +
|
|
"/* DO NOT EDIT IT MANUALLY */\n\n" +
|
|
tmpl
|
|
)
|
|
|
|
for template_field, template_replacement in template.items():
|
|
generated = generated.replace("{{" + template_field + "}}", str(template_replacement))
|
|
|
|
if post_template:
|
|
generated = post_template["function"](generated, template, templates, **post_template["args"])
|
|
|
|
generated_filename = template_file[:template_file.rfind(".")] + template["suffix"] + ".inl"
|
|
with open(generated_filename, 'w', encoding='utf-8', newline='\n') as f:
|
|
f.write(generated)
|
|
|
|
if verbose:
|
|
print(">>> %s" % (generated_filename))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|