shaka-packager/packager/mpd/base/mpd_notifier_util.cc

// 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/mpd/base/mpd_notifier_util.h"

#include "packager/base/strings/string_number_conversions.h"
#include "packager/base/strings/string_util.h"
#include "packager/media/file/file.h"
#include "packager/media/file/file_closer.h"
#include "packager/mpd/base/mpd_utils.h"

namespace shaka {

using media::File;
using media::FileCloser;

bool WriteMpdToFile(const std::string& output_path, MpdBuilder* mpd_builder) {
  CHECK(!output_path.empty());

  std::string mpd;
  if (!mpd_builder->ToString(&mpd)) {
    LOG(ERROR) << "Failed to write MPD to string.";
    return false;
  }

  std::unique_ptr<File, FileCloser> file(File::Open(output_path.c_str(), "w"));
  if (!file) {
    LOG(ERROR) << "Failed to open file for writing: " << output_path;
    return false;
  }

  const char* mpd_char_ptr = mpd.data();
  size_t mpd_bytes_left = mpd.size();
  while (mpd_bytes_left > 0) {
    int64_t length = file->Write(mpd_char_ptr, mpd_bytes_left);
    if (length <= 0) {
      LOG(ERROR) << "Failed to write to file '" << output_path << "' ("
                 << length << ").";
      return false;
    }
    mpd_char_ptr += length;
    mpd_bytes_left -= length;
  }
  // Release the pointer because Close() destructs itself.
  return file.release()->Close();
}

ContentType GetContentType(const MediaInfo& media_info) {
  const bool has_video = media_info.has_video_info();
  const bool has_audio = media_info.has_audio_info();
  const bool has_text = media_info.has_text_info();

  if (MoreThanOneTrue(has_video, has_audio, has_text)) {
    NOTIMPLEMENTED() << "MediaInfo with more than one stream is not supported.";
    return kContentTypeUnknown;
  }
  if (!AtLeastOneTrue(has_video, has_audio, has_text)) {
    LOG(ERROR) << "MediaInfo should contain one audio, video, or text stream.";
    return kContentTypeUnknown;
  }
  return has_video ? kContentTypeVideo
                   : (has_audio ? kContentTypeAudio : kContentTypeText);
}

std::string Uint8VectorToBase64(const std::vector<uint8_t>& input) {
  std::string output;
  std::string input_in_string(input.begin(), input.end());
  base::Base64Encode(input_in_string, &output);
  return output;
}

}  // namespace shaka