core/modules/ArrayList.cppm

module;

#include <cassert>
#include <new>

export module Core.ArrayList;

import Core.AlignedData;
import Core.Meta;
import Core.Types;

export using ::operator new;

export namespace Core {
    template<typename T, size_t N>
    class ArrayList final {
        AlignedType<T> data[N];
        size_t length;

    public:
        ArrayList() : length(0) {
        }

        ArrayList(const ArrayList& other) : ArrayList() {
            copy(other);
        }

        ArrayList(ArrayList&& other) noexcept : ArrayList() {
            move(Core::move(other));
            other.clear();
        }

        ~ArrayList() {
            clear();
        }

        ArrayList& operator=(const ArrayList& other) {
            if(&other != this) {
                clear();
                copy(other);
            }
            return *this;
        }

        ArrayList& operator=(ArrayList&& other) noexcept {
            if(&other != this) {
                clear();
                move(Core::move(other));
                other.clear();
            }
            return *this;
        }

        T* begin() {
            return reinterpret_cast<T*>(data);
        }

        T* end() {
            return begin() + length;
        }

        const T* begin() const {
            return reinterpret_cast<const T*>(data);
        }

        const T* end() const {
            return begin() + length;
        }

        template<typename... Args>
        T* put(Args&&... args) {
            if(length < N) {
                return new(begin() + length++) T(Core::forward<Args>(args)...);
            }
            return nullptr;
        }

        template<typename... Args>
        ArrayList& add(Args&&... args) {
            put(Core::forward<Args>(args)...);
            return *this;
        }

        template<typename... Args>
        T* putAt(size_t t, Args&&... args) {
            if(length >= N) {
                return nullptr;
            } else if(t >= length) {
                return put(Core::forward<Args>(args)...);
            }
            put(Core::move(begin()[length - 1]));
            for(size_t i = length - 2; i > t; i--) {
                (*this)[i] = Core::move((*this)[i - 1]);
            }
            (*this)[t] = Core::move(T(Core::forward<Args>(args)...));
            return begin() + t;
        }

        template<typename... Args>
        ArrayList& addAt(size_t index, Args&&... args) {
            putAt(index, Core::forward<Args>(args)...);
            return *this;
        }

        T& operator[](size_t index) {
            return begin()[index];
        }

        const T& operator[](size_t index) const {
            return begin()[index];
        }

        T& getLast() {
            assert(length > 0);
            return begin()[length - 1];
        }

        const T& getLast() const {
            assert(length > 0);
            return begin()[length - 1];
        }

        size_t getLength() const {
            return length;
        }

        consteval size_t getCapacity() const {
            return N;
        }

        void clear() {
            for(size_t i = 0; i < length; i++) {
                begin()[i].~T();
            }
            length = 0;
        }

        void removeBySwap(size_t index) {
            assert(index < length);
            length--;
            if(index != length) {
                begin()[index] = Core::move(begin()[length]);
            }
            begin()[length].~T();
        }

        void remove(size_t index) {
            assert(index < length);
            length--;
            T* currentT = begin() + index;
            T* endT = end();
            while(currentT != endT) {
                T* nextT = currentT + 1;
                *currentT = Core::move(*nextT);
                currentT = nextT;
            }
            endT->~T();
        }

        void removeLast() {
            removeBySwap(length - 1);
        }

    private:
        void copy(const ArrayList& other) {
            for(size_t i = 0; i < other.length; i++) {
                add(other[i]);
            }
        }

        void move(ArrayList&& other) {
            for(size_t i = 0; i < other.length; i++) {
                add(Core::move(other[i]));
            }
        }
    };
}