react-native-wgpu 0.5.11 → 0.5.14

package/android/src/main/AndroidManifest.xml

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

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

@@ -1,298 +0,0 @@
-package com.webgpu;
-
-import android.annotation.SuppressLint;
-import android.content.Context;
-import android.graphics.Bitmap;
-import android.graphics.Canvas;
-import android.graphics.Paint;
-import android.graphics.PixelFormat;
-import android.graphics.Rect;
-import android.hardware.HardwareBuffer;
-import android.media.Image;
-import android.media.ImageReader;
-import android.os.Build;
-import android.os.Handler;
-import android.os.Looper;
-import android.view.Surface;
-import android.view.View;
-
-import androidx.annotation.NonNull;
-import androidx.annotation.RequiresApi;
-
-import java.nio.ByteBuffer;
-
-@RequiresApi(api = Build.VERSION_CODES.Q)
-@SuppressLint("ViewConstructor")
-public class WebGPUAHBView extends View implements ImageReader.OnImageAvailableListener {
-
-  private static final int MAX_IMAGES = 3;
-  private static final String TAG = "WebGPUAHBView";
-
-  private final WebGPUAPI mApi;
-  private ImageReader mImageReader;
-  private Surface mSurface;
-  private Image mCurrentImage;
-  private Bitmap mCachedBitmap;
-  private final Paint mPaint;
-  private final Handler mHandler;
-  private final Object mImageLock = new Object();
-  private boolean mSurfaceCreated = false;
-  private int mConfiguredWidth = 0;
-  private int mConfiguredHeight = 0;
-
-  public WebGPUAHBView(Context context, WebGPUAPI api) {
-    super(context);
-    mApi = api;
-    mPaint = new Paint(Paint.FILTER_BITMAP_FLAG);
-    mHandler = new Handler(Looper.getMainLooper());
-
-    // Enable hardware acceleration for this view
-    setLayerType(LAYER_TYPE_HARDWARE, null);
-
-    // Make sure we get drawn
-    setWillNotDraw(false);
-  }
-
-  @Override
-  protected void onSizeChanged(int w, int h, int oldw, int oldh) {
-    super.onSizeChanged(w, h, oldw, oldh);
-
-    if (w > 0 && h > 0) {
-      // Recreate ImageReader with new dimensions
-      setupImageReader(w, h);
-    }
-  }
-
-  private void setupImageReader(int width, int height) {
-    // Don't recreate if dimensions haven't changed
-    if (width == mConfiguredWidth && height == mConfiguredHeight && mImageReader != null) {
-      return;
-    }
-
-    // Clean up previous ImageReader
-    cleanupImageReader();
-
-    try {
-      // Create ImageReader with HardwareBuffer support
-      mImageReader = ImageReader.newInstance(
-        width,
-        height,
-        PixelFormat.RGBA_8888,
-        MAX_IMAGES,
-        HardwareBuffer.USAGE_GPU_SAMPLED_IMAGE |
-          HardwareBuffer.USAGE_GPU_COLOR_OUTPUT |
-          HardwareBuffer.USAGE_COMPOSER_OVERLAY |
-          HardwareBuffer.USAGE_CPU_READ_RARELY // For fallback to Bitmap if needed
-      );
-
-      mImageReader.setOnImageAvailableListener(this, mHandler);
-
-      // Get the Surface for WebGPU to render to
-      mSurface = mImageReader.getSurface();
-
-      mConfiguredWidth = width;
-      mConfiguredHeight = height;
-
-      // Notify WebGPU about the new surface
-      if (!mSurfaceCreated) {
-        mApi.surfaceCreated(mSurface);
-        mSurfaceCreated = true;
-      } else {
-        mApi.surfaceChanged(mSurface);
-      }
-
-    } catch (Exception e) {
-      e.printStackTrace();
-      // Fallback to offscreen if ImageReader creation fails
-      mApi.surfaceOffscreen();
-    }
-  }
-
-  @Override
-  public void onImageAvailable(ImageReader reader) {
-    synchronized (mImageLock) {
-      // Close previous image if exists
-      if (mCurrentImage != null) {
-        mCurrentImage.close();
-        mCurrentImage = null;
-      }
-
-      try {
-        // Get the latest image
-        mCurrentImage = reader.acquireLatestImage();
-
-        if (mCurrentImage != null) {
-          // Request a redraw on the UI thread
-          postInvalidateOnAnimation();
-        }
-      } catch (Exception e) {
-        e.printStackTrace();
-      }
-    }
-  }
-
-  @Override
-  protected void onDraw(Canvas canvas) {
-    super.onDraw(canvas);
-
-    synchronized (mImageLock) {
-      if (mCurrentImage == null) {
-        return;
-      }
-
-      try {
-        // Try to use HardwareBuffer directly (most efficient path)
-        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
-          HardwareBuffer hardwareBuffer = mCurrentImage.getHardwareBuffer();
-
-          if (hardwareBuffer != null) {
-            // Draw using HardwareBuffer
-            drawHardwareBuffer(canvas, hardwareBuffer);
-            return;
-          }
-        }
-
-        // Fallback: Convert Image to Bitmap
-        drawImageAsBitmap(canvas, mCurrentImage);
-
-      } catch (Exception e) {
-        e.printStackTrace();
-      }
-    }
-  }
-
-  private void drawHardwareBuffer(Canvas canvas, HardwareBuffer hardwareBuffer) {
-    // On Android Q+, we can create a Bitmap from HardwareBuffer
-    try {
-      // Create or reuse bitmap with matching dimensions
-      if (mCachedBitmap == null ||
-        mCachedBitmap.getWidth() != hardwareBuffer.getWidth() ||
-        mCachedBitmap.getHeight() != hardwareBuffer.getHeight()) {
-
-        if (mCachedBitmap != null) {
-          mCachedBitmap.recycle();
-        }
-
-        // Create a hardware-backed Bitmap from the HardwareBuffer
-        mCachedBitmap = Bitmap.wrapHardwareBuffer(hardwareBuffer, null);
-      } else {
-        // Reuse existing bitmap and update with new HardwareBuffer content
-        mCachedBitmap = Bitmap.wrapHardwareBuffer(hardwareBuffer, null);
-      }
-
-      if (mCachedBitmap != null) {
-        // Draw the bitmap to canvas
-        canvas.drawBitmap(mCachedBitmap, 0, 0, mPaint);
-      }
-    } catch (Exception e) {
-      e.printStackTrace();
-      // Fallback to Image-based rendering
-      drawImageAsBitmap(canvas, mCurrentImage);
-    }
-  }
-
-  private void drawImageAsBitmap(Canvas canvas, Image image) {
-    // Fallback method: manually convert Image to Bitmap
-    if (image.getFormat() != PixelFormat.RGBA_8888) {
-      return;
-    }
-
-    Image.Plane[] planes = image.getPlanes();
-    if (planes.length == 0) {
-      return;
-    }
-
-    ByteBuffer buffer = planes[0].getBuffer();
-    int pixelStride = planes[0].getPixelStride();
-    int rowStride = planes[0].getRowStride();
-    int rowPadding = rowStride - pixelStride * image.getWidth();
-
-    // Create or reuse bitmap
-    int bitmapWidth = image.getWidth() + rowPadding / pixelStride;
-    if (mCachedBitmap == null ||
-      mCachedBitmap.getWidth() != bitmapWidth ||
-      mCachedBitmap.getHeight() != image.getHeight()) {
-
-      if (mCachedBitmap != null) {
-        mCachedBitmap.recycle();
-      }
-      mCachedBitmap = Bitmap.createBitmap(
-        bitmapWidth,
-        image.getHeight(),
-        Bitmap.Config.ARGB_8888
-      );
-    }
-
-    mCachedBitmap.copyPixelsFromBuffer(buffer);
-
-    // Draw only the valid portion (without padding)
-    Rect src = new Rect(0, 0, image.getWidth(), image.getHeight());
-    Rect dst = new Rect(0, 0, getWidth(), getHeight());
-    canvas.drawBitmap(mCachedBitmap, src, dst, mPaint);
-  }
-
-  private void cleanupImageReader() {
-    synchronized (mImageLock) {
-      if (mCurrentImage != null) {
-        mCurrentImage.close();
-        mCurrentImage = null;
-      }
-
-      if (mImageReader != null) {
-        mImageReader.close();
-        mImageReader = null;
-      }
-
-      if (mSurface != null) {
-        mSurface.release();
-        mSurface = null;
-      }
-    }
-  }
-
-  @Override
-  protected void onDetachedFromWindow() {
-    super.onDetachedFromWindow();
-
-    // Notify WebGPU that surface is being destroyed
-    if (mSurfaceCreated) {
-      mApi.surfaceDestroyed();
-      mSurfaceCreated = false;
-    }
-
-    // Clean up resources
-    cleanupImageReader();
-
-    if (mCachedBitmap != null) {
-      mCachedBitmap.recycle();
-      mCachedBitmap = null;
-    }
-  }
-
-  @Override
-  protected void onAttachedToWindow() {
-    super.onAttachedToWindow();
-
-    // Re-setup if we have valid dimensions
-    if (getWidth() > 0 && getHeight() > 0) {
-      setupImageReader(getWidth(), getHeight());
-    }
-  }
-
-  @Override
-  public void setAlpha(float alpha) {
-    super.setAlpha(alpha);
-    mPaint.setAlpha((int) (alpha * 255));
-    invalidate();
-  }
-
-  @Override
-  protected void onVisibilityChanged(@NonNull View changedView, int visibility) {
-    super.onVisibilityChanged(changedView, visibility);
-
-    if (visibility == VISIBLE && mSurface == null && getWidth() > 0 && getHeight() > 0) {
-      // Re-create surface if needed when becoming visible
-      setupImageReader(getWidth(), getHeight());
-    }
-  }
-}

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

