single_segment_segmenter.h

// 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 <cstdint>
#include <memory>
#include <utility>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/base/range.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 {
 
class SingleSegmentSegmenter : public Segmenter {
 public:
  explicit SingleSegmentSegmenter(const MuxerOptions& options);
  ~SingleSegmentSegmenter() override;
 
  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_