react-native-wgpu 0.1.10 → 0.1.12

package/ios/IOSPlatformContext.h

@@ -1,19 +0,0 @@
-#pragma once
-
-#include "PlatformContext.h"
-
-namespace rnwgpu {
-
-class IOSPlatformContext : public PlatformContext {
-public:
-  IOSPlatformContext() = default;
-  ~IOSPlatformContext() = default;
-
-  wgpu::Surface makeSurface(wgpu::Instance instance, void *surface, int width,
-                            int height) override;
-
-  ImageData createImageBitmap(std::string blobId, double offset,
-                              double size) override;
-};
-
-} // namespace rnwgpu

package/ios/IOSPlatformContext.mm

@@ -1,68 +0,0 @@
-#include "IOSPlatformContext.h"
-
-#import <React/RCTBlobManager.h>
-#import <React/RCTBridge+Private.h>
-#import <ReactCommon/RCTTurboModule.h>
-
-#include "RNWebGPUManager.h"
-#include "WebGPUModule.h"
-
-namespace rnwgpu {
-
-wgpu::Surface IOSPlatformContext::makeSurface(wgpu::Instance instance,
-                                              void *surface, int width,
-                                              int height) {
-  wgpu::SurfaceDescriptorFromMetalLayer metalSurfaceDesc;
-  metalSurfaceDesc.layer = surface;
-  wgpu::SurfaceDescriptor surfaceDescriptor;
-  surfaceDescriptor.nextInChain = &metalSurfaceDesc;
-  return instance.CreateSurface(&surfaceDescriptor);
-}
-
-ImageData IOSPlatformContext::createImageBitmap(std::string blobId,
-                                                double offset, double size) {
-  RCTBlobManager *blobManager =
-      [[RCTBridge currentBridge] moduleForClass:RCTBlobManager.class];
-  NSData *blobData =
-      [blobManager resolve:[NSString stringWithUTF8String:blobId.c_str()]
-                    offset:(long)offset
-                      size:(long)size];
-
-  if (!blobData) {
-    throw std::runtime_error("Couldn't retrive blob data");
-  }
-
-  UIImage *image = [UIImage imageWithData:blobData];
-  if (!image) {
-    throw std::runtime_error("Couldn't decode image");
-  }
-
-  CGImageRef cgImage = image.CGImage;
-  size_t width = CGImageGetWidth(cgImage);
-  size_t height = CGImageGetHeight(cgImage);
-  size_t bitsPerComponent = 8;
-  size_t bytesPerRow = width * 4;
-  std::vector<uint8_t> imageData(height * bytesPerRow);
-
-  CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
-  CGContextRef context = CGBitmapContextCreate(
-      imageData.data(), width, height, bitsPerComponent, bytesPerRow,
-      colorSpace, kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big);
-
-  CGContextDrawImage(context, CGRectMake(0, 0, width, height), cgImage);
-
-  // Now imageData contains a copy of the bitmap data
-
-  CGContextRelease(context);
-  CGColorSpaceRelease(colorSpace);
-
-  // Use the copied data
-  ImageData result;
-  result.width = static_cast<int>(width);
-  result.height = static_cast<int>(height);
-  result.data = imageData;
-  result.format = wgpu::TextureFormat::RGBA8Unorm;
-  return result;
-}
-
-} // namespace rnwgpu

package/ios/MetalView.h

@@ -1,10 +0,0 @@
-#pragma once
-
-#import "WebGPUModule.h"
-#import <UIKit/UIKit.h>
-
-@interface MetalView : UIView
-
-@property NSNumber *contextId;
-
-@end

package/ios/MetalView.mm

