shaka-packager/packager/media/chunking/text_chunker.h

53 lines
1.4 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_CHUNKING_TEXT_CHUNKER_H_
#define PACKAGER_MEDIA_CHUNKING_TEXT_CHUNKER_H_
#include <stdint.h>
#include <map>
#include <vector>
#include "packager/media/base/media_handler.h"
#include "packager/status.h"
namespace shaka {
namespace media {
class TextChunker : public MediaHandler {
public:
explicit TextChunker(uint64_t segment_duration_ms);
protected:
Status Process(std::unique_ptr<StreamData> stream_data) override;
Status OnFlushRequest(size_t input_stream_index) override;
private:
TextChunker(const TextChunker&) = delete;
TextChunker& operator=(const TextChunker&) = delete;
using SegmentSamples = std::vector<std::shared_ptr<const TextSample>>;
Status InitializeInternal() override;
Status OnTextSample(std::shared_ptr<const TextSample> sample);
Status DispatchSegmentWithSamples(uint64_t segment,
const SegmentSamples& samples);
uint64_t segment_duration_ms_;
// Mapping of segment number to segment.
std::map<uint64_t, SegmentSamples> segment_map_;
uint64_t head_segment_ = 0;
};
} // namespace media
} // namespace shaka
#endif // PACKAGER_MEDIA_CHUNKING_TEXT_CHUNKER_H_