Add JavaScript bindings and WASM support

Co-authored-by: Matija Kecman <matijak@google.com> Co-authored-by: Sebastian Noreña Rendón <sebas.norena@creativa77.com.ar> Co-authored-by: Kyle Bayes <kylebayes@google.com> PiperOrigin-RevId: 826479070 Change-Id: I25acf36e6c20b091d8492e10798d028ae66f6733
2025-10-31 07:20:18 -07:00
parent 57f7145806
commit 4086261714
58 changed files with 24418 additions and 22 deletions
@@ -0,0 +1,75 @@
+# Copyright 2025 DeepMind Technologies Limited
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Classes used by the parser to generate the structs wrappers and bindings statements."""
+
+import dataclasses
+from typing import Dict, List
+
+from wasm.codegen.helpers import common
+
+
+@dataclasses.dataclass
+class WrappedFieldData:
+  """Data class for struct field definition and binding."""
+
+  # Line for struct field binding
+  binding: str
+
+  # Line for struct field definition
+  definition: str | None = None
+
+  # Initialization code for fields that require it
+  initialization: str | None = None
+
+  # Statement to reset the inner pointer when copying the field
+  ptr_copy_reset: str | None = None
+
+  # Whether the field is a primitive or fixed size
+  is_primitive_or_fixed_size: bool = False
+
+
+@dataclasses.dataclass
+class WrappedStructData:
+  """Data class for struct wrapper definition and binding."""
+
+  # Name of wrapper struct
+  wrap_name: str
+
+  # List of WrappedFieldData for this struct
+  wrapped_fields: List[WrappedFieldData]
+
+  # Struct header code
+  wrapped_header: str
+
+  # Struct source code
+  wrapped_source: str
+
+  # Whether to use shallow copy for this struct
+  use_shallow_copy: bool = True
+
+
+def create_wrapped_structs_set_up_data(
+    struct_names: List[str],
+) -> Dict[str, WrappedStructData]:
+  """Creates a dictionary of WrappedStructData for the given struct names."""
+  return {
+      name: WrappedStructData(
+          wrap_name=common.uppercase_first_letter(name),
+          wrapped_fields=[],
+          wrapped_header="",
+          wrapped_source="",
+      )
+      for name in struct_names
+  }