remoting/test/test_video_renderer.h

   1 // Copyright 2015 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_TEST_TEST_VIDEO_RENDERER_H_
   6 #define REMOTING_TEST_TEST_VIDEO_RENDERER_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "base/threading/thread_checker.h"
  11 #include "media/base/video_frame.h"
  12 #include "remoting/client/video_renderer.h"
  13 #include "remoting/protocol/session_config.h"
  14 #include "remoting/protocol/video_stub.h"
  15
  16 namespace base {
  17 class Thread;
  18 class SingleThreadTaskRunner;
  19 }
  20
  21 namespace webrtc {
  22 class DesktopFrame;
  23 class DesktopRect;
  24 }
  25
  26 namespace remoting {
  27 namespace test {
  28
  29 struct RGBValue;
  30
  31 // Processes video packets as they are received from the remote host. Must be
  32 // used from a thread running a message loop and this class will use that
  33 // message loop to execute the done callbacks passed by the caller of
  34 // ProcessVideoPacket.
  35 class TestVideoRenderer : public VideoRenderer, public protocol::VideoStub {
  36  public:
  37   TestVideoRenderer();
  38   ~TestVideoRenderer() override;
  39
  40   // VideoRenderer interface.
  41   void OnSessionConfig(const protocol::SessionConfig& config) override;
  42   ChromotingStats* GetStats() override;
  43   protocol::VideoStub* GetVideoStub() override;
  44
  45   // protocol::VideoStub interface.
  46   void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet,
  47                           const base::Closure& done) override;
  48
  49   // Initialize a decoder to decode video packets.
  50   void SetCodecForDecoding(const protocol::ChannelConfig::Codec codec);
  51
  52   // Returns a copy of the current frame.
  53   scoped_ptr<webrtc::DesktopFrame> GetCurrentFrameForTest() const;
  54
  55   // Gets a weak pointer for this object.
  56   base::WeakPtr<TestVideoRenderer> GetWeakPtr() {
  57     return weak_factory_.GetWeakPtr();
  58   }
  59
  60   // Set expected rect and average color for comparison and the callback will be
  61   // called when the pattern is matched.
  62   void ExpectAverageColorInRect(
  63       const webrtc::DesktopRect& expected_rect,
  64       const RGBValue& expected_average_color,
  65       const base::Closure& image_pattern_matched_callback);
  66
  67   // Turn on/off saving video frames to disk.
  68   void SaveFrameDataToDisk(bool save_frame_data_to_disk);
  69
  70  private:
  71   // The actual implementation resides in Core class.
  72   class Core;
  73   scoped_ptr<Core> core_;
  74
  75   // Used to ensure TestVideoRenderer methods are called on the same thread.
  76   base::ThreadChecker thread_checker_;
  77
  78   // Used to decode and process video packets.
  79   scoped_ptr<base::Thread>  video_decode_thread_;
  80
  81   // Used to post tasks to video decode thread.
  82   scoped_refptr<base::SingleThreadTaskRunner> video_decode_task_runner_;
  83
  84   // Used to weakly bind |this| to methods.
  85   // NOTE: Weak pointers must be invalidated before all other member variables.
  86   base::WeakPtrFactory<TestVideoRenderer> weak_factory_;
  87
  88   DISALLOW_COPY_AND_ASSIGN(TestVideoRenderer);
  89 };
  90
  91 }  // namespace test
  92 }  // namespace remoting
  93
  94 #endif  // REMOTING_TEST_TEST_VIDEO_RENDERER_H_