Chromium Code Reviews| 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, although | |
| 33 // its identity is not enforced. | |
| 32 // [1] http://www.webmproject.org/docs/container/ | 34 // [1] http://www.webmproject.org/docs/container/ |
| 33 // [2] http://www.matroska.org/technical/specs/index.html | 35 // [2] http://www.matroska.org/technical/specs/index.html |
| 34 // TODO(mcasas): Add support for Audio muxing. | 36 // TODO(mcasas): Add support for Audio muxing. |
| 35 class MEDIA_EXPORT WebmMuxer : public NON_EXPORTED_BASE(mkvmuxer::IMkvWriter) { | 37 class MEDIA_EXPORT WebmMuxer : public NON_EXPORTED_BASE(mkvmuxer::IMkvWriter) { |
| 36 public: | 38 public: |
| 37 // Callback to be called when WebmMuxer is ready to write a chunk of data, | 39 // 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 | 40 // either any file header or a SingleBlock. The chunk is described as a byte |
| 39 // array and a byte length. | 41 // array and a byte length. |
| 40 using WriteDataCB = base::Callback<void(const base::StringPiece&)>; | 42 using WriteDataCB = base::Callback<void(const base::StringPiece&)>; |
| 41 | 43 |
| 42 explicit WebmMuxer(const WriteDataCB& write_data_callback); | 44 explicit WebmMuxer(const WriteDataCB& write_data_callback); |
| 43 ~WebmMuxer() override; | 45 ~WebmMuxer() override; |
| 44 | 46 |
| 45 // Adds a |video_frame| with |encoded_data.data()| to WebM Segment. | 47 // Adds a |video_frame| with |encoded_data.data()| to WebM Segment. |
| 46 // TODO(mcasas): Revisit how |encoded_data| is passed once the whole recording | 48 // TODO(mcasas): Revisit how |encoded_data| is passed once the whole recording |
| 47 // chain is setup (http://crbug.com/262211). | 49 // chain is setup (http://crbug.com/262211). |
| 48 void OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, | 50 void OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, |
| 49 const base::StringPiece& encoded_data, | 51 const base::StringPiece& encoded_data, |
| 50 base::TimeTicks timestamp, | 52 base::TimeTicks timestamp, |
| 51 bool is_key_frame); | 53 bool is_key_frame); |
| 52 | 54 |
| 53 private: | 55 private: |
| 54 friend class WebmMuxerTest; | 56 friend class WebmMuxerTest; |
| 55 | 57 |
| 56 // Creates and adds a new video track. Called upon receiving the first | 58 // 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 | 59 // 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 | 60 // info, although individual frames passed to OnEncodedVideo() can have any |
| 59 // frame size. Returns OnEncodedVideo() callback with a |track_number| bound | 61 // frame size. |
| 60 // in the callback. | |
| 61 void AddVideoTrack(const gfx::Size& frame_size, double frame_rate); | 62 void AddVideoTrack(const gfx::Size& frame_size, double frame_rate); |
| 62 | 63 |
| 63 // IMkvWriter interface. | 64 // IMkvWriter interface. |
| 64 mkvmuxer::int32 Write(const void* buf, mkvmuxer::uint32 len) override; | 65 mkvmuxer::int32 Write(const void* buf, mkvmuxer::uint32 len) override; |
| 65 mkvmuxer::int64 Position() const override; | 66 mkvmuxer::int64 Position() const override; |
| 66 mkvmuxer::int32 Position(mkvmuxer::int64 position) override; | 67 mkvmuxer::int32 Position(mkvmuxer::int64 position) override; |
| 67 bool Seekable() const override; | 68 bool Seekable() const override; |
| 68 void ElementStartNotify(mkvmuxer::uint64 element_id, | 69 void ElementStartNotify(mkvmuxer::uint64 element_id, |
| 69 mkvmuxer::int64 position) override; | 70 mkvmuxer::int64 position) override; |
| 70 | 71 |
| 71 // Used to DCHECK that we are called on the correct thread. | 72 // Used to DCHECK that we are called on the correct thread. |
| 72 base::ThreadChecker thread_checker_; | 73 base::ThreadChecker io_thread_checker_; |
|
miu
2015/09/02 21:28:13
Not sure you need the name change here, since OnEn
mcasas
2015/09/04 02:16:30
Done., but in general IO thread is where the frame
| |
| 73 | 74 |
| 74 // A caller-side identifier to interact with |segment_|, initialised upon | 75 // A caller-side identifier to interact with |segment_|, initialised upon |
| 75 // first frame arrival by AddVideoTrack(). | 76 // first frame arrival by AddVideoTrack(). |
| 76 uint64_t track_index_; | 77 uint64_t track_index_; |
| 77 | 78 |
| 78 // Origin of times for frame timestamps. | 79 // Origin of times for frame timestamps. |
| 79 base::TimeTicks first_frame_timestamp_; | 80 base::TimeTicks first_frame_timestamp_; |
| 80 | 81 |
| 81 // Callback to dump written data as being called by libwebm. | 82 // Callback to dump written data as being called by libwebm. |
| 82 const WriteDataCB write_data_callback_; | 83 const WriteDataCB write_data_callback_; |
| 83 | 84 |
| 84 // Rolling counter of the position in bytes of the written goo. | 85 // Rolling counter of the position in bytes of the written goo. |
| 85 base::CheckedNumeric<mkvmuxer::int64> position_; | 86 base::CheckedNumeric<mkvmuxer::int64> position_; |
| 86 | 87 |
| 87 // The MkvMuxer active element. | 88 // The MkvMuxer active element. |
| 88 mkvmuxer::Segment segment_; | 89 mkvmuxer::Segment segment_; |
| 89 | 90 |
| 90 DISALLOW_COPY_AND_ASSIGN(WebmMuxer); | 91 DISALLOW_COPY_AND_ASSIGN(WebmMuxer); |
| 91 }; | 92 }; |
| 92 | 93 |
| 93 } // namespace media | 94 } // namespace media |
| 94 | 95 |
| 95 #endif // MEDIA_FILTERS_LIBWEBM_MUXER_H_ | 96 #endif // MEDIA_FILTERS_LIBWEBM_MUXER_H_ |
| OLD | NEW |