Add support for adding VideoSinks to VideoTracks.

webrtc/sdk/android/src/jni/native_handle_impl.cc

 /*
  *  Copyright 2015 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include "webrtc/sdk/android/src/jni/native_handle_impl.h"
 
 #include <memory>
 
 #include "webrtc/common_video/include/video_frame_buffer.h"
 #include "webrtc/rtc_base/bind.h"
 #include "webrtc/rtc_base/checks.h"
 #include "webrtc/rtc_base/keep_ref_until_done.h"
 #include "webrtc/rtc_base/logging.h"
 #include "webrtc/rtc_base/scoped_ref_ptr.h"
 #include "webrtc/rtc_base/timeutils.h"
 #include "webrtc/sdk/android/src/jni/classreferenceholder.h"
 #include "webrtc/sdk/android/src/jni/jni_helpers.h"
+#include "webrtc/sdk/android/src/jni/wrapped_native_i420_buffer.h"
 #include "webrtc/system_wrappers/include/aligned_malloc.h"
 
 namespace webrtc_jni {
 
 namespace {
 
 class AndroidVideoI420Buffer : public webrtc::I420BufferInterface {
  public:
   // Wraps an existing reference to a Java VideoBuffer. Retain will not be
   // called but release will be called when the C++ object is destroyed.
@@ skipping 433 matching lines @@
       jni, j_retain_id_, j_release_id_, width, height, j_video_frame_buffer);
 }
 
 JavaVideoFrameFactory::JavaVideoFrameFactory(JNIEnv* jni)
     : j_video_frame_class_(jni, FindClass(jni, "org/webrtc/VideoFrame")) {
   j_video_frame_constructor_id_ =
       GetMethodID(jni, *j_video_frame_class_, "<init>",
                   "(Lorg/webrtc/VideoFrame$Buffer;IJ)V");
 }
 
+static bool IsJavaVideoBuffer(
+    rtc::scoped_refptr<webrtc::VideoFrameBuffer> buffer) {
+  if (buffer->type() != webrtc::VideoFrameBuffer::Type::kNative) {
+    return false;
+  }
+  AndroidVideoFrameBuffer* android_buffer =
+      static_cast<AndroidVideoFrameBuffer*>(buffer.get());
+  return android_buffer->android_type() ==
+         AndroidVideoFrameBuffer::AndroidType::kJavaBuffer;
+}
+
 jobject JavaVideoFrameFactory::ToJavaFrame(
     JNIEnv* jni,
     const webrtc::VideoFrame& frame) const {
-  RTC_DCHECK(frame.video_frame_buffer()->type() ==
-             webrtc::VideoFrameBuffer::Type::kNative);
-  AndroidVideoFrameBuffer* android_buffer =
-      static_cast<AndroidVideoFrameBuffer*>(frame.video_frame_buffer().get());
-  RTC_DCHECK(android_buffer->android_type() ==
-             AndroidVideoFrameBuffer::AndroidType::kJavaBuffer);
-  AndroidVideoBuffer* android_video_buffer =
-      static_cast<AndroidVideoBuffer*>(android_buffer);
-  jobject buffer = android_video_buffer->video_frame_buffer();
+  rtc::scoped_refptr<webrtc::VideoFrameBuffer> buffer =
+      frame.video_frame_buffer();
+  jobject j_buffer;
+  if (IsJavaVideoBuffer(buffer)) {
+    RTC_DCHECK(buffer->type() == webrtc::VideoFrameBuffer::Type::kNative);
+    AndroidVideoFrameBuffer* android_buffer =
+        static_cast<AndroidVideoFrameBuffer*>(buffer.get());
+    RTC_DCHECK(android_buffer->android_type() ==
+               AndroidVideoFrameBuffer::AndroidType::kJavaBuffer);
+    AndroidVideoBuffer* android_video_buffer =
+        static_cast<AndroidVideoBuffer*>(android_buffer);
+    j_buffer = android_video_buffer->video_frame_buffer();
+  } else {
+    j_buffer = WrapI420Buffer(jni, buffer->ToI420());
+  }
   return jni->NewObject(
-      *j_video_frame_class_, j_video_frame_constructor_id_, buffer,
+      *j_video_frame_class_, j_video_frame_constructor_id_, j_buffer,
       static_cast<jint>(frame.rotation()),
       static_cast<jlong>(frame.timestamp_us() * rtc::kNumNanosecsPerMicrosec));
 }
 
 }  // namespace webrtc_jni