react-native-wgpu 0.1.0

package/LICENSE

@@ -0,0 +1,20 @@
+MIT License
+
+Copyright (c) 2024 William Candillon
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,31 @@
+# react-native-wgpu
+
+React Native WebGPU
+
+## Installation
+
+```sh
+npm install react-native-wgpu
+```
+
+## Usage
+
+```js
+import { WgpuView } from "react-native-wgpu";
+
+// ...
+
+<WgpuView color="tomato" />
+```
+
+## Contributing
+
+See the [contributing guide](CONTRIBUTING.md) to learn how to contribute to the repository and the development workflow.
+
+## License
+
+MIT
+
+---
+
+Made with [create-react-native-library](https://github.com/callstack/react-native-builder-bob)

package/android/CMakeLists.txt

@@ -0,0 +1,74 @@
+cmake_minimum_required(VERSION 3.4.1)
+project(RNWGPU)
+
+set (CMAKE_VERBOSE_MAKEFILE ON)
+set (CMAKE_CXX_STANDARD 17)
+
+set (PACKAGE_NAME "react-native-webgpu")
+
+#link_directories(../libs/android/${ANDROID_ABI}/)
+
+if(CMAKE_BUILD_TYPE STREQUAL "Debug")
+    add_definitions(-DDEBUG=1)
+endif()
+
+if(RNF_ENABLE_LOGS)
+    add_definitions(-DRNF_ENABLE_LOGS=1)
+else()
+    add_definitions(-DRNF_ENABLE_LOGS=0)
+endif()
+
+set (WEBGPU_LIBS_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../libs/android/${ANDROID_ABI}")
+
+find_package(ReactAndroid REQUIRED CONFIG)
+find_package(fbjni REQUIRED CONFIG)
+
+add_library(${PACKAGE_NAME} SHARED
+    ./cpp/cpp-adapter.cpp
+    ../cpp/rnwgpu/api/GPU.cpp
+    ../cpp/rnwgpu/api/GPUAdapter.cpp
+    ../cpp/rnwgpu/api/GPUDevice.cpp
+    ../cpp/rnwgpu/api/GPUBuffer.cpp
+    ../cpp/rnwgpu/RNWebGPUManager.cpp
+    ../cpp/jsi/RNFPromise.cpp
+    ../cpp/jsi/RNFHybridObject.cpp
+    ../cpp/jsi/RNFRuntimeCache.cpp
+    ../cpp/jsi/RNFWorkletRuntimeRegistry.cpp
+)
+
+target_include_directories(
+        ${PACKAGE_NAME}
+        PRIVATE
+
+        "${NODE_MODULES_DIR}/react-native/ReactCommon/callinvoker"
+        "${NODE_MODULES_DIR}/react-native/ReactCommon/jsi"
+        "${NODE_MODULES_DIR}/react-native/ReactCommon"
+        "${NODE_MODULES_DIR}/react-native/ReactCommon/react/nativemodule/core"
+        "${NODE_MODULES_DIR}/react-native/ReactCommon/runtimeexecutor"
+        "${NODE_MODULES_DIR}/react-native/ReactAndroid/src/main/jni"
+        "${NODE_MODULES_DIR}/react-native/ReactAndroid/src/main/java/com/facebook/react/turbomodule/core/jni"
+
+        ../cpp
+        ../cpp/rnwgpu
+        ../cpp/rnwgpu/api
+        ../cpp/rnwgpu/api/descriptors
+        ../cpp/jsi
+        ../cpp/webgpu
+
+        ${libfbjni_include_DIRS}
+)
+
+add_library(webgpu_dawn SHARED IMPORTED)
+set_property(TARGET webgpu_dawn PROPERTY IMPORTED_LOCATION "${WEBGPU_LIBS_PATH}/libwebgpu_dawn.so")
+
+# Link
+target_link_libraries(
+        ${PACKAGE_NAME}
+        ${FBJNI_LIBRARY}
+        log
+        jnigraphics
+        android
+        fbjni::fbjni
+        ReactAndroid::jsi
+        webgpu_dawn
+    )

package/android/build.gradle

@@ -0,0 +1,166 @@
+import java.nio.file.Paths
+
+buildscript {
+  repositories {
+    google()
+    mavenCentral()
+  }
+
+  dependencies {
+    classpath "com.android.tools.build:gradle:7.2.1"
+  }
+}
+
+def reactNativeArchitectures() {
+  def value = rootProject.getProperties().get("reactNativeArchitectures")
+  return value ? value.split(",") : ["armeabi-v7a", "x86", "x86_64", "arm64-v8a"]
+}
+
+def isNewArchitectureEnabled() {
+  return rootProject.hasProperty("newArchEnabled") && rootProject.getProperty("newArchEnabled") == "true"
+}
+
+apply plugin: "com.android.library"
+
+if (isNewArchitectureEnabled()) {
+  apply plugin: "com.facebook.react"
+}
+
+static def findNodeModules(baseDir) {
+  // Check if we are building the example app:
+  // Check if ../FabricExample/node_modules exists
+  def exampleAppPath = Paths.get(baseDir.toString(), "..", "FabricExample")
+  def exampleNodeModulesPath = Paths.get(exampleAppPath.toString(), "node_modules")
+  if (exampleNodeModulesPath.toFile().exists()) {
+    return exampleNodeModulesPath.toAbsolutePath()
+  }
+
+  def basePath = baseDir.toPath().normalize()
+  // Node's module resolution algorithm searches up to the root directory,
+  // after which the base path will be null
+  while (basePath) {
+    def nodeModulesPath = Paths.get(basePath.toString(), "node_modules")
+    def reactNativePath = Paths.get(nodeModulesPath.toString(), "react-native")
+    if (nodeModulesPath.toFile().exists() && reactNativePath.toFile().exists()) {
+      return nodeModulesPath.toString()
+    }
+    basePath = basePath.getParent()
+  }
+
+  throw new GradleException("react-native-filament: Failed to find node_modules/ path!")
+}
+
+def nodeModules = findNodeModules(projectDir)
+
+def getExtOrDefault(name) {
+  return rootProject.ext.has(name) ? rootProject.ext.get(name) : project.properties["Wgpu_" + name]
+}
+
+def getExtOrIntegerDefault(name) {
+  return rootProject.ext.has(name) ? rootProject.ext.get(name) : (project.properties["Wgpu_" + name]).toInteger()
+}
+
+def supportsNamespace() {
+  def parsed = com.android.Version.ANDROID_GRADLE_PLUGIN_VERSION.tokenize('.')
+  def major = parsed[0].toInteger()
+  def minor = parsed[1].toInteger()
+
+  // Namespace support was added in 7.3.0
+  return (major == 7 && minor >= 3) || major >= 8
+}
+
+android {
+  if (supportsNamespace()) {
+    namespace "com.wgpu"
+
+    sourceSets {
+      main {
+        manifest.srcFile "src/main/AndroidManifestNew.xml"
+      }
+    }
+  }
+
+  compileSdkVersion getExtOrIntegerDefault("compileSdkVersion")
+
+  defaultConfig {
+    minSdkVersion getExtOrIntegerDefault("minSdkVersion")
+    targetSdkVersion getExtOrIntegerDefault("targetSdkVersion")
+    buildConfigField "boolean", "IS_NEW_ARCHITECTURE_ENABLED", isNewArchitectureEnabled().toString()
+    externalNativeBuild {
+      cmake {
+        cppFlags "-fexceptions", "-frtti", "-std=c++1y", "-DONANDROID"
+        abiFilters (*reactNativeArchitectures())
+        arguments '-DANDROID_STL=c++_shared',
+          "-DNODE_MODULES_DIR=${nodeModules}"
+      }
+    }
+  }
+
+  packagingOptions {
+    exclude "lib/arm64-v8a/libjsi.so"
+    exclude "lib/x86_64/libjsi.so"
+    exclude "lib/x86/libjsi.so"
+    exclude "lib/armeabi-v7a/libjsi.so"
+  }
+
+  buildFeatures {
+    buildConfig true
+    prefab true
+  }
+
+  buildTypes {
+    release {
+      minifyEnabled false
+    }
+  }
+
+  lintOptions {
+    disable "GradleCompatible"
+  }
+
+  compileOptions {
+    sourceCompatibility JavaVersion.VERSION_1_8
+    targetCompatibility JavaVersion.VERSION_1_8
+  }
+
+  sourceSets {
+    main {
+      if (isNewArchitectureEnabled()) {
+        java.srcDirs += [
+          "src/newarch",
+          // This is needed to build Kotlin project with NewArch enabled
+          "${project.buildDir}/generated/source/codegen/java"
+        ]
+      } else {
+        java.srcDirs += ["src/oldarch"]
+      }
+    }
+  }
+
+  externalNativeBuild {
+    cmake {
+      path file('CMakeLists.txt')
+    }
+  }
+}
+
+repositories {
+  mavenCentral()
+  google()
+}
+
+
+dependencies {
+  // For < 0.71, this will be from the local maven repo
+  // For > 0.71, this will be replaced by `com.facebook.react:react-android:$version` by react gradle plugin
+  //noinspection GradleDynamicVersion
+  implementation "com.facebook.react:react-native:+"
+}
+
+if (isNewArchitectureEnabled()) {
+  react {
+    jsRootDir = file("../src/")
+    libraryName = "WgpuView"
+    codegenJavaPackageName = "com.wgpu"
+  }
+}

package/android/cpp/cpp-adapter.cpp

@@ -0,0 +1,12 @@
+#include "Logger.h"
+#include "RNWebGPUManager.h"
+#include <fbjni/fbjni.h>
+#include <jni.h>
+
+#define LOG_TAG "WebGPUModule"
+
+extern "C" JNIEXPORT void JNICALL Java_com_webgpu_WebGPUModule_initializeNative(
+    JNIEnv *env, jobject /* this */, jlong jsRuntime, jobject jsInvokerHolder) {
+  auto runtime = reinterpret_cast<facebook::jsi::Runtime *>(jsRuntime);
+  auto manager = new rnwgpu::RNWebGPUManager(runtime, nullptr);
+}

package/android/gradle.properties

@@ -0,0 +1,5 @@
+Wgpu_kotlinVersion=1.7.0
+Wgpu_minSdkVersion=21
+Wgpu_targetSdkVersion=31
+Wgpu_compileSdkVersion=31
+Wgpu_ndkversion=21.4.7075529

package/android/src/main/AndroidManifest.xml

@@ -0,0 +1,3 @@
+<manifest xmlns:android="http://schemas.android.com/apk/res/android"
+          package="com.webgpu">
+</manifest>

package/android/src/main/AndroidManifestNew.xml

@@ -0,0 +1,2 @@
+<manifest xmlns:android="http://schemas.android.com/apk/res/android">
+</manifest>

package/android/src/main/java/com/webgpu/WebGPUModule.java

@@ -0,0 +1,39 @@
+package com.webgpu;
+
+import com.facebook.jni.HybridData;
+import com.facebook.proguard.annotations.DoNotStrip;
+import com.facebook.react.bridge.ReactApplicationContext;
+import com.facebook.react.bridge.ReactContextBaseJavaModule;
+import com.facebook.react.bridge.JavaScriptContextHolder;
+import com.facebook.react.bridge.ReactMethod;
+import com.facebook.react.common.annotations.FrameworkAPI;
+import com.facebook.react.turbomodule.core.CallInvokerHolderImpl;
+import com.facebook.react.turbomodule.core.interfaces.CallInvokerHolder;
+
+@FrameworkAPI public class WebGPUModule extends ReactContextBaseJavaModule {
+    static {
+        System.loadLibrary("react-native-webgpu"); // Load the C++ library
+    }
+
+    public WebGPUModule(ReactApplicationContext reactContext) {
+        super(reactContext);
+        // Initialize the C++ module
+        initialize();
+    }
+
+    @Override
+    public String getName() {
+        return "WebGPUModule";
+    }
+
+    @ReactMethod(isBlockingSynchronousMethod = true)
+    public void install() {
+      ReactApplicationContext context = getReactApplicationContext();
+      JavaScriptContextHolder jsContext = context.getJavaScriptContextHolder();
+      CallInvokerHolder callInvokerHolder = context.getCatalystInstance().getJSCallInvokerHolder();
+      initializeNative(jsContext.get(),  (CallInvokerHolderImpl) callInvokerHolder);
+    }
+
+    @DoNotStrip
+    private native void initializeNative(long jsRuntime, CallInvokerHolderImpl jsInvoker);
+}

package/android/src/main/java/com/webgpu/WebGPUView.java

@@ -0,0 +1,24 @@
+package com.webgpu;
+
+import androidx.annotation.Nullable;
+
+import android.content.Context;
+import android.util.AttributeSet;
+
+import android.view.View;
+
+public class WebGPUView extends View {
+
+  public WebGPUView(Context context) {
+    super(context);
+  }
+
+  public WebGPUView(Context context, @Nullable AttributeSet attrs) {
+    super(context, attrs);
+  }
+
+  public WebGPUView(Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
+    super(context, attrs, defStyleAttr);
+  }
+
+}

package/android/src/main/java/com/webgpu/WebGPUViewManager.java

@@ -0,0 +1,31 @@
+package com.webgpu;
+
+import android.graphics.Color;
+
+import androidx.annotation.Nullable;
+
+import com.facebook.react.module.annotations.ReactModule;
+import com.facebook.react.uimanager.ThemedReactContext;
+import com.facebook.react.uimanager.annotations.ReactProp;
+
+@ReactModule(name = WebGPUViewManager.NAME)
+public class WebGPUViewManager extends WebGPUViewManagerSpec<WebGPUView> {
+
+  public static final String NAME = "WebGPUView";
+
+  @Override
+  public String getName() {
+    return NAME;
+  }
+
+  @Override
+  public WebGPUView createViewInstance(ThemedReactContext context) {
+    return new WebGPUView(context);
+  }
+
+  @Override
+  @ReactProp(name = "color")
+  public void setColor(WebGPUView view, @Nullable String color) {
+    view.setBackgroundColor(Color.parseColor(color));
+  }
+}

package/android/src/main/java/com/webgpu/WebGPUViewPackage.java

@@ -0,0 +1,26 @@
+
+package com.webgpu;
+
+import com.facebook.react.ReactPackage;
+import com.facebook.react.bridge.NativeModule;
+import com.facebook.react.bridge.ReactApplicationContext;
+import com.facebook.react.uimanager.ViewManager;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+public class WebGPUViewPackage implements ReactPackage {
+  @Override
+  public List<ViewManager> createViewManagers(ReactApplicationContext reactContext) {
+    List<ViewManager> viewManagers = new ArrayList<>();
+    viewManagers.add(new WebGPUViewManager());
+    return viewManagers;
+  }
+
+
+  @Override
+  public List<NativeModule> createNativeModules(ReactApplicationContext reactContext) {
+      return Arrays.asList(new WebGPUModule(reactContext));
+  }
+}

package/android/src/newarch/WgpuViewManagerSpec.java

@@ -0,0 +1,24 @@
+package com.webgpu;
+
+import android.view.View;
+
+import androidx.annotation.Nullable;
+
+import com.facebook.react.uimanager.SimpleViewManager;
+import com.facebook.react.uimanager.ViewManagerDelegate;
+import com.facebook.react.viewmanagers.WgpuViewManagerDelegate;
+import com.facebook.react.viewmanagers.WgpuViewManagerInterface;
+
+public abstract class WgpuViewManagerSpec<T extends View> extends SimpleViewManager<T> implements WgpuViewManagerInterface<T> {
+  private final ViewManagerDelegate<T> mDelegate;
+
+  public WgpuViewManagerSpec() {
+    mDelegate = new WgpuViewManagerDelegate(this);
+  }
+
+  @Nullable
+  @Override
+  protected ViewManagerDelegate<T> getDelegate() {
+    return mDelegate;
+  }
+}

package/android/src/oldarch/WebGPUViewManagerSpec.java

@@ -0,0 +1,11 @@
+package com.webgpu;
+
+import android.view.View;
+
+import androidx.annotation.Nullable;
+
+import com.facebook.react.uimanager.SimpleViewManager;
+
+public abstract class WebGPUViewManagerSpec<T extends View> extends SimpleViewManager<T> {
+  public abstract void setColor(T view, @Nullable String value);
+}

package/cpp/Logger.h

@@ -0,0 +1,82 @@
+//
+// Created by Christian Falch on 26/08/2021.
+//
+
+#pragma once
+
+#include <jsi/jsi.h>
+#include <string>
+
+#if defined(ANDROID) || defined(__ANDROID__)
+#include <android/log.h>
+#endif
+
+#ifdef TARGET_OS_IPHONE
+#include <syslog.h>
+#endif
+
+namespace rnwgpu {
+
+namespace jsi = facebook::jsi;
+
+class Logger {
+public:
+  /**
+   * Logs message to console
+   * @param message Message to be written out
+   */
+  static void logToConsole(std::string message) {
+#if defined(ANDROID) || defined(__ANDROID__)
+    __android_log_write(ANDROID_LOG_INFO, "RNSkia", message.c_str());
+#endif
+
+#ifdef TARGET_OS_IPHONE
+    syslog(LOG_ERR, "%s\n", message.c_str());
+#endif
+  }
+
+  /**
+   * Logs to console
+   * @param fmt Format string
+   * @param ... Arguments to format string
+   */
+  static void logToConsole(const char *fmt, ...) {
+    va_list args;
+    va_start(args, fmt);
+
+    static char buffer[512];
+    vsnprintf(buffer, sizeof(buffer), fmt, args);
+#if defined(ANDROID) || defined(__ANDROID__)
+    __android_log_write(ANDROID_LOG_INFO, "RNSkia", buffer);
+#endif
+#ifdef TARGET_OS_IPHONE
+    syslog(LOG_ERR, "RNSKIA: %s\n", buffer);
+#endif
+    va_end(args);
+  }
+
+  static void logToJavascriptConsole(jsi::Runtime &runtime,
+                                     const std::string &message) {
+    auto console = Logger::getJavascriptConsole(runtime).asObject(runtime);
+    auto log = console.getPropertyAsFunction(runtime, "log");
+    log.call(runtime, jsi::String::createFromUtf8(runtime, message));
+  }
+
+  static void warnToJavascriptConsole(jsi::Runtime &runtime,
+                                      const std::string &message) {
+    auto console = Logger::getJavascriptConsole(runtime).asObject(runtime);
+    auto warn = console.getPropertyAsFunction(runtime, "warn");
+    warn.call(runtime, jsi::String::createFromUtf8(runtime, message));
+  }
+
+private:
+  static jsi::Value getJavascriptConsole(jsi::Runtime &runtime) {
+    auto console = runtime.global().getProperty(runtime, "console");
+    if (console.isUndefined() || console.isNull()) {
+      throw jsi::JSError(runtime, "Could not find console object.");
+      return jsi::Value::undefined();
+    }
+    return console;
+  }
+};
+} // namespace rnwgpu