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

76 lines
2.4 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/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_closer.h"
#include "packager/media/file/file.h"
#include "packager/mpd/base/mpd_utils.h"
namespace edash_packager {
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;
}
scoped_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 edash_packager