| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef REMOTING_CLIENT_RECTANGLE_UPDATE_DECODER_H_ | |
| 6 #define REMOTING_CLIENT_RECTANGLE_UPDATE_DECODER_H_ | |
| 7 | |
| 8 #include <list> | |
| 9 | |
| 10 #include "base/callback_forward.h" | |
| 11 #include "base/memory/ref_counted.h" | |
| 12 #include "base/memory/scoped_ptr.h" | |
| 13 #include "remoting/codec/video_decoder.h" | |
| 14 #include "remoting/client/chromoting_stats.h" | |
| 15 #include "remoting/client/frame_consumer_proxy.h" | |
| 16 #include "remoting/client/frame_producer.h" | |
| 17 #include "remoting/protocol/video_stub.h" | |
| 18 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" | |
| 19 | |
| 20 namespace base { | |
| 21 class SingleThreadTaskRunner; | |
| 22 } // namespace base | |
| 23 | |
| 24 namespace remoting { | |
| 25 | |
| 26 class ChromotingStats; | |
| 27 | |
| 28 namespace protocol { | |
| 29 class SessionConfig; | |
| 30 } // namespace protocol | |
| 31 | |
| 32 // TODO(ajwong): Re-examine this API, especially with regards to how error | |
| 33 // conditions on each step are reported. Should they be CHECKs? Logs? Other? | |
| 34 // TODO(sergeyu): Rename this class. | |
| 35 class RectangleUpdateDecoder | |
| 36 : public base::RefCountedThreadSafe<RectangleUpdateDecoder>, | |
| 37 public FrameProducer, | |
| 38 public protocol::VideoStub { | |
| 39 public: | |
| 40 // Creates an update decoder on |main_task_runner_| and |decode_task_runner_|, | |
| 41 // outputting to |consumer|. The |main_task_runner_| is responsible for | |
| 42 // receiving and queueing packets. The |decode_task_runner_| is responsible | |
| 43 // for decoding the video packets. | |
| 44 // TODO(wez): Replace the ref-counted proxy with an owned FrameConsumer. | |
| 45 RectangleUpdateDecoder( | |
| 46 scoped_refptr<base::SingleThreadTaskRunner> main_task_runner, | |
| 47 scoped_refptr<base::SingleThreadTaskRunner> decode_task_runner, | |
| 48 scoped_refptr<FrameConsumerProxy> consumer); | |
| 49 | |
| 50 // Initializes decoder with the information from the protocol config. | |
| 51 void Initialize(const protocol::SessionConfig& config); | |
| 52 | |
| 53 // FrameProducer implementation. These methods may be called before we are | |
| 54 // Initialize()d, or we know the source screen size. | |
| 55 virtual void DrawBuffer(webrtc::DesktopFrame* buffer) OVERRIDE; | |
| 56 virtual void InvalidateRegion(const webrtc::DesktopRegion& region) OVERRIDE; | |
| 57 virtual void RequestReturnBuffers(const base::Closure& done) OVERRIDE; | |
| 58 virtual void SetOutputSizeAndClip( | |
| 59 const webrtc::DesktopSize& view_size, | |
| 60 const webrtc::DesktopRect& clip_area) OVERRIDE; | |
| 61 virtual const webrtc::DesktopRegion* GetBufferShape() OVERRIDE; | |
| 62 | |
| 63 // VideoStub implementation. | |
| 64 virtual void ProcessVideoPacket(scoped_ptr<VideoPacket> packet, | |
| 65 const base::Closure& done) OVERRIDE; | |
| 66 | |
| 67 // Return the stats recorded by this client. | |
| 68 ChromotingStats* GetStats(); | |
| 69 | |
| 70 private: | |
| 71 friend class base::RefCountedThreadSafe<RectangleUpdateDecoder>; | |
| 72 virtual ~RectangleUpdateDecoder(); | |
| 73 | |
| 74 // Paints the invalidated region to the next available buffer and returns it | |
| 75 // to the consumer. | |
| 76 void SchedulePaint(); | |
| 77 void DoPaint(); | |
| 78 | |
| 79 // Decodes the contents of |packet|. DecodePacket may keep a reference to | |
| 80 // |packet| so the |packet| must remain alive and valid until |done| is | |
| 81 // executed. | |
| 82 void DecodePacket(scoped_ptr<VideoPacket> packet, const base::Closure& done); | |
| 83 | |
| 84 // Callback method when a VideoPacket is processed. |decode_start| contains | |
| 85 // the timestamp when the packet will start to be processed. | |
| 86 void OnPacketDone(base::Time decode_start, const base::Closure& done); | |
| 87 | |
| 88 scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_; | |
| 89 scoped_refptr<base::SingleThreadTaskRunner> decode_task_runner_; | |
| 90 scoped_refptr<FrameConsumerProxy> consumer_; | |
| 91 scoped_ptr<VideoDecoder> decoder_; | |
| 92 | |
| 93 // Remote screen size in pixels. | |
| 94 webrtc::DesktopSize source_size_; | |
| 95 | |
| 96 // Vertical and horizontal DPI of the remote screen. | |
| 97 webrtc::DesktopVector source_dpi_; | |
| 98 | |
| 99 // The current dimensions of the frame consumer view. | |
| 100 webrtc::DesktopSize view_size_; | |
| 101 webrtc::DesktopRect clip_area_; | |
| 102 | |
| 103 // The drawing buffers supplied by the frame consumer. | |
| 104 std::list<webrtc::DesktopFrame*> buffers_; | |
| 105 | |
| 106 // Flag used to coalesce runs of SchedulePaint()s into a single DoPaint(). | |
| 107 bool paint_scheduled_; | |
| 108 | |
| 109 ChromotingStats stats_; | |
| 110 | |
| 111 // Keep track of the most recent sequence number bounced back from the host. | |
| 112 int64 latest_sequence_number_; | |
| 113 }; | |
| 114 | |
| 115 } // namespace remoting | |
| 116 | |
| 117 #endif // REMOTING_CLIENT_RECTANGLE_UPDATE_DECODER_H_ | |
| OLD | NEW |