@@ -1,20 +0,0 @@
-package com.webgpu;
-
-import android.view.Surface;
-
-import com.facebook.proguard.annotations.DoNotStrip;
-
-public interface WebGPUAPI {
-
-  void surfaceCreated(
-    Surface surface
-  );
-
-  void surfaceChanged(
-    Surface surface
-  );
-
-  void surfaceDestroyed();
-
-  void surfaceOffscreen();
-}

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

@@ -1,61 +0,0 @@
-package com.webgpu;
-
-import android.content.Context;
-import android.view.Surface;
-import android.view.View;
-import com.facebook.proguard.annotations.DoNotStrip;
-
-public abstract class WebGPUBaseView extends View {
-  protected Integer mContextId;
-
-  public WebGPUBaseView(Context context, Integer contextId) {
-    super(context);
-    mContextId = contextId;
-  }
-
-  protected void handleSurfaceCreate(Surface surface) {
-    float density = getResources().getDisplayMetrics().density;
-    float scaledWidth = getWidth() / density;
-    float scaledHeight = getHeight() / density;
-    onSurfaceCreate(surface, mContextId, scaledWidth, scaledHeight);
-  }
-
-  protected void handleSurfaceChanged(Surface surface) {
-    float density = getResources().getDisplayMetrics().density;
-    float scaledWidth = getWidth() / density;
-    float scaledHeight = getHeight() / density;
-    onSurfaceChanged(surface, mContextId, scaledWidth, scaledHeight);
-  }
-
-  protected void handleSurfaceDestroyed() {
-    onSurfaceDestroy(mContextId);
-  }
-
-  @Override
-  protected void onLayout(boolean changed, int left, int top, int right, int bottom) {
-    super.onLayout(changed, left, top, right, bottom);
-  }
-
-
-  @DoNotStrip
-  private native void onSurfaceCreate(
-    Surface surface,
-    int contextId,
-    float width,
-    float height
-  );
-
-  @DoNotStrip
-  private native void onSurfaceChanged(
-    Surface surface,
-    int contextId,
-    float width,
-    float height
-  );
-
-  @DoNotStrip
-  private native void onSurfaceDestroy(int contextId);
-
-  @DoNotStrip
-  protected native void switchToOffscreenSurface(int contextId);
-}

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

