#include "statstate.hpp"

#include "esmreader.hpp"
#include "esmwriter.hpp"

namespace ESM
{
    template<typename T>
    StatState<T>::StatState() : mBase(0), mMod(0), mCurrent(0), mDamage(0), mProgress(0) {}

    template<typename T>
    void StatState<T>::load(ESMReader &esm, bool intFallback)
    {
        // We changed stats values from integers to floats; ensure backwards compatibility
        if (intFallback)
        {
            int base = 0;
            esm.getHNT(base, "STBA");
            mBase = static_cast<float>(base);

            int mod = 0;
            esm.getHNOT(mod, "STMO");
            mMod = static_cast<float>(mod);

            int current = 0;
            esm.getHNOT(current, "STCU");
            mCurrent = static_cast<float>(current);

            // mDamage was changed to a float; ensure backwards compatibility
            int oldDamage = 0;
            esm.getHNOT(oldDamage, "STDA");
            mDamage = static_cast<float>(oldDamage);
        }
        else
        {
            mBase = 0;
            esm.getHNT(mBase, "STBA");

            mMod = 0;
            esm.getHNOT(mMod, "STMO");

            mCurrent = 0;
            esm.getHNOT(mCurrent, "STCU");

            mDamage = 0;
            esm.getHNOT(mDamage, "STDF");

            mProgress = 0;
        }

        esm.getHNOT(mDamage, "STDF");

        mProgress = 0;
        esm.getHNOT(mProgress, "STPR");
    }

    template<typename T>
    void StatState<T>::save(ESMWriter &esm) const
    {
        esm.writeHNT("STBA", mBase);

        if (mMod != 0)
            esm.writeHNT("STMO", mMod);

        if (mCurrent)
            esm.writeHNT("STCU", mCurrent);

        if (mDamage)
            esm.writeHNT("STDF", mDamage);

        if (mProgress)
            esm.writeHNT("STPR", mProgress);
    }
}

template struct ESM::StatState<int>;
template struct ESM::StatState<float>;