seek_head.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_SEEK_HEAD_H_
#define PACKAGER_MEDIA_FORMATS_WEBM_SEEK_HEAD_H_
 
#include <cstdint>
#include <vector>
 
#include <mkvmuxer/mkvmuxer.h>
#include <mkvmuxer/mkvmuxertypes.h>
 
namespace shaka {
namespace media {
 
class SeekHead {
 public:
  SeekHead();
  ~SeekHead();
 
  bool Write(mkvmuxer::IMkvWriter* writer);
  bool WriteVoid(mkvmuxer::IMkvWriter* writer);
 
  void set_cluster_pos(uint64_t pos) { cluster_pos_ = pos; }
  void set_cues_pos(uint64_t pos) { cues_pos_ = pos; }
  void set_info_pos(uint64_t pos) { info_pos_ = pos; }
  void set_tracks_pos(uint64_t pos) { tracks_pos_ = pos; }
 
 private:
  SeekHead(const SeekHead&) = delete;
  SeekHead& operator=(const SeekHead&) = delete;
 
  struct SeekElement {
    mkvmuxer::uint64 id;
    mkvmuxer::uint64 position;
    mkvmuxer::uint64 size;
 
    SeekElement(uint64_t seek_id, uint64_t seek_position)
        : id(seek_id), position(seek_position), size(0) {}
  };
 
  // Create seek element vector from positions.
  std::vector<SeekElement> CreateSeekElements();
 
  // In practice, these positions, if set, will never be 0, so we use a zero
  // value to denote that they are not set.
  uint64_t cluster_pos_ = 0;
  uint64_t cues_pos_ = 0;
  uint64_t info_pos_ = 0;
  uint64_t tracks_pos_ = 0;
  bool wrote_void_ = false;
  const uint64_t total_void_size_ = 0;
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_WEBM_SEEK_HEAD_H_