| OLD | NEW | 
|---|
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be | 
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. | 
| 4 | 4 | 
| 5 #ifndef MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 5 #ifndef MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 
| 6 #define MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 6 #define MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 
| 7 | 7 | 
| 8 #include <set> | 8 #include <set> | 
| 9 | 9 | 
| 10 #include "base/callback.h" | 10 #include "base/callback.h" | 
| 11 #include "base/gtest_prod_util.h" | 11 #include "base/gtest_prod_util.h" | 
| 12 #include "base/numerics/safe_math.h" | 12 #include "base/numerics/safe_math.h" | 
| 13 #include "base/strings/string_piece.h" | 13 #include "base/strings/string_piece.h" | 
| 14 #include "base/threading/thread_checker.h" | 14 #include "base/threading/thread_checker.h" | 
| 15 #include "base/time/time.h" | 15 #include "base/time/time.h" | 
| 16 #include "media/base/media_export.h" | 16 #include "media/base/media_export.h" | 
| 17 #include "third_party/libwebm/source/mkvmuxer.hpp" | 17 #include "third_party/libwebm/source/mkvmuxer.hpp" | 
| 18 #include "ui/gfx/geometry/size.h" | 18 #include "ui/gfx/geometry/size.h" | 
| 19 | 19 | 
| 20 namespace media { | 20 namespace media { | 
| 21 | 21 | 
| 22 class VideoFrame; | 22 class VideoFrame; | 
| 23 | 23 | 
| 24 // Adapter class to manage a WebM container [1], a simplified version of a | 24 // Adapter class to manage a WebM container [1], a simplified version of a | 
| 25 // Matroska container [2], composed of an EBML header, and a single Segment | 25 // Matroska container [2], composed of an EBML header, and a single Segment | 
| 26 // including at least a Track Section and a number of SimpleBlocks each | 26 // including at least a Track Section and a number of SimpleBlocks each | 
| 27 // containing a single encoded video frame. WebM container has no Trailer. | 27 // containing a single encoded video frame. WebM container has no Trailer. | 
| 28 // Clients will push encoded VPx video frames one by one via the | 28 // Clients will push encoded VPx video frames one by one via the | 
| 29 // OnEncodedVideo(). libwebm will eventually ping the WriteDataCB passed on | 29 // OnEncodedVideo(). libwebm will eventually ping the WriteDataCB passed on | 
| 30 // contructor with the wrapped encoded data. All operations must happen in a | 30 // contructor with the wrapped encoded data. | 
| 31 // single thread, where WebmMuxer is created and destroyed. | 31 // WebmMuxer is created/destroyed on a thread, usually the Main Render thread, | 
|  | 32 // and receives OnEncodedVideo()s on another thread, usually Render IO. | 
| 32 // [1] http://www.webmproject.org/docs/container/ | 33 // [1] http://www.webmproject.org/docs/container/ | 
| 33 // [2] http://www.matroska.org/technical/specs/index.html | 34 // [2] http://www.matroska.org/technical/specs/index.html | 
| 34 // TODO(mcasas): Add support for Audio muxing. | 35 // TODO(mcasas): Add support for Audio muxing. | 
| 35 class MEDIA_EXPORT WebmMuxer : public NON_EXPORTED_BASE(mkvmuxer::IMkvWriter) { | 36 class MEDIA_EXPORT WebmMuxer : public NON_EXPORTED_BASE(mkvmuxer::IMkvWriter) { | 
| 36  public: | 37  public: | 
| 37   // Callback to be called when WebmMuxer is ready to write a chunk of data, | 38   // Callback to be called when WebmMuxer is ready to write a chunk of data, | 
| 38   // either any file header or a SingleBlock. The chunk is described as a byte | 39   // either any file header or a SingleBlock. The chunk is described as a byte | 
| 39   // array and a byte length. | 40   // array and a byte length. | 
| 40   using WriteDataCB = base::Callback<void(const base::StringPiece&)>; | 41   using WriteDataCB = base::Callback<void(const base::StringPiece&)>; | 
| 41 | 42 | 
| 42   explicit WebmMuxer(const WriteDataCB& write_data_callback); | 43   explicit WebmMuxer(const WriteDataCB& write_data_callback); | 
| 43   ~WebmMuxer() override; | 44   ~WebmMuxer() override; | 
| 44 | 45 | 
| 45   // Adds a |video_frame| with |encoded_data.data()| to WebM Segment. | 46   // Adds a |video_frame| with |encoded_data.data()| to WebM Segment. | 
| 46   // TODO(mcasas): Revisit how |encoded_data| is passed once the whole recording | 47   // TODO(mcasas): Revisit how |encoded_data| is passed once the whole recording | 
| 47   // chain is setup (http://crbug.com/262211). | 48   // chain is setup (http://crbug.com/262211). | 
| 48   void OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, | 49   void OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, | 
| 49                       const base::StringPiece& encoded_data, | 50                       const base::StringPiece& encoded_data, | 
| 50                       base::TimeTicks timestamp, | 51                       base::TimeTicks timestamp, | 
| 51                       bool is_key_frame); | 52                       bool is_key_frame); | 
| 52 | 53 | 
| 53  private: | 54  private: | 
| 54   friend class WebmMuxerTest; | 55   friend class WebmMuxerTest; | 
| 55 | 56 | 
| 56   // Creates and adds a new video track. Called upon receiving the first | 57   // Creates and adds a new video track. Called upon receiving the first | 
| 57   // frame of a given Track, adds |frame_size| and |frame_rate| to the Segment | 58   // frame of a given Track, adds |frame_size| and |frame_rate| to the Segment | 
| 58   // info, although individual frames passed to OnEncodedVideo() can have any | 59   // info, although individual frames passed to OnEncodedVideo() can have any | 
| 59   // frame size. Returns OnEncodedVideo() callback with a |track_number| bound | 60   // frame size. | 
| 60   // in the callback. |  | 
| 61   void AddVideoTrack(const gfx::Size& frame_size, double frame_rate); | 61   void AddVideoTrack(const gfx::Size& frame_size, double frame_rate); | 
| 62 | 62 | 
| 63   // IMkvWriter interface. | 63   // IMkvWriter interface. | 
| 64   mkvmuxer::int32 Write(const void* buf, mkvmuxer::uint32 len) override; | 64   mkvmuxer::int32 Write(const void* buf, mkvmuxer::uint32 len) override; | 
| 65   mkvmuxer::int64 Position() const override; | 65   mkvmuxer::int64 Position() const override; | 
| 66   mkvmuxer::int32 Position(mkvmuxer::int64 position) override; | 66   mkvmuxer::int32 Position(mkvmuxer::int64 position) override; | 
| 67   bool Seekable() const override; | 67   bool Seekable() const override; | 
| 68   void ElementStartNotify(mkvmuxer::uint64 element_id, | 68   void ElementStartNotify(mkvmuxer::uint64 element_id, | 
| 69                           mkvmuxer::int64 position) override; | 69                           mkvmuxer::int64 position) override; | 
| 70 | 70 | 
| 71   // Used to DCHECK that we are called on the correct thread. | 71   // Used to DCHECK that we are called on the correct thread. | 
| 72   base::ThreadChecker thread_checker_; | 72   base::ThreadChecker io_thread_checker_; | 
| 73 | 73 | 
| 74   // A caller-side identifier to interact with |segment_|, initialised upon | 74   // A caller-side identifier to interact with |segment_|, initialised upon | 
| 75   // first frame arrival by AddVideoTrack(). | 75   // first frame arrival by AddVideoTrack(). | 
| 76   uint64_t track_index_; | 76   uint64_t track_index_; | 
| 77 | 77 | 
| 78   // Origin of times for frame timestamps. | 78   // Origin of times for frame timestamps. | 
| 79   base::TimeTicks first_frame_timestamp_; | 79   base::TimeTicks first_frame_timestamp_; | 
| 80 | 80 | 
| 81   // Callback to dump written data as being called by libwebm. | 81   // Callback to dump written data as being called by libwebm. | 
| 82   const WriteDataCB write_data_callback_; | 82   const WriteDataCB write_data_callback_; | 
| 83 | 83 | 
| 84   // Rolling counter of the position in bytes of the written goo. | 84   // Rolling counter of the position in bytes of the written goo. | 
| 85   base::CheckedNumeric<mkvmuxer::int64> position_; | 85   base::CheckedNumeric<mkvmuxer::int64> position_; | 
| 86 | 86 | 
| 87   // The MkvMuxer active element. | 87   // The MkvMuxer active element. | 
| 88   mkvmuxer::Segment segment_; | 88   mkvmuxer::Segment segment_; | 
| 89 | 89 | 
| 90   DISALLOW_COPY_AND_ASSIGN(WebmMuxer); | 90   DISALLOW_COPY_AND_ASSIGN(WebmMuxer); | 
| 91 }; | 91 }; | 
| 92 | 92 | 
| 93 }  // namespace media | 93 }  // namespace media | 
| 94 | 94 | 
| 95 #endif  // MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 95 #endif  // MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 
| OLD | NEW | 
|---|