98 lines
3.1 KiB
C++
98 lines
3.1 KiB
C++
// Copyright 2015 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/webm/multi_segment_segmenter.h"
|
|
|
|
#include "packager/media/base/media_stream.h"
|
|
#include "packager/media/base/muxer_options.h"
|
|
#include "packager/media/base/muxer_util.h"
|
|
#include "packager/media/base/stream_info.h"
|
|
#include "packager/media/event/muxer_listener.h"
|
|
#include "packager/third_party/libwebm/src/mkvmuxer.hpp"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
namespace webm {
|
|
MultiSegmentSegmenter::MultiSegmentSegmenter(const MuxerOptions& options)
|
|
: Segmenter(options), num_segment_(0) {}
|
|
|
|
MultiSegmentSegmenter::~MultiSegmentSegmenter() {}
|
|
|
|
bool MultiSegmentSegmenter::GetInitRangeStartAndEnd(uint64_t* start,
|
|
uint64_t* end) {
|
|
return false;
|
|
}
|
|
|
|
bool MultiSegmentSegmenter::GetIndexRangeStartAndEnd(uint64_t* start,
|
|
uint64_t* end) {
|
|
return false;
|
|
}
|
|
|
|
Status MultiSegmentSegmenter::DoInitialize() {
|
|
std::unique_ptr<MkvWriter> writer(new MkvWriter);
|
|
Status status = writer->Open(options().output_file_name);
|
|
if (!status.ok())
|
|
return status;
|
|
writer_ = std::move(writer);
|
|
return WriteSegmentHeader(0, writer_.get());
|
|
}
|
|
|
|
Status MultiSegmentSegmenter::DoFinalize() {
|
|
Status status = FinalizeSegment();
|
|
status.Update(writer_->Close());
|
|
return status;
|
|
}
|
|
|
|
Status MultiSegmentSegmenter::FinalizeSegment() {
|
|
if (!cluster()->Finalize())
|
|
return Status(error::FILE_FAILURE, "Error finalizing segment.");
|
|
|
|
if (muxer_listener()) {
|
|
const uint64_t size = cluster()->Size();
|
|
const uint64_t start_webm_timecode = cluster()->timecode();
|
|
const uint64_t start_timescale = FromWebMTimecode(start_webm_timecode);
|
|
muxer_listener()->OnNewSegment(writer_->file()->file_name(),
|
|
start_timescale,
|
|
cluster_length_in_time_scale(), size);
|
|
}
|
|
|
|
VLOG(1) << "WEBM file '" << writer_->file()->file_name() << "' finalized.";
|
|
return Status::OK;
|
|
}
|
|
|
|
Status MultiSegmentSegmenter::NewSubsegment(uint64_t start_timescale) {
|
|
if (cluster() && !cluster()->Finalize())
|
|
return Status(error::FILE_FAILURE, "Error finalizing segment.");
|
|
|
|
uint64_t start_webm_timecode = FromBMFFTimescale(start_timescale);
|
|
return SetCluster(start_webm_timecode, 0, writer_.get());
|
|
}
|
|
|
|
Status MultiSegmentSegmenter::NewSegment(uint64_t start_timescale) {
|
|
if (cluster()) {
|
|
Status temp = FinalizeSegment();
|
|
if (!temp.ok())
|
|
return temp;
|
|
}
|
|
|
|
// Create a new file for the new segment.
|
|
std::string segment_name =
|
|
GetSegmentName(options().segment_template, start_timescale, num_segment_,
|
|
options().bandwidth);
|
|
writer_.reset(new MkvWriter);
|
|
Status status = writer_->Open(segment_name);
|
|
if (!status.ok())
|
|
return status;
|
|
num_segment_++;
|
|
|
|
uint64_t start_webm_timecode = FromBMFFTimescale(start_timescale);
|
|
return SetCluster(start_webm_timecode, 0, writer_.get());
|
|
}
|
|
|
|
} // namespace webm
|
|
} // namespace media
|
|
} // namespace shaka
|