| 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 |