react-native-suuqencode 0.1.0 → 0.1.2

package/Suuqencode.podspec

@@ -16,7 +16,7 @@ Pod::Spec.new do |s|
   s.source_files = "ios/**/*.{h,m,mm,cpp}"
   s.private_header_files = "ios/**/*.h"
   s.frameworks = "VideoToolbox", "CoreMedia", "CoreFoundation"
-
+  s.dependency "SuuqeDMABuf"
 
   install_modules_dependencies(s)
 end

package/ios/Suuqencode.mm

@@ -1,10 +1,13 @@
 #import "Suuqencode.h"
+#import <SuuqeDMABuf/DMABuf.h>
 
-@interface Suuqencode()
+@interface Suuqencode ()
 
-@property (nonatomic) VTCompressionSessionRef compressionSession;
-@property (nonatomic) dispatch_queue_t encodeQueue;
-@property (nonatomic) int frameCount;
+@property(nonatomic) VTCompressionSessionRef compressionSession;
+@property(nonatomic) dispatch_queue_t encodeQueue;
+@property(nonatomic) int frameCount;
+
+- (void)sendEncodedData:(NSData *)data;
 
 @end
 
@@ -12,113 +15,132 @@
 
 RCT_EXPORT_MODULE()
 
-- (instancetype)init
-{
-    self = [super init];
-    if (self) {
-        _encodeQueue = dispatch_queue_create("com.suuqencode.encodequeue", DISPATCH_QUEUE_SERIAL);
-    }
-    return self;
-}
-
-RCT_EXPORT_METHOD(encode:(NSString *)base64Bitmap width:(int)width height:(int)height)
-{
-    dispatch_async(_encodeQueue, ^{
-        if (!self.compressionSession) {
-            [self setupCompressionSessionWithWidth:width height:height];
-        }
-
-        NSData *bitmapData = [[NSData alloc] initWithBase64EncodedString:base64Bitmap options:0];
-        if (!bitmapData) {
-            NSLog(@"Invalid base64 bitmap string");
-            return;
-        }
-
-        CVPixelBufferRef pixelBuffer = NULL;
-        CVReturn status = CVPixelBufferCreate(kCFAllocatorDefault, width, height, kCVPixelFormatType_32ARGB, nil, &pixelBuffer);
-        if (status != kCVReturnSuccess) {
-            NSLog(@"Failed to create CVPixelBuffer");
-            return;
-        }
-
-        CVPixelBufferLockBaseAddress(pixelBuffer, 0);
-        void *pixelData = CVPixelBufferGetBaseAddress(pixelBuffer);
-        memcpy(pixelData, [bitmapData bytes], [bitmapData length]);
-        CVPixelBufferUnlockBaseAddress(pixelBuffer, 0);
-
-        CMTime presentationTimeStamp = CMTimeMake(self.frameCount++, 30);
-        VTEncodeInfoFlags flags;
-
-        VTCompressionSessionEncodeFrame(self.compressionSession, pixelBuffer, presentationTimeStamp, kCMTimeInvalid, NULL, NULL, &flags);
-        CVPixelBufferRelease(pixelBuffer);
-    });
-}
-
-- (void)setupCompressionSessionWithWidth:(int)width height:(int)height {
-    VTCompressionSessionCreate(NULL, width, height, kCMVideoCodecType_H264, NULL, NULL, NULL, compressionOutputCallback, (__bridge void *)(self), &_compressionSession);
-
-    VTSessionSetProperty(_compressionSession, kVTCompressionPropertyKey_RealTime, kCFBooleanTrue);
-    VTSessionSetProperty(_compressionSession, kVTCompressionPropertyKey_ProfileLevel, kVTProfileLevel_H264_Baseline_AutoLevel);
-    VTSessionSetProperty(_compressionSession, kVTCompressionPropertyKey_AverageBitRate, (__bridge CFTypeRef)@(width * height * 10));
-    VTSessionSetProperty(_compressionSession, kVTCompressionPropertyKey_MaxKeyFrameInterval, (__bridge CFTypeRef)@(20));
-
-    VTCompressionSessionPrepareToEncodeFrames(_compressionSession);
+- (instancetype)init {
+  self = [super init];
+  if (self) {
+    _encodeQueue = dispatch_queue_create("com.suuqencode.encodequeue",
+                                         DISPATCH_QUEUE_SERIAL);
+  }
+  return self;
 }
 
-void compressionOutputCallback(void *outputCallbackRefCon, void *sourceFrameRefCon, OSStatus status, VTEncodeInfoFlags infoFlags, CMSampleBufferRef sampleBuffer) {
-    if (status != noErr) {
-        NSLog(@"Error encoding frame: %d", (int)status);
-        return;
-    }
+RCT_EXPORT_METHOD(startEncode) {
+  [DMABuf setFrameChangeCallback:^{
+    void *buf = [DMABuf buf];
+    int width = [DMABuf width];
+    int height = [DMABuf height];
 
-    if (!CMSampleBufferDataIsReady(sampleBuffer)) {
-        return;
+    if (!buf) {
+      return;
     }
 
-    Suuqencode *encoder = (__bridge Suuqencode *)outputCallbackRefCon;
+    dispatch_async(self.encodeQueue, ^{
+      if (!self.compressionSession) {
+        [self setupCompressionSessionWithWidth:width height:height];
+      }
 
-    bool isKeyFrame = !CFDictionaryContainsKey( (CFDictionaryRef)CFArrayGetValueAtIndex(CMSampleBufferGetSampleAttachmentsArray(sampleBuffer, true), 0), (const void *)kCMSampleAttachmentKey_NotSync);
+      CVPixelBufferRef pixelBuffer = NULL;
+      CVReturn status = CVPixelBufferCreateWithBytes(
+          kCFAllocatorDefault, width, height, kCVPixelFormatType_32BGRA, buf,
+          width * 4, NULL, NULL, NULL, &pixelBuffer);
 
-    if (isKeyFrame)
-    {
-        CMFormatDescriptionRef format = CMSampleBufferGetFormatDescription(sampleBuffer);
-        const uint8_t *sparameterSet;
-        size_t sparameterSetSize, sparameterSetCount;
-        CMVideoFormatDescriptionGetH264ParameterSetAtIndex(format, 0, &sparameterSet, &sparameterSetSize, &sparameterSetCount, 0 );
-
-        const uint8_t *pparameterSet;
-        size_t pparameterSetSize, pparameterSetCount;
-        CMVideoFormatDescriptionGetH264ParameterSetAtIndex(format, 1, &pparameterSet, &pparameterSetSize, &pparameterSetCount, 0 );
+      if (status != kCVReturnSuccess) {
+        NSLog(@"Failed to create CVPixelBuffer");
+        return;
+      }
 
-        NSData *sps = [NSData dataWithBytes:sparameterSet length:sparameterSetSize];
-        NSData *pps = [NSData dataWithBytes:pparameterSet length:pparameterSetSize];
+      CMTime presentationTimeStamp = CMTimeMake(self.frameCount++, 30);
+      VTEncodeInfoFlags flags;
 
-        [encoder sendEncodedData:sps];
-        [encoder sendEncodedData:pps];
-    }
+      VTCompressionSessionEncodeFrame(self.compressionSession, pixelBuffer,
+                                      presentationTimeStamp, kCMTimeInvalid,
+                                      NULL, NULL, &flags);
+      CVPixelBufferRelease(pixelBuffer);
+    });
+  }];
+}
 
-    CMBlockBufferRef dataBuffer = CMSampleBufferGetDataBuffer(sampleBuffer);
-    size_t length, totalLength;
-    char *dataPointer;
-    CMBlockBufferGetDataPointer(dataBuffer, 0, &length, &totalLength, &dataPointer);
+- (void)setupCompressionSessionWithWidth:(int)width height:(int)height {
+  VTCompressionSessionCreate(NULL, width, height, kCMVideoCodecType_H264, NULL,
+                             NULL, NULL, compressionOutputCallback,
+                             (__bridge void *)(self), &_compressionSession);
+
+  VTSessionSetProperty(_compressionSession, kVTCompressionPropertyKey_RealTime,
+                       kCFBooleanTrue);
+  VTSessionSetProperty(_compressionSession,
+                       kVTCompressionPropertyKey_ProfileLevel,
+                       kVTProfileLevel_H264_Baseline_AutoLevel);
+  VTSessionSetProperty(_compressionSession,
+                       kVTCompressionPropertyKey_AverageBitRate,
+                       (__bridge CFTypeRef) @(width * height * 10));
+  VTSessionSetProperty(_compressionSession,
+                       kVTCompressionPropertyKey_MaxKeyFrameInterval,
+                       (__bridge CFTypeRef) @(20));
+
+  VTCompressionSessionPrepareToEncodeFrames(_compressionSession);
+}
 
-    NSData *naluData = [NSData dataWithBytes:dataPointer length:length];
-    [encoder sendEncodedData:naluData];
+void compressionOutputCallback(void *outputCallbackRefCon,
+                               void *sourceFrameRefCon, OSStatus status,
+                               VTEncodeInfoFlags infoFlags,
+                               CMSampleBufferRef sampleBuffer) {
+  if (status != noErr) {
+    NSLog(@"Error encoding frame: %d", (int)status);
+    return;
+  }
+
+  if (!CMSampleBufferDataIsReady(sampleBuffer)) {
+    return;
+  }
+
+  Suuqencode *encoder = (__bridge Suuqencode *)outputCallbackRefCon;
+
+  bool isKeyFrame = !CFDictionaryContainsKey(
+      (CFDictionaryRef)CFArrayGetValueAtIndex(
+          CMSampleBufferGetSampleAttachmentsArray(sampleBuffer, true), 0),
+      (const void *)kCMSampleAttachmentKey_NotSync);
+
+  if (isKeyFrame) {
+    CMFormatDescriptionRef format =
+        CMSampleBufferGetFormatDescription(sampleBuffer);
+    const uint8_t *sparameterSet;
+    size_t sparameterSetSize, sparameterSetCount;
+    CMVideoFormatDescriptionGetH264ParameterSetAtIndex(
+        format, 0, &sparameterSet, &sparameterSetSize, &sparameterSetCount, 0);
+
+    const uint8_t *pparameterSet;
+    size_t pparameterSetSize, pparameterSetCount;
+    CMVideoFormatDescriptionGetH264ParameterSetAtIndex(
+        format, 1, &pparameterSet, &pparameterSetSize, &pparameterSetCount, 0);
+
+    NSData *sps = [NSData dataWithBytes:sparameterSet length:sparameterSetSize];
+    NSData *pps = [NSData dataWithBytes:pparameterSet length:pparameterSetSize];
+
+    [encoder sendEncodedData:sps];
+    [encoder sendEncodedData:pps];
+  }
+
+  CMBlockBufferRef dataBuffer = CMSampleBufferGetDataBuffer(sampleBuffer);
+  size_t length, totalLength;
+  char *dataPointer;
+  CMBlockBufferGetDataPointer(dataBuffer, 0, &length, &totalLength,
+                              &dataPointer);
+
+  NSData *naluData = [NSData dataWithBytes:dataPointer length:length];
+  [encoder sendEncodedData:naluData];
 }
 
 - (void)sendEncodedData:(NSData *)data {
-    NSString *base64Encoded = [data base64EncodedStringWithOptions:0];
-    [self sendEventWithName:@"onEncodedData" body:base64Encoded];
+  NSString *base64Encoded = [data base64EncodedStringWithOptions:0];
+  [self sendEventWithName:@"onEncodedData" body:base64Encoded];
 }
 
-- (NSArray<NSString *> *)supportedEvents
-{
-    return @[@"onEncodedData"];
+- (NSArray<NSString *> *)supportedEvents {
+  return @[ @"onEncodedData" ];
 }
 
-+ (BOOL)requiresMainQueueSetup
-{
-    return NO;
++ (BOOL)requiresMainQueueSetup {
+  return NO;
 }
 
 @end

package/lib/typescript/src/NativeSuuqencode.d.ts

@@ -1,6 +1,6 @@
 import { type TurboModule } from 'react-native';
 export interface Spec extends TurboModule {
-    encode(base64Bitmap: string, width: number, height: number): void;
+    startEncode(): void;
     addListener(eventName: string): void;
     removeListeners(count: number): void;
 }

package/lib/typescript/src/NativeSuuqencode.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"NativeSuuqencode.d.ts","sourceRoot":"","sources":["../../../src/NativeSuuqencode.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,KAAK,WAAW,EAAE,MAAM,cAAc,CAAC;AAErE,MAAM,WAAW,IAAK,SAAQ,WAAW;IACvC,MAAM,CAAC,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAClE,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtC;;AAED,wBAAoE"}
+{"version":3,"file":"NativeSuuqencode.d.ts","sourceRoot":"","sources":["../../../src/NativeSuuqencode.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuB,KAAK,WAAW,EAAE,MAAM,cAAc,CAAC;AAErE,MAAM,WAAW,IAAK,SAAQ,WAAW;IACvC,WAAW,IAAI,IAAI,CAAC;IACpB,WAAW,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtC;;AAED,wBAAoE"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-suuqencode",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "h264 hardware encode support for iOS, developed by Suuqe Llc.",
   "main": "./lib/module/index.js",
   "types": "./lib/typescript/src/index.d.ts",

package/src/NativeSuuqencode.ts

@@ -1,7 +1,7 @@
 import { TurboModuleRegistry, type TurboModule } from 'react-native';
 
 export interface Spec extends TurboModule {
-  encode(base64Bitmap: string, width: number, height: number): void;
+  startEncode(): void;
   addListener(eventName: string): void;
   removeListeners(count: number): void;
 }