#ifndef STORM_ARRAY_TS_FIXED_ARRAY_HPP
#define STORM_ARRAY_TS_FIXED_ARRAY_HPP

#include "storm/Memory.hpp"
#include "storm/array/TSBaseArray.hpp"
#include <cstdint>

template <class T>
class TSFixedArray : public TSBaseArray<T> {
    public:
    ~TSFixedArray();
    TSFixedArray<T>& operator=(const TSFixedArray<T>& source);
    void ReallocAndClearData(uint32_t count);
    void ReallocData(uint32_t count);
    void Set(uint32_t count, const T* data);
    void SetCount(uint32_t count);
};

template <class T>
TSFixedArray<T>::~TSFixedArray() {
    for (uint32_t i = 0; i < this->Count(); i++) {
        auto element = &this->operator[](i);
        element->~T();
    }

    if (this->Ptr()) {
        SMemFree(this->Ptr(), this->MemFileName(), this->MemLineNo(), 0x0);
    }
}

template <class T>
TSFixedArray<T>& TSFixedArray<T>::operator=(const TSFixedArray<T>& source) {
    if (this != &source) {
        this->Set(source.Count(), source.Ptr());
    }

  return *this;
}

template <class T>
void TSFixedArray<T>::ReallocAndClearData(uint32_t count) {
    this->m_alloc = count;

    if (this->m_data || count) {
        void* m = SMemReAlloc(this->m_data, sizeof(T) * count, this->MemFileName(), this->MemLineNo(), 0x0);
        this->m_data = static_cast<T*>(m);
    }
}

template <class T>
void TSFixedArray<T>::ReallocData(uint32_t count) {
    T* oldData = this->m_data;

    if (count < this->m_count) {
        for (uint32_t i = count; i < this->m_count; i++) {
            T* element = &this->m_data[i];
            delete element;
        }
    }

    this->m_alloc = count;

    void* v6 = SMemReAlloc(oldData, sizeof(T) * count, nullptr, 0, 0x10);
    this->m_data = (T*)v6;

    if (!v6) {
        this->m_data = (T*)SMemAlloc(sizeof(T) * count, nullptr, 0, 0x0);

        if (oldData) {
            uint32_t smallestCount = count >= this->m_count ? this->m_count : count;

            for (uint32_t i = 0; i < smallestCount; i++) {
                T* v8 = &this->m_data[i];

                if (v8) {
                    *v8 = oldData[i];
                }
            }

            SMemFree(oldData, nullptr, 0, 0x0);
        }
    }
}

template <class T>
void TSFixedArray<T>::Set(uint32_t count, const T* data) {
    this->ReallocAndClearData(count);

    for (uint32_t i; i < count; i++) {
        new (&this->m_data[i]) T(data[i]);
    }

    this->m_count = count;
}

template <class T>
void TSFixedArray<T>::SetCount(uint32_t count) {
    if (count != this->m_count) {
        if (count) {
            this->ReallocData(count);

            for (uint32_t i = this->m_count; i < count; i++) {
                new (&this->m_data[i]) T();
            }

            this->m_count = count;
        } else {
            this->Clear();
        }
    }
}

#endif