@@ -1,50 +0,0 @@
-package com.webgpu;
-
-import android.util.Log;
-
-import androidx.annotation.OptIn;
-
-import java.util.HashSet;
-import java.util.Set;
-
-import com.facebook.proguard.annotations.DoNotStrip;
-import com.facebook.react.bridge.ReactApplicationContext;
-import com.facebook.react.bridge.JavaScriptContextHolder;
-import com.facebook.react.bridge.ReactMethod;
-import com.facebook.react.common.annotations.FrameworkAPI;
-import com.facebook.react.module.annotations.ReactModule;
-import com.facebook.react.modules.blob.BlobModule;
-import com.facebook.react.modules.blob.BlobProvider;
-import com.facebook.react.turbomodule.core.CallInvokerHolderImpl;
-import com.facebook.react.turbomodule.core.interfaces.CallInvokerHolder;
-
-@ReactModule(name = WebGPUModule.NAME)
-public class WebGPUModule extends NativeWebGPUModuleSpec {
-  static {
-      System.loadLibrary("react-native-wgpu"); // Load the C++ library
-  }
-
-  public WebGPUModule(ReactApplicationContext reactContext) {
-    super(reactContext);
-    // Initialize the C++ module
-    initialize();
-  }
-
-  @OptIn(markerClass = FrameworkAPI.class)
-  @ReactMethod(isBlockingSynchronousMethod = true)
-  public boolean install() {
-    ReactApplicationContext context = getReactApplicationContext();
-    JavaScriptContextHolder jsContext = context.getJavaScriptContextHolder();
-    CallInvokerHolder callInvokerHolder = context.getCatalystInstance().getJSCallInvokerHolder();
-    BlobModule blobModule = getReactApplicationContext().getNativeModule(BlobModule.class);
-    if (blobModule == null) {
-      throw new RuntimeException("React Native's BlobModule was not found!");
-    }
-    initializeNative(jsContext.get(), (CallInvokerHolderImpl) callInvokerHolder, blobModule);
-    return true;
-  }
-
-  @OptIn(markerClass = FrameworkAPI.class)
-  @DoNotStrip
-  private native void initializeNative(long jsRuntime, CallInvokerHolderImpl jsInvoker, BlobModule blobModule);
-}

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

