138 lines
3.7 KiB
Python
138 lines
3.7 KiB
Python
import os
|
|
import argparse
|
|
import json
|
|
import re
|
|
from keyword import iskeyword
|
|
from collections import defaultdict
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
|
|
BASEDIR = os.path.dirname(__file__)
|
|
env = Environment(
|
|
loader=FileSystemLoader(f"{BASEDIR}/pool_arrays_templates"),
|
|
trim_blocks=True,
|
|
lstrip_blocks=True,
|
|
)
|
|
|
|
|
|
class TypeItem:
|
|
def __init__(self, **kwargs):
|
|
self.__dict__.update(**kwargs)
|
|
|
|
|
|
TYPES = [
|
|
# Base types
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_int_array",
|
|
py_pool=f"PoolIntArray",
|
|
gd_value=f"pandemonium_int",
|
|
py_value=f"pandemonium_int",
|
|
is_base_type=True,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_real_array",
|
|
py_pool=f"PoolRealArray",
|
|
gd_value=f"pandemonium_real",
|
|
py_value=f"pandemonium_real",
|
|
is_base_type=True,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool="pandemonium_pool_byte_array",
|
|
py_pool="PoolByteArray",
|
|
gd_value="uint8_t",
|
|
py_value="uint8_t",
|
|
is_base_type=True,
|
|
is_stack_only=True,
|
|
),
|
|
# Stack only builtin types
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector2_array",
|
|
py_pool=f"PoolVector2Array",
|
|
gd_value=f"pandemonium_vector2",
|
|
py_value=f"Vector2",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector2i_array",
|
|
py_pool=f"PoolVector2iArray",
|
|
gd_value=f"pandemonium_vector2i",
|
|
py_value=f"Vector2i",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector3_array",
|
|
py_pool=f"PoolVector3Array",
|
|
gd_value=f"pandemonium_vector3",
|
|
py_value=f"Vector3",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector3i_array",
|
|
py_pool=f"PoolVector3iArray",
|
|
gd_value=f"pandemonium_vector3i",
|
|
py_value=f"Vector3i",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector4_array",
|
|
py_pool=f"PoolVector4Array",
|
|
gd_value=f"pandemonium_vector4",
|
|
py_value=f"Vector4",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_vector4i_array",
|
|
py_pool=f"PoolVector4iArray",
|
|
gd_value=f"pandemonium_vector4i",
|
|
py_value=f"Vector4i",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
TypeItem(
|
|
gd_pool=f"pandemonium_pool_color_array",
|
|
py_pool=f"PoolColorArray",
|
|
gd_value=f"pandemonium_color",
|
|
py_value=f"Color",
|
|
is_base_type=False,
|
|
is_stack_only=True,
|
|
),
|
|
# Stack&heap builtin types
|
|
TypeItem(
|
|
gd_pool="pandemonium_pool_string_array",
|
|
py_pool="PoolStringArray",
|
|
gd_value="pandemonium_string",
|
|
py_value="GDString",
|
|
is_base_type=False,
|
|
is_stack_only=False,
|
|
),
|
|
]
|
|
|
|
|
|
def generate_pool_array(output_path):
|
|
template = env.get_template("pool_arrays.tmpl.pyx")
|
|
out = template.render(types=TYPES)
|
|
with open(output_path, "w") as fd:
|
|
fd.write(out)
|
|
|
|
pxd_output_path = output_path.rsplit(".", 1)[0] + ".pxd"
|
|
template = env.get_template("pool_arrays.tmpl.pxd")
|
|
out = template.render(types=TYPES)
|
|
with open(pxd_output_path, "w") as fd:
|
|
fd.write(out)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(
|
|
description="Generate pandemonium pool_x_array builtins bindings files"
|
|
)
|
|
parser.add_argument("--output", "-o", default=None)
|
|
args = parser.parse_args()
|
|
generate_pool_array(args.output or f"pool_arrays.pyx")
|