buffer_writer.h

// Copyright 2014 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 PACKAGER_MEDIA_BASE_BUFFER_WRITER_H_
#define PACKAGER_MEDIA_BASE_BUFFER_WRITER_H_

#include <vector>

#include "packager/base/macros.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_