@@ -1,41 +0,0 @@
-package com.webgpu;
-
-import android.annotation.SuppressLint;
-import android.content.Context;
-import android.view.SurfaceHolder;
-import android.view.SurfaceView;
-
-import androidx.annotation.NonNull;
-
-@SuppressLint("ViewConstructor")
-public class WebGPUSurfaceView extends SurfaceView implements SurfaceHolder.Callback {
-
-  WebGPUAPI mApi;
-
-  public WebGPUSurfaceView(Context context, WebGPUAPI api) {
-    super(context);
-    mApi = api;
-    getHolder().addCallback(this);
-  }
-
-  @Override
-  protected void onDetachedFromWindow() {
-    super.onDetachedFromWindow();
-    mApi.surfaceDestroyed();
-  }
-
-  @Override
-  public void surfaceCreated(@NonNull SurfaceHolder holder) {
-    mApi.surfaceCreated(holder.getSurface());
-  }
-
-  @Override
-  public void surfaceChanged(@NonNull SurfaceHolder holder, int format, int width, int height) {
-    mApi.surfaceChanged(holder.getSurface());
-  }
-
-  @Override
-  public void surfaceDestroyed(@NonNull SurfaceHolder holder) {
-    mApi.surfaceOffscreen();
-  }
-}

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

