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. |
44 segment_.Finalize(); | |
45 } | 38 } |
46 | 39 |
47 void WebmMuxer::OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, | 40 void WebmMuxer::OnEncodedVideo(const scoped_refptr<VideoFrame>& video_frame, |
48 const base::StringPiece& encoded_data, | 41 const base::StringPiece& encoded_data, |
49 base::TimeTicks timestamp, | 42 base::TimeTicks timestamp, |
50 bool is_key_frame) { | 43 bool is_key_frame) { |
| 44 DVLOG(1) << __FUNCTION__ << " - " << encoded_data.size() << "B"; |
51 DCHECK(thread_checker_.CalledOnValidThread()); | 45 DCHECK(thread_checker_.CalledOnValidThread()); |
52 if (!track_index_) { | 46 if (!track_index_) { |
53 // |track_index_|, cannot be zero (!), initialize WebmMuxer in that case. | 47 // |track_index_|, cannot be zero (!), initialize WebmMuxer in that case. |
54 // http://www.matroska.org/technical/specs/index.html#Tracks | 48 // http://www.matroska.org/technical/specs/index.html#Tracks |
55 AddVideoTrack(video_frame->visible_rect().size(), | 49 AddVideoTrack(video_frame->visible_rect().size(), |
56 GetFrameRate(video_frame)); | 50 GetFrameRate(video_frame)); |
57 first_frame_timestamp_ = timestamp; | 51 first_frame_timestamp_ = timestamp; |
58 } | 52 } |
59 segment_.AddFrame(reinterpret_cast<const uint8_t*>(encoded_data.data()), | 53 segment_.AddFrame(reinterpret_cast<const uint8_t*>(encoded_data.data()), |
60 encoded_data.size(), | 54 encoded_data.size(), |
61 track_index_, | 55 track_index_, |
62 (timestamp - first_frame_timestamp_).InMicroseconds() * | 56 (timestamp - first_frame_timestamp_).InMicroseconds() * |
63 base::Time::kNanosecondsPerMicrosecond, | 57 base::Time::kNanosecondsPerMicrosecond, |
64 is_key_frame); | 58 is_key_frame); |
65 } | 59 } |
66 | 60 |
67 void WebmMuxer::AddVideoTrack(const gfx::Size& frame_size, double frame_rate) { | 61 void WebmMuxer::AddVideoTrack(const gfx::Size& frame_size, double frame_rate) { |
68 DCHECK(thread_checker_.CalledOnValidThread()); | 62 DCHECK(thread_checker_.CalledOnValidThread()); |
| 63 |
| 64 segment_.Init(this); |
| 65 segment_.set_mode(mkvmuxer::Segment::kLive); |
| 66 segment_.OutputCues(false); |
| 67 |
| 68 mkvmuxer::SegmentInfo* const info = segment_.GetSegmentInfo(); |
| 69 info->set_writing_app("Chrome"); |
| 70 info->set_muxing_app("Chrome"); |
| 71 |
69 DCHECK_EQ(track_index_, 0u); | 72 DCHECK_EQ(track_index_, 0u); |
70 track_index_ = | 73 track_index_ = |
71 segment_.AddVideoTrack(frame_size.width(), frame_size.height(), 0); | 74 segment_.AddVideoTrack(frame_size.width(), frame_size.height(), 0); |
72 DCHECK_GT(track_index_, 0u); | 75 DCHECK_GT(track_index_, 0u); |
73 | 76 |
74 mkvmuxer::VideoTrack* const video_track = | 77 mkvmuxer::VideoTrack* const video_track = |
75 reinterpret_cast<mkvmuxer::VideoTrack*>( | 78 reinterpret_cast<mkvmuxer::VideoTrack*>( |
76 segment_.GetTrackByNumber(track_index_)); | 79 segment_.GetTrackByNumber(track_index_)); |
77 DCHECK(video_track); | 80 DCHECK(video_track); |
78 video_track->set_codec_id(mkvmuxer::Tracks::kVp8CodecId); | 81 video_track->set_codec_id(mkvmuxer::Tracks::kVp8CodecId); |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 } | 115 } |
113 | 116 |
114 void WebmMuxer::ElementStartNotify(mkvmuxer::uint64 element_id, | 117 void WebmMuxer::ElementStartNotify(mkvmuxer::uint64 element_id, |
115 mkvmuxer::int64 position) { | 118 mkvmuxer::int64 position) { |
116 // This method gets pinged before items are sent to |write_data_callback_|. | 119 // This method gets pinged before items are sent to |write_data_callback_|. |
117 DCHECK_GE(position, position_.ValueOrDefault(0)) | 120 DCHECK_GE(position, position_.ValueOrDefault(0)) |
118 << "Can't go back in a live WebM stream."; | 121 << "Can't go back in a live WebM stream."; |
119 } | 122 } |
120 | 123 |
121 } // namespace media | 124 } // namespace media |
OLD | NEW |