Seele/src/Engine/Containers/Array.h

#pragma once
#include "EngineTypes.h"
#include <initializer_list>
#include <iterator>
#include <assert.h>
#include <memory_resource>
#include <boost/serialization/serialization.hpp>

#ifndef DEFAULT_ALLOC_SIZE
#define DEFAULT_ALLOC_SIZE 16
#endif

namespace Seele
{
	enum class Init_t {
		NO_INIT
	};
	template <typename T>
	struct Array
	{
	public:
		
		Array()
			: arraySize(0)
			, allocated(DEFAULT_ALLOC_SIZE)
		{
			_data = new T[DEFAULT_ALLOC_SIZE];
			assert(_data != nullptr);
			//std::memset(_data, 0, sizeof(T) * DEFAULT_ALLOC_SIZE);
			markIteratorDirty();
		}
		Array(Init_t)
			: arraySize(0)
			, allocated(0)
			, _data(nullptr)
			, beginIt(Iterator(nullptr))
			, endIt(Iterator(nullptr))
		{
		}
		Array(size_t size, T value = T())
			: arraySize(size)
			, allocated(size)
		{
			_data = new T[size];
			assert(_data != nullptr);
			for (size_t i = 0; i < size; ++i)
			{
				assert(i < size);
				_data[i] = value;
			}
			markIteratorDirty();
		}
		Array(std::initializer_list<T> init)
			: arraySize(init.size())
			, allocated(init.size())
		{
			_data = new T[init.size()];
			auto it = init.begin();
			for (size_t i = 0; it != init.end(); i++, it++)
			{
				assert(i < init.size());
				_data[i] = *it;
			}
			markIteratorDirty();
		}
		Array(const Array &other)
			: arraySize(other.arraySize)
			, allocated(other.allocated)
		{
			_data = new T[other.allocated];
			assert(_data != nullptr);
			markIteratorDirty();
			std::copy(other.begin(), other.end(), begin());
		}
		Array(Array &&other) noexcept
			: arraySize(std::move(other.arraySize))
			, allocated(std::move(other.allocated))
		{
			_data = other._data;
			other._data = nullptr;
			other.allocated = 0;
			other.arraySize = 0;
			markIteratorDirty();
		}
		Array &operator=(const Array &other) noexcept
		{
			if (this != &other)
			{
				if (_data != nullptr)
				{
					if(other.arraySize > allocated)
					{
						delete[] _data;
						_data = new T[other.allocated];
						allocated = other.allocated;
					}
				}
				arraySize = other.arraySize;
				markIteratorDirty();
				std::copy(other.begin(), other.end(), begin());
			}
			return *this;
		}
		Array &operator=(Array &&other) noexcept
		{
			if (this != &other)
			{
				if (_data != nullptr)
				{
					delete[] _data;
					_data = nullptr;
				}
				allocated = std::move(other.allocated);
				arraySize = std::move(other.arraySize);
				_data = other._data;
				other._data = nullptr;
				markIteratorDirty();
			}
			return *this;
		}
		~Array()
		{
			if (_data)
			{
				delete[] _data;
				_data = nullptr;
			}
		}
		template <typename X>
		class IteratorBase
		{
		public:
			using iterator_category = std::random_access_iterator_tag;
			using value_type = X;
			using difference_type = std::ptrdiff_t;
			using reference = X&;
			using pointer = X*;

			IteratorBase(X *x = nullptr)
				: p(x)
			{
			}
			reference operator*() const
			{
				return *p;
			}
			pointer operator->() const
			{
				return p;
			}
			inline bool operator!=(const IteratorBase &other) const
			{
				return p != other.p;
			}
			inline bool operator==(const IteratorBase &other) const
			{
				return p == other.p;
			}
			inline int operator-(const IteratorBase &other) const
			{
				return (int)(p - other.p);
			}
			IteratorBase &operator++()
			{
				p++;
				return *this;
			}
			IteratorBase &operator--()
			{
				p--;
				return *this;
			}
			IteratorBase operator++(int)
			{
				IteratorBase tmp(*this);
				++*this;
				return tmp;
			}
			IteratorBase operator--(int)
			{
				IteratorBase tmp(*this);
				--*this;
				return tmp;
			}

