gdnative_python/generation/generate_pool_arrays.py

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"godot_pool_int_array",
        py_pool=f"PoolIntArray",
        gd_value=f"godot_int",
        py_value=f"godot_int",
        is_base_type=True,
        is_stack_only=True,
    ),
    TypeItem(
        gd_pool=f"godot_pool_real_array",
        py_pool=f"PoolRealArray",
        gd_value=f"godot_real",
        py_value=f"godot_real",
        is_base_type=True,
        is_stack_only=True,
    ),
    TypeItem(
        gd_pool="godot_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"godot_pool_vector2_array",
        py_pool=f"PoolVector2Array",
        gd_value=f"godot_vector2",
        py_value=f"Vector2",
        is_base_type=False,
        is_stack_only=True,
    ),
    TypeItem(
        gd_pool=f"godot_pool_vector3_array",
        py_pool=f"PoolVector3Array",
        gd_value=f"godot_vector3",
        py_value=f"Vector3",
        is_base_type=False,
        is_stack_only=True,
    ),
    TypeItem(
        gd_pool=f"godot_pool_color_array",
        py_pool=f"PoolColorArray",
        gd_value=f"godot_color",
        py_value=f"Color",
        is_base_type=False,
        is_stack_only=True,
    ),
    # Stack&heap builtin types
    TypeItem(
        gd_pool="godot_pool_string_array",
        py_pool="PoolStringArray",
        gd_value="godot_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 godot 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")