Add AssetLoader for attachments

src/ObjLoading/Game/T6/AssetLoaders/AssetLoaderWeaponAttachment.cpp

@@ -0,0 +1,138 @@
+#include "AssetLoaderWeaponAttachment.h"
+
+#include <cstring>
+#include <iostream>
+#include <cassert>
+
+#include "Game/T6/ObjConstantsT6.h"
+#include "Game/T6/T6.h"
+#include "Game/T6/InfoString/EnumStrings.h"
+#include "Game/T6/InfoString/InfoStringToStructConverter.h"
+#include "Game/T6/InfoString/WeaponAttachmentFields.h"
+#include "InfoString/InfoString.h"
+
+using namespace T6;
+
+namespace T6
+{
+    eAttachmentPoint attachmentPointByAttachmentTable[]
+    {
+        ATTACHMENT_POINT_NONE,      // none
+        ATTACHMENT_POINT_TOP,       // acog
+        ATTACHMENT_POINT_TRIGGER,   // dualclip
+        ATTACHMENT_POINT_TOP,       // dualoptic
+        ATTACHMENT_POINT_BOTTOM,    // dw
+        ATTACHMENT_POINT_MUZZLE,    // extbarrel
+        ATTACHMENT_POINT_TRIGGER,   // extclip
+        ATTACHMENT_POINT_TRIGGER,   // extramags
+        ATTACHMENT_POINT_GUNPERK,   // fastads
+        ATTACHMENT_POINT_TOP,       // fastreload
+        ATTACHMENT_POINT_TRIGGER,   // fmj
+        ATTACHMENT_POINT_BOTTOM,    // gl
+        ATTACHMENT_POINT_BOTTOM,    // grip
+        ATTACHMENT_POINT_TOP,       // holo
+        ATTACHMENT_POINT_BOTTOM,    // ir
+        ATTACHMENT_POINT_BOTTOM,    // is
+        ATTACHMENT_POINT_GUNPERK,   // longbreath
+        ATTACHMENT_POINT_BOTTOM,    // mk
+        ATTACHMENT_POINT_TOP,       // mms
+        ATTACHMENT_POINT_TOP,       // rangefinder
+        ATTACHMENT_POINT_TOP,       // reflex
+        ATTACHMENT_POINT_MUZZLE,    // rf
+        ATTACHMENT_POINT_BOTTOM,    // sf
+        ATTACHMENT_POINT_MUZZLE,    // silencer
+        ATTACHMENT_POINT_TRIGGER,   // stackfire
+        ATTACHMENT_POINT_GUNPERK,   // stalker
+        ATTACHMENT_POINT_GUNPERK,   // steadyaim
+        ATTACHMENT_POINT_GUNPERK,   // swayreduc
+        ATTACHMENT_POINT_TOP,       // tacknife
+        ATTACHMENT_POINT_TOP,       // vzoom
+    };
+
+    static_assert(std::extent<decltype(attachmentPointByAttachmentTable)>::value == ATTACHMENT_TYPE_COUNT);
+
+    class InfoStringToWeaponAttachmentConverter final : public InfoStringToStructConverter
+    {
+    protected:
+        bool ConvertExtensionField(const cspField_t& field, const std::string& value) override
+        {
+            switch (static_cast<attachmentFieldType_t>(field.iFieldType))
+            {
+            case AFT_ATTACHMENTTYPE:
+                return ConvertEnumInt(value, field.iOffset, szAttachmentTypeNames, std::extent<decltype(szAttachmentTypeNames)>::value);
+
+            case AFT_PENETRATE_TYPE:
+                return ConvertEnumInt(value, field.iOffset, penetrateTypeNames, std::extent<decltype(penetrateTypeNames)>::value);
+
+            case AFT_FIRETYPE:
+                return ConvertEnumInt(value, field.iOffset, szWeapFireTypeNames, std::extent<decltype(szWeapFireTypeNames)>::value);
+
+            default:
+                assert(false);
+                return false;
+            }
+        }
+
+    public:
+        InfoStringToWeaponAttachmentConverter(const InfoString& infoString, WeaponAttachment* weaponAttachment, ZoneScriptStrings& zoneScriptStrings, MemoryManager* memory, IAssetLoadingManager* manager,
+            const cspField_t* fields, const size_t fieldCount)
+            : InfoStringToStructConverter(infoString, weaponAttachment, zoneScriptStrings, memory, manager, fields, fieldCount)
+        {
+        }
+    };
+}
+
+void AssetLoaderWeaponAttachment::CalculateAttachmentFields(WeaponAttachment* attachment)
+{
+    // attachmentPoint
+    if(static_cast<unsigned>(attachment->attachmentType) < ATTACHMENT_TYPE_COUNT)
+    {
+        attachment->attachmentPoint = attachmentPointByAttachmentTable[attachment->attachmentType];
+    }
+}
+
+void* AssetLoaderWeaponAttachment::CreateEmptyAsset(const std::string& assetName, MemoryManager* memory)
+{
+    auto* attachment = memory->Create<WeaponAttachment>();
+    memset(attachment, 0, sizeof(WeaponAttachment));
+    CalculateAttachmentFields(attachment);
+    attachment->szInternalName = memory->Dup(assetName.c_str());
+    return attachment;
+}
+
+bool AssetLoaderWeaponAttachment::CanLoadFromRaw() const
+{
+    return true;
+}
+
+bool AssetLoaderWeaponAttachment::LoadFromRaw(const std::string& assetName, ISearchPath* searchPath, MemoryManager* memory, IAssetLoadingManager* manager, Zone* zone) const
+{
+    const auto fileName = "attachment/" + assetName;
+    const auto file = searchPath->Open(fileName);
+    if (!file.IsOpen())
+        return false;
+
+    InfoString infoString;
+    if (!infoString.FromStream(ObjConstants::INFO_STRING_PREFIX_WEAPON_ATTACHMENT, *file.m_stream))
+    {
+        std::cout << "Failed to read attachment raw file: \"" << fileName << "\"" << std::endl;
+        return true;
+    }
+
+    auto* attachment = memory->Create<WeaponAttachment>();
+    memset(attachment, 0, sizeof(WeaponAttachment));
+
+    InfoStringToWeaponAttachmentConverter converter(infoString, attachment, zone->m_script_strings, memory, manager, attachment_fields, std::extent<decltype(attachment_fields)>::value);
+    if (!converter.Convert())
+    {
+        std::cout << "Failed to parse attachment raw file: \"" << fileName << "\"" << std::endl;
+        return true;
+    }
+
+    CalculateAttachmentFields(attachment);
+    attachment->szInternalName = memory->Dup(assetName.c_str());
+
+    manager->AddAsset(ASSET_TYPE_ATTACHMENT, assetName, attachment, converter.GetDependencies(), converter.GetUsedScriptStrings());
+
+    return true;
+}