62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
|
// Copyright 2018 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/media/formats/webvtt/text_padder.h"
|
||
|
|
|
||
|
|
#include <algorithm>
|
||
|
|
|
||
|
|
#include "packager/status_macros.h"
|
||
|
|
|
||
|
|
namespace shaka {
|
||
|
|
namespace media {
|
||
|
|
namespace {
|
||
|
|
const uint64_t kStreamIndex = 0;
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
TextPadder::TextPadder(int64_t duration_ms) : duration_ms_(duration_ms) {}
|
||
|
|
|
||
|
|
Status TextPadder::InitializeInternal() {
|
||
|
|
return Status::OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
Status TextPadder::Process(std::unique_ptr<StreamData> data) {
|
||
|
|
DCHECK_EQ(data->stream_index, kStreamIndex);
|
||
|
|
const bool is_text_sample =
|
||
|
|
data->stream_data_type == StreamDataType::kTextSample;
|
||
|
|
return is_text_sample ? OnTextSample(std::move(data))
|
||
|
|
: Dispatch(std::move(data));
|
||
|
|
}
|
||
|
|
|
||
|
|
Status TextPadder::OnFlushRequest(size_t index) {
|
||
|
|
if (duration_ms_ > max_end_time_ms_) {
|
||
|
|
std::shared_ptr<TextSample> filler = std::make_shared<TextSample>();
|
||
|
|
filler->SetTime(max_end_time_ms_, duration_ms_);
|
||
|
|
RETURN_IF_ERROR(
|
||
|
|
MediaHandler::DispatchTextSample(kStreamIndex, std::move(filler)));
|
||
|
|
}
|
||
|
|
|
||
|
|
return FlushDownstream(index);
|
||
|
|
}
|
||
|
|
|
||
|
|
Status TextPadder::OnTextSample(std::unique_ptr<StreamData> data) {
|
||
|
|
const TextSample& sample = *data->text_sample;
|
||
|
|
|
||
|
|
// Check if there will be a gap between samples if we just dispatch this
|
||
|
|
// sample right away. If there will be one, create an empty sample that will
|
||
|
|
// fill in that gap.
|
||
|
|
if (sample.start_time() > max_end_time_ms_) {
|
||
|
|
std::shared_ptr<TextSample> filler = std::make_shared<TextSample>();
|
||
|
|
filler->SetTime(max_end_time_ms_, sample.start_time());
|
||
|
|
RETURN_IF_ERROR(
|
||
|
|
MediaHandler::DispatchTextSample(kStreamIndex, std::move(filler)));
|
||
|
|
}
|
||
|
|
|
||
|
|
max_end_time_ms_ = std::max(max_end_time_ms_, sample.EndTime());
|
||
|
|
return Dispatch(std::move(data));
|
||
|
|
}
|
||
|
|
} // namespace media
|
||
|
|
} // namespace shaka
|