diff options
Diffstat (limited to 'lib/binarytools')
| -rw-r--r-- | lib/binarytools/BinaryReader.cpp | 4 | ||||
| -rw-r--r-- | lib/binarytools/BinaryReader.h | 1 | ||||
| -rw-r--r-- | lib/binarytools/BinaryWriter.cpp | 6 | ||||
| -rw-r--r-- | lib/binarytools/BinaryWriter.h | 5 |
4 files changed, 15 insertions, 1 deletions
diff --git a/lib/binarytools/BinaryReader.cpp b/lib/binarytools/BinaryReader.cpp index f802885..4afeb17 100644 --- a/lib/binarytools/BinaryReader.cpp +++ b/lib/binarytools/BinaryReader.cpp @@ -8,6 +8,10 @@ LUS::BinaryReader::BinaryReader(char* nBuffer, size_t nBufferSize) { mStream = std::make_shared<MemoryStream>(nBuffer, nBufferSize); } +LUS::BinaryReader::BinaryReader(uint8_t* nBuffer, size_t nBufferSize) { + mStream = std::make_shared<MemoryStream>((char*) nBuffer, nBufferSize); +} + LUS::BinaryReader::BinaryReader(Stream* nStream) { mStream.reset(nStream); } diff --git a/lib/binarytools/BinaryReader.h b/lib/binarytools/BinaryReader.h index 02e65cc..a98742f 100644 --- a/lib/binarytools/BinaryReader.h +++ b/lib/binarytools/BinaryReader.h @@ -13,6 +13,7 @@ namespace LUS { class BinaryReader { public: BinaryReader(char* nBuffer, size_t nBufferSize); + BinaryReader(uint8_t* nBuffer, size_t nBufferSize); BinaryReader(Stream* nStream); BinaryReader(std::shared_ptr<Stream> nStream); diff --git a/lib/binarytools/BinaryWriter.cpp b/lib/binarytools/BinaryWriter.cpp index 6e0140f..4d48a0e 100644 --- a/lib/binarytools/BinaryWriter.cpp +++ b/lib/binarytools/BinaryWriter.cpp @@ -1,5 +1,6 @@ #include "BinaryWriter.h" #include "MemoryStream.h" +#include <iostream> LUS::BinaryWriter::BinaryWriter() { mStream = std::make_shared<MemoryStream>(); @@ -153,3 +154,8 @@ void LUS::BinaryWriter::WriteByte(char value) { std::vector<char> LUS::BinaryWriter::ToVector() { return mStream->ToVector(); } + +void LUS::BinaryWriter::Finish(std::ostream& output) { + auto data = mStream->ToVector(); + output.write(data.data(), data.size()); +}
\ No newline at end of file diff --git a/lib/binarytools/BinaryWriter.h b/lib/binarytools/BinaryWriter.h index dca683f..28fdd09 100644 --- a/lib/binarytools/BinaryWriter.h +++ b/lib/binarytools/BinaryWriter.h @@ -39,8 +39,11 @@ class BinaryWriter { std::vector<char> ToVector(); - protected: + void Finish(std::ostream &output); + +protected: std::shared_ptr<Stream> mStream; Endianness mEndianness = Endianness::Native; + }; } // namespace LUS |
