2017-05-22 20:28:10 +00:00
|
|
|
// Copyright 2017 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/packager.h"
|
|
|
|
|
|
|
|
#include "packager/app/libcrypto_threading.h"
|
|
|
|
#include "packager/app/packager_util.h"
|
|
|
|
#include "packager/app/stream_descriptor.h"
|
2017-05-23 02:41:26 +00:00
|
|
|
#include "packager/base/at_exit.h"
|
2017-05-22 20:28:10 +00:00
|
|
|
#include "packager/base/files/file_path.h"
|
|
|
|
#include "packager/base/logging.h"
|
|
|
|
#include "packager/base/path_service.h"
|
|
|
|
#include "packager/base/strings/stringprintf.h"
|
|
|
|
#include "packager/base/threading/simple_thread.h"
|
|
|
|
#include "packager/base/time/clock.h"
|
2017-07-10 18:26:22 +00:00
|
|
|
#include "packager/file/file.h"
|
2017-05-22 20:28:10 +00:00
|
|
|
#include "packager/hls/base/hls_notifier.h"
|
|
|
|
#include "packager/hls/base/simple_hls_notifier.h"
|
|
|
|
#include "packager/media/base/container_names.h"
|
|
|
|
#include "packager/media/base/fourccs.h"
|
|
|
|
#include "packager/media/base/key_source.h"
|
|
|
|
#include "packager/media/base/muxer_options.h"
|
|
|
|
#include "packager/media/base/muxer_util.h"
|
|
|
|
#include "packager/media/chunking/chunking_handler.h"
|
|
|
|
#include "packager/media/crypto/encryption_handler.h"
|
|
|
|
#include "packager/media/demuxer/demuxer.h"
|
2017-08-29 20:42:33 +00:00
|
|
|
#include "packager/media/event/combined_muxer_listener.h"
|
2017-05-22 20:28:10 +00:00
|
|
|
#include "packager/media/event/hls_notify_muxer_listener.h"
|
|
|
|
#include "packager/media/event/mpd_notify_muxer_listener.h"
|
|
|
|
#include "packager/media/event/vod_media_info_dump_muxer_listener.h"
|
|
|
|
#include "packager/media/formats/mp2t/ts_muxer.h"
|
|
|
|
#include "packager/media/formats/mp4/mp4_muxer.h"
|
|
|
|
#include "packager/media/formats/webm/webm_muxer.h"
|
|
|
|
#include "packager/media/trick_play/trick_play_handler.h"
|
|
|
|
#include "packager/mpd/base/dash_iop_mpd_notifier.h"
|
|
|
|
#include "packager/mpd/base/media_info.pb.h"
|
|
|
|
#include "packager/mpd/base/mpd_builder.h"
|
|
|
|
#include "packager/mpd/base/simple_mpd_notifier.h"
|
2017-05-23 02:41:26 +00:00
|
|
|
#include "packager/version/version.h"
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
namespace shaka {
|
|
|
|
|
|
|
|
// TODO(kqyang): Clean up namespaces.
|
|
|
|
using media::Demuxer;
|
|
|
|
using media::KeySource;
|
|
|
|
using media::MuxerOptions;
|
|
|
|
|
|
|
|
namespace media {
|
|
|
|
namespace {
|
|
|
|
|
|
|
|
const char kMediaInfoSuffix[] = ".media_info";
|
|
|
|
|
|
|
|
// TODO(rkuroiwa): Write TTML and WebVTT parser (demuxing) for a better check
|
|
|
|
// and for supporting live/segmenting (muxing). With a demuxer and a muxer,
|
|
|
|
// CreateRemuxJobs() shouldn't treat text as a special case.
|
|
|
|
std::string DetermineTextFileFormat(const std::string& file) {
|
|
|
|
std::string content;
|
|
|
|
if (!File::ReadFileToString(file.c_str(), &content)) {
|
|
|
|
LOG(ERROR) << "Failed to open file " << file
|
|
|
|
<< " to determine file format.";
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
MediaContainerName container_name = DetermineContainer(
|
|
|
|
reinterpret_cast<const uint8_t*>(content.data()), content.size());
|
|
|
|
if (container_name == CONTAINER_WEBVTT) {
|
|
|
|
return "vtt";
|
|
|
|
} else if (container_name == CONTAINER_TTML) {
|
|
|
|
return "ttml";
|
|
|
|
}
|
|
|
|
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
|
|
|
|
MediaContainerName GetOutputFormat(const StreamDescriptor& descriptor) {
|
|
|
|
MediaContainerName output_format = CONTAINER_UNKNOWN;
|
|
|
|
if (!descriptor.output_format.empty()) {
|
|
|
|
output_format = DetermineContainerFromFormatName(descriptor.output_format);
|
|
|
|
if (output_format == CONTAINER_UNKNOWN) {
|
|
|
|
LOG(ERROR) << "Unable to determine output format from '"
|
|
|
|
<< descriptor.output_format << "'.";
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
const std::string& output_name = descriptor.output.empty()
|
|
|
|
? descriptor.segment_template
|
|
|
|
: descriptor.output;
|
|
|
|
if (output_name.empty())
|
|
|
|
return CONTAINER_UNKNOWN;
|
|
|
|
output_format = DetermineContainerFromFileName(output_name);
|
|
|
|
if (output_format == CONTAINER_UNKNOWN) {
|
|
|
|
LOG(ERROR) << "Unable to determine output format from '" << output_name
|
|
|
|
<< "'.";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return output_format;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ValidateStreamDescriptor(bool dump_stream_info,
|
|
|
|
const StreamDescriptor& descriptor) {
|
|
|
|
// Validate and insert the descriptor
|
|
|
|
if (descriptor.input.empty()) {
|
|
|
|
LOG(ERROR) << "Stream input not specified.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
if (!dump_stream_info && descriptor.stream_selector.empty()) {
|
|
|
|
LOG(ERROR) << "Stream stream_selector not specified.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// We should have either output or segment_template specified.
|
|
|
|
const bool output_specified =
|
|
|
|
!descriptor.output.empty() || !descriptor.segment_template.empty();
|
|
|
|
if (!output_specified) {
|
2017-05-23 02:41:26 +00:00
|
|
|
if (!dump_stream_info) {
|
2017-05-22 20:28:10 +00:00
|
|
|
LOG(ERROR) << "Stream output not specified.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
const MediaContainerName output_format = GetOutputFormat(descriptor);
|
|
|
|
if (output_format == CONTAINER_UNKNOWN)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (output_format == MediaContainerName::CONTAINER_MPEG2TS) {
|
|
|
|
if (descriptor.segment_template.empty()) {
|
|
|
|
LOG(ERROR) << "Please specify segment_template. Single file TS output "
|
|
|
|
"is not supported.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
// Note that MPEG2 TS doesn't need a separate initialization segment, so
|
|
|
|
// output field is not needed.
|
|
|
|
if (!descriptor.output.empty()) {
|
|
|
|
LOG(WARNING) << "TS init_segment '" << descriptor.output
|
|
|
|
<< "' ignored. TS muxer does not support initialization "
|
|
|
|
"segment generation.";
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if (descriptor.output.empty()) {
|
|
|
|
LOG(ERROR) << "init_segment is required for format " << output_format;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ValidateParams(const PackagingParams& packaging_params,
|
|
|
|
const std::vector<StreamDescriptor>& stream_descriptors) {
|
|
|
|
if (!packaging_params.chunking_params.segment_sap_aligned &&
|
|
|
|
packaging_params.chunking_params.subsegment_sap_aligned) {
|
|
|
|
LOG(ERROR) << "Setting segment_sap_aligned to false but "
|
|
|
|
"subsegment_sap_aligned to true is not allowed.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (stream_descriptors.empty()) {
|
|
|
|
LOG(ERROR) << "Stream descriptors cannot be empty.";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
// On demand profile generates single file segment while live profile
|
|
|
|
// generates multiple segments specified using segment template.
|
|
|
|
const bool on_demand_dash_profile =
|
|
|
|
stream_descriptors.begin()->segment_template.empty();
|
|
|
|
for (const auto& descriptor : stream_descriptors) {
|
|
|
|
if (on_demand_dash_profile != descriptor.segment_template.empty()) {
|
|
|
|
LOG(ERROR) << "Inconsistent stream descriptor specification: "
|
|
|
|
"segment_template should be specified for none or all "
|
|
|
|
"stream descriptors.";
|
|
|
|
return false;
|
|
|
|
}
|
2017-05-23 02:41:26 +00:00
|
|
|
if (!ValidateStreamDescriptor(packaging_params.test_params.dump_stream_info,
|
|
|
|
descriptor))
|
2017-05-22 20:28:10 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
if (packaging_params.output_media_info && !on_demand_dash_profile) {
|
|
|
|
// TODO(rkuroiwa, kqyang): Support partial media info dump for live.
|
|
|
|
NOTIMPLEMENTED() << "ERROR: --output_media_info is only supported for "
|
|
|
|
"on-demand profile (not using segment_template).";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
class StreamDescriptorCompareFn {
|
|
|
|
public:
|
|
|
|
bool operator()(const StreamDescriptor& a, const StreamDescriptor& b) {
|
|
|
|
if (a.input == b.input) {
|
|
|
|
if (a.stream_selector == b.stream_selector)
|
|
|
|
// Stream with high trick_play_factor is at the beginning.
|
|
|
|
return a.trick_play_factor > b.trick_play_factor;
|
|
|
|
else
|
|
|
|
return a.stream_selector < b.stream_selector;
|
|
|
|
}
|
|
|
|
|
|
|
|
return a.input < b.input;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
/// Sorted list of StreamDescriptor.
|
|
|
|
typedef std::multiset<StreamDescriptor, StreamDescriptorCompareFn>
|
|
|
|
StreamDescriptorList;
|
|
|
|
|
|
|
|
// A fake clock that always return time 0 (epoch). Should only be used for
|
|
|
|
// testing.
|
|
|
|
class FakeClock : public base::Clock {
|
|
|
|
public:
|
|
|
|
base::Time Now() override { return base::Time(); }
|
|
|
|
};
|
|
|
|
|
2017-08-16 17:25:53 +00:00
|
|
|
class Job : public base::SimpleThread {
|
2017-05-22 20:28:10 +00:00
|
|
|
public:
|
2017-08-23 17:30:43 +00:00
|
|
|
Job(const std::string& name, std::shared_ptr<OriginHandler> work)
|
2017-08-17 16:25:11 +00:00
|
|
|
: SimpleThread(name),
|
|
|
|
work_(work),
|
|
|
|
wait_(base::WaitableEvent::ResetPolicy::MANUAL,
|
|
|
|
base::WaitableEvent::InitialState::NOT_SIGNALED) {}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-08-16 17:25:53 +00:00
|
|
|
void Initialize() {
|
|
|
|
DCHECK(work_);
|
|
|
|
status_ = work_->Initialize();
|
|
|
|
}
|
|
|
|
|
|
|
|
void Cancel() {
|
|
|
|
DCHECK(work_);
|
|
|
|
work_->Cancel();
|
|
|
|
}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-08-17 16:25:11 +00:00
|
|
|
const Status& status() const { return status_; }
|
|
|
|
|
|
|
|
base::WaitableEvent* wait() { return &wait_; }
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
private:
|
2017-08-16 17:25:53 +00:00
|
|
|
Job(const Job&) = delete;
|
|
|
|
Job& operator=(const Job&) = delete;
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
void Run() override {
|
2017-08-16 17:25:53 +00:00
|
|
|
DCHECK(work_);
|
|
|
|
status_ = work_->Run();
|
2017-08-17 16:25:11 +00:00
|
|
|
wait_.Signal();
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
2017-08-16 17:25:53 +00:00
|
|
|
std::shared_ptr<OriginHandler> work_;
|
2017-05-22 20:28:10 +00:00
|
|
|
Status status_;
|
2017-08-17 16:25:11 +00:00
|
|
|
|
|
|
|
base::WaitableEvent wait_;
|
2017-05-22 20:28:10 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
bool StreamInfoToTextMediaInfo(const StreamDescriptor& stream_descriptor,
|
|
|
|
const MuxerOptions& stream_muxer_options,
|
|
|
|
MediaInfo* text_media_info) {
|
|
|
|
const std::string& language = stream_descriptor.language;
|
|
|
|
const std::string format = DetermineTextFileFormat(stream_descriptor.input);
|
|
|
|
if (format.empty()) {
|
|
|
|
LOG(ERROR) << "Failed to determine the text file format for "
|
|
|
|
<< stream_descriptor.input;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!File::Copy(stream_descriptor.input.c_str(),
|
|
|
|
stream_muxer_options.output_file_name.c_str())) {
|
|
|
|
LOG(ERROR) << "Failed to copy the input file (" << stream_descriptor.input
|
|
|
|
<< ") to output file (" << stream_muxer_options.output_file_name
|
|
|
|
<< ").";
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
text_media_info->set_media_file_name(stream_muxer_options.output_file_name);
|
|
|
|
text_media_info->set_container_type(MediaInfo::CONTAINER_TEXT);
|
|
|
|
|
|
|
|
if (stream_muxer_options.bandwidth != 0) {
|
|
|
|
text_media_info->set_bandwidth(stream_muxer_options.bandwidth);
|
|
|
|
} else {
|
|
|
|
// Text files are usually small and since the input is one file; there's no
|
|
|
|
// way for the player to do ranged requests. So set this value to something
|
|
|
|
// reasonable.
|
|
|
|
const int kDefaultTextBandwidth = 256;
|
|
|
|
text_media_info->set_bandwidth(kDefaultTextBandwidth);
|
|
|
|
}
|
|
|
|
|
|
|
|
MediaInfo::TextInfo* text_info = text_media_info->mutable_text_info();
|
|
|
|
text_info->set_format(format);
|
|
|
|
if (!language.empty())
|
|
|
|
text_info->set_language(language);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<Muxer> CreateOutputMuxer(const MuxerOptions& options,
|
|
|
|
MediaContainerName container) {
|
|
|
|
if (container == CONTAINER_WEBM) {
|
|
|
|
return std::shared_ptr<Muxer>(new webm::WebMMuxer(options));
|
|
|
|
} else if (container == CONTAINER_MPEG2TS) {
|
|
|
|
return std::shared_ptr<Muxer>(new mp2t::TsMuxer(options));
|
|
|
|
} else {
|
|
|
|
DCHECK_EQ(container, CONTAINER_MOV);
|
|
|
|
return std::shared_ptr<Muxer>(new mp4::MP4Muxer(options));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-29 20:39:43 +00:00
|
|
|
std::shared_ptr<MediaHandler> CreateCryptoHandler(
|
|
|
|
const PackagingParams& packaging_params,
|
|
|
|
const StreamDescriptor& stream,
|
|
|
|
KeySource* key_source) {
|
|
|
|
if (stream.skip_encryption) {
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!key_source) {
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make a copy so that we can modify it for this specific stream.
|
|
|
|
EncryptionParams encryption_params = packaging_params.encryption_params;
|
|
|
|
|
|
|
|
// Use Sample AES in MPEG2TS.
|
|
|
|
// TODO(kqyang): Consider adding a new flag to enable Sample AES as we
|
|
|
|
// will support CENC in TS in the future.
|
|
|
|
if (GetOutputFormat(stream) == CONTAINER_MPEG2TS) {
|
|
|
|
VLOG(1) << "Use Apple Sample AES encryption for MPEG2TS.";
|
|
|
|
encryption_params.protection_scheme = kAppleSampleAesProtectionScheme;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!encryption_params.stream_label_func) {
|
|
|
|
const int kDefaultMaxSdPixels = 768 * 576;
|
|
|
|
const int kDefaultMaxHdPixels = 1920 * 1080;
|
|
|
|
const int kDefaultMaxUhd1Pixels = 4096 * 2160;
|
|
|
|
encryption_params.stream_label_func = std::bind(
|
|
|
|
&Packager::DefaultStreamLabelFunction, kDefaultMaxSdPixels,
|
|
|
|
kDefaultMaxHdPixels, kDefaultMaxUhd1Pixels, std::placeholders::_1);
|
|
|
|
}
|
|
|
|
|
|
|
|
return std::make_shared<EncryptionHandler>(encryption_params, key_source);
|
|
|
|
}
|
|
|
|
|
2017-08-23 17:30:43 +00:00
|
|
|
Status CreateRemuxJobs(const StreamDescriptorList& stream_descriptors,
|
|
|
|
const PackagingParams& packaging_params,
|
|
|
|
FakeClock* fake_clock,
|
|
|
|
KeySource* encryption_key_source,
|
|
|
|
MpdNotifier* mpd_notifier,
|
|
|
|
hls::HlsNotifier* hls_notifier,
|
|
|
|
std::vector<std::unique_ptr<Job>>* jobs) {
|
2017-05-22 20:28:10 +00:00
|
|
|
// No notifiers OR (mpd_notifier XOR hls_notifier); which is NAND.
|
|
|
|
DCHECK(!(mpd_notifier && hls_notifier));
|
2017-08-16 17:25:53 +00:00
|
|
|
DCHECK(jobs);
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-08-16 17:25:53 +00:00
|
|
|
std::shared_ptr<Demuxer> demuxer;
|
2017-05-22 20:28:10 +00:00
|
|
|
std::shared_ptr<TrickPlayHandler> trick_play_handler;
|
|
|
|
|
|
|
|
std::string previous_input;
|
|
|
|
std::string previous_stream_selector;
|
|
|
|
int stream_number = 0;
|
|
|
|
for (StreamDescriptorList::const_iterator
|
|
|
|
stream_iter = stream_descriptors.begin();
|
|
|
|
stream_iter != stream_descriptors.end();
|
|
|
|
++stream_iter, ++stream_number) {
|
|
|
|
MediaContainerName output_format = GetOutputFormat(*stream_iter);
|
|
|
|
|
|
|
|
// Process stream descriptor.
|
2017-08-18 18:57:34 +00:00
|
|
|
MuxerOptions stream_muxer_options;
|
|
|
|
stream_muxer_options.mp4_params = packaging_params.mp4_output_params;
|
|
|
|
stream_muxer_options.temp_dir = packaging_params.temp_dir;
|
2017-05-22 20:28:10 +00:00
|
|
|
stream_muxer_options.output_file_name = stream_iter->output;
|
|
|
|
if (!stream_iter->segment_template.empty()) {
|
|
|
|
if (!ValidateSegmentTemplate(stream_iter->segment_template)) {
|
2017-08-23 17:30:43 +00:00
|
|
|
return Status(
|
|
|
|
error::INVALID_ARGUMENT,
|
|
|
|
"Invalid segment template: " + stream_iter->segment_template);
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
stream_muxer_options.segment_template = stream_iter->segment_template;
|
|
|
|
}
|
|
|
|
stream_muxer_options.bandwidth = stream_iter->bandwidth;
|
|
|
|
|
|
|
|
if (stream_iter->stream_selector == "text" &&
|
|
|
|
output_format != CONTAINER_MOV) {
|
|
|
|
MediaInfo text_media_info;
|
|
|
|
if (!StreamInfoToTextMediaInfo(*stream_iter, stream_muxer_options,
|
|
|
|
&text_media_info)) {
|
2017-08-23 17:30:43 +00:00
|
|
|
return Status(error::INVALID_ARGUMENT,
|
|
|
|
"Could not create media info for stream.");
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (mpd_notifier) {
|
|
|
|
uint32_t unused;
|
|
|
|
if (!mpd_notifier->NotifyNewContainer(text_media_info, &unused)) {
|
|
|
|
LOG(ERROR) << "Failed to process text file " << stream_iter->input;
|
|
|
|
} else {
|
|
|
|
mpd_notifier->Flush();
|
|
|
|
}
|
|
|
|
} else if (packaging_params.output_media_info) {
|
|
|
|
VodMediaInfoDumpMuxerListener::WriteMediaInfoToFile(
|
|
|
|
text_media_info,
|
|
|
|
stream_muxer_options.output_file_name + kMediaInfoSuffix);
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (stream_iter->input != previous_input) {
|
|
|
|
// New remux job needed. Create demux and job thread.
|
2017-08-16 17:25:53 +00:00
|
|
|
demuxer = std::make_shared<Demuxer>(stream_iter->input);
|
|
|
|
|
2017-05-23 02:41:26 +00:00
|
|
|
demuxer->set_dump_stream_info(
|
|
|
|
packaging_params.test_params.dump_stream_info);
|
2017-05-22 20:28:10 +00:00
|
|
|
if (packaging_params.decryption_params.key_provider !=
|
|
|
|
KeyProvider::kNone) {
|
|
|
|
std::unique_ptr<KeySource> decryption_key_source(
|
|
|
|
CreateDecryptionKeySource(packaging_params.decryption_params));
|
2017-08-23 17:30:43 +00:00
|
|
|
if (!decryption_key_source) {
|
|
|
|
return Status(
|
|
|
|
error::INVALID_ARGUMENT,
|
|
|
|
"Must define decryption key source when defining key provider");
|
|
|
|
}
|
2017-05-22 20:28:10 +00:00
|
|
|
demuxer->SetKeySource(std::move(decryption_key_source));
|
|
|
|
}
|
2017-08-16 17:25:53 +00:00
|
|
|
jobs->emplace_back(new media::Job("RemuxJob", demuxer));
|
2017-05-22 20:28:10 +00:00
|
|
|
trick_play_handler.reset();
|
|
|
|
previous_input = stream_iter->input;
|
|
|
|
// Skip setting up muxers if output is not needed.
|
|
|
|
if (stream_iter->output.empty() && stream_iter->segment_template.empty())
|
|
|
|
continue;
|
|
|
|
}
|
2017-08-16 17:25:53 +00:00
|
|
|
DCHECK(!jobs->empty());
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
// Each stream selector requires an individual trick play handler.
|
|
|
|
// E.g., an input with two video streams needs two trick play handlers.
|
|
|
|
// TODO(hmchen): add a test case in packager_test.py for two video streams
|
|
|
|
// input.
|
|
|
|
if (stream_iter->stream_selector != previous_stream_selector) {
|
|
|
|
previous_stream_selector = stream_iter->stream_selector;
|
|
|
|
trick_play_handler.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<Muxer> muxer(
|
|
|
|
CreateOutputMuxer(stream_muxer_options, output_format));
|
2017-05-23 02:41:26 +00:00
|
|
|
if (packaging_params.test_params.inject_fake_clock)
|
|
|
|
muxer->set_clock(fake_clock);
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-08-29 20:42:33 +00:00
|
|
|
std::list<std::unique_ptr<MuxerListener>> muxer_listeners;
|
2017-05-22 20:28:10 +00:00
|
|
|
DCHECK(!(packaging_params.output_media_info && mpd_notifier));
|
|
|
|
if (packaging_params.output_media_info) {
|
|
|
|
const std::string output_media_info_file_name =
|
|
|
|
stream_muxer_options.output_file_name + kMediaInfoSuffix;
|
2017-08-29 20:42:33 +00:00
|
|
|
muxer_listeners.emplace_back(
|
|
|
|
new VodMediaInfoDumpMuxerListener(output_media_info_file_name));
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
2017-08-29 20:42:33 +00:00
|
|
|
|
2017-05-22 20:28:10 +00:00
|
|
|
if (mpd_notifier) {
|
2017-08-29 20:42:33 +00:00
|
|
|
muxer_listeners.emplace_back(new MpdNotifyMuxerListener(mpd_notifier));
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (hls_notifier) {
|
|
|
|
// TODO(rkuroiwa): Do some smart stuff to group the audios, e.g. detect
|
|
|
|
// languages.
|
|
|
|
std::string group_id = stream_iter->hls_group_id;
|
|
|
|
std::string name = stream_iter->hls_name;
|
|
|
|
std::string hls_playlist_name = stream_iter->hls_playlist_name;
|
|
|
|
if (group_id.empty())
|
|
|
|
group_id = "audio";
|
|
|
|
if (name.empty())
|
|
|
|
name = base::StringPrintf("stream_%d", stream_number);
|
|
|
|
if (hls_playlist_name.empty())
|
|
|
|
hls_playlist_name = base::StringPrintf("stream_%d.m3u8", stream_number);
|
|
|
|
|
2017-08-29 20:39:43 +00:00
|
|
|
muxer_listeners.emplace_back(new HlsNotifyMuxerListener(
|
|
|
|
hls_playlist_name, name, group_id, hls_notifier));
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
2017-08-29 20:42:33 +00:00
|
|
|
if (!muxer_listeners.empty()) {
|
|
|
|
std::unique_ptr<CombinedMuxerListener> combined_muxer_listener(
|
|
|
|
new CombinedMuxerListener(&muxer_listeners));
|
|
|
|
muxer->SetMuxerListener(std::move(combined_muxer_listener));
|
|
|
|
}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
// Create a new trick_play_handler. Note that the stream_decriptors
|
|
|
|
// are sorted so that for the same input and stream_selector, the main
|
|
|
|
// stream is always the last one following the trick play streams.
|
|
|
|
if (stream_iter->trick_play_factor > 0) {
|
|
|
|
if (!trick_play_handler) {
|
|
|
|
trick_play_handler.reset(new TrickPlayHandler());
|
|
|
|
}
|
|
|
|
trick_play_handler->SetHandlerForTrickPlay(stream_iter->trick_play_factor,
|
|
|
|
std::move(muxer));
|
|
|
|
if (trick_play_handler->IsConnected())
|
|
|
|
continue;
|
|
|
|
} else if (trick_play_handler) {
|
|
|
|
trick_play_handler->SetHandlerForMainStream(std::move(muxer));
|
|
|
|
DCHECK(trick_play_handler->IsConnected());
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<std::shared_ptr<MediaHandler>> handlers;
|
|
|
|
|
2017-07-27 17:23:27 +00:00
|
|
|
auto chunking_handler =
|
|
|
|
std::make_shared<ChunkingHandler>(packaging_params.chunking_params);
|
2017-05-22 20:28:10 +00:00
|
|
|
handlers.push_back(chunking_handler);
|
|
|
|
|
2017-08-29 20:39:43 +00:00
|
|
|
std::shared_ptr<MediaHandler> crypto_handler = CreateCryptoHandler(
|
|
|
|
packaging_params, *stream_iter, encryption_key_source);
|
|
|
|
if (crypto_handler) {
|
|
|
|
handlers.push_back(crypto_handler);
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// If trick_play_handler is available, muxer should already be connected to
|
|
|
|
// trick_play_handler.
|
|
|
|
if (trick_play_handler) {
|
|
|
|
handlers.push_back(trick_play_handler);
|
|
|
|
} else {
|
|
|
|
handlers.push_back(std::move(muxer));
|
|
|
|
}
|
|
|
|
|
2017-08-29 20:39:43 +00:00
|
|
|
Status status;
|
|
|
|
status.Update(
|
|
|
|
demuxer->SetHandler(stream_iter->stream_selector, chunking_handler));
|
2017-05-22 20:28:10 +00:00
|
|
|
status.Update(ConnectHandlers(handlers));
|
|
|
|
|
|
|
|
if (!status.ok()) {
|
2017-08-23 17:30:43 +00:00
|
|
|
return status;
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
if (!stream_iter->language.empty())
|
2017-08-29 20:39:43 +00:00
|
|
|
demuxer->SetLanguageOverride(stream_iter->stream_selector,
|
|
|
|
stream_iter->language);
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Initialize processing graph.
|
2017-08-23 17:30:43 +00:00
|
|
|
Status status;
|
2017-08-16 17:25:53 +00:00
|
|
|
for (const std::unique_ptr<Job>& job : *jobs) {
|
|
|
|
job->Initialize();
|
2017-08-23 17:30:43 +00:00
|
|
|
status.Update(job->status());
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
2017-08-23 17:30:43 +00:00
|
|
|
return status;
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
2017-08-17 16:25:11 +00:00
|
|
|
Status RunJobs(const std::vector<std::unique_ptr<Job>>& jobs) {
|
|
|
|
// We need to store the jobs and the waits separately in order to use the
|
|
|
|
// |WaitMany| function. |WaitMany| takes an array of WaitableEvents but we
|
|
|
|
// need to access the jobs in order to join the thread and check the status.
|
|
|
|
// The indexes needs to be check in sync or else we won't be able to relate a
|
|
|
|
// WaitableEvent back to the job.
|
|
|
|
std::vector<Job*> active_jobs;
|
|
|
|
std::vector<base::WaitableEvent*> active_waits;
|
|
|
|
|
|
|
|
// Start every job and add it to the active jobs list so that we can wait
|
|
|
|
// on each one.
|
|
|
|
for (auto& job : jobs) {
|
2017-05-22 20:28:10 +00:00
|
|
|
job->Start();
|
|
|
|
|
2017-08-17 16:25:11 +00:00
|
|
|
active_jobs.push_back(job.get());
|
|
|
|
active_waits.push_back(job->wait());
|
|
|
|
}
|
|
|
|
|
2017-05-22 20:28:10 +00:00
|
|
|
// Wait for all jobs to complete or an error occurs.
|
|
|
|
Status status;
|
2017-08-17 16:25:11 +00:00
|
|
|
while (status.ok() && active_jobs.size()) {
|
|
|
|
// Wait for an event to finish and then update our status so that we can
|
|
|
|
// quit if something has gone wrong.
|
2017-08-23 17:30:43 +00:00
|
|
|
const size_t done =
|
|
|
|
base::WaitableEvent::WaitMany(active_waits.data(), active_waits.size());
|
2017-08-17 16:25:11 +00:00
|
|
|
Job* job = active_jobs[done];
|
|
|
|
|
|
|
|
job->Join();
|
|
|
|
status.Update(job->status());
|
|
|
|
|
|
|
|
// Remove the job and the wait from our tracking.
|
|
|
|
active_jobs.erase(active_jobs.begin() + done);
|
|
|
|
active_waits.erase(active_waits.begin() + done);
|
|
|
|
}
|
|
|
|
|
|
|
|
// If the main loop has exited and there are still jobs running,
|
|
|
|
// we need to cancel them and clean-up.
|
|
|
|
for (auto& job : active_jobs) {
|
|
|
|
job->Cancel();
|
|
|
|
}
|
|
|
|
|
|
|
|
for (auto& job : active_jobs) {
|
|
|
|
job->Join();
|
|
|
|
}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace
|
|
|
|
} // namespace media
|
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
struct Packager::PackagerInternal {
|
2017-05-22 20:28:10 +00:00
|
|
|
media::FakeClock fake_clock;
|
|
|
|
std::unique_ptr<KeySource> encryption_key_source;
|
|
|
|
std::unique_ptr<MpdNotifier> mpd_notifier;
|
|
|
|
std::unique_ptr<hls::HlsNotifier> hls_notifier;
|
2017-08-16 17:25:53 +00:00
|
|
|
std::vector<std::unique_ptr<media::Job>> jobs;
|
2017-05-22 20:28:10 +00:00
|
|
|
};
|
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
Packager::Packager() {}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
Packager::~Packager() {}
|
2017-05-22 20:28:10 +00:00
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
Status Packager::Initialize(
|
2017-05-22 20:28:10 +00:00
|
|
|
const PackagingParams& packaging_params,
|
|
|
|
const std::vector<StreamDescriptor>& stream_descriptors) {
|
2017-05-23 02:41:26 +00:00
|
|
|
// Needed by base::WorkedPool used in ThreadedIoFile.
|
|
|
|
static base::AtExitManager exit;
|
2017-05-22 20:28:10 +00:00
|
|
|
static media::LibcryptoThreading libcrypto_threading;
|
|
|
|
|
|
|
|
if (internal_)
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Already initialized.");
|
|
|
|
|
|
|
|
if (!media::ValidateParams(packaging_params, stream_descriptors))
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Invalid packaging params.");
|
|
|
|
|
2017-05-23 02:41:26 +00:00
|
|
|
if (!packaging_params.test_params.injected_library_version.empty()) {
|
|
|
|
SetPackagerVersionForTesting(
|
|
|
|
packaging_params.test_params.injected_library_version);
|
|
|
|
}
|
|
|
|
|
2017-05-22 20:28:10 +00:00
|
|
|
std::unique_ptr<PackagerInternal> internal(new PackagerInternal);
|
|
|
|
|
|
|
|
// Create encryption key source if needed.
|
|
|
|
if (packaging_params.encryption_params.key_provider != KeyProvider::kNone) {
|
2017-07-05 23:47:55 +00:00
|
|
|
internal->encryption_key_source = CreateEncryptionKeySource(
|
|
|
|
static_cast<media::FourCC>(
|
|
|
|
packaging_params.encryption_params.protection_scheme),
|
|
|
|
packaging_params.encryption_params);
|
2017-05-22 20:28:10 +00:00
|
|
|
if (!internal->encryption_key_source)
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Failed to create key source.");
|
|
|
|
}
|
|
|
|
|
|
|
|
const MpdParams& mpd_params = packaging_params.mpd_params;
|
|
|
|
if (!mpd_params.mpd_output.empty()) {
|
2017-08-02 20:37:47 +00:00
|
|
|
const bool on_demand_dash_profile =
|
|
|
|
stream_descriptors.begin()->segment_template.empty();
|
|
|
|
MpdOptions mpd_options = media::GetMpdOptions(on_demand_dash_profile,
|
|
|
|
packaging_params.mpd_params);
|
2017-05-22 20:28:10 +00:00
|
|
|
if (mpd_params.generate_dash_if_iop_compliant_mpd) {
|
2017-07-27 18:49:50 +00:00
|
|
|
internal->mpd_notifier.reset(new DashIopMpdNotifier(mpd_options));
|
2017-05-22 20:28:10 +00:00
|
|
|
} else {
|
2017-07-27 18:49:50 +00:00
|
|
|
internal->mpd_notifier.reset(new SimpleMpdNotifier(mpd_options));
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
if (!internal->mpd_notifier->Init()) {
|
|
|
|
LOG(ERROR) << "MpdNotifier failed to initialize.";
|
|
|
|
return Status(error::INVALID_ARGUMENT,
|
|
|
|
"Failed to initialize MpdNotifier.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
const HlsParams& hls_params = packaging_params.hls_params;
|
|
|
|
if (!hls_params.master_playlist_output.empty()) {
|
|
|
|
base::FilePath master_playlist_path(
|
|
|
|
base::FilePath::FromUTF8Unsafe(hls_params.master_playlist_output));
|
|
|
|
base::FilePath master_playlist_name = master_playlist_path.BaseName();
|
|
|
|
|
|
|
|
internal->hls_notifier.reset(new hls::SimpleHlsNotifier(
|
2017-06-30 20:42:46 +00:00
|
|
|
hls_params.playlist_type, hls_params.time_shift_buffer_depth,
|
|
|
|
hls_params.base_url,
|
2017-05-22 20:28:10 +00:00
|
|
|
master_playlist_path.DirName().AsEndingWithSeparator().AsUTF8Unsafe(),
|
|
|
|
master_playlist_name.AsUTF8Unsafe()));
|
|
|
|
}
|
|
|
|
|
|
|
|
media::StreamDescriptorList stream_descriptor_list;
|
|
|
|
for (const StreamDescriptor& descriptor : stream_descriptors)
|
|
|
|
stream_descriptor_list.insert(descriptor);
|
2017-08-23 17:30:43 +00:00
|
|
|
Status status = media::CreateRemuxJobs(
|
|
|
|
stream_descriptor_list, packaging_params, &internal->fake_clock,
|
|
|
|
internal->encryption_key_source.get(), internal->mpd_notifier.get(),
|
|
|
|
internal->hls_notifier.get(), &internal->jobs);
|
|
|
|
|
|
|
|
if (status.ok()) {
|
|
|
|
internal_ = std::move(internal);
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
2017-08-23 17:30:43 +00:00
|
|
|
|
|
|
|
return status;
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
Status Packager::Run() {
|
2017-05-22 20:28:10 +00:00
|
|
|
if (!internal_)
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Not yet initialized.");
|
2017-08-17 16:25:11 +00:00
|
|
|
Status status = media::RunJobs(internal_->jobs);
|
2017-05-22 20:28:10 +00:00
|
|
|
if (!status.ok())
|
|
|
|
return status;
|
|
|
|
|
|
|
|
if (internal_->hls_notifier) {
|
|
|
|
if (!internal_->hls_notifier->Flush())
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Failed to flush Hls.");
|
|
|
|
}
|
|
|
|
if (internal_->mpd_notifier) {
|
|
|
|
if (!internal_->mpd_notifier->Flush())
|
|
|
|
return Status(error::INVALID_ARGUMENT, "Failed to flush Mpd.");
|
|
|
|
}
|
|
|
|
return Status::OK;
|
|
|
|
}
|
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
void Packager::Cancel() {
|
2017-05-22 20:28:10 +00:00
|
|
|
if (!internal_) {
|
|
|
|
LOG(INFO) << "Not yet initialized. Return directly.";
|
|
|
|
return;
|
|
|
|
}
|
2017-08-16 17:25:53 +00:00
|
|
|
for (const std::unique_ptr<media::Job>& job : internal_->jobs)
|
|
|
|
job->Cancel();
|
2017-05-22 20:28:10 +00:00
|
|
|
}
|
|
|
|
|
2017-06-13 22:16:08 +00:00
|
|
|
std::string Packager::GetLibraryVersion() {
|
2017-05-23 02:41:26 +00:00
|
|
|
return GetPackagerVersion();
|
|
|
|
}
|
|
|
|
|
2017-07-05 23:47:55 +00:00
|
|
|
std::string Packager::DefaultStreamLabelFunction(
|
|
|
|
int max_sd_pixels,
|
|
|
|
int max_hd_pixels,
|
|
|
|
int max_uhd1_pixels,
|
|
|
|
const EncryptionParams::EncryptedStreamAttributes& stream_attributes) {
|
|
|
|
if (stream_attributes.stream_type ==
|
|
|
|
EncryptionParams::EncryptedStreamAttributes::kAudio)
|
|
|
|
return "AUDIO";
|
|
|
|
if (stream_attributes.stream_type ==
|
|
|
|
EncryptionParams::EncryptedStreamAttributes::kVideo) {
|
|
|
|
const int pixels = stream_attributes.oneof.video.width *
|
|
|
|
stream_attributes.oneof.video.height;
|
2017-08-23 17:30:43 +00:00
|
|
|
if (pixels <= max_sd_pixels)
|
|
|
|
return "SD";
|
|
|
|
if (pixels <= max_hd_pixels)
|
|
|
|
return "HD";
|
|
|
|
if (pixels <= max_uhd1_pixels)
|
|
|
|
return "UHD1";
|
2017-07-05 23:47:55 +00:00
|
|
|
return "UHD2";
|
|
|
|
}
|
|
|
|
return "";
|
|
|
|
}
|
|
|
|
|
2017-05-22 20:28:10 +00:00
|
|
|
} // namespace shaka
|