87 lines
3.0 KiB
C
87 lines
3.0 KiB
C
|
// Copyright 2017 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
|
||
|
|
|
||
|
|
#ifndef PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_TEXT_HANDLER_H_
|
||
|
|
#define PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_TEXT_HANDLER_H_
|
||
|
|
|
||
|
|
#include <stdint.h>
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "packager/media/base/media_handler.h"
|
||
|
|
#include "packager/media/base/muxer_options.h"
|
||
|
|
#include "packager/media/event/muxer_listener.h"
|
||
|
|
|
||
|
|
namespace shaka {
|
||
|
|
namespace media {
|
||
|
|
|
||
|
|
// WebVttOutputHandler is the base class for all WebVtt text output handlers.
|
||
|
|
// It handles taking in the samples and writing the text out, but relies on
|
||
|
|
// sub classes to handle the logic of when and where to write the information.
|
||
|
|
class WebVttOutputHandler : public MediaHandler {
|
||
|
|
public:
|
||
|
|
WebVttOutputHandler() = default;
|
||
|
|
virtual ~WebVttOutputHandler() = default;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
virtual Status OnStreamInfo(const StreamInfo& info) = 0;
|
||
|
|
virtual Status OnSegmentInfo(const SegmentInfo& info) = 0;
|
||
|
|
virtual Status OnTextSample(const TextSample& sample) = 0;
|
||
|
|
virtual Status OnStreamEnd() = 0;
|
||
|
|
|
||
|
|
// Top level functions for output. These functions should be used by
|
||
|
|
// subclasses to write to files.
|
||
|
|
void WriteCue(const std::string& id,
|
||
|
|
uint64_t start,
|
||
|
|
uint64_t end,
|
||
|
|
const std::string& settings,
|
||
|
|
const std::string& payload);
|
||
|
|
// Writes the current state of the current segment to disk. This will
|
||
|
|
// reset the internal state and set it up for the next segment.
|
||
|
|
Status WriteSegmentToFile(const std::string& filename);
|
||
|
|
|
||
|
|
private:
|
||
|
|
WebVttOutputHandler(const WebVttOutputHandler&) = delete;
|
||
|
|
WebVttOutputHandler& operator=(const WebVttOutputHandler&) = delete;
|
||
|
|
|
||
|
|
Status InitializeInternal() override;
|
||
|
|
Status Process(std::unique_ptr<StreamData> stream_data) override;
|
||
|
|
Status OnFlushRequest(size_t input_stream_index) override;
|
||
|
|
|
||
|
|
// A buffer of characters waiting to be written to a file.
|
||
|
|
std::string buffer_;
|
||
|
|
};
|
||
|
|
|
||
|
|
// This WebVttt output handler should only be used when the source WebVTT
|
||
|
|
// content needs to be segmented across multiple files.
|
||
|
|
class WebVttSegmentedOutputHandler : public WebVttOutputHandler {
|
||
|
|
public:
|
||
|
|
WebVttSegmentedOutputHandler(const MuxerOptions& muxer_options,
|
||
|
|
std::unique_ptr<MuxerListener> muxer_listener);
|
||
|
|
|
||
|
|
private:
|
||
|
|
Status OnStreamInfo(const StreamInfo& info) override;
|
||
|
|
Status OnSegmentInfo(const SegmentInfo& info) override;
|
||
|
|
Status OnTextSample(const TextSample& sample) override;
|
||
|
|
Status OnStreamEnd() override;
|
||
|
|
|
||
|
|
Status OnSegmentEnded();
|
||
|
|
|
||
|
|
void GoToNextSegment(uint64_t start_time_ms);
|
||
|
|
|
||
|
|
const MuxerOptions muxer_options_;
|
||
|
|
std::unique_ptr<MuxerListener> muxer_listener_;
|
||
|
|
|
||
|
|
// Sum together all segment durations so we know how long the stream is.
|
||
|
|
uint64_t total_duration_ms_ = 0;
|
||
|
|
uint32_t segment_index_ = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace media
|
||
|
|
} // namespace shaka
|
||
|
|
|
||
|
|
#endif // PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_TEXT_HANDLER_H_
|