mirror of
https://github.com/izzy2lost/Torch.git
synced 2026-03-26 17:00:59 -07:00
50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "lib/binarytools/endianness.h"
|
|
#include "lib/binarytools/Stream.h"
|
|
|
|
namespace LUS {
|
|
class BinaryWriter {
|
|
public:
|
|
BinaryWriter();
|
|
BinaryWriter(Stream* nStream);
|
|
BinaryWriter(std::shared_ptr<Stream> nStream);
|
|
|
|
void SetEndianness(Torch::Endianness endianness);
|
|
|
|
std::shared_ptr<Stream> GetStream();
|
|
uint64_t GetBaseAddress();
|
|
uint64_t GetLength();
|
|
void Seek(int32_t offset, SeekOffsetType seekType);
|
|
void Close();
|
|
|
|
void Write(int8_t value);
|
|
void Write(uint8_t value);
|
|
void Write(int16_t value);
|
|
void Write(uint16_t value);
|
|
void Write(int32_t value);
|
|
void Write(int32_t valueA, int32_t valueB);
|
|
void Write(uint32_t value);
|
|
void Write(int64_t value);
|
|
void Write(uint64_t value);
|
|
void Write(float value);
|
|
void Write(double value);
|
|
void Write(const std::string& str, bool writeLength = true);
|
|
void Write(char* srcBuffer, size_t length);
|
|
void WriteByte(char value);
|
|
|
|
std::vector<char> ToVector();
|
|
|
|
void Finish(std::ostream &output);
|
|
|
|
protected:
|
|
std::shared_ptr<Stream> mStream;
|
|
Torch::Endianness mEndianness = Torch::Endianness::Native;
|
|
|
|
};
|
|
} // namespace LUS
|