81 lines
2.4 KiB
C++
81 lines
2.4 KiB
C++
// 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 MEDIA_BASE_BUFFER_WRITER_H_
|
|
#define MEDIA_BASE_BUFFER_WRITER_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "packager/base/macros.h"
|
|
#include "packager/media/base/status.h"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
class File;
|
|
|
|
/// A simple buffer writer implementation which appends various data types to
|
|
/// buffer.
|
|
class BufferWriter {
|
|
public:
|
|
BufferWriter();
|
|
/// Construct the object with a reserved capacity.
|
|
/// @param reserved_size_in_bytes is intended for optimization and is
|
|
/// not a hard limit. It does not affect the actual size of the buffer,
|
|
/// which still starts from zero.
|
|
explicit BufferWriter(size_t reserved_size_in_bytes);
|
|
~BufferWriter();
|
|
|
|
/// These convenience functions append the integers (in network byte order,
|
|
/// i.e. big endian) of various size and signedness to the end of the buffer.
|
|
/// @{
|
|
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);
|
|
/// @}
|
|
|
|
/// Append the least significant @a num_bytes of @a v to buffer.
|
|
/// @param num_bytes should not be larger than sizeof(@a v), i.e. 8 on a
|
|
/// 64-bit system.
|
|
void AppendNBytes(uint64_t v, size_t num_bytes);
|
|
|
|
void AppendVector(const std::vector<uint8_t>& v);
|
|
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(); }
|
|
/// @return Underlying buffer. Behavior is undefined if the buffer size is 0.
|
|
const uint8_t* Buffer() const { return buf_.data(); }
|
|
|
|
/// Write the buffer to file. The internal buffer will be cleared after
|
|
/// writing.
|
|
/// @param file should not be NULL.
|
|
/// @return OK on success.
|
|
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 // MEDIA_BASE_BUFFER_WRITER_H_
|