#include "PhysPresetInfoStringDumperT5.h"

#include "Game/T5/InfoString/InfoStringFromStructConverter.h"
#include "Game/T5/ObjConstantsT5.h"
#include "Game/T5/PhysPreset/PhysPresetFields.h"
#include "PhysPreset/PhysPresetCommon.h"

#include <algorithm>
#include <cassert>
#include <cmath>
#include <type_traits>

using namespace T5;

namespace
{
    class InfoStringFromPhysPresetConverter final : public InfoStringFromStructConverter
    {
    protected:
        void FillFromExtensionField(const cspField_t& field) override
        {
            assert(false);
        }

    public:
        InfoStringFromPhysPresetConverter(const PhysPresetInfo* structure,
                                          const cspField_t* fields,
                                          const size_t fieldCount,
                                          std::function<std::string(scr_string_t)> scriptStringValueCallback)
            : InfoStringFromStructConverter(structure, fields, fieldCount, std::move(scriptStringValueCallback))
        {
        }
    };

    void CopyToPhysPresetInfo(const PhysPreset* physPreset, PhysPresetInfo* physPresetInfo)
    {
        physPresetInfo->mass = std::clamp(physPreset->mass * 1000.0f, 1.0f, 2000.0f);
        physPresetInfo->bounce = physPreset->bounce;

        if (physPreset->friction >= MAX_FRICTION)
        {
            physPresetInfo->isFrictionInfinity = 1;
            physPresetInfo->friction = 0;
        }
        else
        {
            physPresetInfo->isFrictionInfinity = 0;
            physPresetInfo->friction = physPreset->friction;
        }

        physPresetInfo->bulletForceScale = physPreset->bulletForceScale;
        physPresetInfo->explosiveForceScale = physPreset->explosiveForceScale;
        physPresetInfo->piecesSpreadFraction = physPreset->piecesSpreadFraction;
        physPresetInfo->piecesUpwardVelocity = physPreset->piecesUpwardVelocity;
        physPresetInfo->canFloat = physPreset->canFloat;
        physPresetInfo->gravityScale = std::clamp(physPreset->gravityScale, 0.01f, 10.0f);
        physPresetInfo->centerOfMassOffset = physPreset->centerOfMassOffset;
        physPresetInfo->buoyancyBoxMin = physPreset->buoyancyBoxMin;
        physPresetInfo->buoyancyBoxMax = physPreset->buoyancyBoxMax;
    }

    InfoString CreateInfoString(const XAssetInfo<PhysPreset>& asset)
    {
        auto* physPresetInfo = new PhysPresetInfo;
        CopyToPhysPresetInfo(asset.Asset(), physPresetInfo);

        InfoStringFromPhysPresetConverter converter(physPresetInfo,
                                                    phys_preset_fields,
                                                    std::extent_v<decltype(phys_preset_fields)>,
                                                    [asset](const scr_string_t scrStr) -> std::string
                                                    {
                                                        assert(scrStr < asset.m_zone->m_script_strings.Count());
                                                        if (scrStr >= asset.m_zone->m_script_strings.Count())
                                                            return "";

                                                        return asset.m_zone->m_script_strings[scrStr];
                                                    });

        return converter.Convert();
    }
} // namespace

namespace phys_preset
{
    void InfoStringDumperT5::DumpAsset(AssetDumpingContext& context, const XAssetInfo<AssetPhysPreset::Type>& asset)
    {
        // Only dump raw when no gdt available
        if (context.m_gdt)
        {
            const auto infoString = CreateInfoString(asset);
            GdtEntry gdtEntry(asset.m_name, GDF_FILENAME_PHYS_PRESET);
            infoString.ToGdtProperties(INFO_STRING_PREFIX_PHYS_PRESET, gdtEntry);
            context.m_gdt->WriteEntry(gdtEntry);
        }
        else
        {
            const auto assetFile = context.OpenAssetFile(GetFileNameForAssetName(asset.m_name));

            if (!assetFile)
                return;

            auto& stream = *assetFile;
            const auto infoString = CreateInfoString(asset);
            const auto stringValue = infoString.ToString(INFO_STRING_PREFIX_PHYS_PRESET);
            stream.write(stringValue.c_str(), stringValue.size());
        }
    }
} // namespace phys_preset