feat: chunk sorting and parsing on REFR records

tiltedphoques · Jan 13, 2022 · 8c196f5 · 8c196f5
1 parent 173e3ee
commit 8c196f5
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/Code/es_loader/ESLoader.cpp b/Code/es_loader/ESLoader.cpp
@@ -1,6 +1,9 @@
 #include "ESLoader.h"
 
 #include <filesystem>
+
+#include <Records/REFR.h>
+
 namespace fs = std::filesystem;
 
 ESLoader::ESLoader(String aDirectory) 
@@ -64,7 +67,20 @@ void ESLoader::LoadFiles()
 {
     for (const auto& filename : m_esmFilenames)
     {
-        m_standardPlugins.push_back(TESFile(filename));
+        if (filename.filename().string() != "Skyrim.esm")
+            continue;
+
+        //m_standardPlugins.push_back(TESFile(filename));
+        TESFile skyrimEsm(filename);
+
+        const Vector<REFR*>& refs = skyrimEsm.GetObjectReferences();
+
+        Vector<REFR::Data> refrData;
+        for (REFR* ref : refs)
+        {
+            refrData.push_back(ref->ParseChunks());
+        }
+        spdlog::info("refrData count: {}", refrData.size());
     }
 }
 

diff --git a/Code/es_loader/TESFile.cpp b/Code/es_loader/TESFile.cpp
@@ -4,6 +4,7 @@
 #include <fstream>
 
 #include "Records/Record.h"
+#include "Records/REFR.h"
 
 TESFile::TESFile(const std::filesystem::path& acPath)
 {
@@ -66,8 +67,9 @@ bool TESFile::ReadGroupOrRecord(Buffer::Reader& aReader) noexcept
         switch (record->GetType())
         {
         case FormEnum::REFR:
-        case FormEnum::ACHR:
-            m_objectReferences.push_back(record);
+        //case FormEnum::ACHR:
+            REFR* recordREFR = reinterpret_cast<REFR*>(record);
+            m_objectReferences[record->GetFormId()] = recordREFR;
             break;
         case FormEnum::CELL:
             m_cells.push_back(record);

diff --git a/Code/es_loader/TESFile.h b/Code/es_loader/TESFile.h
@@ -3,6 +3,7 @@
 #include <Records/Group.h>
 
 class Record;
+class REFR;
 
 class TESFile
 {
@@ -25,6 +26,11 @@ class TESFile
 
     template<class T> Vector<T> GetRecords() noexcept;
 
+    const Vector<REFR*>& GetObjectReferences() const noexcept
+    {
+        return m_objectReferences;
+    }
+
 private:
     void BuildFormIdRecordMap() noexcept;
     bool ReadGroupOrRecord(Buffer::Reader& aReader) noexcept;
@@ -40,6 +46,6 @@ class TESFile
         uint16_t m_liteId;
     };
 
-    Vector<Record*> m_objectReferences;
+    Map<uint32_t, REFR*> m_objectReferences;
     Vector<Record*> m_cells;
 };