@@ -1,40 +0,0 @@
-#import "MetalView.h"
-#import "SurfaceUtils.h"
-#import "webgpu_cpp.h"
-#import <React/RCTViewManager.h>
-
-@implementation MetalView {
-  BOOL _isConfigured;
-}
-
-+ (Class)layerClass {
-  return [CAMetalLayer class];
-}
-
-// Paper only method
-// TODO: this method is wrong, it appears to call configureSurface instead of
-// updateSurface sometimes
-- (void)reactSetFrame:(CGRect)frame {
-  [super reactSetFrame:frame];
-  if (!_isConfigured) {
-    [SurfaceUtils configureSurface:self.layer
-                              size:self.frame.size
-                         contextId:[_contextId intValue]];
-    _isConfigured = YES;
-  } else {
-    [SurfaceUtils updateSurface:[_contextId intValue] size:self.frame.size];
-  }
-}
-
-- (void)willMoveToSuperview:(UIView *)newSuperview {
-  [super willMoveToSuperview:newSuperview];
-
-  if (newSuperview == nil) {
-    // The view is being removed from its superview
-    // Add your cleanup code here
-    [SurfaceUtils cleanupSurface:[_contextId intValue]];
-    _isConfigured = NO;
-  }
-}
-
-@end

package/ios/SurfaceUtils.h

@@ -1,15 +0,0 @@
-#pragma once
-
-#import <UIKit/UIKit.h>
-
-@interface SurfaceUtils : NSObject
-
-+ (void)configureSurface:(CALayer *)layer
-                    size:(CGSize)size
-               contextId:(int)contextId;
-
-+ (void)updateSurface:(int)contextId size:(CGSize)size;
-
-+ (void)cleanupSurface:(int)contextId;
-
-@end

package/ios/SurfaceUtils.mm

@@ -1,34 +0,0 @@
-#import "SurfaceUtils.h"
-#import "MetalView.h"
-#import "WebGPUModule.h"
-#import "WebGPUView.h"
-
-@implementation SurfaceUtils
-
-+ (void)configureSurface:(CALayer *)layer
-                    size:(CGSize)size
-               contextId:(int)contextId {
-  std::shared_ptr<rnwgpu::RNWebGPUManager> manager = [WebGPUModule getManager];
-  void *nativeSurface = (__bridge void *)layer;
-  auto &registry = rnwgpu::SurfaceRegistry::getInstance();
-  auto gpu = manager->_gpu;
-  auto surface = manager->_platformContext->makeSurface(
-      gpu, nativeSurface, size.width, size.height);
-  registry.getSurfaceInfoOrCreate(contextId, gpu, size.width, size.height)
-      ->switchToOnscreen(nativeSurface, surface);
-}
-
-+ (void)updateSurface:(int)contextId size:(CGSize)size {
-  // std::shared_ptr<rnwgpu::RNWebGPUManager> manager = [WebGPUModule
-  // getManager];
-  auto &registry = rnwgpu::SurfaceRegistry::getInstance();
-  registry.getSurfaceInfo(contextId)->resize(size.width, size.height);
-}
-
-+ (void)cleanupSurface:(int)contextId {
-  auto &registry = rnwgpu::SurfaceRegistry::getInstance();
-  // Remove the surface info from the registry
-  registry.removeSurfaceInfo(contextId);
-}
-
-@end

package/ios/WebGPUModule.h

@@ -1,19 +0,0 @@
-#pragma once
-
-#import "RNWebGPUManager.h"
-#import <React/RCTBridgeModule.h>
-#import <React/RCTEventEmitter.h>
-
-#ifdef RCT_NEW_ARCH_ENABLED
-#import <RNWgpuViewSpec/RNWgpuViewSpec.h>
-@interface WebGPUModule : RCTEventEmitter <NativeWebGPUModuleSpec>
-#else
-@interface WebGPUModule : RCTEventEmitter <RCTBridgeModule>
-#endif
-
-@property(nonatomic, weak) RCTBridge *bridge;
-@property(nonatomic, weak) RCTModuleRegistry *moduleRegistry;
-
-+ (std::shared_ptr<rnwgpu::RNWebGPUManager>)getManager;
-
-@end

package/ios/WebGPUModule.mm

