pure python js_webgpu backend using pyodide

codegen/__init__.py

@@ -1,20 +1,24 @@
 import io
 
 from .utils import print, PrintToFile
-from . import apiwriter, apipatcher, wgpu_native_patcher, idlparser, hparser
+from . import apiwriter, apipatcher, wgpu_native_patcher, idlparser, hparser, jswriter
 from .files import file_cache
 
 
-def main():
+def main(do_api=True, do_wgpu_native=True, do_js=False):
     """Codegen entry point. This will populate the file cache with the
     new code, but not write it to disk."""
 
     log = io.StringIO()
     with PrintToFile(log):
         print("# Code generation report")
         prepare()
-        update_api()
-        update_wgpu_native()
+        if do_api:
+            update_api()
+        if do_wgpu_native:
+            update_wgpu_native()
+        if do_js:
+            update_js()
         file_cache.write("resources/codegen_report.md", log.getvalue())
 
 
@@ -63,3 +67,19 @@ def update_wgpu_native():
     code1 = file_cache.read("backends/wgpu_native/_api.py")
     code2 = wgpu_native_patcher.patch_wgpu_native_backend(code1)
     file_cache.write("backends/wgpu_native/_api.py", code2)
+
+
+def update_js():
+    """
+    Writes? (maybe updates later) the JS webgpu backend API.
+    """
+
+    print("## Writing backends/js_webgpu/_api.py")
+
+    code = jswriter.generate_js_webgpu_api()
+    # TODO: run the code against a patcher that adds hand written API diff methods
+
+    file_cache.write("backends/js_webgpu/_api.py", code)
+
+
+

codegen/__main__.py

@@ -5,6 +5,14 @@
 
 import os
 import sys
+from argparse import ArgumentParser
+
+parser = ArgumentParser(description="Run code generation and patching.")
+parser.add_argument("--skip-api", action="store_false", dest="api", help="Skip API patching")
+parser.add_argument("--skip-wgpu-native", action="store_false", dest="wgpu_native", help="Skip wgpu-native backend updates")
+# TODO: this is highly confusing to be swawpped around, but I didn't want to break the existing functionality.
+# maybe better to have a --skip a,b,c and --only e,f,g kinda CLI with all default?
+parser.add_argument("--js", action="store_true", dest="js", help="Generate JS backend updates")
 
 
 # Little trick to allow running this file as a script
@@ -15,5 +23,8 @@
 
 
 if __name__ == "__main__":
-    main()
+    # maybe have args here to just run the js writer?
+    args = parser.parse_args()
+    main(do_api=args.api, do_wgpu_native=args.wgpu_native, do_js=args.js)
+    # TODO: we need to actually patch the codegen report if we skip patching some parts of it...
     file_cache.write_changed_files_to_disk()

codegen/files.py

@@ -35,6 +35,7 @@ class FileCache:
         "structs.py",
         "backends/wgpu_native/_api.py",
         "backends/wgpu_native/_mappings.py",
+        "backends/js_webgpu/_api.py", # TODO: maybe this file should be more like _mappings
         "resources/codegen_report.md",
     ]
 

codegen/idlparser.py

@@ -7,8 +7,7 @@
 identify and remove code paths that are no longer used.
 """
 
-from typing import Literal
-
+from typing import Dict, Literal
 from codegen.utils import print
 from codegen.files import read_file
 
@@ -145,7 +144,7 @@ def peek_line(self):
 
     def parse(self, verbose=True):
         self._interfaces = {}
-        self.classes = {}
+        self.classes:Dict[str, Interface] = {}
         self.structs = {}
         self.flags = {}
         self.enums = {}
@@ -239,6 +238,7 @@ def resolve_type(self, typename) -> str:
             "ImageData": "ArrayLike",
             "VideoFrame": "ArrayLike",
             "AllowSharedBufferSource": "ArrayLike",
+            "[AllowShared] Uint32Array": "ArrayLike",
             "GPUPipelineConstantValue": "float",
             "GPUExternalTexture": "object",
             "undefined": "None",