IFileStream: added GetPos and SetPos methods

Author: TheMostDiligent

Common/interface/BasicFileStream.hpp

@@ -64,6 +64,10 @@ class BasicFileStream : public ObjectBase<IFileStream>
 
     virtual size_t DILIGENT_CALL_TYPE GetSize() override;
 
+    virtual size_t DILIGENT_CALL_TYPE GetPos() override;
+
+    virtual bool DILIGENT_CALL_TYPE SetPos(size_t Offset, int Origin) override;
+
     virtual bool DILIGENT_CALL_TYPE IsValid() override;
 
 private:

Common/interface/MemoryFileStream.hpp

@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2023 Diligent Graphics LLC
+ *  Copyright 2019-2024 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -60,6 +60,10 @@ class MemoryFileStream final : public ObjectBase<IFileStream>
 
     virtual size_t DILIGENT_CALL_TYPE GetSize() override final;
 
+    virtual size_t DILIGENT_CALL_TYPE GetPos() override;
+
+    virtual bool DILIGENT_CALL_TYPE SetPos(size_t Offset, int Origin) override;
+
     virtual bool DILIGENT_CALL_TYPE IsValid() override final;
 
     static RefCntAutoPtr<MemoryFileStream> Create(IDataBlob* pData);

Common/src/BasicFileStream.cpp

@@ -77,4 +77,14 @@ size_t BasicFileStream::GetSize()
     return m_FileWrpr->GetSize();
 }
 
+size_t BasicFileStream::GetPos()
+{
+    return m_FileWrpr->GetPos();
+}
+
+bool BasicFileStream::SetPos(size_t Offset, int Origin)
+{
+    return m_FileWrpr->SetPos(Offset, static_cast<FilePosOrigin>(Origin));
+}
+
 } // namespace Diligent

Common/src/MemoryFileStream.cpp

@@ -89,4 +89,29 @@ size_t MemoryFileStream::GetSize()
     return m_DataBlob->GetSize();
 }
 
+size_t MemoryFileStream::GetPos()
+{
+    return m_CurrentOffset;
+}
+
+bool MemoryFileStream::SetPos(size_t Offset, int Origin)
+{
+    switch (static_cast<FilePosOrigin>(Origin))
+    {
+        case FilePosOrigin::Start:
+            m_CurrentOffset = Offset;
+            break;
+
+        case FilePosOrigin::Curr:
+            m_CurrentOffset += Offset;
+            break;
+
+        case FilePosOrigin::End:
+            m_CurrentOffset = m_DataBlob->GetSize() + Offset;
+            break;
+    }
+
+    return true;
+}
+
 } // namespace Diligent

Primitives/interface/FileStream.h

@@ -1,5 +1,5 @@
 /*
- *  Copyright 2019-2023 Diligent Graphics LLC
+ *  Copyright 2019-2024 Diligent Graphics LLC
  *  Copyright 2015-2019 Egor Yusov
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
@@ -67,6 +67,12 @@ DILIGENT_BEGIN_INTERFACE(IFileStream, IObject)
 
     VIRTUAL size_t METHOD(GetSize)(THIS) PURE;
 
+    VIRTUAL size_t METHOD(GetPos)(THIS) PURE;
+
+    VIRTUAL bool METHOD(SetPos)(THIS_
+                                size_t Offset,
+                                int    Origin) PURE;
+
     VIRTUAL bool METHOD(IsValid)(THIS) PURE;
 };
 DILIGENT_END_INTERFACE
@@ -81,6 +87,8 @@ DILIGENT_END_INTERFACE
 #    define IFileStream_ReadBlob(This, ...) CALL_IFACE_METHOD(FileStream, ReadBlob, This, __VA_ARGS__)
 #    define IFileStream_Write(This, ...)    CALL_IFACE_METHOD(FileStream, Write,    This, __VA_ARGS__)
 #    define IFileStream_GetSize(This)       CALL_IFACE_METHOD(FileStream, GetSize,  This)
+#    define IFileStream_GetPos(This)        CALL_IFACE_METHOD(FileStream, GetPos,   This)
+#    define IFileStream_SetPos(This, ...)   CALL_IFACE_METHOD(FileStream, SetPos,   This, __VA_ARGS__)
 #    define IFileStream_IsValid(This)       CALL_IFACE_METHOD(FileStream, IsValid,  This)
 
 // clang-format on