104 lines
3.0 KiB
C++
104 lines
3.0 KiB
C++
// Copyright 2014 Google LLC. 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/mpd/util/mpd_writer.h"
|
|
|
|
#include <absl/flags/flag.h>
|
|
#include <glog/logging.h>
|
|
#include <google/protobuf/text_format.h>
|
|
#include <stdint.h>
|
|
|
|
#include "packager/file/file.h"
|
|
#include "packager/mpd/base/mpd_builder.h"
|
|
#include "packager/mpd/base/mpd_notifier.h"
|
|
#include "packager/mpd/base/mpd_utils.h"
|
|
#include "packager/mpd/base/simple_mpd_notifier.h"
|
|
|
|
ABSL_FLAG(bool,
|
|
generate_dash_if_iop_compliant_mpd,
|
|
true,
|
|
"Try to generate DASH-IF IOP compliant MPD. This is best effort "
|
|
"and does not guarantee compliance.");
|
|
|
|
namespace shaka {
|
|
|
|
namespace {
|
|
|
|
// Factory that creates SimpleMpdNotifier instances.
|
|
class SimpleMpdNotifierFactory : public MpdNotifierFactory {
|
|
public:
|
|
SimpleMpdNotifierFactory() {}
|
|
~SimpleMpdNotifierFactory() override {}
|
|
|
|
std::unique_ptr<MpdNotifier> Create(const MpdOptions& mpd_options) override {
|
|
return std::unique_ptr<MpdNotifier>(new SimpleMpdNotifier(mpd_options));
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
MpdWriter::MpdWriter() : notifier_factory_(new SimpleMpdNotifierFactory()) {}
|
|
MpdWriter::~MpdWriter() {}
|
|
|
|
bool MpdWriter::AddFile(const std::string& media_info_path) {
|
|
std::string file_content;
|
|
if (!File::ReadFileToString(media_info_path.c_str(), &file_content)) {
|
|
LOG(ERROR) << "Failed to read " << media_info_path << " to string.";
|
|
return false;
|
|
}
|
|
|
|
MediaInfo media_info;
|
|
if (!::google::protobuf::TextFormat::ParseFromString(file_content,
|
|
&media_info)) {
|
|
LOG(ERROR) << "Failed to parse " << file_content << " to MediaInfo.";
|
|
return false;
|
|
}
|
|
|
|
media_infos_.push_back(media_info);
|
|
return true;
|
|
}
|
|
|
|
void MpdWriter::AddBaseUrl(const std::string& base_url) {
|
|
base_urls_.push_back(base_url);
|
|
}
|
|
|
|
bool MpdWriter::WriteMpdToFile(const char* file_name) {
|
|
CHECK(file_name);
|
|
MpdOptions mpd_options;
|
|
mpd_options.mpd_params.base_urls = base_urls_;
|
|
mpd_options.mpd_params.mpd_output = file_name;
|
|
mpd_options.mpd_params.generate_dash_if_iop_compliant_mpd =
|
|
absl::GetFlag(FLAGS_generate_dash_if_iop_compliant_mpd);
|
|
std::unique_ptr<MpdNotifier> notifier =
|
|
notifier_factory_->Create(mpd_options);
|
|
if (!notifier->Init()) {
|
|
LOG(ERROR) << "failed to initialize MpdNotifier.";
|
|
return false;
|
|
}
|
|
|
|
for (const MediaInfo& media_info : media_infos_) {
|
|
uint32_t unused_conatiner_id;
|
|
if (!notifier->NotifyNewContainer(media_info, &unused_conatiner_id)) {
|
|
LOG(ERROR) << "Failed to add MediaInfo for media file: "
|
|
<< media_info.media_file_name();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (!notifier->Flush()) {
|
|
LOG(ERROR) << "Failed to flush MPD notifier.";
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void MpdWriter::SetMpdNotifierFactoryForTest(
|
|
std::unique_ptr<MpdNotifierFactory> factory) {
|
|
notifier_factory_ = std::move(factory);
|
|
}
|
|
|
|
} // namespace shaka
|