remoting/protocol/webrtc_video_stream.h

// Copyright 2015 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_PROTOCOL_WEBRTC_VIDEO_STREAM_H_
#define REMOTING_PROTOCOL_WEBRTC_VIDEO_STREAM_H_

#include <stdint.h>

#include <memory>

#include "base/callback.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/threading/thread_checker.h"
#include "base/timer/timer.h"
#include "remoting/codec/video_encoder.h"
#include "remoting/protocol/video_stream.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"

namespace webrtc {
class MediaStreamInterface;
class PeerConnectionInterface;
class PeerConnectionFactoryInterface;
class VideoTrackInterface;
}  // namespace webrtc

namespace remoting {
namespace protocol {

class WebrtcVideoCapturerAdapter;
class WebrtcTransport;

class WebrtcVideoStream : public VideoStream,
                          public webrtc::DesktopCapturer::Callback {
 public:
  WebrtcVideoStream();
  ~WebrtcVideoStream() override;

  bool Start(
      std::unique_ptr<webrtc::DesktopCapturer> desktop_capturer,
      WebrtcTransport* webrtc_transport,
      scoped_refptr<base::SingleThreadTaskRunner> encode_task_runner,
      std::unique_ptr<VideoEncoder> video_encoder);

  // VideoStream interface.
  void Pause(bool pause) override;
  void OnInputEventReceived(int64_t event_timestamp) override;
  void SetLosslessEncode(bool want_lossless) override;
  void SetLosslessColor(bool want_lossless) override;
  void SetObserver(Observer* observer) override;

 private:
  // webrtc::DesktopCapturer::Callback interface.
  void OnCaptureResult(webrtc::DesktopCapturer::Result result,
                       std::unique_ptr<webrtc::DesktopFrame> frame) override;

  // Starts |capture_timer_|.
  void StartCaptureTimer();

  // Called by |capture_timer_|.
  void CaptureNextFrame();

  void OnFrameEncoded(std::unique_ptr<VideoPacket> packet);

  void SetKeyFrameRequest();
  bool ClearAndGetKeyFrameRequest();
  void SetTargetBitrate(int bitrate);

  scoped_refptr<webrtc::PeerConnectionInterface> peer_connection_;
  scoped_refptr<webrtc::MediaStreamInterface> stream_;

  bool key_frame_request_ = false;
  uint32_t target_bitrate_kbps_ = 1000; // Initial bitrate.

  bool received_first_frame_request_ = false;

  bool capture_pending_ = false;
  bool encode_pending_ = false;

  // Last time capture was started.
  base::TimeTicks last_capture_started_ticks_;

  webrtc::DesktopSize frame_size_;
  webrtc::DesktopVector frame_dpi_;
  Observer* observer_ = nullptr;

  // Main task runner.
  scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_;

  // Task runner used to run |encoder_|.
  scoped_refptr<base::SingleThreadTaskRunner> encode_task_runner_;

  // Capturer used to capture the screen.
  std::unique_ptr<webrtc::DesktopCapturer> capturer_;

  std::unique_ptr<base::RepeatingTimer> capture_timer_;

  // Used to send across encoded frames.
  WebrtcTransport* webrtc_transport_ = nullptr;

  // Used to encode captured frames. Always accessed on the encode thread.
  std::unique_ptr<VideoEncoder> encoder_;

  base::ThreadChecker thread_checker_;

  base::WeakPtrFactory<WebrtcVideoStream> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(WebrtcVideoStream);
};

}  // namespace protocol
}  // namespace remoting

#endif  // REMOTING_PROTOCOL_WEBRTC_VIDEO_STREAM_H_