@@ -1,93 +0,0 @@
-#import "WebGPUModule.h"
-#import "GPUCanvasContext.h"
-#include "IOSPlatformContext.h"
-
-#import <React/RCTBridge+Private.h>
-#import <React/RCTLog.h>
-#import <React/RCTUIManagerUtils.h>
-#import <ReactCommon/RCTTurboModule.h>
-#import <jsi/jsi.h>
-#import <memory>
-
-namespace jsi = facebook::jsi;
-namespace react = facebook::react;
-
-@implementation WebGPUModule
-
-RCT_EXPORT_MODULE(WebGPUModule)
-
-static std::shared_ptr<rnwgpu::RNWebGPUManager> webgpuManager;
-
-+ (std::shared_ptr<rnwgpu::RNWebGPUManager>)getManager {
-  return webgpuManager;
-}
-
-#pragma Setup and invalidation
-
-+ (BOOL)requiresMainQueueSetup {
-  return YES;
-}
-
-- (void)invalidate {
-  // if (_webgpuManager != nil) {
-  //   [_webgpuManager invalidate];
-  // }
-  webgpuManager = nil;
-}
-
-- (std::shared_ptr<rnwgpu::RNWebGPUManager>)getManager {
-  return webgpuManager;
-}
-
-RCT_EXPORT_BLOCKING_SYNCHRONOUS_METHOD(install) {
-  if (webgpuManager != nil) {
-    // Already initialized, ignore call.
-    return @true;
-  }
-  RCTCxxBridge *cxxBridge = (RCTCxxBridge *)[RCTBridge currentBridge];
-  if (!cxxBridge.runtime) {
-    NSLog(@"Failed to install react-native-filament: RCTBridge is not a "
-          @"RCTCxxBridge!");
-    return [NSNumber numberWithBool:NO];
-  }
-
-  jsi::Runtime *runtime = (jsi::Runtime *)cxxBridge.runtime;
-  if (!runtime) {
-    NSLog(@"Failed to install react-native-filament: jsi::Runtime* was null!");
-    return [NSNumber numberWithBool:NO];
-  }
-  std::shared_ptr<react::CallInvoker> jsInvoker = cxxBridge.jsCallInvoker;
-  if (!jsInvoker) {
-    NSLog(@"Failed to install react-native-filament: react::CallInvoker was "
-          @"null!");
-    return [NSNumber numberWithBool:NO];
-  }
-
-  if (!jsInvoker) {
-    jsInvoker = cxxBridge.jsCallInvoker;
-  }
-  std::shared_ptr<rnwgpu::PlatformContext> platformContext =
-      std::make_shared<rnwgpu::IOSPlatformContext>();
-  // TODO: remove allocation here
-  webgpuManager = std::make_shared<rnwgpu::RNWebGPUManager>(runtime, jsInvoker,
-                                                            platformContext);
-  return @true;
-}
-
-// RCT_EXPORT_BLOCKING_SYNCHRONOUS_METHOD(createSurfaceContext
-//                                        : (double)contextId) {
-//   int contextIdInt = contextId;
-//   RCTCxxBridge *cxxBridge = (RCTCxxBridge *)self.bridge;
-//   auto runtime = (jsi::Runtime *)cxxBridge.runtime;
-
-//   return @true;
-// }
-
-#ifdef RCT_NEW_ARCH_ENABLED
-- (std::shared_ptr<facebook::react::TurboModule>)getTurboModule:
-    (const facebook::react::ObjCTurboModule::InitParams &)params {
-  return std::make_shared<facebook::react::NativeWebGPUModuleSpecJSI>(params);
-}
-#endif
-
-@end

package/ios/WebGPUView.h

@@ -1,18 +0,0 @@
-#ifdef RCT_NEW_ARCH_ENABLED
-#pragma once
-
-#import "MetalView.h"
-#import <React/RCTViewComponentView.h>
-#import <UIKit/UIKit.h>
-
-NS_ASSUME_NONNULL_BEGIN
-
-@interface WebGPUView : RCTViewComponentView
-+ (void)registerMetalView:(MetalView *)metalView
-            withContextId:(NSNumber *)contextId;
-+ (bool)isContextRegisterd:(NSNumber *)contextId;
-@end
-
-NS_ASSUME_NONNULL_END
-
-#endif /* RCT_NEW_ARCH_ENABLED */

