Merge pull request #212 from Hoikas/performance

Improve PRP read performance.
H-uru · Jun 7, 2021 · e435db7 · e435db7
2 parents 3460b08 + bf26756
commit e435db7
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 13 deletions.
diff --git a/core/ResManager/plKeyCollector.cpp b/core/ResManager/plKeyCollector.cpp
@@ -44,7 +44,7 @@ plKeyCollector::~plKeyCollector()
 plKey plKeyCollector::findKey(const plKey& match)
 {
     plKey key;
-    std::vector<plKey> kList = getKeys(match->getLocation(), match->getType());
+    const auto& kList = keys[match->getLocation()][match->getType()];
     for (unsigned int i=0; i < kList.size(); i++) {
         if (*kList[i] == *match) {
             key = kList[i];

diff --git a/core/Stream/pfSizedStream.cpp b/core/Stream/pfSizedStream.cpp
@@ -18,7 +18,7 @@
 #include "Debug/plDebug.h"
 
 pfSizedStream::pfSizedStream(hsStream* S, uint32_t len)
-    : fBase(S), fLength(len)
+    : fBase(S), fLength(len), fPos()
 {
     if (S) {
         ver = S->getVer();
@@ -41,37 +41,43 @@ void pfSizedStream::seek(uint32_t pos)
     }
 
     fBase->seek(fBegin + pos);
+    fPos = pos;
 }
 
 void pfSizedStream::skip(int32_t count)
 {
-    if (pos() + count > fLength) { // pos() is the index in the sub-stream
+    if (fPos + count > fLength) { // fPos is the index in the sub-stream
         throw hsFileReadException(__FILE__, __LINE__,
                          ST::format("Seek out of range: {} bytes requested, {} available",
-                         count, (fLength - pos())).c_str());
+                         count, (fLength - fPos)).c_str());
     }
 
     fBase->skip(count);
+    fPos += count;
 }
 
 size_t pfSizedStream::read(size_t size, void* buf)
 {
-    if (pos() + size > fLength) { // pos() is the index in the sub-stream
+    if (fPos + size > fLength) { // fPos is the index in the sub-stream
         throw hsFileReadException(__FILE__, __LINE__,
                          ST::format("Read past end of sized stream: {} bytes requested, {} available",
-                         size, (fLength - pos())).c_str());
+                         size, (fLength - fPos)).c_str());
     }
 
-    return fBase->read(size, buf);
+    size_t nread = fBase->read(size, buf);
+    fPos += nread;
+    return nread;
 }
 
 size_t pfSizedStream::write(size_t size, const void* buf)
 {
-    if (pos() + size > fLength) { // pos() is the index in the sub-stream
+    if (fPos + size > fLength) { // fPos is the index in the sub-stream
         throw hsFileReadException(__FILE__, __LINE__,
                          ST::format("Write past end of sized stream: {} bytes requested, {} available",
-                         size, (fLength - pos())).c_str());
+                         size, (fLength - fPos)).c_str());
     }
 
-    return fBase->write(size, buf);
+    size_t nwrite = fBase->write(size, buf);
+    fPos += nwrite;
+    return nwrite;
 }
diff --git a/core/Stream/pfSizedStream.h b/core/Stream/pfSizedStream.h
@@ -19,20 +19,21 @@
 
 #include "hsStream.h"
 
-class PLASMA_DLL pfSizedStream : public hsStream
+class PLASMA_DLL pfSizedStream HS_FINAL : public hsStream
 {
 private:
     hsStream* fBase;
     uint32_t fLength; //!< the length of the substream - not the end position of it!
     uint32_t fBegin;
+    uint32_t fPos;
 
 public:
     pfSizedStream(hsStream* S, uint32_t len);
     ~pfSizedStream() { } // Do NOT free fBase!!!
 
     uint32_t size() const HS_OVERRIDE { return fLength; }
-    uint32_t pos() const HS_OVERRIDE { return fBase->pos() - fBegin; }
-    bool eof() const HS_OVERRIDE { return fBase->eof() || pos() == fLength; }
+    uint32_t pos() const HS_OVERRIDE { return fPos; }
+    bool eof() const HS_OVERRIDE { return fPos == fLength || fBase->eof(); }
 
     void seek(uint32_t pos) HS_OVERRIDE;
     void skip(int32_t count) HS_OVERRIDE;