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 #include "media/capture/webm_muxer.h" | 5 #include "media/capture/webm_muxer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "media/base/limits.h" | 8 #include "media/base/limits.h" |
9 #include "media/base/video_frame.h" | 9 #include "media/base/video_frame.h" |
10 | 10 |
(...skipping 10 matching lines...) Expand all Loading... |
21 frame_rate > media::limits::kMaxFramesPerSecond) { | 21 frame_rate > media::limits::kMaxFramesPerSecond) { |
22 frame_rate = kDefaultFrameRate; | 22 frame_rate = kDefaultFrameRate; |
23 } | 23 } |
24 return frame_rate; | 24 return frame_rate; |
25 } | 25 } |
26 | 26 |
27 WebmMuxer::WebmMuxer(const WriteDataCB& write_data_callback) | 27 WebmMuxer::WebmMuxer(const WriteDataCB& write_data_callback) |
28 : track_index_(0), | 28 : track_index_(0), |
29 write_data_callback_(write_data_callback), | 29 write_data_callback_(write_data_callback), |
30 position_(0) { | 30 position_(0) { |
31 DCHECK(thread_checker_.CalledOnValidThread()); | |
32 DCHECK(!write_data_callback_.is_null()); | 31 DCHECK(!write_data_callback_.is_null()); |
33 segment_.Init(this); | 32 // Creation is done on a different thread than main activities. |
34 segment_.set_mode(mkvmuxer::Segment::kLive); | 33 thread_checker_.DetachFromThread(); |
35 segment_.OutputCues(false); | |
36 | |
37 mkvmuxer::SegmentInfo* const info = segment_.GetSegmentInfo(); | |
38 info->set_writing_app("Chrome"); | |
39 info->set_muxing_app("Chrome"); | |
40 } | 34 } |
41 | 35 |
42 WebmMuxer::~WebmMuxer() { | 36 WebmMuxer::~WebmMuxer() { |
43 DCHECK(thread_checker_.CalledOnValidThread()); | 37 // No need to segment_.Finalize() since is not Seekable(), i.e. a live stream, |
| 38 // but is good practice. |
44 segment_.Finalize(); | 39 segment_.Finalize(); |
45 } | 40 } |
46 | 41 |
47 void WebmMuxer::OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, | 42 void WebmMuxer::OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, |
48 const base::StringPiece& encoded_data, | 43 const base::StringPiece& encoded_data, |
49 base::TimeTicks timestamp, | 44 base::TimeTicks timestamp, |
50 bool is_key_frame) { | 45 bool is_key_frame) { |
| 46 DVLOG(1) << __FUNCTION__ << " - " << encoded_data.size() << "B"; |
51 DCHECK(thread_checker_.CalledOnValidThread()); | 47 DCHECK(thread_checker_.CalledOnValidThread()); |
52 if (!track_index_) { | 48 if (!track_index_) { |
53 // |track_index_|, cannot be zero (!), initialize WebmMuxer in that case. | 49 // |track_index_|, cannot be zero (!), initialize WebmMuxer in that case. |
54 // http://www.matroska.org/technical/specs/index.html#Tracks | 50 // http://www.matroska.org/technical/specs/index.html#Tracks |
55 AddVideoTrack(video_frame->visible_rect().size(), | 51 AddVideoTrack(video_frame->visible_rect().size(), |
56 GetFrameRate(video_frame)); | 52 GetFrameRate(video_frame)); |
57 first_frame_timestamp_ = timestamp; | 53 first_frame_timestamp_ = timestamp; |
58 } | 54 } |
59 segment_.AddFrame(reinterpret_cast<const uint8_t*>(encoded_data.data()), | 55 segment_.AddFrame(reinterpret_cast<const uint8_t*>(encoded_data.data()), |
60 encoded_data.size(), | 56 encoded_data.size(), |
61 track_index_, | 57 track_index_, |
62 (timestamp - first_frame_timestamp_).InMicroseconds() * | 58 (timestamp - first_frame_timestamp_).InMicroseconds() * |
63 base::Time::kNanosecondsPerMicrosecond, | 59 base::Time::kNanosecondsPerMicrosecond, |
64 is_key_frame); | 60 is_key_frame); |
65 } | 61 } |
66 | 62 |
67 void WebmMuxer::AddVideoTrack(const gfx::Size& frame_size, double frame_rate) { | 63 void WebmMuxer::AddVideoTrack(const gfx::Size& frame_size, double frame_rate) { |
68 DCHECK(thread_checker_.CalledOnValidThread()); | 64 DCHECK(thread_checker_.CalledOnValidThread()); |
69 DCHECK_EQ(track_index_, 0u); | 65 DCHECK_EQ(track_index_, 0u) << "WebmMuxer can only be initialised once."; |
| 66 |
| 67 segment_.Init(this); |
| 68 segment_.set_mode(mkvmuxer::Segment::kLive); |
| 69 segment_.OutputCues(false); |
| 70 |
| 71 mkvmuxer::SegmentInfo* const info = segment_.GetSegmentInfo(); |
| 72 info->set_writing_app("Chrome"); |
| 73 info->set_muxing_app("Chrome"); |
| 74 |
70 track_index_ = | 75 track_index_ = |
71 segment_.AddVideoTrack(frame_size.width(), frame_size.height(), 0); | 76 segment_.AddVideoTrack(frame_size.width(), frame_size.height(), 0); |
72 DCHECK_GT(track_index_, 0u); | 77 DCHECK_GT(track_index_, 0u); |
73 | 78 |
74 mkvmuxer::VideoTrack* const video_track = | 79 mkvmuxer::VideoTrack* const video_track = |
75 reinterpret_cast<mkvmuxer::VideoTrack*>( | 80 reinterpret_cast<mkvmuxer::VideoTrack*>( |
76 segment_.GetTrackByNumber(track_index_)); | 81 segment_.GetTrackByNumber(track_index_)); |
77 DCHECK(video_track); | 82 DCHECK(video_track); |
78 video_track->set_codec_id(mkvmuxer::Tracks::kVp8CodecId); | 83 video_track->set_codec_id(mkvmuxer::Tracks::kVp8CodecId); |
79 DCHECK_EQ(video_track->crop_right(), 0ull); | 84 DCHECK_EQ(video_track->crop_right(), 0ull); |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 } | 117 } |
113 | 118 |
114 void WebmMuxer::ElementStartNotify(mkvmuxer::uint64 element_id, | 119 void WebmMuxer::ElementStartNotify(mkvmuxer::uint64 element_id, |
115 mkvmuxer::int64 position) { | 120 mkvmuxer::int64 position) { |
116 // This method gets pinged before items are sent to |write_data_callback_|. | 121 // This method gets pinged before items are sent to |write_data_callback_|. |
117 DCHECK_GE(position, position_.ValueOrDefault(0)) | 122 DCHECK_GE(position, position_.ValueOrDefault(0)) |
118 << "Can't go back in a live WebM stream."; | 123 << "Can't go back in a live WebM stream."; |
119 } | 124 } |
120 | 125 |
121 } // namespace media | 126 } // namespace media |
OLD | NEW |