Files
Torch/lib/binarytools/MemoryStream.h
2023-08-17 00:35:58 -06:00

35 lines
774 B
C++

#pragma once
#include <memory>
#include <vector>
#include "Stream.h"
namespace LUS {
class MemoryStream : public Stream {
public:
MemoryStream();
MemoryStream(char* nBuffer, size_t nBufferSize);
~MemoryStream();
uint64_t GetLength() override;
void Seek(int32_t offset, SeekOffsetType seekType) override;
std::unique_ptr<char[]> Read(size_t length) override;
void Read(const char* dest, size_t length) override;
int8_t ReadByte() override;
void Write(char* srcBuffer, size_t length) override;
void WriteByte(int8_t value) override;
std::vector<char> ToVector() override;
void Flush() override;
void Close() override;
protected:
std::vector<char> mBuffer;
std::size_t mBufferSize;
};
} // namespace LUS