From a3c1fb0b9b257ac600237c774608647171f8dccc Mon Sep 17 00:00:00 2001
From: CloudWebRTC <duanweiwei1982@gmail.com>
Date: Thu, 8 Jun 2023 13:46:51 +0800
Subject: [PATCH] fix: add WrappedVideoDecoderFactory.java. (#74)

* fix: add WrappedVideoDecoderFactory.java.

* fix compile for android.
---
 sdk/android/BUILD.gn                          |  1 +
 .../webrtc/WrappedVideoDecoderFactory.java    | 75 +++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 sdk/android/api/org/webrtc/WrappedVideoDecoderFactory.java

diff --git a/sdk/android/BUILD.gn b/sdk/android/BUILD.gn
index c4fd34a2a3..596dbc9693 100644
--- a/sdk/android/BUILD.gn
+++ b/sdk/android/BUILD.gn
@@ -364,6 +364,7 @@ if (is_android) {
     sources = [
       "api/org/webrtc/DefaultVideoDecoderFactory.java",
       "api/org/webrtc/DefaultVideoEncoderFactory.java",
+      "api/org/webrtc/WrappedVideoDecoderFactory.java",
     ]
 
     deps = [
diff --git a/sdk/android/api/org/webrtc/WrappedVideoDecoderFactory.java b/sdk/android/api/org/webrtc/WrappedVideoDecoderFactory.java
new file mode 100644
index 0000000000..a7acd37289
--- /dev/null
+++ b/sdk/android/api/org/webrtc/WrappedVideoDecoderFactory.java
@@ -0,0 +1,75 @@
+/*
+ * Copyright 2023 LiveKit
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.webrtc;
+
+import android.media.MediaCodecInfo;
+import androidx.annotation.Nullable;
+
+import java.util.Arrays;
+import java.util.LinkedHashSet;
+
+public class WrappedVideoDecoderFactory implements VideoDecoderFactory {
+    public WrappedVideoDecoderFactory(@Nullable EglBase.Context eglContext) {
+        this.hardwareVideoDecoderFactory = new HardwareVideoDecoderFactory(eglContext);
+        this.platformSoftwareVideoDecoderFactory = new PlatformSoftwareVideoDecoderFactory(eglContext);
+    }
+
+    private final VideoDecoderFactory hardwareVideoDecoderFactory;
+    private final VideoDecoderFactory hardwareVideoDecoderFactoryWithoutEglContext = new HardwareVideoDecoderFactory(null)   ;
+    private final VideoDecoderFactory softwareVideoDecoderFactory = new SoftwareVideoDecoderFactory();
+    @Nullable
+    private final VideoDecoderFactory platformSoftwareVideoDecoderFactory;
+
+    @Override
+    public VideoDecoder createDecoder(VideoCodecInfo codecType) {
+        VideoDecoder softwareDecoder = this.softwareVideoDecoderFactory.createDecoder(codecType);
+        VideoDecoder hardwareDecoder = this.hardwareVideoDecoderFactory.createDecoder(codecType);
+        if (softwareDecoder == null && this.platformSoftwareVideoDecoderFactory != null) {
+            softwareDecoder = this.platformSoftwareVideoDecoderFactory.createDecoder(codecType);
+        }
+
+        if(hardwareDecoder != null && disableSurfaceTextureFrame(hardwareDecoder.getImplementationName())) {
+            hardwareDecoder.release();
+            hardwareDecoder = this.hardwareVideoDecoderFactoryWithoutEglContext.createDecoder(codecType);
+        }
+
+        if (hardwareDecoder != null && softwareDecoder != null) {
+            return new VideoDecoderFallback(softwareDecoder, hardwareDecoder);
+        } else {
+            return hardwareDecoder != null ? hardwareDecoder : softwareDecoder;
+        }
+    }
+
+    private boolean disableSurfaceTextureFrame(String name) {
+        if (name.startsWith("OMX.qcom.") || name.startsWith("OMX.hisi.")) {
+            return true;
+        }
+        return false;
+    }
+
+    @Override
+    public VideoCodecInfo[] getSupportedCodecs() {
+        LinkedHashSet<VideoCodecInfo> supportedCodecInfos = new LinkedHashSet();
+        supportedCodecInfos.addAll(Arrays.asList(this.softwareVideoDecoderFactory.getSupportedCodecs()));
+        supportedCodecInfos.addAll(Arrays.asList(this.hardwareVideoDecoderFactory.getSupportedCodecs()));
+        if (this.platformSoftwareVideoDecoderFactory != null) {
+            supportedCodecInfos.addAll(Arrays.asList(this.platformSoftwareVideoDecoderFactory.getSupportedCodecs()));
+        }
+
+        return (VideoCodecInfo[])supportedCodecInfos.toArray(new VideoCodecInfo[supportedCodecInfos.size()]);
+    }
+}