buffer_writer.h

// Copyright 2014 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_BASE_BUFFER_WRITER_H_
#define PACKAGER_MEDIA_BASE_BUFFER_WRITER_H_
 
#include <cstdint>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/status.h>
 
namespace shaka {
 
class File;
 
namespace media {
 
class BufferWriter {
 public:
  BufferWriter();
  explicit BufferWriter(size_t reserved_size_in_bytes);
  ~BufferWriter();
 
  void AppendInt(uint8_t v);
  void AppendInt(uint16_t v);
  void AppendInt(uint32_t v);
  void AppendInt(uint64_t v);
  void AppendInt(int16_t v);
  void AppendInt(int32_t v);
  void AppendInt(int64_t v);
 
  void AppendNBytes(uint64_t v, size_t num_bytes);
 
  void AppendVector(const std::vector<uint8_t>& v);
  void AppendString(const std::string& s);
  void AppendArray(const uint8_t* buf, size_t size);
  void AppendBuffer(const BufferWriter& buffer);
 
  void Swap(BufferWriter* buffer) { buf_.swap(buffer->buf_); }
  void SwapBuffer(std::vector<uint8_t>* buffer) { buf_.swap(*buffer); }
 
  void Clear() { buf_.clear(); }
  size_t Size() const { return buf_.size(); }
  const uint8_t* Buffer() const { return buf_.data(); }
 
  Status WriteToFile(File* file);
 
 private:
  // Internal implementation of multi-byte write.
  template <typename T>
  void AppendInternal(T v);
 
  std::vector<uint8_t> buf_;
 
  DISALLOW_COPY_AND_ASSIGN(BufferWriter);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_BUFFER_WRITER_H_