76 lines
2.4 KiB
C++
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.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
|