mirror of https://github.com/AxioDL/nod.git
60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
#include <cstdio>
|
|
#include "nod/Util.hpp"
|
|
#include "nod/IDiscIO.hpp"
|
|
#include "nod/IFileIO.hpp"
|
|
|
|
namespace nod {
|
|
|
|
class DiscIOISO : public IDiscIO {
|
|
std::unique_ptr<IFileIO> m_fio;
|
|
|
|
public:
|
|
DiscIOISO(SystemStringView fpin) : m_fio(NewFileIO(fpin)) {}
|
|
|
|
class ReadStream : public IReadStream {
|
|
friend class DiscIOISO;
|
|
std::unique_ptr<IFileIO::IReadStream> fp;
|
|
ReadStream(std::unique_ptr<IFileIO::IReadStream>&& fpin, bool& err) : fp(std::move(fpin)) {
|
|
if (!fp)
|
|
err = true;
|
|
}
|
|
|
|
public:
|
|
uint64_t read(void* buf, uint64_t length) { return fp->read(buf, length); }
|
|
uint64_t position() const { return fp->position(); }
|
|
void seek(int64_t offset, int whence) { fp->seek(offset, whence); }
|
|
};
|
|
|
|
std::unique_ptr<IReadStream> beginReadStream(uint64_t offset) const {
|
|
bool Err = false;
|
|
auto ret = std::unique_ptr<IReadStream>(new ReadStream(m_fio->beginReadStream(offset), Err));
|
|
if (Err)
|
|
return {};
|
|
return ret;
|
|
}
|
|
|
|
class WriteStream : public IWriteStream {
|
|
friend class DiscIOISO;
|
|
std::unique_ptr<IFileIO::IWriteStream> fp;
|
|
WriteStream(std::unique_ptr<IFileIO::IWriteStream>&& fpin, bool& err) : fp(std::move(fpin)) {
|
|
if (!fp)
|
|
err = true;
|
|
}
|
|
|
|
public:
|
|
uint64_t write(const void* buf, uint64_t length) { return fp->write(buf, length); }
|
|
};
|
|
|
|
std::unique_ptr<IWriteStream> beginWriteStream(uint64_t offset) const {
|
|
bool Err = false;
|
|
auto ret = std::unique_ptr<IWriteStream>(new WriteStream(m_fio->beginWriteStream(offset), Err));
|
|
if (Err)
|
|
return {};
|
|
return ret;
|
|
}
|
|
};
|
|
|
|
std::unique_ptr<IDiscIO> NewDiscIOISO(SystemStringView path) { return std::unique_ptr<IDiscIO>(new DiscIOISO(path)); }
|
|
|
|
} // namespace nod
|