		private:
			X *p;
		};
		typedef IteratorBase<T> Iterator;
		typedef IteratorBase<const T> ConstIterator;

		bool operator==(const Array &other)
		{
			return _data == other._data;
		}

		bool operator!=(const Array &other)
		{
			return !(*this == other);
		}

		Iterator find(const T &item)
		{
			for (uint32 i = 0; i < arraySize; ++i)
			{
				if (_data[i] == item)
				{
					return Iterator(&_data[i]);
				}
			}
			return endIt;
		}
		Iterator find(T&& item)
		{
			for (uint32 i = 0; i < arraySize; ++i)
			{
				if (_data[i] == item)
				{
					return Iterator(&_data[i]);
				}
			}
			return endIt;	
		}
		Iterator begin() const
		{
			return beginIt;
		}
		Iterator end() const
		{
			return endIt;
		}
		ConstIterator cbegin() const
		{
			return beginIt;
		}
		ConstIterator cend() const
		{
			return endIt;
		}
		T &add(const T &item = T())
		{
			return addInternal(item);
		}
		T &add(T&& item)
		{
			return addInternal(std::forward<T>(item));
		}
		T &addUnique(const T &item = T())
		{
			Iterator it;
			if((it = std::move(find(item))) != endIt)
			{
				return *it;
			}
			return addInternal(item);
		}
		template<typename... args>
		T &emplace(args... arguments)
		{
			if (arraySize == allocated)
			{
				size_t newSize = arraySize + 1;
				allocated = calculateGrowth(newSize);
				T *tempArray = new T[allocated];
				assert(tempArray != nullptr);
				for (size_t i = 0; i < arraySize; ++i)
				{
					tempArray[i] = _data[i];
				}
				delete[] _data;
				_data = tempArray;
			}
			_data[arraySize++] = T(arguments...);
			markIteratorDirty();
			return _data[arraySize - 1];
		}
		void remove(Iterator it, bool keepOrder = true)
		{
			remove(it - beginIt, keepOrder);
		}
		void remove(int index, bool keepOrder = true)
		{
			if (keepOrder)
			{
				for(uint32 i = index; i < arraySize-1; ++i)
				{
					_data[i] = std::move(_data[i+1]);
				}
			}
			else
			{
				_data[index] = std::move(_data[arraySize - 1]);
			}
			arraySize--;
			markIteratorDirty();
		}
		void clear()
		{
			delete[] _data;
			_data = nullptr;
			arraySize = 0;
			allocated = 0;
			markIteratorDirty();
		}
		void resize(size_t newSize)
		{
			if (newSize <= allocated)
			{
				arraySize = newSize;
			}
			else
			{
				T *newData = new T[newSize];
				assert(newData != nullptr);
				allocated = newSize;
				for(uint32 i = 0; i < arraySize; ++i)
				{
					newData[i] = std::move(_data[i]);
				}
				arraySize = newSize;
				delete _data;
				_data = newData;
			}
			markIteratorDirty();
		}
		inline size_t indexOf(Iterator iterator)
		{
			return iterator - beginIt;
		}
		inline size_t indexOf(ConstIterator iterator) const
		{
			return iterator.p - beginIt.p;
		}
		inline size_t indexOf(T& t)
		{
			return indexOf(find(t));
		}
		inline size_t indexOf(const T& t) const
		{
			return indexOf(find(t));
		}
		inline size_t size() const
		{
			return arraySize;
		}
		inline size_t empty() const
		{
			return arraySize == 0;
		}
		inline size_t capacity() const
		{
			return allocated;
		}
		inline T *data() const
		{
			return _data;
		}
		inline T &back() const
		{
			return _data[arraySize - 1];
		}
		void pop()
		{
			arraySize--;
			markIteratorDirty();
		}
		constexpr inline T &operator[](size_t index)
		{
			assert(index < arraySize);
			return _data[index];
		}
		constexpr inline const T &operator[](size_t index) const
		{
			assert(index < arraySize);
			return _data[index];
		}
	private:
		size_t calculateGrowth(size_t newSize) const
		{
			const size_t oldCapacity = capacity();

			if (oldCapacity > SIZE_MAX - oldCapacity / 2)
			{
				return newSize; // geometric growth would overflow
			}

			const size_t geometric = oldCapacity + oldCapacity / 2;

			if (geometric < newSize)
			{
				return newSize; // geometric growth would be insufficient
			}

			return geometric; // geometric growth is sufficient
		}
		void markIteratorDirty()
		{
			beginIt = Iterator(_data);
			endIt = Iterator(_data + arraySize);
		}
		template<typename Type>
		T& addInternal(Type&& t)
		{
			if (arraySize == allocated)
			{
				size_t newSize = arraySize + 1;
				allocated = calculateGrowth(newSize);
				T *tempArray = new T[allocated];
				assert(tempArray != nullptr);
				for (size_t i = 0; i < arraySize; ++i)
				{
					tempArray[i] = std::forward<Type>(_data[i]);
				}
				delete[] _data;
				_data = tempArray;
			}
			_data[arraySize++] = std::forward<Type>(t);
			markIteratorDirty();
			return _data[arraySize - 1];
		}
		friend class boost::serialization::access;
		template<class Archive>
		void serialize(Archive& ar, const unsigned int version)
		{
			ar & arraySize;
			resize(arraySize);
			for(size_t i = 0; i < arraySize; ++i)
				ar & _data[i];
			markIteratorDirty();
		}
		size_t arraySize = 0;
		size_t allocated = 0;
		Iterator beginIt;
		Iterator endIt;
		T *_data = nullptr;
	};

