shaka-packager/packager/mpd/util/mpd_writer.cc

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