package/ios/WebGPUView.mm

@@ -1,86 +0,0 @@
-#ifdef RCT_NEW_ARCH_ENABLED
-#import "WebGPUView.h"
-
-#import <react/renderer/components/RNWgpuViewSpec/EventEmitters.h>
-#import <react/renderer/components/RNWgpuViewSpec/Props.h>
-#import <react/renderer/components/RNWgpuViewSpec/RCTComponentViewHelpers.h>
-
-#import "MetalView.h"
-#import "RCTFabricComponentsPlugins.h"
-#import "Utils.h"
-#import "WebGPUModule.h"
-#import "WebGPUViewComponentDescriptor.h"
-
-using namespace facebook::react;
-
-@interface WebGPUView () <RCTWebGPUViewViewProtocol>
-
-@end
-
-@implementation WebGPUView {
-  NSNumber *_contextId;
-}
-
-static NSMutableDictionary<NSNumber *, MetalView *> *metalViewRegistry =
-    [NSMutableDictionary new];
-
-+ (ComponentDescriptorProvider)componentDescriptorProvider {
-  return concreteComponentDescriptorProvider<WebGPUViewComponentDescriptor>();
-}
-
-+ (void)registerMetalView:(MetalView *)metalView
-            withContextId:(NSNumber *)contextId {
-  metalViewRegistry[contextId] = metalView;
-}
-
-+ (bool)isContextRegisterd:(NSNumber *)contextId {
-  return metalViewRegistry[contextId] != nil;
-}
-
-- (instancetype)initWithFrame:(CGRect)frame {
-  if (self = [super initWithFrame:frame]) {
-    static const auto defaultProps = std::make_shared<const WebGPUViewProps>();
-    _props = defaultProps;
-  }
-
-  return self;
-}
-
-- (void)prepareForRecycle {
-  [super prepareForRecycle];
-  self.contentView = nil;
-  if ([metalViewRegistry objectForKey:_contextId] != nil) {
-    [metalViewRegistry removeObjectForKey:_contextId];
-  }
-}
-
-- (void)updateProps:(const facebook::react::Props::Shared &)props
-           oldProps:(const facebook::react::Props::Shared &)oldProps {
-  const auto &oldViewProps =
-      *std::static_pointer_cast<const WebGPUViewProps>(_props);
-  const auto &newViewProps =
-      *std::static_pointer_cast<const WebGPUViewProps>(props);
-
-  if (newViewProps.contextId != oldViewProps.contextId) {
-    _contextId = [[NSNumber alloc] initWithInt:newViewProps.contextId];
-  }
-
-  [super updateProps:props oldProps:oldProps];
-}
-
-- (void)updateLayoutMetrics:
-            (const facebook::react::LayoutMetrics &)layoutMetrics
-           oldLayoutMetrics:
-               (const facebook::react::LayoutMetrics &)oldLayoutMetrics {
-  [super updateLayoutMetrics:layoutMetrics oldLayoutMetrics:oldLayoutMetrics];
-  if (!self.contentView) {
-    const auto &props =
-        *std::static_pointer_cast<WebGPUViewProps const>(_props);
-    self.contentView = metalViewRegistry[@(props.contextId)];
-  }
-}
-
-Class<RCTComponentViewProtocol> WebGPUViewCls(void) { return WebGPUView.class; }
-
-@end
-#endif

package/ios/WebGPUViewComponentDescriptor.h

