shaka-packager/packager/media/formats/webm/single_segment_segmenter.h

74 lines
2.3 KiB
C++

// Copyright 2015 Google LLC. 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_WEBM_SINGLE_SEGMENT_SEGMENTER_H_
#define PACKAGER_MEDIA_FORMATS_WEBM_SINGLE_SEGMENT_SEGMENTER_H_
#include <memory>
#include <packager/macros/classes.h>
#include <packager/media/formats/webm/mkv_writer.h>
#include <packager/media/formats/webm/segmenter.h>
#include <packager/status.h>
namespace shaka {
namespace media {
struct MuxerOptions;
namespace webm {
/// An implementation of a Segmenter for a single-segment. This assumes that
/// the output file is seekable. For non-seekable files, use
/// TwoPassSingleSegmentSegmenter.
class SingleSegmentSegmenter : public Segmenter {
public:
explicit SingleSegmentSegmenter(const MuxerOptions& options);
~SingleSegmentSegmenter() override;
/// @name Segmenter implementation overrides.
/// @{
Status FinalizeSegment(int64_t start_timestamp,
int64_t duration_timestamp,
bool is_subsegment,
int64_t segment_number) override;
bool GetInitRangeStartAndEnd(uint64_t* start, uint64_t* end) override;
bool GetIndexRangeStartAndEnd(uint64_t* start, uint64_t* end) override;
std::vector<Range> GetSegmentRanges() override;
/// @}
protected:
MkvWriter* writer() { return writer_.get(); }
uint64_t init_end() { return init_end_; }
void set_init_end(uint64_t end) { init_end_ = end; }
void set_index_start(uint64_t start) { index_start_ = start; }
void set_index_end(uint64_t end) { index_end_ = end; }
void set_writer(std::unique_ptr<MkvWriter> writer) {
writer_ = std::move(writer);
}
// Segmenter implementation overrides.
Status DoInitialize() override;
Status DoFinalize() override;
private:
// Segmenter implementation overrides.
Status NewSegment(int64_t start_timestamp, bool is_subsegment) override;
std::unique_ptr<MkvWriter> writer_;
uint64_t init_end_;
uint64_t index_start_;
uint64_t index_end_;
DISALLOW_COPY_AND_ASSIGN(SingleSegmentSegmenter);
};
} // namespace webm
} // namespace media
} // namespace shaka
#endif // PACKAGER_MEDIA_FORMATS_WEBM_SINGLE_SEGMENT_SEGMENTER_H_