	template <typename T, size_t N>
	struct StaticArray
	{
	public:	
		template <typename X>
		class IteratorBase
		{
		public:
			using iterator_category = std::random_access_iterator_tag;
			using value_type = X;
			using difference_type = std::ptrdiff_t;
			using reference = X&;
			using pointer = X*;

			IteratorBase(X *x = nullptr)
				: p(x)
			{
			}
			reference operator*() const
			{
				return *p;
			}
			pointer operator->() const
			{
				return p;
			}
			inline bool operator!=(const IteratorBase &other)
			{
				return p != other.p;
			}
			inline bool operator==(const IteratorBase &other)
			{
				return p == other.p;
			}
			IteratorBase &operator++()
			{
				p++;
				return *this;
			}
			IteratorBase operator++(int)
			{
				IteratorBase tmp(*this);
				++*this;
				return tmp;
			}
			IteratorBase &operator--()
			{
				p--;
				return *this;
			}
			IteratorBase operator--(int)
			{
				IteratorBase tmp(*this);
				--*this;
				return tmp;
			}
			

		private:
			X *p;
		};
		using value_type = T;
		using size_type = size_t;
		using difference_type = std::ptrdiff_t;
		using pointer = T*;
		using const_pointer = const T*;
		using reference = T&;
		using const_reference = const T&;

		using iterator = IteratorBase<T>;
		using const_iterator = IteratorBase<const T>;

		using reverse_iterator = std::reverse_iterator<iterator>;
		using const_reverse_iterator = std::reverse_iterator<const_iterator>;

		StaticArray()
		{
			beginIt = iterator(_data);
			endIt = iterator(_data + N);
		}
		StaticArray(T value)
		{
			for (int i = 0; i < N; ++i)
			{
				_data[i] = value;
			}
			beginIt = iterator(_data);
			endIt = iterator(_data + N);
		}
		~StaticArray()
		{
		}
		
		inline size_type size() const
		{
			return N;
		}
		inline pointer data()
		{
			return _data;
		}
		inline const_pointer data() const
		{
			return _data;
		}
		constexpr reference operator[](size_type index) noexcept
		{
			assert(index < N);
			return _data[index];
		}
		constexpr const_reference operator[](size_type index) const noexcept
		{
			assert(index < N);
			return _data[index];
		}
		iterator begin()
		{
			return beginIt;
		}
		iterator end()
		{
			return endIt;
		}
		const_iterator begin() const
		{
			return beginIt;
		}
		const_iterator end() const
		{
			return beginIt;
		}
	private:
		T _data[N];
		iterator beginIt;
		iterator endIt;
		friend class boost::serialization::access;
		template<class Archive>
		void serialize(Archive& ar, const unsigned int version)
		{
			ar & version;
			ar & N;
			ar & _data;
		}
	};
} // namespace Seele