1 files changed, 183 insertions, 0 deletions
diff --git a/libs/game_bethesda/src/games/morrowind/morrowindsavegame.cpp b/libs/game_bethesda/src/games/morrowind/morrowindsavegame.cpp
new file mode 100644
index 0000000..9588563
--- /dev/null
+++ b/libs/game_bethesda/src/games/morrowind/morrowindsavegame.cpp
@@ -0,0 +1,183 @@
+#include "morrowindsavegame.h"
+
+#include <QPixmap>
+#include <QRegularExpression>
+#include <filesystem>
+
+MorrowindSaveGame::MorrowindSaveGame(QString const& fileName, GameMorrowind const* game)
+    : GamebryoSaveGame(fileName, game)
+{
+  std::filesystem::path realFile(fileName.toStdWString());
+  QString realFileName = QString::fromStdWString(realFile.filename().wstring());
+  m_SaveNumber = realFileName.mid(4, 5).remove(QRegularExpression("0+$")).toInt();
+
+  FileWrapper file(fileName, "TES3");
+  QStringList dummyPlugins;
+  fetchInformationFields(file, m_SaveName, dummyPlugins, m_PCCurrentHealth,
+                         m_PCCMaxHealth, m_PCLocation, m_GameDays, m_PCName);
+}
+
+QString MorrowindSaveGame::getName() const
+{
+  return QString("%1, #%2, %3").arg(m_PCName).arg(m_SaveNumber).arg(m_PCLocation);
+}
+
+unsigned short MorrowindSaveGame::getPCLevel() const
+{
+  return dynamic_cast<MorrowindDataFields*>(m_DataFields.value().get())->PCLevel;
+}
+
+// Fetch easy-to-access information.
+void MorrowindSaveGame::fetchInformationFields(FileWrapper& file, QString& saveName,
+                                               QStringList& plugins,
+                                               float& playerCurrentHealth,
+                                               float& playerMaxHealth,
+                                               QString& playerLocation, float& gameDays,
+                                               QString& playerName) const
+{
+  file.skip<uint32_t>(3);       // data size
+  file.skip<unsigned char>(4);  // HEDR tag
+  file.skip<uint32_t>();        // header size
+  file.skip<float>();           // header version
+  file.skip<uint32_t>();  // following data chunk size? seems to be 9 groupings of 32
+                          // bytes
+  file.skip<unsigned char>(32);  // Author empty for save files
+
+  // The defined save name. This is technically the description, but is likely only
+  // 31+\0 chars max.
+  {
+    std::vector<char> saveNameBuffer(256);  // 31 char save name with a null terminator
+    file.read(saveNameBuffer.data(), 256);
+    saveName = QString::fromLatin1(saveNameBuffer.data(), -1).trimmed();
+  }
+
+  file.skip<uint32_t>();  // NumRecords (for the entire save)
+  std::vector<char> buffer(255);
+  file.read(buffer.data(), 4);
+  // Parse the MAST/DATA records
+  while (QString::fromLatin1(buffer.data(), 4) == "MAST") {
+    uint32_t len;
+    file.read(len);                 // Length of master name
+    file.read(buffer.data(), len);  // Name of master
+    QString name = QString::fromLatin1(buffer.data(), len - 1);
+    file.skip<unsigned char>(4);  // DATA record
+
+    // Typically size 8 - contains length of master data for version checking
+    file.read(len);  // Length
+    file.skip<unsigned char>(len);
+
+    file.read(buffer.data(), 4);  // Get next record type
+    plugins.push_back(name);
+  }
+
+  // Start of GMDT
+  file.skip<uint32_t>();  // size of record
+
+  file.read(playerCurrentHealth);
+  file.read(playerMaxHealth);
+
+  file.skip<double>();  // current stam?
+  file.skip<double>();  // max stam?
+  // file.skip<double>(2); // unknown values
+
+  std::fill(buffer.begin(), buffer.end(), '\0');
+  file.read(buffer.data(), 64);
+  playerLocation = QString::fromLatin1(buffer.data(), -1).trimmed();
+
+  file.read(gameDays);
+
+  std::fill(buffer.begin(), buffer.end(), '\0');
+  file.read(buffer.data(), 32);
+  playerName = QString::fromLatin1(buffer.data(), -1).trimmed();
+
+  // End of GMDT
+}
+
+std::unique_ptr<GamebryoSaveGame::DataFields> MorrowindSaveGame::fetchDataFields() const
+{
+  FileWrapper file(getFilepath(), "TES3");
+  std::vector<char> buffer(255);
+
+  std::unique_ptr<MorrowindDataFields> fields = std::make_unique<MorrowindDataFields>();
+
+  {
+    QString dummy;
+    float dummyF;
+    fetchInformationFields(file, dummy, fields->Plugins, dummyF, dummyF, dummy, dummyF,
+                           dummy);
+  }
+
+  file.skip<unsigned char>(28);  // Skip the SCRD
+  // I believe this tells the engine what color each pixel represents and the bitness of
+  // the image
+
+  // Start of screenshot
+  file.skip<unsigned char>(4);  // SCRS
+  file.skip<uint32_t>();        // Size of screenshot always 65536 (128x128x4) RGBA8888
+
+  QImage image       = readImageBGRA(file, 128, 128, 0, 1);
+  fields->Screenshot = image.scaled(252, 192);
+
+  // definitively have to use another method to access the player level
+  // it is stored in the fifth byte of the NPDT subrecord of the first NPC_ record
+
+  // Globals, Scripts, Regions
+  // file.skip<unsigned char>();
+  std::vector<char> buff(4);
+  file.read(buff.data(), 4);
+  while (QString::fromLatin1(buff.data(), 4) != "NPC_") {
+    uint32_t len;
+    file.read(len);
+    file.skip<unsigned char>(8 + len);
+    file.read(buff.data(), 4);
+  }
+  while (QString::fromLatin1(buff.data(), 4) == "NPC_") {
+    uint32_t size;
+    file.read(size);
+    file.skip<unsigned long>(3);
+    uint32_t len;
+    file.read(len);
+    file.read(buffer.data(), len);
+    if (QString::fromLatin1(buffer.data(), len - 1) == "player") {
+      file.read(buff.data(), 4);
+      while (QString::fromLatin1(buff.data(), 4) != "NPDT") {
+        uint32_t len;
+        file.read(len);
+        file.skip<unsigned char>(len);
+        file.read(buff.data(), 4);
+      }
+      file.skip<unsigned long>();
+      file.read(fields->PCLevel);
+    } else {
+      file.skip<unsigned char>(size - len - 8);
+    }
+  }
+
+  return fields;
+}
+
+QImage MorrowindSaveGame::readImageBGRA(GamebryoSaveGame::FileWrapper& file,
+                                        unsigned long width, unsigned long height,
+                                        int scale = 0, bool alpha = false) const
+{
+  QImage image(width, height, QImage::Format_RGBA8888);
+  for (unsigned long h = 0; h < width; h++) {
+    for (unsigned long w = 0; w < width; w++) {
+      uint8_t blue;
+      file.read(blue);
+      uint8_t green;
+      file.read(green);
+      uint8_t red;
+      file.read(red);
+      uint8_t alpha;
+      file.read(alpha);
+      alpha = 255 - alpha;
+      QColor color(red, green, blue, alpha);
+      image.setPixel(w, h, color.rgba());
+    }
+  }
+  if (scale != 0)
+    return image.copy().scaledToWidth(scale);
+  else
+    return image.copy();
+}