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

71 lines
2.2 KiB
C++

// Copyright 2015 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 MEDIA_FORMATS_WEBM_SINGLE_SEGMENT_SEGMENTER_H_
#define MEDIA_FORMATS_WEBM_SINGLE_SEGMENT_SEGMENTER_H_
#include "packager/media/formats/webm/segmenter.h"
#include <memory>
#include "packager/media/base/status.h"
#include "packager/media/formats/webm/mkv_writer.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(uint64_t start_timescale,
uint64_t duration_timescale,
bool is_subsegment) override;
bool GetInitRangeStartAndEnd(uint64_t* start, uint64_t* end) override;
bool GetIndexRangeStartAndEnd(uint64_t* start, uint64_t* end) 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(std::unique_ptr<MkvWriter> writer) override;
Status DoFinalize() override;
private:
// Segmenter implementation overrides.
Status NewSegment(uint64_t start_timescale, 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 // MEDIA_FORMATS_WEBM_SINGLE_SEGMENT_SEGMENTER_H_