@@ -1,79 +0,0 @@
-package com.webgpu;
-
-import android.annotation.SuppressLint;
-import android.content.Context;
-import android.graphics.PixelFormat;
-import android.os.Build;
-import android.view.Surface;
-import android.view.SurfaceControl;
-import android.view.SurfaceHolder;
-import android.view.SurfaceView;
-
-import androidx.annotation.NonNull;
-import androidx.annotation.RequiresApi;
-
-
-@SuppressLint("ViewConstructor")
-@RequiresApi(api = Build.VERSION_CODES.Q)
-public class WebGPUSurfaceViewWithSC extends SurfaceView implements SurfaceHolder.Callback {
-
-  WebGPUAPI mApi;
-  SurfaceControl mSurfaceControl;
-  Surface mSurface;
-
-  public WebGPUSurfaceViewWithSC(Context context, WebGPUAPI api) {
-    super(context);
-    mApi = api;
-    getHolder().addCallback(this);
-  }
-
-  @Override
-  protected void onLayout(boolean changed, int left, int top, int right, int bottom) {
-    super.onLayout(changed, left, top, right, bottom);
-
-  }
-
-  @Override
-  protected void onDetachedFromWindow() {
-    super.onDetachedFromWindow();
-  }
-
-  @Override
-  public void surfaceCreated(@NonNull SurfaceHolder holder) {
-    if (mSurfaceControl != null) {
-      SurfaceControl.Transaction tr = new SurfaceControl.Transaction();
-      tr.setVisibility(mSurfaceControl, true);
-      tr.reparent(mSurfaceControl, getSurfaceControl());
-      tr.apply();
-    } else {
-      SurfaceControl.Builder scb = new SurfaceControl.Builder();
-      scb.setName("WebGPUView");
-      scb.setOpaque(true);
-      scb.setBufferSize(getWidth(), getHeight());
-      scb.setParent(getSurfaceControl());
-      scb.setFormat(PixelFormat.RGBA_8888);
-      mSurfaceControl = scb.build();
-      mSurface = new Surface(mSurfaceControl);
-      mApi.surfaceCreated(mSurface);
-    }
-    SurfaceControl.Transaction tr = new SurfaceControl.Transaction();
-    tr.setVisibility(mSurfaceControl, true);
-    tr.apply();
-  }
-
-  @Override
-  public void surfaceChanged(@NonNull SurfaceHolder holder, int format, int width, int height) {
-    mApi.surfaceChanged(mSurface);
-    SurfaceControl.Transaction tr = new SurfaceControl.Transaction();
-    tr.setVisibility(mSurfaceControl, true);
-    tr.setBufferSize(mSurfaceControl, getWidth(), getHeight());
-    tr.apply();
-  }
-
-  @Override
-  public void surfaceDestroyed(@NonNull SurfaceHolder holder) {
-    SurfaceControl.Transaction tr = new SurfaceControl.Transaction();
-    tr.reparent(mSurfaceControl, null);
-    tr.apply();
-  }
-}

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

