filedevice: Implement FileDeviceStreamSrc

CMakeLists.txt

@@ -66,11 +66,13 @@ add_library(sead OBJECT
   include/filedevice/seadArchiveFileDevice.h
   include/filedevice/seadFileDevice.h
   include/filedevice/seadFileDeviceMgr.h
+  include/filedevice/seadFileDeviceStreamSrc.h
   include/filedevice/seadMainFileDevice.h
   include/filedevice/seadPath.h
   modules/src/filedevice/seadArchiveFileDevice.cpp
   modules/src/filedevice/seadFileDevice.cpp
   modules/src/filedevice/seadFileDeviceMgr.cpp
+  modules/src/filedevice/seadFileDeviceStreamSrc.cpp
   modules/src/filedevice/seadMainFileDevice.cpp
   modules/src/filedevice/seadPath.cpp
 

include/filedevice/seadFileDeviceStreamSrc.h

@@ -0,0 +1,31 @@
+#pragma once
+
+#include <basis/seadTypes.h>
+#include <filedevice/seadFileDevice.h>
+#include <stream/seadStreamSrc.h>
+
+namespace sead
+{
+class FileDeviceStreamSrc : public StreamSrc
+{
+public:
+    FileDeviceStreamSrc();
+    FileDeviceStreamSrc(FileHandle* fileHandle);
+    ~FileDeviceStreamSrc() override;
+
+    u32 read(void* buffer, u32 size) override;
+    u32 write(const void* buffer, u32 size) override;
+    u32 skip(s32 offset) override;
+    void rewind() override;
+    bool isEOF() override;
+
+    void setFileHandle(sead::FileHandle* fileHandle);
+
+private:
+    FileHandle* mFileHandle = nullptr;
+    u32 mStartingPos = 0;
+    FileHandle _18;
+    bool mIsHandleOpen = false;
+    u32 mFileSize = 0;
+};
+}  // namespace sead

modules/src/filedevice/seadFileDeviceStreamSrc.cpp

@@ -0,0 +1,56 @@
+#include <filedevice/seadFileDeviceStreamSrc.h>
+
+namespace sead
+{
+FileDeviceStreamSrc::FileDeviceStreamSrc() = default;
+
+FileDeviceStreamSrc::FileDeviceStreamSrc(FileHandle* fileHandle)
+    : mFileHandle(fileHandle), mStartingPos(fileHandle->getCurrentSeekPos()),
+      mFileSize(fileHandle->getFileSize())
+{
+}
+
+FileDeviceStreamSrc::~FileDeviceStreamSrc()
+{
+    if (mIsHandleOpen && mFileHandle)
+        mFileHandle->close();
+}
+
+u32 FileDeviceStreamSrc::read(void* buffer, u32 size)
+{
+    return mFileHandle->read(static_cast<u8*>(buffer), size);
+}
+
+u32 FileDeviceStreamSrc::write(const void* buffer, u32 size)
+{
+    return mFileHandle->write(static_cast<const u8*>(buffer), size);
+}
+
+u32 FileDeviceStreamSrc::skip(s32 offset)
+{
+    if (!mFileHandle->seek(offset, FileDevice::cSeekOrigin_Current))
+        return 0;
+    return offset;
+}
+
+void FileDeviceStreamSrc::rewind()
+{
+    mFileHandle->seek(mStartingPos, FileDevice::cSeekOrigin_Begin);
+}
+
+bool FileDeviceStreamSrc::isEOF()
+{
+    return mFileHandle->getCurrentSeekPos() >= mFileSize;
+}
+
+void FileDeviceStreamSrc::setFileHandle(sead::FileHandle* fileHandle)
+{
+    mFileHandle = fileHandle;
+    if (fileHandle)
+    {
+        mStartingPos = fileHandle->getCurrentSeekPos();
+        mFileSize = fileHandle->getFileSize();
+    }
+}
+
+}  // namespace sead