@byteplus/react-native-live-push 1.1.3-rc.1 → 1.1.3-rc.3

package/android/src/main/java/com/volcengine/velive/rn/push/mixer/YuvHelper.java

@@ -0,0 +1,165 @@
+package com.volcengine.velive.rn.push.mixer;
+
+import android.content.Context;
+import android.graphics.Bitmap;
+import android.graphics.Matrix;
+import android.renderscript.Allocation;
+import android.renderscript.Element;
+import android.renderscript.RenderScript;
+import android.renderscript.ScriptIntrinsicYuvToRGB;
+import android.renderscript.Type;
+
+import java.nio.ByteBuffer;
+
+public class YuvHelper {
+    private static RenderScript globalRS = null;
+    public static synchronized RenderScript getGlobalRS(Context context) {
+        if (globalRS == null) {
+            globalRS = RenderScript.create(context.getApplicationContext());
+        }
+        return globalRS;
+    }
+    public static void destroyGlobalRS() {
+        if (globalRS != null) {
+            globalRS.destroy();
+            globalRS = null;
+        }
+    }
+    public static class NV21ToBitmap {
+        private RenderScript rs;
+        private ScriptIntrinsicYuvToRGB yuvToRgbIntrinsic;
+        private Type.Builder yuvType, rgbaType;
+
+        public NV21ToBitmap(Context context) {
+            rs = YuvHelper.getGlobalRS(context);
+            yuvToRgbIntrinsic = ScriptIntrinsicYuvToRGB.create(rs, Element.U8_4(rs));
+            yuvType = new Type.Builder(rs, Element.U8(rs));
+            rgbaType = new Type.Builder(rs, Element.RGBA_8888(rs));
+        }
+
+        public void cleanup() {
+            try {
+                if (yuvToRgbIntrinsic != null) {
+                    yuvToRgbIntrinsic.destroy();
+                    yuvToRgbIntrinsic = null;
+                }
+                // 不销毁全局 rs
+                // rs = null;
+            } catch (Exception e) {
+                // Ignore
+            }
+        }
+
+        public Bitmap nv21ToBitmap(byte[] nv21, int width, int height) {
+            Allocation in = null;
+            Allocation out = null;
+            try {
+                yuvType.setX(nv21.length);
+                in = Allocation.createTyped(rs, yuvType.create(), Allocation.USAGE_SCRIPT);
+                in.copyFrom(nv21);
+
+                rgbaType.setX(width).setY(height);
+                out = Allocation.createTyped(rs, rgbaType.create(), Allocation.USAGE_SCRIPT);
+                yuvToRgbIntrinsic.setInput(in);
+                yuvToRgbIntrinsic.forEach(out);
+
+                Bitmap bmp = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
+                out.copyTo(bmp);
+
+                return bmp;
+            } finally {
+                if (in != null) {
+                    in.destroy();
+                }
+                if (out != null) {
+                    out.destroy();
+                }
+            }
+        }
+    }
+
+    public static void NV21toI420(ByteBuffer input, ByteBuffer output, final int width, final int height) {
+        final int size = width * height;
+        final int quarter = size / 4;
+        final int v0 = size + quarter;
+
+        System.arraycopy(input.array(), 0, output.array(), 0, size);
+
+        for (int u = size, v = v0, o = size; u < v0; u++, v++, o += 2) {
+            output.array()[v] = input.array()[o];
+            output.array()[u] = input.array()[o + 1];
+        }
+        output.limit(width * height * 3 / 2);
+        output.position(0);
+    }
+
+    public static void NV21toNV12(ByteBuffer input, ByteBuffer output, final int width, final int height) {
+        final int size = width * height;
+
+        System.arraycopy(input.array(), 0, output.array(), 0, size);
+        for (int i = size; i < 1.5f * size; i += 2) {
+            output.array()[i] = input.array()[i + 1];
+            output.array()[i + 1] = input.array()[i];
+        }
+        output.position(0);
+        output.limit(width * height * 3 / 2);
+    }
+
+    public static void NV12toI420(ByteBuffer input, ByteBuffer output, final int width, final int height) {
+        final int size = width * height;
+        final int quarter = size / 4;
+        final int v0 = size + quarter;
+
+        System.arraycopy(input.array(), 0, output.array(), 0, size);
+
+        for (int u = size, v = v0, o = size; u < v0; u++, v++, o += 2) {
+            output.array()[v] = input.array()[o + 1];
+            output.array()[u] = input.array()[o];
+        }
+    }
+
+    public static void I420toNV21(ByteBuffer input, ByteBuffer output, final int width, final int height) {
+        final int size = width * height;
+        final int quarter = size / 4;
+        final int v0 = size + quarter;
+
+        System.arraycopy(input.array(), 0, output.array(), 0, size);
+
+        for (int u = size, v = v0, o = size; u < v0; u++, v++, o += 2) {
+            output.array()[o + 1] = input.array()[v];
+            output.array()[o] = input.array()[u];
+        }
+    }
+
+    public static Bitmap rotateBitmap(Bitmap origin, float alpha, boolean horizontalMirror, boolean verticalMirror) {
+        if (origin == null || origin.isRecycled()) {
+            return null;
+        }
+        
+        try {
+            int width = origin.getWidth();
+            int height = origin.getHeight();
+            Matrix matrix = new Matrix();
+            float sx = 1;
+            float sy = 1;
+            if (horizontalMirror) {
+                sx = -1;
+            }
+            if (verticalMirror) {
+                sy = -1;
+            }
+            matrix.postScale(sx, sy);
+            matrix.postRotate(alpha);
+            
+            if (origin.isRecycled()) {
+                return null;
+            }
+            
+            return Bitmap.createBitmap(origin, 0, 0, width, height, matrix, false);
+        } catch (IllegalStateException e) {
+            return null;
+        } catch (Exception e) {
+            return null;
+        }
+    }
+}

