PackUnpackImpl.h

Go to the documentation of this file.

#pragma once
 
#include "RE/A/Array.h"
#include "RE/B/BSTSmartPointer.h"
#include "RE/P/PackUnpack.h"
#include "RE/T/TypeInfo.h"
#include "RE/V/Variable.h"
#include "RE/V/VirtualMachine.h"
 
// DO NOT INCLUDE EXCEPT IN MEGA HEADER!!!
 
namespace RE
{
    namespace BSScript
    {
        // T requires:
        //  * begin()
        //  * end()
        //  * size()
        //  * value_type
        //  * input iterator:
        //      * weakly incrementable
        //      * indirectly readable
        template <
            class T,
            class U,
            std::enable_if_t<
                is_array_v<U>,
                int>>
        void PackValue(Variable* a_dst, T&& a_src)
        {
            assert(a_dst);
            a_dst->SetNone();
 
            auto vm = Internal::VirtualMachine::GetSingleton();
            if (!vm) {
                assert(false);
                return;
            }
 
            BSTSmartPointer<Array> array;
            TypeInfo               typeInfo(GetRawType<typename U::value_type>());
            if (!vm->CreateArray(typeInfo, static_cast<std::uint32_t>(a_src.size()), array) || !array) {
                assert(false);
                return;
            }
 
            auto          it = a_src.begin();
            auto          end = a_src.end();
            std::uint32_t i = 0;
            while (it != end) {
                if constexpr (std::is_same_v<U, std::vector<bool>>) {
                    (*array)[i++].Pack(static_cast<bool>(*it));
                } else {
                    (*array)[i++].Pack(*it);
                }
                ++it;
            }
 
            a_dst->SetArray(std::move(array));
        }
 
        // T requires:
        //  * default constructible
        //  * destructible
        //  * value_type
        //  * push_back(value_type)
        template <
            class T,
            std::enable_if_t<
                is_array_v<T>,
                int>>
        [[nodiscard]] T UnpackValue(const Variable* a_src)
        {
            assert(a_src);
 
            std::remove_const_t<T> container;
            if (a_src->IsNoneObject()) {
                return container;
            }
 
            auto array = a_src->GetArray();
            if (!array) {
                return container;
            }
 
            for (auto& elem : *array) {
                container.push_back(elem.Unpack<typename T::value_type>());
            }
 
            return container;
        }
    }
}