• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2013 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 MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
6 #define MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
7 
8 #include "base/memory/scoped_ptr.h"
9 #include "base/memory/weak_ptr.h"
10 #include "base/message_loop/message_loop.h"
11 #include "media/cast/cast_config.h"
12 #include "media/cast/cast_environment.h"
13 #include "media/cast/video_sender/codecs/vp8/vp8_encoder.h"
14 
15 namespace media {
16 class VideoFrame;
17 }
18 
19 namespace media {
20 namespace cast {
21 
22 // This object is called external from the main cast thread and internally from
23 // the video encoder thread.
24 class VideoEncoder : public VideoEncoderController {
25  public:
26   typedef base::Callback<void(scoped_ptr<EncodedVideoFrame>,
27                               const base::TimeTicks&)> FrameEncodedCallback;
28 
29   VideoEncoder(scoped_refptr<CastEnvironment> cast_environment,
30                const VideoSenderConfig& video_config,
31                uint8 max_unacked_frames);
32 
33   virtual ~VideoEncoder();
34 
35   // Called from the main cast thread. This function post the encode task to the
36   // video encoder thread;
37   // The video_frame must be valid until the closure callback is called.
38   // The closure callback is called from the video encoder thread as soon as
39   // the encoder is done with the frame; it does not mean that the encoded frame
40   // has been sent out.
41   // Once the encoded frame is ready the frame_encoded_callback is called.
42   bool EncodeVideoFrame(const scoped_refptr<media::VideoFrame>& video_frame,
43                         const base::TimeTicks& capture_time,
44                         const FrameEncodedCallback& frame_encoded_callback);
45 
46  protected:
47   struct CodecDynamicConfig {
48     bool key_frame_requested;
49     uint32 latest_frame_id_to_reference;
50     int bit_rate;
51   };
52 
53   // The actual encode, called from the video encoder thread.
54   void EncodeVideoFrameEncoderThread(
55       const scoped_refptr<media::VideoFrame>& video_frame,
56       const base::TimeTicks& capture_time,
57       const CodecDynamicConfig& dynamic_config,
58       const FrameEncodedCallback& frame_encoded_callback);
59 
60   // The following functions are called from the main cast thread.
61   virtual void SetBitRate(int new_bit_rate) OVERRIDE;
62   virtual void SkipNextFrame(bool skip_next_frame) OVERRIDE;
63   virtual void GenerateKeyFrame() OVERRIDE;
64   virtual void LatestFrameIdToReference(uint32 frame_id) OVERRIDE;
65   virtual int NumberOfSkippedFrames() const OVERRIDE;
66 
67  private:
68   friend class base::RefCountedThreadSafe<VideoEncoder>;
69 
70   const VideoSenderConfig video_config_;
71   scoped_refptr<CastEnvironment> cast_environment_;
72   scoped_ptr<Vp8Encoder> vp8_encoder_;
73   CodecDynamicConfig dynamic_config_;
74   bool skip_next_frame_;
75   int skip_count_;
76 
77   DISALLOW_COPY_AND_ASSIGN(VideoEncoder);
78 };
79 
80 }  // namespace cast
81 }  // namespace media
82 
83 #endif  // MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
84