@@ -1,61 +0,0 @@
-#ifdef RCT_NEW_ARCH_ENABLED
-
-#pragma once
-
-#include <react/debug/react_native_assert.h>
-#include <react/renderer/components/RNWgpuViewSpec/ShadowNodes.h>
-#include <react/renderer/components/view/ConcreteViewShadowNode.h>
-#include <react/renderer/core/ConcreteComponentDescriptor.h>
-#include <react/renderer/core/LayoutContext.h>
-
-#import "MetalView.h"
-#import "SurfaceUtils.h"
-#import "WebGPUView.h"
-
-namespace facebook {
-namespace react {
-
-class WebGPUViewCustomShadowNode final : public WebGPUViewShadowNode {
-
-public:
-  using ConcreteViewShadowNode::ConcreteViewShadowNode;
-
-  void layout(LayoutContext layoutContext) override {
-    YogaLayoutableShadowNode::layout(layoutContext);
-    configureSurface();
-  }
-
-  void configureSurface() {
-    const auto &viewProps =
-        *std::static_pointer_cast<WebGPUViewProps const>(props_);
-    if ([WebGPUView isContextRegisterd:@(viewProps.contextId)]) {
-      return;
-    }
-    __block MetalView *metalView;
-    __block CALayer *layer;
-    dispatch_sync(dispatch_get_main_queue(), ^{
-      metalView = [[MetalView alloc] init];
-      layer = metalView.layer;
-    });
-    [WebGPUView registerMetalView:metalView
-                    withContextId:@(viewProps.contextId)];
-
-    // TODO: use physical size
-    float width = layoutMetrics_.frame.size.width;
-    float height = layoutMetrics_.frame.size.height;
-    [SurfaceUtils configureSurface:layer
-                              size:CGSizeMake(width, height)
-                         contextId:viewProps.contextId];
-  }
-};
-
-class WebGPUViewComponentDescriptor final
-    : public ConcreteComponentDescriptor<WebGPUViewCustomShadowNode> {
-public:
-  using ConcreteComponentDescriptor::ConcreteComponentDescriptor;
-};
-
-} // namespace react
-} // namespace facebook
-
-#endif // RCT_NEW_ARCH_ENABLED

package/ios/WebGPUViewManager.mm

@@ -1,23 +0,0 @@
-#import "MetalView.h"
-#import "RCTBridge.h"
-#import "WebGPUModule.h"
-#import <React/RCTUIManager.h>
-#import <React/RCTViewManager.h>
-
-@interface WebGPUViewManager : RCTViewManager
-@end
-
-@implementation WebGPUViewManager
-
-RCT_EXPORT_MODULE(WebGPUView)
-
-- (UIView *)view {
-  return [MetalView new];
-}
-
-RCT_CUSTOM_VIEW_PROPERTY(contextId, NSNumber, UIView) {
-  NSNumber *contextId = [RCTConvert NSNumber:json];
-  [(MetalView *)view setContextId:contextId];
-}
-
-@end

package/src/hooks.ts

@@ -1,53 +0,0 @@
-import { useEffect, useRef, useState } from "react";
-
-import type { RNCanvasContext, CanvasRef, NativeCanvas } from "./Canvas";
-
-type Unsubscribe = () => void;
-
-export const useSurface = () => {
-  const [surface, setSurface] = useState<NativeCanvas | null>(null);
-  const ref = useCanvasEffect(() => {
-    const sur = ref.current!.getNativeSurface();
-    setSurface(sur);
-  });
-  return { ref, surface };
-};
-
-export const useGPUContext = () => {
-  const [context, setContext] = useState<RNCanvasContext | null>(null);
-  const ref = useCanvasEffect(() => {
-    const ctx = ref.current!.getContext("webgpu")!;
-    setContext(ctx);
-  });
-  return { ref, context };
-};
-
-export const useCanvasEffect = (
-  effect: () =>
-    | void
-    | Unsubscribe
-    | Promise<Unsubscribe | void>
-    | Promise<void>,
-) => {
-  const unsub = useRef<Unsubscribe | null | Promise<Unsubscribe | void>>(null);
-  const ref = useRef<CanvasRef>(null);
-  useEffect(() => {
-    ref.current!.whenReady(async () => {
-      const sub = effect();
-      if (sub) {
-        unsub.current = sub;
-      }
-    });
-    return () => {
-      if (unsub.current) {
-        if (unsub.current instanceof Promise) {
-          unsub.current.then((sub) => sub && sub());
-        } else {
-          unsub.current();
-        }
-      }
-    };
-    // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, []);
-  return ref;
-};