Add support for drawing video onto a Java ByteBuffer

remoting/client/jni/frame_consumer_impl.cc

Index: remoting/client/jni/frame_consumer_impl.cc
diff --git a/remoting/client/jni/frame_consumer_impl.cc b/remoting/client/jni/frame_consumer_impl.cc
new file mode 100644
index 0000000000000000000000000000000000000000..30c54bae680a4f8f200dcd7568a3ab53bfd3e5d5
--- /dev/null
+++ b/remoting/client/jni/frame_consumer_impl.cc
@@ -0,0 +1,85 @@
+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "remoting/client/jni/frame_consumer_impl.h"
+
+#include "base/android/jni_android.h"
+#include "base/logging.h"
+#include "media/base/yuv_convert.h"
+#include "remoting/client/frame_producer.h"
+#include "remoting/client/jni/chromoting_jni.h"
+#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
+
+namespace {
+
+class DirectDesktopFrame : public webrtc::BasicDesktopFrame {
+ public:
+  DirectDesktopFrame(int width, int height);
+
+  jobject buffer() const {
+    return buffer_;
+  }
+
+ private:
+  jobject buffer_;
+};
+
+DirectDesktopFrame::DirectDesktopFrame(int width, int height)
+    : webrtc::BasicDesktopFrame(webrtc::DesktopSize(width, height)) {
+  JNIEnv* env = base::android::AttachCurrentThread();
+  buffer_ = env->NewDirectByteBuffer(data(), stride()*height);
+}
+
+}  // namespace
+
+namespace remoting {
+
+FrameConsumerImpl::FrameConsumerImpl()
+    : frame_producer_(NULL),
+      buffer_(NULL) {
+  media::InitializeCPUSpecificYUVConversions();
+}
+
+FrameConsumerImpl::~FrameConsumerImpl() {
+  ReturnBuffer(buffer_);
+}
+
+void FrameConsumerImpl::set_frame_producer(FrameProducer* producer) {
+  frame_producer_ = producer;
+}
+
+void FrameConsumerImpl::ApplyBuffer(const SkISize& view_size,
+                                    const SkIRect& clip_area,
+                                    webrtc::DesktopFrame* buffer,
+                                    const SkRegion& region) {
+  ChromotingJni::GetInstance()->RedrawCanvas();
+  // Give |frame_producer_| back the same buffer so that |buffer| and |buffer_|
+  // are always the same exact pointer.
+  frame_producer_->DrawBuffer(buffer);
+}
+
+void FrameConsumerImpl::ReturnBuffer(webrtc::DesktopFrame* buffer) {
+  delete buffer;
+  // |buffer_| is the same pointer as |buffer|, so we just invalidated it.
+  buffer_ = NULL;
+}
+
+void FrameConsumerImpl::SetSourceSize(const SkISize& source_size,
+                                      const SkIPoint& dpi) {
+  view_size_ = source_size;
+  clip_area_ = SkIRect::MakeSize(source_size);
+  frame_producer_->SetOutputSizeAndClip(view_size_, clip_area_);
+
+  if (!buffer_)
+    buffer_ = new DirectDesktopFrame(source_size.width(),
+                                     source_size.height());
+
+  ChromotingJni::GetInstance()->UpdateImageBuffer(
+      source_size.width(),
+      source_size.height(),
+      static_cast<DirectDesktopFrame*>(buffer_)->buffer());
+  frame_producer_->DrawBuffer(buffer_);
+}
+
+}  // namespace remoting