From 7ecf01874300ef8415154774db3ed9c51ffef4f3 Mon Sep 17 00:00:00 2001
From: Alexei Kotov <alexdobrohotov@yandex.ru>
Date: Wed, 8 Feb 2023 22:58:47 +0300
Subject: [PATCH] Add NiBlendInterpolator

---
 components/nif/controller.cpp | 93 +++++++++++++++++++++++++++++++++++
 components/nif/controller.hpp | 31 ++++++++++++
 2 files changed, 124 insertions(+)

diff --git a/components/nif/controller.cpp b/components/nif/controller.cpp
index 16d553cec8..47d1cbed8d 100644
--- a/components/nif/controller.cpp
+++ b/components/nif/controller.cpp
@@ -502,4 +502,97 @@ namespace Nif
         data.post(nif);
     }
 
+    void NiBlendInterpolator::read(NIFStream* nif)
+    {
+        if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 112))
+            mManagerControlled = nif->getChar() & 1;
+        size_t numInterps = 0;
+        if (nif->getVersion() <= NIFStream::generateVersion(10, 1, 0, 109))
+        {
+            numInterps = nif->getUShort();
+            mArrayGrowBy = nif->getUShort();
+        }
+        else
+        {
+            numInterps = nif->getChar();
+            if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 112))
+            {
+                mWeightThreshold = nif->getFloat();
+                if (!mManagerControlled)
+                {
+                    mInterpCount = nif->getChar();
+                    mSingleIndex = nif->getChar();
+                    mHighPriority = nif->getChar();
+                    mNextHighPriority = nif->getChar();
+                    mSingleTime = nif->getFloat();
+                    mHighWeightsSum = nif->getFloat();
+                    mNextHighWeightsSum = nif->getFloat();
+                    mHighEaseSpinner = nif->getFloat();
+                }
+            }
+        }
+
+        if (!mManagerControlled)
+        {
+            mItems.resize(numInterps);
+            for (Item& item : mItems)
+                item.read(nif);
+        }
+
+        if (nif->getVersion() <= NIFStream::generateVersion(10, 1, 0, 111))
+        {
+            mManagerControlled = nif->getBoolean();
+            mWeightThreshold = nif->getFloat();
+            mOnlyUseHighestWeight = nif->getBoolean();
+            if (nif->getVersion() <= NIFStream::generateVersion(10, 1, 0, 109))
+            {
+                mInterpCount = nif->getUShort();
+                mSingleIndex = nif->getUShort();
+            }
+            else
+            {
+                mInterpCount = nif->getChar();
+                mSingleIndex = nif->getChar();
+            }
+            if (nif->getVersion() >= NIFStream::generateVersion(10, 1, 0, 108))
+            {
+                mSingleInterpolator.read(nif);
+                mSingleTime = nif->getFloat();
+            }
+            if (nif->getVersion() <= NIFStream::generateVersion(10, 1, 0, 109))
+            {
+                mHighPriority = nif->getInt();
+                mNextHighPriority = nif->getInt();
+            }
+            else
+            {
+                mHighPriority = nif->getChar();
+                mNextHighPriority = nif->getChar();
+            }
+        }
+    }
+
+    void NiBlendInterpolator::post(Reader& nif)
+    {
+        for (Item& item : mItems)
+            item.post(nif);
+        mSingleInterpolator.post(nif);
+    }
+
+    void NiBlendInterpolator::Item::read(NIFStream* nif)
+    {
+        mInterpolator.read(nif);
+        mWeight = nif->getFloat();
+        mNormalizedWeight = nif->getFloat();
+        if (nif->getVersion() <= NIFStream::generateVersion(10, 1, 0, 109))
+            mPriority = nif->getInt();
+        else
+            mPriority = nif->getChar();
+        mEaseSpinner = nif->getFloat();
+    }
+
+    void NiBlendInterpolator::Item::post(Reader& nif)
+    {
+        mInterpolator.post(nif);
+    }
 }
diff --git a/components/nif/controller.hpp b/components/nif/controller.hpp
index a5213a9be7..d5b8fad4d3 100644
--- a/components/nif/controller.hpp
+++ b/components/nif/controller.hpp
@@ -345,5 +345,36 @@ namespace Nif
         void post(Reader& nif) override;
     };
 
+    // Abstract
+    struct NiBlendInterpolator : public NiInterpolator
+    {
+        struct Item
+        {
+            NiInterpolatorPtr mInterpolator;
+            float mWeight, mNormalizedWeight;
+            int mPriority;
+            float mEaseSpinner;
+
+            void read(NIFStream* nif);
+            void post(Reader& nif);
+        };
+
+        bool mManagerControlled{ false };
+        bool mOnlyUseHighestWeight{ false };
+        unsigned short mArrayGrowBy{ 0 };
+        float mWeightThreshold;
+        unsigned short mInterpCount;
+        unsigned short mSingleIndex;
+        int mHighPriority, mNextHighPriority;
+        float mSingleTime;
+        float mHighWeightsSum, mNextHighWeightsSum;
+        float mHighEaseSpinner;
+        std::vector<Item> mItems;
+        NiInterpolatorPtr mSingleInterpolator;
+
+        void read(NIFStream* nif) override;
+        void post(Reader& nif) override;
+    };
+
 } // Namespace
 #endif