Index: remoting/client/dual_buffer_frame_consumer.h |
diff --git a/remoting/client/dual_buffer_frame_consumer.h b/remoting/client/dual_buffer_frame_consumer.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a1f609f4f0510701295cf882e25ab30b85f6538c |
--- /dev/null |
+++ b/remoting/client/dual_buffer_frame_consumer.h |
@@ -0,0 +1,82 @@ |
+// Copyright 2016 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. |
+ |
+#ifndef REMOTING_CLIENT_DUAL_BUFFER_FRAME_CONSUMER_H_ |
+#define REMOTING_CLIENT_DUAL_BUFFER_FRAME_CONSUMER_H_ |
+ |
+#include "base/callback.h" |
+#include "base/macros.h" |
+#include "base/memory/ref_counted.h" |
+#include "base/memory/weak_ptr.h" |
+#include "base/single_thread_task_runner.h" |
+#include "base/threading/thread_checker.h" |
+#include "remoting/protocol/frame_consumer.h" |
+#include "third_party/webrtc/modules/desktop_capture/desktop_region.h" |
+ |
+namespace webrtc { |
+class SharedDesktopFrame; |
+} // namespace webrtc |
+ |
+namespace remoting { |
+ |
+// This class continuously uses two BasicDesktopFrame as buffer for decoding |
+// updated regions until the resolution is changed. |
+// This class should be used and destroyed on the same thread. If |task_runner| |
+// is null |callback| will be run directly upon the stack of DrawFrame, |
+// otherwise a task will be posted to feed the callback on the thread of |
+// |task_runner|. |
+// Only areas bound by updated_region() on the buffer are considered valid to |
+// |callback|. Please use RequestFullDesktopFrame() if you want to get a full |
+// desktop frame. |
+class DualBufferFrameConsumer : public protocol::FrameConsumer { |
+ public: |
+ // RenderCallback(decoded_frame, done) |
+ // |done| should be run after it is rendered. Can be called on any thread. |
+ using RenderCallback = |
+ base::Callback<void(std::unique_ptr<webrtc::DesktopFrame>, |
+ const base::Closure&)>; |
+ DualBufferFrameConsumer( |
+ const RenderCallback& callback, |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner, |
+ PixelFormat format); |
+ ~DualBufferFrameConsumer() override; |
+ |
+ // Feeds the callback on the right thread with a BasicDesktopFrame that merges |
+ // updates from buffer_[0] and buffer_[1]. Do nothing if no updates have |
+ // received yet. |
+ void RequestFullDesktopFrame(); |
+ |
+ // FrameConsumer interface. |
+ std::unique_ptr<webrtc::DesktopFrame> AllocateFrame( |
+ const webrtc::DesktopSize& size) override; |
+ void DrawFrame(std::unique_ptr<webrtc::DesktopFrame> frame, |
+ const base::Closure& done) override; |
+ PixelFormat GetPixelFormat() override; |
+ |
+ base::WeakPtr<DualBufferFrameConsumer> GetWeakPtr(); |
+ |
+ private: |
+ void RunRenderCallback(std::unique_ptr<webrtc::DesktopFrame> frame, |
+ const base::Closure& done); |
+ |
+ std::unique_ptr<webrtc::SharedDesktopFrame> buffers_[2]; |
+ |
+ // Represents dirty regions that are currently in buffers_[1]. Will be used |
+ // when calling RequestFullDesktopFrame() to construct the full desktop frame. |
+ webrtc::DesktopRegion buffer_1_mask_; |
+ |
+ int current_buffer_ = 0; |
+ |
+ RenderCallback callback_; |
+ scoped_refptr<base::SingleThreadTaskRunner> task_runner_; |
+ PixelFormat pixel_format_; |
+ base::ThreadChecker thread_checker_; |
+ base::WeakPtr<DualBufferFrameConsumer> weak_ptr_; |
+ base::WeakPtrFactory<DualBufferFrameConsumer> weak_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(DualBufferFrameConsumer); |
+}; |
+ |
+} // namespace remoting |
+#endif // REMOTING_CLIENT_DUAL_BUFFER_FRAME_CONSUMER_H_ |