Refactor WASM binding directory structure
* All tests moved into wasm/codegen/tests folder * Merged wasm/codegen/helpers/ into wasm/codegen/generators/ PiperOrigin-RevId: 829471173 Change-Id: I2dbc5d9351771817ec260c7ddf87c31e66a9eabe
This commit is contained in:
committed by
Copybara-Service
parent
44220fcc51
commit
4e46db8903
+2
-61
@@ -14,73 +14,14 @@
|
||||
|
||||
"""Generates Javascript/TypeScript bindings for MuJoCo."""
|
||||
|
||||
import os
|
||||
|
||||
from introspect import ast_nodes
|
||||
from introspect import enums as introspect_enums
|
||||
from introspect import functions as introspect_functions
|
||||
|
||||
from wasm.codegen.generators import enums
|
||||
from wasm.codegen.generators import functions
|
||||
from wasm.codegen.generators import structs
|
||||
from wasm.codegen.helpers import common
|
||||
from wasm.codegen.helpers import constants as _constants
|
||||
from wasm.codegen.helpers import functions as function_utils
|
||||
|
||||
|
||||
class BindingBuilder:
|
||||
"""Builds WASM bindings for MuJoCo."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
template_path_cc: str,
|
||||
):
|
||||
with open(template_path_cc, "r") as f:
|
||||
self.content_cc = f.readlines()
|
||||
|
||||
self.markers_and_content = []
|
||||
|
||||
def set_enums(self):
|
||||
"""Generates and sets the enum bindings."""
|
||||
generator = enums.Generator(introspect_enums.ENUMS)
|
||||
self.markers_and_content += generator.generate()
|
||||
return self
|
||||
|
||||
def set_structs(self):
|
||||
"""Generates and sets the struct bindings."""
|
||||
generator = structs.Generator()
|
||||
self.markers_and_content += generator.generate()
|
||||
return self
|
||||
|
||||
def set_functions(self):
|
||||
"""Generates and sets the function wrappers and bindings."""
|
||||
|
||||
functions_to_bind: dict[str, ast_nodes.FunctionDecl] = {}
|
||||
for name, func in introspect_functions.FUNCTIONS.items():
|
||||
if not function_utils.is_excluded_function_name(name):
|
||||
if name not in _constants.BOUNDCHECK_FUNCS:
|
||||
functions_to_bind[name] = func
|
||||
|
||||
generator = functions.Generator(functions_to_bind)
|
||||
self.markers_and_content += generator.generate()
|
||||
return self
|
||||
|
||||
def to_string(self) -> str:
|
||||
for marker, content in self.markers_and_content:
|
||||
self.content_cc = common.replace_lines_containing_marker(
|
||||
self.content_cc, marker, content
|
||||
)
|
||||
return "".join(self.content_cc)
|
||||
|
||||
def build(self, generated_path_cc: str):
|
||||
common.write_to_file(generated_path_cc, self.to_string())
|
||||
from wasm.codegen.generators import binding_builder
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
template_file = "wasm/codegen/templates/bindings.cc"
|
||||
generated_file = "wasm/codegen/generated/bindings.cc"
|
||||
|
||||
builder = BindingBuilder(template_file)
|
||||
builder = binding_builder.BindingBuilder(template_file)
|
||||
builder.set_enums()
|
||||
builder.set_structs()
|
||||
builder.set_functions()
|
||||
|
||||
Reference in New Issue
Block a user