219 lines
7.9 KiB
C++
219 lines
7.9 KiB
C++
// Copyright 2014 Google Inc. All rights reserved.
|
||
//
|
||
// Use of this source code is governed by a BSD-style
|
||
// license that can be found in the LICENSE file or at
|
||
// https://developers.google.com/open-source/licenses/bsd
|
||
|
||
#include "packager/media/formats/mp4/multi_segment_segmenter.h"
|
||
|
||
#include <algorithm>
|
||
|
||
#include "packager/base/strings/string_number_conversions.h"
|
||
#include "packager/base/strings/string_util.h"
|
||
#include "packager/file/file.h"
|
||
#include "packager/file/file_closer.h"
|
||
#include "packager/media/base/buffer_writer.h"
|
||
#include "packager/media/base/muxer_options.h"
|
||
#include "packager/media/base/muxer_util.h"
|
||
#include "packager/media/event/muxer_listener.h"
|
||
#include "packager/media/formats/mp4/box_definitions.h"
|
||
#include "packager/media/formats/mp4/key_frame_info.h"
|
||
|
||
namespace shaka {
|
||
namespace media {
|
||
namespace mp4 {
|
||
|
||
MultiSegmentSegmenter::MultiSegmentSegmenter(const MuxerOptions& options,
|
||
std::unique_ptr<FileType> ftyp,
|
||
std::unique_ptr<Movie> moov)
|
||
: Segmenter(options, std::move(ftyp), std::move(moov)),
|
||
styp_(new SegmentType),
|
||
num_segments_(0) {
|
||
// Use the same brands for styp as ftyp.
|
||
styp_->major_brand = Segmenter::ftyp()->major_brand;
|
||
styp_->compatible_brands = Segmenter::ftyp()->compatible_brands;
|
||
// Replace 'cmfc' with 'cmfs' for CMAF segments compatibility.
|
||
std::replace(styp_->compatible_brands.begin(), styp_->compatible_brands.end(),
|
||
FOURCC_cmfc, FOURCC_cmfs);
|
||
}
|
||
|
||
MultiSegmentSegmenter::~MultiSegmentSegmenter() {}
|
||
|
||
bool MultiSegmentSegmenter::GetInitRange(size_t* offset, size_t* size) {
|
||
VLOG(1) << "MultiSegmentSegmenter outputs init segment: "
|
||
<< options().output_file_name;
|
||
return false;
|
||
}
|
||
|
||
bool MultiSegmentSegmenter::GetIndexRange(size_t* offset, size_t* size) {
|
||
VLOG(1) << "MultiSegmentSegmenter does not have index range.";
|
||
return false;
|
||
}
|
||
|
||
std::vector<Range> MultiSegmentSegmenter::GetSegmentRanges() {
|
||
VLOG(1) << "MultiSegmentSegmenter does not have media segment ranges.";
|
||
return std::vector<Range>();
|
||
}
|
||
|
||
Status MultiSegmentSegmenter::DoInitialize() {
|
||
return WriteInitSegment();
|
||
}
|
||
|
||
Status MultiSegmentSegmenter::DoFinalize() {
|
||
// Update init segment with media duration set.
|
||
Status status = WriteInitSegment();
|
||
if (status.ok())
|
||
SetComplete();
|
||
return status;
|
||
}
|
||
|
||
Status MultiSegmentSegmenter::DoFinalizeSegment() {
|
||
DCHECK(sidx());
|
||
// earliest_presentation_time is the earliest presentation time of any
|
||
// access unit in the reference stream in the first subsegment.
|
||
// It will be re-calculated later when subsegments are finalized.
|
||
sidx()->earliest_presentation_time =
|
||
sidx()->references[0].earliest_presentation_time;
|
||
|
||
if (options().mp4_params.num_subsegments_per_sidx <= 0)
|
||
return WriteSegment();
|
||
|
||
// sidx() contains pre-generated segment references with one reference per
|
||
// fragment. Calculate |num_fragments_per_subsegment| and combine
|
||
// pre-generated references into final subsegment references.
|
||
size_t num_fragments = sidx()->references.size();
|
||
size_t num_fragments_per_subsegment =
|
||
(num_fragments - 1) / options().mp4_params.num_subsegments_per_sidx + 1;
|
||
if (num_fragments_per_subsegment <= 1)
|
||
return WriteSegment();
|
||
|
||
size_t frag_index = 0;
|
||
size_t subseg_index = 0;
|
||
std::vector<SegmentReference>& refs = sidx()->references;
|
||
uint64_t first_sap_time =
|
||
refs[0].sap_delta_time + refs[0].earliest_presentation_time;
|
||
for (size_t i = 1; i < num_fragments; ++i) {
|
||
refs[subseg_index].referenced_size += refs[i].referenced_size;
|
||
refs[subseg_index].subsegment_duration += refs[i].subsegment_duration;
|
||
refs[subseg_index].earliest_presentation_time =
|
||
std::min(refs[subseg_index].earliest_presentation_time,
|
||
refs[i].earliest_presentation_time);
|
||
if (refs[subseg_index].sap_type == SegmentReference::TypeUnknown &&
|
||
refs[i].sap_type != SegmentReference::TypeUnknown) {
|
||
refs[subseg_index].sap_type = refs[i].sap_type;
|
||
first_sap_time =
|
||
refs[i].sap_delta_time + refs[i].earliest_presentation_time;
|
||
}
|
||
if (++frag_index >= num_fragments_per_subsegment) {
|
||
// Calculate sap delta time w.r.t. sidx_->earliest_presentation_time.
|
||
if (refs[subseg_index].sap_type != SegmentReference::TypeUnknown) {
|
||
refs[subseg_index].sap_delta_time =
|
||
first_sap_time - refs[subseg_index].earliest_presentation_time;
|
||
}
|
||
if (++i >= num_fragments)
|
||
break;
|
||
refs[++subseg_index] = refs[i];
|
||
first_sap_time =
|
||
refs[i].sap_delta_time + refs[i].earliest_presentation_time;
|
||
frag_index = 1;
|
||
}
|
||
}
|
||
|
||
refs.resize(options().mp4_params.num_subsegments_per_sidx);
|
||
|
||
// earliest_presentation_time is the earliest presentation time of any
|
||
// access unit in the reference stream in the first subsegment.
|
||
sidx()->earliest_presentation_time = refs[0].earliest_presentation_time;
|
||
|
||
return WriteSegment();
|
||
}
|
||
|
||
Status MultiSegmentSegmenter::WriteInitSegment() {
|
||
DCHECK(ftyp());
|
||
DCHECK(moov());
|
||
// Generate the output file with init segment.
|
||
std::unique_ptr<File, FileCloser> file(
|
||
File::Open(options().output_file_name.c_str(), "w"));
|
||
if (!file) {
|
||
return Status(error::FILE_FAILURE,
|
||
"Cannot open file for write " + options().output_file_name);
|
||
}
|
||
std::unique_ptr<BufferWriter> buffer(new BufferWriter);
|
||
ftyp()->Write(buffer.get());
|
||
moov()->Write(buffer.get());
|
||
return buffer->WriteToFile(file.get());
|
||
}
|
||
|
||
Status MultiSegmentSegmenter::WriteSegment() {
|
||
DCHECK(sidx());
|
||
DCHECK(fragment_buffer());
|
||
DCHECK(styp_);
|
||
|
||
std::unique_ptr<BufferWriter> buffer(new BufferWriter());
|
||
std::unique_ptr<File, FileCloser> file;
|
||
std::string file_name;
|
||
if (options().segment_template.empty()) {
|
||
// Append the segment to output file if segment template is not specified.
|
||
file_name = options().output_file_name.c_str();
|
||
file.reset(File::Open(file_name.c_str(), "a"));
|
||
if (!file) {
|
||
return Status(error::FILE_FAILURE, "Cannot open file for append " +
|
||
options().output_file_name);
|
||
}
|
||
} else {
|
||
file_name = GetSegmentName(options().segment_template,
|
||
sidx()->earliest_presentation_time,
|
||
num_segments_++, options().bandwidth);
|
||
file.reset(File::Open(file_name.c_str(), "w"));
|
||
if (!file) {
|
||
return Status(error::FILE_FAILURE,
|
||
"Cannot open file for write " + file_name);
|
||
}
|
||
styp_->Write(buffer.get());
|
||
}
|
||
|
||
// If num_subsegments_per_sidx is negative, no SIDX box is generated.
|
||
if (options().mp4_params.num_subsegments_per_sidx >= 0)
|
||
sidx()->Write(buffer.get());
|
||
|
||
const size_t segment_header_size = buffer->Size();
|
||
const size_t segment_size = segment_header_size + fragment_buffer()->Size();
|
||
DCHECK_NE(segment_size, 0u);
|
||
|
||
Status status = buffer->WriteToFile(file.get());
|
||
if (status.ok()) {
|
||
if (muxer_listener()) {
|
||
for (const KeyFrameInfo& key_frame_info : key_frame_infos()) {
|
||
muxer_listener()->OnKeyFrame(
|
||
key_frame_info.timestamp,
|
||
segment_header_size + key_frame_info.start_byte_offset,
|
||
key_frame_info.size);
|
||
}
|
||
}
|
||
status = fragment_buffer()->WriteToFile(file.get());
|
||
}
|
||
|
||
if (!status.ok())
|
||
return status;
|
||
|
||
uint64_t segment_duration = 0;
|
||
// ISO/IEC 23009-1:2012: the value shall be identical to sum of the the
|
||
// values of all Subsegment_duration fields in the first ‘sidx’ box.
|
||
for (size_t i = 0; i < sidx()->references.size(); ++i)
|
||
segment_duration += sidx()->references[i].subsegment_duration;
|
||
|
||
UpdateProgress(segment_duration);
|
||
if (muxer_listener()) {
|
||
muxer_listener()->OnSampleDurationReady(sample_duration());
|
||
muxer_listener()->OnNewSegment(file_name,
|
||
sidx()->earliest_presentation_time,
|
||
segment_duration, segment_size);
|
||
}
|
||
|
||
return Status::OK;
|
||
}
|
||
|
||
} // namespace mp4
|
||
} // namespace media
|
||
} // namespace shaka
|