@@ -1,46 +0,0 @@
-package com.webgpu;
-
-import android.annotation.SuppressLint;
-import android.content.Context;
-import android.graphics.SurfaceTexture;
-import android.view.Surface;
-import android.view.TextureView;
-import androidx.annotation.NonNull;
-
-import org.w3c.dom.Text;
-
-@SuppressLint("ViewConstructor")
-public class WebGPUTextureView extends TextureView implements TextureView.SurfaceTextureListener {
-
-  WebGPUAPI mApi;
-
-  public WebGPUTextureView(Context context, WebGPUAPI api) {
-    super(context);
-    mApi = api;
-    setOpaque(false);
-    setSurfaceTextureListener(this);
-  }
-
-  @Override
-  public void onSurfaceTextureAvailable(@NonNull SurfaceTexture surfaceTexture, int width, int height) {
-    Surface surface = new Surface(surfaceTexture);
-    mApi.surfaceCreated(surface);
-  }
-
-  @Override
-  public void onSurfaceTextureSizeChanged(@NonNull SurfaceTexture surfaceTexture, int width, int height) {
-    Surface surface = new Surface(surfaceTexture);
-    mApi.surfaceChanged(surface);
-  }
-
-  @Override
-  public boolean onSurfaceTextureDestroyed(@NonNull SurfaceTexture surfaceTexture) {
-    mApi.surfaceDestroyed();
-    return true;
-  }
-
-  @Override
-  public void onSurfaceTextureUpdated(@NonNull SurfaceTexture surfaceTexture) {
-    // No implementation needed
-  }
-}

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

@@ -1,108 +0,0 @@
-package com.webgpu;
-
-import android.content.Context;
-import android.os.Build;
-import android.view.Surface;
-import android.view.View;
-
-import com.facebook.proguard.annotations.DoNotStrip;
-import com.facebook.react.uimanager.ThemedReactContext;
-import com.facebook.react.views.view.ReactViewGroup;
-
-public class WebGPUView extends ReactViewGroup implements WebGPUAPI {
-
-
-  private int mContextId;
-  private boolean mTransparent = false;
-  private WebGPUModule mModule;
-  private View mView = null;
-
-  WebGPUView(Context context) {
-    super(context);
-  }
-
-  public void setContextId(int contextId) {
-    if (mModule == null) {
-      Context context = getContext();
-      if (context instanceof ThemedReactContext) {
-        mModule = ((ThemedReactContext) context).getReactApplicationContext().getNativeModule(WebGPUModule.class);
-      }
-    }
-    mContextId = contextId;
-  }
-
-  public void setTransparent(boolean value) {
-    Context ctx = getContext();
-    if (value != mTransparent || mView == null) {
-      if (mView != null) {
-        removeView(mView);
-      }
-      mTransparent = value;
-      if (mTransparent) {
-//        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
-//          mView = new WebGPUAHBView(ctx, this);
-//        } else {
-          mView = new WebGPUTextureView(ctx, this);
-//        }
-      } else {
-        mView = new WebGPUSurfaceView(ctx, this);
-      }
-      addView(mView);
-    }
-  }
-
-  @Override
-  protected void onLayout(boolean changed, int left, int top, int right, int bottom) {
-    super.onLayout(changed, left, top, right, bottom);
-    mView.layout(0, 0, this.getMeasuredWidth(), this.getMeasuredHeight());
-  }
-
-  @Override
-  public void surfaceCreated(Surface surface) {
-    float density = getResources().getDisplayMetrics().density;
-    float width = getWidth() / density;
-    float height = getHeight() / density;
-    onSurfaceCreate(surface, mContextId, width, height);
-  }
-
-  @Override
-  public void surfaceChanged(Surface surface) {
-    float density = getResources().getDisplayMetrics().density;
-    float width = getWidth() / density;
-    float height = getHeight() / density;
-    onSurfaceChanged(surface, mContextId, width, height);
-  }
-
-  @Override
-  public void surfaceDestroyed() {
-    onSurfaceDestroy(mContextId);
-  }
-
-  @Override
-  public void surfaceOffscreen() {
-    switchToOffscreenSurface(mContextId);
-  }
-
-  @DoNotStrip
-  private native void onSurfaceCreate(
-    Surface surface,
-    int contextId,
-    float width,
-    float height
-  );
-
-  @DoNotStrip
-  private native void onSurfaceChanged(
-    Surface surface,
-    int contextId,
-    float width,
-    float height
-  );
-
-  @DoNotStrip
-  private native void onSurfaceDestroy(int contextId);
-
-  @DoNotStrip
-  private native void switchToOffscreenSurface(int contextId);
-
-}