1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- // 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/memory/ref_counted.h"
- #include "base/memory/weak_ptr.h"
- #include "base/task/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"
- #include "third_party/webrtc/modules/desktop_capture/shared_desktop_frame.h"
- 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::RepeatingCallback<void(std::unique_ptr<webrtc::DesktopFrame>,
- base::OnceClosure)>;
- DualBufferFrameConsumer(
- RenderCallback callback,
- scoped_refptr<base::SingleThreadTaskRunner> task_runner,
- PixelFormat format);
- DualBufferFrameConsumer(const DualBufferFrameConsumer&) = delete;
- DualBufferFrameConsumer& operator=(const DualBufferFrameConsumer&) = delete;
- ~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,
- base::OnceClosure done) override;
- PixelFormat GetPixelFormat() override;
- base::WeakPtr<DualBufferFrameConsumer> GetWeakPtr();
- private:
- void RunRenderCallback(std::unique_ptr<webrtc::DesktopFrame> frame,
- base::OnceClosure 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_{this};
- };
- } // namespace remoting
- #endif // REMOTING_CLIENT_DUAL_BUFFER_FRAME_CONSUMER_H_
|