61 lines
2.0 KiB
C++
61 lines
2.0 KiB
C++
|
#include "packager/media/formats/webvtt/cue.h"
|
||
|
|
|
||
|
|
#include "packager/base/strings/string_util.h"
|
||
|
|
|
||
|
|
namespace shaka {
|
||
|
|
namespace media {
|
||
|
|
|
||
|
|
Cue::Cue() : start_time(0), duration(0) {}
|
||
|
|
Cue::~Cue() {}
|
||
|
|
|
||
|
|
// Mapping:
|
||
|
|
// comment --> side data (and side data only sample)
|
||
|
|
// settings --> side data
|
||
|
|
// start_time --> pts
|
||
|
|
std::shared_ptr<MediaSample> CueToMediaSample(const Cue& cue) {
|
||
|
|
const bool kKeyFrame = true;
|
||
|
|
if (!cue.comment.empty()) {
|
||
|
|
const std::string comment = base::JoinString(cue.comment, "\n");
|
||
|
|
return MediaSample::FromMetadata(
|
||
|
|
reinterpret_cast<const uint8_t*>(comment.data()), comment.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
const std::string payload = base::JoinString(cue.payload, "\n");
|
||
|
|
std::shared_ptr<MediaSample> media_sample = MediaSample::CopyFrom(
|
||
|
|
reinterpret_cast<const uint8_t*>(payload.data()), payload.size(),
|
||
|
|
reinterpret_cast<const uint8_t*>(cue.settings.data()),
|
||
|
|
cue.settings.size(), !kKeyFrame);
|
||
|
|
|
||
|
|
media_sample->set_config_id(cue.identifier);
|
||
|
|
media_sample->set_pts(cue.start_time);
|
||
|
|
media_sample->set_duration(cue.duration);
|
||
|
|
return media_sample;
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO(rkuroiwa): Cue gets converted to MediaSample in WebVttMediaParser and
|
||
|
|
// then back to Cue in the muxer. Consider making MediaSample a protobuf or make
|
||
|
|
// Cue a protobuf and (ab)use MediaSample::data() to store serialized Cue.
|
||
|
|
Cue MediaSampleToCue(const MediaSample& sample) {
|
||
|
|
Cue cue;
|
||
|
|
if (sample.data_size() == 0) {
|
||
|
|
std::string comment(sample.side_data(),
|
||
|
|
sample.side_data() + sample.side_data_size());
|
||
|
|
cue.comment.push_back(comment);
|
||
|
|
return cue;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string payload(sample.data(), sample.data() + sample.data_size());
|
||
|
|
cue.payload.push_back(payload);
|
||
|
|
cue.identifier.assign(sample.config_id());
|
||
|
|
cue.start_time = sample.pts();
|
||
|
|
cue.duration = sample.duration();
|
||
|
|
if (sample.side_data_size() != 0) {
|
||
|
|
cue.settings.assign(sample.side_data(),
|
||
|
|
sample.side_data() + sample.side_data_size());
|
||
|
|
}
|
||
|
|
return cue;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace media
|
||
|
|
} // namespace shaka
|