package/ios/VeLiveMixerHelper.h

@@ -0,0 +1,44 @@
+//
+//  VeLiveMixerHelper.h
+//  react-native-live-push
+//
+//  Created by ByteDance on 2024/12/09.
+//
+
+#import <Foundation/Foundation.h>
+#import "VeLiveMixerView.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+// 前向声明
+@class VeLivePusher;
+
+// 混流管理器助手类 - 完全对齐Android MixerManager.java
+@interface VeLiveMixerHelper : NSObject <VeLiveBitmapCaptureCallback>
+
+// 对齐Android的static cachedMixedViews
+@property (class, nonatomic, strong) NSMutableDictionary<NSString *, VeLiveMixerUIView *> *cachedMixedViews;
+
+@property (nonatomic, weak) VeLivePusher *pusher;
+
+- (instancetype)initWithPusher:(VeLivePusher *)pusher;
+
+// Public方法 - 完全对齐Android MixerManager的public接口
+- (void)setPusher:(VeLivePusher *)pusher;
+- (int)addView:(NSString *)viewId config:(NSDictionary *)viewInfo;
+- (BOOL)updateView:(NSString *)viewId config:(NSDictionary *)viewInfo;
+- (BOOL)removeView:(NSString *)viewId;
+- (void)captureView:(NSString *)viewId;
+
+// Static方法 - 只保留Android中实际存在的
++ (void)setupCallbackForMixerView:(VeLiveMixerUIView *)mixerView streamId:(int)streamId;
++ (void)sendBitmapToMixerStatic:(int)streamId bitmap:(UIImage *)bitmap;
++ (VeLiveMixerHelper *)findActiveMixerManager;
++ (void)onViewReady:(NSString *)viewId mixerView:(VeLiveMixerUIView *)mixerView;
+
+// 清理资源
+- (void)cleanup;
+
+@end
+
+NS_ASSUME_NONNULL_END