53 lines
1.5 KiB
C++
53 lines
1.5 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_SEEK_HEAD_H_
|
|
#define MEDIA_FORMATS_WEBM_SEEK_HEAD_H_
|
|
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
|
|
#include "base/macros.h"
|
|
#include "packager/third_party/libwebm/src/mkvmuxer.hpp"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
/// Used to write the SeekHead to the output stream. This supports non-seekable
|
|
/// files and setting the values before write; this also supports updating.
|
|
class SeekHead {
|
|
public:
|
|
SeekHead();
|
|
~SeekHead();
|
|
|
|
/// Writes the seek head to the given writer. This should only be called
|
|
/// once. For seekable files, use WriteVoid first, then call this method.
|
|
bool Write(mkvmuxer::IMkvWriter* writer);
|
|
/// Writes a void element large enough to fit the SeekHead.
|
|
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:
|
|
uint64_t GetPayloadSize(std::vector<uint64_t>* data);
|
|
|
|
int64_t cluster_pos_;
|
|
int64_t cues_pos_;
|
|
int64_t info_pos_;
|
|
int64_t tracks_pos_;
|
|
bool wrote_void_;
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(SeekHead);
|
|
};
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|
|
|
|
#endif // MEDIA_FORMATS_WEBM_SEEK_HEAD_H_
|