#if defined(WITH_ANGELSCRIPT)

#include "RegisterArray.h"
#include "../../Common.h"

#include <new>
#include <stdlib.h>
#include <algorithm>	// std::sort

#include <Containers/StringConcatenable.h>

#define AS_USE_ACCESSORS 1

using namespace Death::Containers;
using namespace Death::Containers::Literals;

namespace Jazz2::Scripting
{
	struct SArrayBuffer {
		asDWORD maxElements;
		asDWORD numElements;
		asBYTE  data[1];
	};

	struct SArrayCache {
		asIScriptFunction* cmpFunc;
		asIScriptFunction* eqFunc;
		int cmpFuncReturnCode; // To allow better error message in case of multiple matches
		int eqFuncReturnCode;
	};

	// We just define a number here that we assume nobody else is using for
	// object type user data. The add-ons have reserved the numbers 1000
	// through 1999 for this purpose, so we should be fine.
	constexpr asPWORD ARRAY_CACHE = 1000;

	static void CleanupTypeInfoArrayCache(asITypeInfo* type)
	{
		SArrayCache* cache = reinterpret_cast<SArrayCache*>(type->GetUserData(ARRAY_CACHE));
		if (cache != nullptr) {
			cache->~SArrayCache();
			asFreeMem(cache);
		}
	}

	CScriptArray* CScriptArray::Create(asITypeInfo* ti, asUINT length)
	{
		// Allocate the memory
		void* mem = asAllocMem(sizeof(CScriptArray));
		if (mem == nullptr) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
			return nullptr;
		}

		// Initialize the object
		CScriptArray* a = new(mem) CScriptArray(length, ti);

		return a;
	}

	CScriptArray* CScriptArray::Create(asITypeInfo* ti, void* initList)
	{
		// Allocate the memory
		void* mem = asAllocMem(sizeof(CScriptArray));
		if (mem == nullptr) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
			return nullptr;
		}

		// Initialize the object
		CScriptArray* a = new(mem) CScriptArray(ti, initList);

		return a;
	}

	CScriptArray* CScriptArray::Create(asITypeInfo* ti, asUINT length, void* defVal)
	{
		// Allocate the memory
		void* mem = asAllocMem(sizeof(CScriptArray));
		if (mem == nullptr) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
			return nullptr;
		}

		// Initialize the object
		CScriptArray* a = new(mem) CScriptArray(length, defVal, ti);

		return a;
	}

	CScriptArray* CScriptArray::Create(asITypeInfo* ti)
	{
		return CScriptArray::Create(ti, asUINT(0));
	}

	// This optional callback is called when the template type is first used by the compiler.
	// It allows the application to validate if the template can be instantiated for the requested
	// subtype at compile time, instead of at runtime. The output argument dontGarbageCollect
	// allow the callback to tell the engine if the template instance type shouldn't be garbage collected,
	// i.e. no asOBJ_GC flag.
	static bool ScriptArrayTemplateCallback(asITypeInfo* ti, bool& dontGarbageCollect)
	{
		// Make sure the subtype can be instantiated with a default factory/constructor,
		// otherwise we won't be able to instantiate the elements.
		int typeId = ti->GetSubTypeId();
		if (typeId == asTYPEID_VOID)
			return false;

		if ((typeId & asTYPEID_MASK_OBJECT) && !(typeId & asTYPEID_OBJHANDLE)) {
			asITypeInfo* subtype = ti->GetEngine()->GetTypeInfoById(typeId);
			asDWORD flags = subtype->GetFlags();
			if ((flags & asOBJ_VALUE) && !(flags & asOBJ_POD)) {
				// Verify that there is a default constructor
				bool found = false;
				for (asUINT n = 0; n < subtype->GetBehaviourCount(); n++) {
					asEBehaviours beh;
					asIScriptFunction* func = subtype->GetBehaviourByIndex(n, &beh);
					if (beh != asBEHAVE_CONSTRUCT) continue;

					if (func->GetParamCount() == 0) {
						// Found the default constructor
						found = true;
						break;
					}
				}

				if (!found) {
					// There is no default constructor
					// TODO: Should format the message to give the name of the subtype for better understanding
					ti->GetEngine()->WriteMessage("array", 0, 0, asMSGTYPE_ERROR, "The subtype has no default constructor");
					return false;
				}
			} else if ((flags & asOBJ_REF)) {
				bool found = false;

				// If value assignment for ref type has been disabled then the array
				// can be created if the type has a default factory function
				if (!ti->GetEngine()->GetEngineProperty(asEP_DISALLOW_VALUE_ASSIGN_FOR_REF_TYPE)) {
					// Verify that there is a default factory
					for (asUINT n = 0; n < subtype->GetFactoryCount(); n++) {
						asIScriptFunction* func = subtype->GetFactoryByIndex(n);
						if (func->GetParamCount() == 0) {
							// Found the default factory
							found = true;
							break;
						}
					}
				}

				if (!found) {
					// No default factory
					char buffer[512];
					snprintf(buffer, sizeof(buffer), "The subtype '%s' has no default factory", subtype ? subtype->GetEngine()->GetTypeDeclaration(subtype->GetTypeId()) : "UNKNOWN");
					ti->GetEngine()->WriteMessage("array", 0, 0, asMSGTYPE_ERROR, buffer);
					return false;
				}
			}

			// If the object type is not garbage collected then the array also doesn't need to be
			if (!(flags & asOBJ_GC)) {
				dontGarbageCollect = true;
			}
		} else if (!(typeId & asTYPEID_OBJHANDLE)) {
			// Arrays with primitives cannot form circular references,
			// thus there is no need to garbage collect them
			dontGarbageCollect = true;
		} else {
			ASSERT(typeId & asTYPEID_OBJHANDLE);

			// It is not necessary to set the array as garbage collected for all handle types.
			// If it is possible to determine that the handle cannot refer to an object type
			// that can potentially form a circular reference with the array then it is not
			// necessary to make the array garbage collected.
			asITypeInfo* subtype = ti->GetEngine()->GetTypeInfoById(typeId);
			asDWORD flags = subtype->GetFlags();
			if (!(flags & asOBJ_GC)) {
				if ((flags & asOBJ_SCRIPT_OBJECT)) {
					// Even if a script class is by itself not garbage collected, it is possible
					// that classes that derive from it may be, so it is not possible to know
					// that no circular reference can occur.
					if ((flags & asOBJ_NOINHERIT)) {
						// A script class declared as final cannot be inherited from, thus
						// we can be certain that the object cannot be garbage collected.
						dontGarbageCollect = true;
					}
				} else {
					// For application registered classes we assume the application knows
					// what it is doing and don't mark the array as garbage collected unless
					// the type is also garbage collected.
					dontGarbageCollect = true;
				}
			}
		}

		// The type is ok
		return true;
	}

	void RegisterArray(asIScriptEngine* engine)
	{
		int r;

		// Register the object type user data clean up
		engine->SetTypeInfoUserDataCleanupCallback(CleanupTypeInfoArrayCache, ARRAY_CACHE);

		// Register the array type as a template
		r = engine->RegisterObjectType("array<class T>", 0, asOBJ_REF | asOBJ_GC | asOBJ_TEMPLATE); RETURN_ASSERT(r >= 0);

		// Register a callback for validating the subtype before it is used
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_TEMPLATE_CALLBACK, "bool f(int&in, bool&out)", asFUNCTION(ScriptArrayTemplateCallback), asCALL_CDECL); RETURN_ASSERT(r >= 0);

		// Templates receive the object type as the first parameter. To the script writer this is hidden
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_FACTORY, "array<T>@ f(int&in)", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*), CScriptArray*), asCALL_CDECL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_FACTORY, "array<T>@ f(int&in, uint length) explicit", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*, asUINT), CScriptArray*), asCALL_CDECL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_FACTORY, "array<T>@ f(int&in, uint length, const T &in value)", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*, asUINT, void*), CScriptArray*), asCALL_CDECL); RETURN_ASSERT(r >= 0);

		// Register the factory that will be used for initialization lists
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_LIST_FACTORY, "array<T>@ f(int&in type, int&in list) {repeat T}", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*, void*), CScriptArray*), asCALL_CDECL); RETURN_ASSERT(r >= 0);

		// The memory management methods
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_ADDREF, "void f()", asMETHOD(CScriptArray, AddRef), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_RELEASE, "void f()", asMETHOD(CScriptArray, Release), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

		// The index operator returns the template subtype
		r = engine->RegisterObjectMethod("array<T>", "T &opIndex(uint index)", asMETHODPR(CScriptArray, At, (asUINT), void*), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "const T &opIndex(uint index) const", asMETHODPR(CScriptArray, At, (asUINT) const, const void*), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

		// The assignment operator
		r = engine->RegisterObjectMethod("array<T>", "array<T> &opAssign(const array<T>&in)", asMETHOD(CScriptArray, operator=), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

		// Other methods
		r = engine->RegisterObjectMethod("array<T>", "void insertAt(uint index, const T&in value)", asMETHODPR(CScriptArray, InsertAt, (asUINT, void*), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void insertAt(uint index, const array<T>& arr)", asMETHODPR(CScriptArray, InsertAt, (asUINT, const CScriptArray&), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void insertLast(const T&in value)", asMETHOD(CScriptArray, InsertLast), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void removeAt(uint index)", asMETHOD(CScriptArray, RemoveAt), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void removeLast()", asMETHOD(CScriptArray, RemoveLast), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void removeRange(uint start, uint count)", asMETHOD(CScriptArray, RemoveRange), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// TODO: Should length() and resize() be deprecated as the property accessors do the same thing?
		// TODO: Register as size() for consistency with other types
#if AS_USE_ACCESSORS != 1
		r = engine->RegisterObjectMethod("array<T>", "uint length() const", asMETHOD(CScriptArray, GetSize), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
#endif
		r = engine->RegisterObjectMethod("array<T>", "void reserve(uint length)", asMETHOD(CScriptArray, Reserve), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void resize(uint length)", asMETHODPR(CScriptArray, Resize, (asUINT), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void sortAsc()", asMETHODPR(CScriptArray, SortAsc, (), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void sortAsc(uint startAt, uint count)", asMETHODPR(CScriptArray, SortAsc, (asUINT, asUINT), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void sortDesc()", asMETHODPR(CScriptArray, SortDesc, (), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void sortDesc(uint startAt, uint count)", asMETHODPR(CScriptArray, SortDesc, (asUINT, asUINT), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void reverse()", asMETHOD(CScriptArray, Reverse), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// The token 'if_handle_then_const' tells the engine that if the type T is a handle, then it should refer to a read-only object
		r = engine->RegisterObjectMethod("array<T>", "int find(const T&in if_handle_then_const value) const", asMETHODPR(CScriptArray, Find, (void*) const, int), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// TODO: It should be "int find(const T&in value, uint startAt = 0) const"
		r = engine->RegisterObjectMethod("array<T>", "int find(uint startAt, const T&in if_handle_then_const value) const", asMETHODPR(CScriptArray, Find, (asUINT, void*) const, int), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "int findByRef(const T&in if_handle_then_const value) const", asMETHODPR(CScriptArray, FindByRef, (void*) const, int), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// TODO: It should be "int findByRef(const T&in value, uint startAt = 0) const"
		r = engine->RegisterObjectMethod("array<T>", "int findByRef(uint startAt, const T&in if_handle_then_const value) const", asMETHODPR(CScriptArray, FindByRef, (asUINT, void*) const, int), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "bool opEquals(const array<T>&in) const", asMETHOD(CScriptArray, operator==), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		
		r = engine->RegisterObjectMethod("array<T>", "bool isEmpty() const", asMETHOD(CScriptArray, IsEmpty), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "bool empty() const", asMETHOD(CScriptArray, IsEmpty), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

		// Sort with callback for comparison
		r = engine->RegisterFuncdef("bool array<T>::less(const T&in if_handle_then_const a, const T&in if_handle_then_const b)"); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void sort(const less &in, uint startAt = 0, uint count = uint(-1))", asMETHODPR(CScriptArray, Sort, (asIScriptFunction*, asUINT, asUINT), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

#if AS_USE_STLNAMES != 1 && AS_USE_ACCESSORS == 1
		// Register virtual properties
		r = engine->RegisterObjectMethod("array<T>", "uint get_length() const property", asMETHOD(CScriptArray, GetSize), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void set_length(uint) property", asMETHODPR(CScriptArray, Resize, (asUINT), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
#endif

		// Register GC behaviours in case the array needs to be garbage collected
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_GETREFCOUNT, "int f()", asMETHOD(CScriptArray, GetRefCount), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_SETGCFLAG, "void f()", asMETHOD(CScriptArray, SetFlag), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_GETGCFLAG, "bool f()", asMETHOD(CScriptArray, GetFlag), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_ENUMREFS, "void f(int&in)", asMETHOD(CScriptArray, EnumReferences), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectBehaviour("array<T>", asBEHAVE_RELEASEREFS, "void f(int&in)", asMETHOD(CScriptArray, ReleaseAllHandles), asCALL_THISCALL); RETURN_ASSERT(r >= 0);

#if AS_USE_STLNAMES == 1
		// Same as length
		r = engine->RegisterObjectMethod("array<T>", "uint size() const", asMETHOD(CScriptArray, GetSize), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// Same as isEmpty
		r = engine->RegisterObjectMethod("array<T>", "bool empty() const", asMETHOD(CScriptArray, IsEmpty), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// Same as insertLast
		r = engine->RegisterObjectMethod("array<T>", "void push_back(const T&in)", asMETHOD(CScriptArray, InsertLast), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// Same as removeLast
		r = engine->RegisterObjectMethod("array<T>", "void pop_back()", asMETHOD(CScriptArray, RemoveLast), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// Same as insertAt
		r = engine->RegisterObjectMethod("array<T>", "void insert(uint index, const T&in value)", asMETHODPR(CScriptArray, InsertAt, (asUINT, void*), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		r = engine->RegisterObjectMethod("array<T>", "void insert(uint index, const array<T>& arr)", asMETHODPR(CScriptArray, InsertAt, (asUINT, const CScriptArray&), void), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
		// Same as removeAt
		r = engine->RegisterObjectMethod("array<T>", "void erase(uint)", asMETHOD(CScriptArray, RemoveAt), asCALL_THISCALL); RETURN_ASSERT(r >= 0);
#endif

		r = engine->RegisterDefaultArrayType("array<T>"); RETURN_ASSERT(r >= 0);
	}

	CScriptArray& CScriptArray::operator=(const CScriptArray& other)
	{
		// Only perform the copy if the array types are the same
		if (&other != this && other.GetArrayObjectType() == GetArrayObjectType()) {
			// Make sure the arrays are of the same size
			Resize(other.buffer->numElements);

			// Copy the value of each element
			CopyBuffer(buffer, other.buffer);
		}

		return *this;
	}

	CScriptArray::CScriptArray(asITypeInfo* ti, void* buf)
	{
		// The object type should be the template instance of the array
		ASSERT(ti != nullptr && StringView(ti->GetName()) == "array"_s);

		refCount = 1;
		gcFlag = false;
		objType = ti;
		objType->AddRef();
		buffer = 0;

		Precache();

		asIScriptEngine* engine = ti->GetEngine();

		// Determine element size
		if (subTypeId & asTYPEID_MASK_OBJECT) {
			elementSize = sizeof(asPWORD);
		} else {
			elementSize = engine->GetSizeOfPrimitiveType(subTypeId);
		}

		// Determine the initial size from the buffer
		asUINT length = *(asUINT*)buf;

		// Make sure the array size isn't too large for us to handle
		if (!CheckMaxSize(length)) {
			// Don't continue with the initialization
			return;
		}

		// Copy the values of the array elements from the buffer
		if ((ti->GetSubTypeId() & asTYPEID_MASK_OBJECT) == 0) {
			CreateBuffer(&buffer, length);

			// Copy the values of the primitive type into the internal buffer
			if (length > 0) {
				memcpy(At(0), (((asUINT*)buf) + 1), length * elementSize);
			}
		} else if (ti->GetSubTypeId() & asTYPEID_OBJHANDLE) {
			CreateBuffer(&buffer, length);

			// Copy the handles into the internal buffer
			if (length > 0) {
				memcpy(At(0), (((asUINT*)buf) + 1), length * elementSize);
			}

			// With object handles it is safe to clear the memory in the received buffer
			// instead of increasing the ref count. It will save time both by avoiding the
			// call the increase ref, and also relieve the engine from having to release
			// its references too
			memset((((asUINT*)buf) + 1), 0, length * elementSize);
		} else if (ti->GetSubType()->GetFlags() & asOBJ_REF) {
			// Only allocate the buffer, but not the objects
			subTypeId |= asTYPEID_OBJHANDLE;
			CreateBuffer(&buffer, length);
			subTypeId &= ~asTYPEID_OBJHANDLE;

			// Copy the handles into the internal buffer
			if (length > 0) {
				memcpy(buffer->data, (((asUINT*)buf) + 1), length * elementSize);
			}

			// For ref types we can do the same as for handles, as they are
			// implicitly stored as handles.
			memset((((asUINT*)buf) + 1), 0, length * elementSize);
		} else {
			// TODO: Optimize by calling the copy constructor of the object instead of
			//       constructing with the default constructor and then assigning the value
			// TODO: With C++11 ideally we should be calling the move constructor, instead
			//       of the copy constructor as the engine will just discard the objects in the
			//       buffer afterwards.
			CreateBuffer(&buffer, length);

			// For value types we need to call the opAssign for each individual object
			for (asUINT n = 0; n < length; n++) {
				void* obj = At(n);
				asBYTE* srcObj = (asBYTE*)buf;
				srcObj += 4 + n * ti->GetSubType()->GetSize();
				engine->AssignScriptObject(obj, srcObj, ti->GetSubType());
			}
		}

		// Notify the GC of the successful creation
		if (objType->GetFlags() & asOBJ_GC) {
			objType->GetEngine()->NotifyGarbageCollectorOfNewObject(this, objType);
		}
	}

	CScriptArray::CScriptArray(asUINT length, asITypeInfo* ti)
	{
		// The object type should be the template instance of the array
		ASSERT(ti != nullptr && StringView(ti->GetName()) == "array"_s);

		refCount = 1;
		gcFlag = false;
		objType = ti;
		objType->AddRef();
		buffer = 0;

		Precache();

		// Determine element size
		if (subTypeId & asTYPEID_MASK_OBJECT) {
			elementSize = sizeof(asPWORD);
		} else {
			elementSize = objType->GetEngine()->GetSizeOfPrimitiveType(subTypeId);
		}

		// Make sure the array size isn't too large for us to handle
		if (!CheckMaxSize(length)) {
			// Don't continue with the initialization
			return;
		}

		CreateBuffer(&buffer, length);

		// Notify the GC of the successful creation
		if (objType->GetFlags() & asOBJ_GC) {
			objType->GetEngine()->NotifyGarbageCollectorOfNewObject(this, objType);
		}
	}

	CScriptArray::CScriptArray(const CScriptArray& other)
	{
		refCount = 1;
		gcFlag = false;
		objType = other.objType;
		objType->AddRef();
		buffer = 0;

		Precache();

		elementSize = other.elementSize;

		if (objType->GetFlags() & asOBJ_GC) {
			objType->GetEngine()->NotifyGarbageCollectorOfNewObject(this, objType);
		}

		CreateBuffer(&buffer, 0);

		// Copy the content
		*this = other;
	}

	CScriptArray::CScriptArray(asUINT length, void* defVal, asITypeInfo* ti)
	{
		// The object type should be the template instance of the array
		ASSERT(ti != nullptr && StringView(ti->GetName()) == "array"_s);

		refCount = 1;
		gcFlag = false;
		objType = ti;
		objType->AddRef();
		buffer = 0;

		Precache();

		// Determine element size
		if (subTypeId & asTYPEID_MASK_OBJECT) {
			elementSize = sizeof(asPWORD);
		} else {
			elementSize = objType->GetEngine()->GetSizeOfPrimitiveType(subTypeId);
		}

		// Make sure the array size isn't too large for us to handle
		if (!CheckMaxSize(length)) {
			// Don't continue with the initialization
			return;
		}

		CreateBuffer(&buffer, length);

		// Notify the GC of the successful creation
		if (objType->GetFlags() & asOBJ_GC) {
			objType->GetEngine()->NotifyGarbageCollectorOfNewObject(this, objType);
		}

		// Initialize the elements with the default value
		for (asUINT n = 0; n < GetSize(); n++) {
			SetValue(n, defVal);
		}
	}

	void CScriptArray::SetValue(asUINT index, void* value)
	{
		// At() will take care of the out-of-bounds checking, though
		// if called from the application then nothing will be done
		void* ptr = At(index);
		if (ptr == nullptr) return;

		if ((subTypeId & ~asTYPEID_MASK_SEQNBR) && !(subTypeId & asTYPEID_OBJHANDLE)) {
			asITypeInfo* subType = objType->GetSubType();
			if (subType->GetFlags() & asOBJ_ASHANDLE) {
				// For objects that should work as handles we must use the opHndlAssign method
				// TODO: Must support alternative syntaxes as well
				// TODO: Move the lookup of the opHndlAssign method to Precache() so it is only done once
				String decl = StringView(subType->GetName()) + "& opHndlAssign(const "_s + StringView(subType->GetName()) + "&in)"_s;
				asIScriptFunction* func = subType->GetMethodByDecl(decl.data());
				if (func != nullptr) {
					// TODO: Reuse active context if existing
					asIScriptEngine* engine = objType->GetEngine();
					asIScriptContext* ctx = engine->RequestContext();
					ctx->Prepare(func);
					ctx->SetObject(ptr);
					ctx->SetArgAddress(0, value);
					// TODO: Handle errors
					ctx->Execute();
					engine->ReturnContext(ctx);
				} else {
					// opHndlAssign doesn't exist, so try ordinary value assign instead
					objType->GetEngine()->AssignScriptObject(ptr, value, subType);
				}
			} else {
				objType->GetEngine()->AssignScriptObject(ptr, value, subType);
			}
		} else if (subTypeId & asTYPEID_OBJHANDLE) {
			void* tmp = *(void**)ptr;
			*(void**)ptr = *(void**)value;
			objType->GetEngine()->AddRefScriptObject(*(void**)value, objType->GetSubType());
			if (tmp != nullptr)
				objType->GetEngine()->ReleaseScriptObject(tmp, objType->GetSubType());
		} else if (subTypeId == asTYPEID_BOOL ||
				 subTypeId == asTYPEID_INT8 ||
				 subTypeId == asTYPEID_UINT8) {
			*(char*)ptr = *(char*)value;
		} else if (subTypeId == asTYPEID_INT16 ||
				 subTypeId == asTYPEID_UINT16) {
			*(short*)ptr = *(short*)value;
		} else if (subTypeId == asTYPEID_INT32 ||
				 subTypeId == asTYPEID_UINT32 ||
				 subTypeId == asTYPEID_FLOAT ||
				 subTypeId > asTYPEID_DOUBLE) { // enums have a type id larger than doubles
			*(int*)ptr = *(int*)value;
		} else if (subTypeId == asTYPEID_INT64 ||
				 subTypeId == asTYPEID_UINT64 ||
				 subTypeId == asTYPEID_DOUBLE) {
			*(double*)ptr = *(double*)value;
		}
	}

	CScriptArray::~CScriptArray()
	{
		if (buffer != nullptr) {
			DeleteBuffer(buffer);
			buffer = 0;
		}
		if (objType != nullptr) objType->Release();
	}

	asUINT CScriptArray::GetSize() const
	{
		return buffer->numElements;
	}

	bool CScriptArray::IsEmpty() const
	{
		return buffer->numElements == 0;
	}

	void CScriptArray::Reserve(asUINT maxElements)
	{
		if (maxElements <= buffer->maxElements)
			return;

		if (!CheckMaxSize(maxElements))
			return;

		// Allocate memory for the buffer
		SArrayBuffer* newBuffer = reinterpret_cast<SArrayBuffer*>(asAllocMem(sizeof(SArrayBuffer) - 1 + elementSize * maxElements));
		if (newBuffer != nullptr) {
			newBuffer->numElements = buffer->numElements;
			newBuffer->maxElements = maxElements;
		} else {
			// Out of memory
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
			return;
		}

		// As objects in arrays of objects are not stored inline, it is safe to use memcpy here
		// since we're just copying the pointers to objects and not the actual objects.
		memcpy(newBuffer->data, buffer->data, buffer->numElements * elementSize);

		// Release the old buffer
		asFreeMem(buffer);

		buffer = newBuffer;
	}

	void CScriptArray::Resize(asUINT numElements)
	{
		if (!CheckMaxSize(numElements)) {
			return;
		}
		Resize((int)numElements - (int)buffer->numElements, (asUINT)-1);
	}

	void CScriptArray::RemoveRange(asUINT start, asUINT count)
	{
		if (count == 0) {
			return;
		}

		if (buffer == nullptr || start > buffer->numElements) {
			// If this is called from a script we raise a script exception
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}
			return;
		}

		// Cap count to the end of the array
		if (start + count > buffer->numElements) {
			count = buffer->numElements - start;
		}

		// Destroy the elements that are being removed
		Destruct(buffer, start, start + count);

		// Compact the elements
		// As objects in arrays of objects are not stored inline, it is safe to use memmove here
		// since we're just copying the pointers to objects and not the actual objects.
		memmove(buffer->data + start * elementSize, buffer->data + (start + count) * elementSize, (buffer->numElements - start - count) * elementSize);
		buffer->numElements -= count;
	}

	// Internal
	void CScriptArray::Resize(int delta, asUINT at)
	{
		if (delta < 0) {
			if (-delta > (int)buffer->numElements) {
				delta = -(int)buffer->numElements;
			}
			if (at > buffer->numElements + delta) {
				at = buffer->numElements + delta;
			}
		} else if (delta > 0) {
			// Make sure the array size isn't too large for us to handle
			if (delta > 0 && !CheckMaxSize(buffer->numElements + delta)) {
				return;
			}
			if (at > buffer->numElements) {
				at = buffer->numElements;
			}
		}

		if (delta == 0) return;

		if (buffer->maxElements < buffer->numElements + delta) {
			// Allocate memory for the buffer
			SArrayBuffer* newBuffer = reinterpret_cast<SArrayBuffer*>(asAllocMem(sizeof(SArrayBuffer) - 1 + elementSize * (buffer->numElements + delta)));
			if (newBuffer != nullptr) {
				newBuffer->numElements = buffer->numElements + delta;
				newBuffer->maxElements = newBuffer->numElements;
			} else {
				// Out of memory
				asIScriptContext* ctx = asGetActiveContext();
				if (ctx != nullptr) {
					ctx->SetException("Out of memory");
				}
				return;
			}

			// As objects in arrays of objects are not stored inline, it is safe to use memcpy here
			// since we're just copying the pointers to objects and not the actual objects.
			memcpy(newBuffer->data, buffer->data, at * elementSize);
			if (at < buffer->numElements) {
				memcpy(newBuffer->data + (at + delta) * elementSize, buffer->data + at * elementSize, (buffer->numElements - at) * elementSize);
			}
			// Initialize the new elements with default values
			Construct(newBuffer, at, at + delta);

			// Release the old buffer
			asFreeMem(buffer);

			buffer = newBuffer;
		} else if (delta < 0) {
			Destruct(buffer, at, at - delta);
			// As objects in arrays of objects are not stored inline, it is safe to use memmove here
			// since we're just copying the pointers to objects and not the actual objects.
			memmove(buffer->data + at * elementSize, buffer->data + (at - delta) * elementSize, (buffer->numElements - (at - delta)) * elementSize);
			buffer->numElements += delta;
		} else {
			// As objects in arrays of objects are not stored inline, it is safe to use memmove here
			// since we're just copying the pointers to objects and not the actual objects.
			memmove(buffer->data + (at + delta) * elementSize, buffer->data + at * elementSize, (buffer->numElements - at) * elementSize);
			Construct(buffer, at, at + delta);
			buffer->numElements += delta;
		}
	}

	// internal
	bool CScriptArray::CheckMaxSize(asUINT numElements)
	{
		// This code makes sure the size of the buffer that is allocated
		// for the array doesn't overflow and becomes smaller than requested

		asUINT maxSize = 0xFFFFFFFFul - sizeof(SArrayBuffer) + 1;
		if (elementSize > 0) {
			maxSize /= elementSize;
		}
		if (numElements > maxSize) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Too large array size");
			}
			return false;
		}

		return true;
	}

	asITypeInfo* CScriptArray::GetArrayObjectType() const
	{
		return objType;
	}

	int CScriptArray::GetArrayTypeId() const
	{
		return objType->GetTypeId();
	}

	int CScriptArray::GetElementTypeId() const
	{
		return subTypeId;
	}

	void CScriptArray::InsertAt(asUINT index, void* value)
	{
		if (index > buffer->numElements) {
			// If this is called from a script we raise a script exception
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}
			return;
		}

		// Make room for the new element
		Resize(1, index);

		// Set the value of the new element
		SetValue(index, value);
	}

	void CScriptArray::InsertAt(asUINT index, const CScriptArray& arr)
	{
		if (index > buffer->numElements) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}
			return;
		}

		if (objType != arr.objType) {
			// This shouldn't really be possible to happen when
			// called from a script, but let's check for it anyway
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Mismatching array types");
			}
			return;
		}

		asUINT elements = arr.GetSize();
		Resize(elements, index);
		if (&arr != this) {
			for (asUINT n = 0; n < arr.GetSize(); n++) {
				// This const cast is allowed, since we know the
				// value will only be used to make a copy of it
				void* value = const_cast<void*>(arr.At(n));
				SetValue(index + n, value);
			}
		} else {
			// The array that is being inserted is the same as this one.
			// So we should iterate over the elements before the index,
			// and then the elements after
			for (asUINT n = 0; n < index; n++) {
				// This const cast is allowed, since we know the
				// value will only be used to make a copy of it
				void* value = const_cast<void*>(arr.At(n));
				SetValue(index + n, value);
			}

			for (asUINT n = index + elements, m = 0; n < arr.GetSize(); n++, m++) {
				// This const cast is allowed, since we know the
				// value will only be used to make a copy of it
				void* value = const_cast<void*>(arr.At(n));
				SetValue(index + index + m, value);
			}
		}
	}

	void CScriptArray::InsertLast(void* value)
	{
		InsertAt(buffer->numElements, value);
	}

	void CScriptArray::RemoveAt(asUINT index)
	{
		if (index >= buffer->numElements) {
			// If this is called from a script we raise a script exception
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}
			return;
		}

		// Remove the element
		Resize(-1, index);
	}

	void CScriptArray::RemoveLast()
	{
		RemoveAt(buffer->numElements - 1);
	}

	// Return a pointer to the array element. Returns 0 if the index is out of bounds
	const void* CScriptArray::At(asUINT index) const
	{
		if (buffer == nullptr || index >= buffer->numElements) {
			// If this is called from a script we raise a script exception
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Index out of bounds");
			}
			return 0;
		}

		if ((subTypeId & asTYPEID_MASK_OBJECT) && !(subTypeId & asTYPEID_OBJHANDLE)) {
			return *(void**)(buffer->data + elementSize * index);
		} else {
			return buffer->data + elementSize * index;
		}
	}
	void* CScriptArray::At(asUINT index)
	{
		return const_cast<void*>(const_cast<const CScriptArray*>(this)->At(index));
	}

	void* CScriptArray::GetBuffer()
	{
		return buffer->data;
	}


	// internal
	void CScriptArray::CreateBuffer(SArrayBuffer** buf, asUINT numElements)
	{
		*buf = reinterpret_cast<SArrayBuffer*>(asAllocMem(sizeof(SArrayBuffer) - 1 + elementSize * numElements));

		if ((*buf) != nullptr) {
			(*buf)->numElements = numElements;
			(*buf)->maxElements = numElements;
			Construct(*buf, 0, numElements);
		} else {
			// Oops, out of memory
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
		}
	}

	// internal
	void CScriptArray::DeleteBuffer(SArrayBuffer* buf)
	{
		Destruct(buf, 0, buf->numElements);

		// Free the buffer
		asFreeMem(buf);
	}

	// internal
	void CScriptArray::Construct(SArrayBuffer* buf, asUINT start, asUINT end)
	{
		if ((subTypeId & asTYPEID_MASK_OBJECT) && !(subTypeId & asTYPEID_OBJHANDLE)) {
			// Create an object using the default constructor/factory for each element
			void** max = (void**)(buf->data + end * sizeof(void*));
			void** d = (void**)(buf->data + start * sizeof(void*));

			asIScriptEngine* engine = objType->GetEngine();
			asITypeInfo* subType = objType->GetSubType();

			for (; d < max; d++) {
				*d = (void*)engine->CreateScriptObject(subType);
				if (*d == 0) {
					// Set the remaining entries to null so the destructor
					// won't attempt to destroy invalid objects later
					memset(d, 0, sizeof(void*) * (max - d));

					// There is no need to set an exception on the context,
					// as CreateScriptObject has already done that
					return;
				}
			}
		} else {
			// Set all elements to zero whether they are handles or primitives
			void* d = (void*)(buf->data + start * elementSize);
			memset(d, 0, (end - start) * elementSize);
		}
	}

	// internal
	void CScriptArray::Destruct(SArrayBuffer* buf, asUINT start, asUINT end)
	{
		if (subTypeId & asTYPEID_MASK_OBJECT) {
			asIScriptEngine* engine = objType->GetEngine();

			void** max = (void**)(buf->data + end * sizeof(void*));
			void** d = (void**)(buf->data + start * sizeof(void*));

			for (; d < max; d++) {
				if (*d) {
					engine->ReleaseScriptObject(*d, objType->GetSubType());
				}
			}
		}
	}


	// internal
	bool CScriptArray::Less(const void* a, const void* b, bool asc)
	{
		if (!asc) {
			// Swap items
			const void* TEMP = a;
			a = b;
			b = TEMP;
		}

		if (!(subTypeId & ~asTYPEID_MASK_SEQNBR)) {
			// Simple compare of values
			switch (subTypeId) {
#define COMPARE(T) *((T*)a) < *((T*)b)
				case asTYPEID_BOOL:   return COMPARE(bool);
				case asTYPEID_INT8:   return COMPARE(asINT8);
				case asTYPEID_INT16:  return COMPARE(asINT16);
				case asTYPEID_INT32:  return COMPARE(asINT32);
				case asTYPEID_INT64:  return COMPARE(asINT64);
				case asTYPEID_UINT8:  return COMPARE(asBYTE);
				case asTYPEID_UINT16: return COMPARE(asWORD);
				case asTYPEID_UINT32: return COMPARE(asDWORD);
				case asTYPEID_UINT64: return COMPARE(asQWORD);
				case asTYPEID_FLOAT:  return COMPARE(float);
				case asTYPEID_DOUBLE: return COMPARE(double);
				default: return COMPARE(signed int); // All enums fall in this case
#undef COMPARE
			}
		}

		return false;
	}

	void CScriptArray::Reverse()
	{
		asUINT size = GetSize();

		if (size >= 2) {
			asBYTE TEMP[16];

			for (asUINT i = 0; i < size / 2; i++) {
				Copy(TEMP, GetArrayItemPointer(i));
				Copy(GetArrayItemPointer(i), GetArrayItemPointer(size - i - 1));
				Copy(GetArrayItemPointer(size - i - 1), TEMP);
			}
		}
	}

	bool CScriptArray::operator==(const CScriptArray& other) const
	{
		if (objType != other.objType) {
			return false;
		}
		if (GetSize() != other.GetSize()) {
			return false;
		}

		asIScriptContext* cmpContext = 0;
		bool isNested = false;

		if (subTypeId & ~asTYPEID_MASK_SEQNBR) {
			// Try to reuse the active context
			cmpContext = asGetActiveContext();
			if (cmpContext != nullptr) {
				if (cmpContext->GetEngine() == objType->GetEngine() && cmpContext->PushState() >= 0) {
					isNested = true;
				} else {
					cmpContext = nullptr;
				}
			}
			if (cmpContext == nullptr) {
				// TODO: Ideally this context would be retrieved from a pool, so we don't have to
				//       create a new one everytime. We could keep a context with the array object
				//       but that would consume a lot of resources as each context is quite heavy.
				cmpContext = objType->GetEngine()->CreateContext();
			}
		}

		// Check if all elements are equal
		bool isEqual = true;
		SArrayCache* cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
		for (asUINT n = 0; n < GetSize(); n++)
			if (!Equals(At(n), other.At(n), cmpContext, cache)) {
				isEqual = false;
				break;
			}

		if (cmpContext != nullptr) {
			if (isNested) {
				asEContextState state = cmpContext->GetState();
				cmpContext->PopState();
				if (state == asEXECUTION_ABORTED) {
					cmpContext->Abort();
				}
			} else {
				cmpContext->Release();
			}
		}

		return isEqual;
	}

	// internal
	bool CScriptArray::Equals(const void* a, const void* b, asIScriptContext* ctx, SArrayCache* cache) const
	{
		if (!(subTypeId & ~asTYPEID_MASK_SEQNBR)) {
			// Simple compare of values
			switch (subTypeId) {
#define COMPARE(T) *((T*)a) == *((T*)b)
				case asTYPEID_BOOL:   return COMPARE(bool);
				case asTYPEID_INT8:   return COMPARE(asINT8);
				case asTYPEID_INT16:  return COMPARE(asINT16);
				case asTYPEID_INT32:  return COMPARE(asINT32);
				case asTYPEID_INT64:  return COMPARE(asINT64);
				case asTYPEID_UINT8:  return COMPARE(asBYTE);
				case asTYPEID_UINT16: return COMPARE(asWORD);
				case asTYPEID_UINT32: return COMPARE(asDWORD);
				case asTYPEID_UINT64: return COMPARE(asQWORD);
				case asTYPEID_FLOAT:  return COMPARE(float);
				case asTYPEID_DOUBLE: return COMPARE(double);
				default: return COMPARE(signed int); // All enums fall here. TODO: update this when enums can have different sizes and types
#undef COMPARE
			}
		} else {
			int r = 0;

			if (subTypeId & asTYPEID_OBJHANDLE) {
				// Allow the find to work even if the array contains null handles
				if (*(void**)a == *(void**)b) return true;
			}

			// Execute object opEquals if available
			if (cache != nullptr && cache->eqFunc != nullptr) {
				// TODO: Add proper error handling
				r = ctx->Prepare(cache->eqFunc); ASSERT(r >= 0);

				if (subTypeId & asTYPEID_OBJHANDLE) {
					r = ctx->SetObject(*((void**)a)); ASSERT(r >= 0);
					r = ctx->SetArgObject(0, *((void**)b)); ASSERT(r >= 0);
				} else {
					r = ctx->SetObject((void*)a); ASSERT(r >= 0);
					r = ctx->SetArgObject(0, (void*)b); ASSERT(r >= 0);
				}

				r = ctx->Execute();

				if (r == asEXECUTION_FINISHED) {
					return ctx->GetReturnByte() != 0;
				}
				return false;
			}

			// Execute object opCmp if available
			if (cache != nullptr && cache->cmpFunc != nullptr) {
				// TODO: Add proper error handling
				r = ctx->Prepare(cache->cmpFunc); ASSERT(r >= 0);

				if (subTypeId & asTYPEID_OBJHANDLE) {
					r = ctx->SetObject(*((void**)a)); ASSERT(r >= 0);
					r = ctx->SetArgObject(0, *((void**)b)); ASSERT(r >= 0);
				} else {
					r = ctx->SetObject((void*)a); ASSERT(r >= 0);
					r = ctx->SetArgObject(0, (void*)b); ASSERT(r >= 0);
				}

				r = ctx->Execute();

				if (r == asEXECUTION_FINISHED) {
					return (int)ctx->GetReturnDWord() == 0;
				}
				return false;
			}
		}

		return false;
	}

	int CScriptArray::FindByRef(void* ref) const
	{
		return FindByRef(0, ref);
	}

	int CScriptArray::FindByRef(asUINT startAt, void* ref) const
	{
		// Find the matching element by its reference
		asUINT size = GetSize();
		if (subTypeId & asTYPEID_OBJHANDLE) {
			// Dereference the pointer
			ref = *(void**)ref;
			for (asUINT i = startAt; i < size; i++) {
				if (*(void**)At(i) == ref) {
					return i;
				}
			}
		} else {
			// Compare the reference directly
			for (asUINT i = startAt; i < size; i++) {
				if (At(i) == ref) {
					return i;
				}
			}
		}

		return -1;
	}

	int CScriptArray::Find(void* value) const
	{
		return Find(0, value);
	}

	int CScriptArray::Find(asUINT startAt, void* value) const
	{
		// Check if the subtype really supports find()
		// TODO: Can't this be done at compile time too by the template callback
		SArrayCache* cache = 0;
		if (subTypeId & ~asTYPEID_MASK_SEQNBR) {
			cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
			if (cache == nullptr || (cache->cmpFunc == 0 && cache->eqFunc == 0)) {
				asIScriptContext* ctx = asGetActiveContext();
				asITypeInfo* subType = objType->GetEngine()->GetTypeInfoById(subTypeId);

				// Throw an exception
				if (ctx != nullptr) {
					char tmp[512];

					if (cache && cache->eqFuncReturnCode == asMULTIPLE_FUNCTIONS) {
#if defined(DEATH_TARGET_MSVC)
						sprintf_s(tmp, arraySize(tmp), "Type '%s' has multiple matching opEquals or opCmp methods", subType->GetName());
#else
						sprintf(tmp, "Type '%s' has multiple matching opEquals or opCmp methods", subType->GetName());
#endif
					} else {
#if defined(DEATH_TARGET_MSVC)
						sprintf_s(tmp, arraySize(tmp), "Type '%s' does not have a matching opEquals or opCmp method", subType->GetName());
#else
						sprintf(tmp, "Type '%s' does not have a matching opEquals or opCmp method", subType->GetName());
#endif
					}
					ctx->SetException(tmp);
				}

				return -1;
			}
		}

		asIScriptContext* cmpContext = 0;
		bool isNested = false;

		if (subTypeId & ~asTYPEID_MASK_SEQNBR) {
			// Try to reuse the active context
			cmpContext = asGetActiveContext();
			if (cmpContext != nullptr) {
				if (cmpContext->GetEngine() == objType->GetEngine() && cmpContext->PushState() >= 0) {
					isNested = true;
				} else {
					cmpContext = nullptr;
				}
			}
			if (cmpContext == nullptr) {
				// TODO: Ideally this context would be retrieved from a pool, so we don't have to
				//       create a new one everytime. We could keep a context with the array object
				//       but that would consume a lot of resources as each context is quite heavy.
				cmpContext = objType->GetEngine()->CreateContext();
			}
		}

		// Find the matching element
		int ret = -1;
		asUINT size = GetSize();

		for (asUINT i = startAt; i < size; i++) {
			// value passed by reference
			if (Equals(At(i), value, cmpContext, cache)) {
				ret = (int)i;
				break;
			}
		}

		if (cmpContext != nullptr) {
			if (isNested) {
				asEContextState state = cmpContext->GetState();
				cmpContext->PopState();
				if (state == asEXECUTION_ABORTED) {
					cmpContext->Abort();
				}
			} else {
				cmpContext->Release();
			}
		}

		return ret;
	}



	// internal
	// Copy object handle or primitive value
	// Even in arrays of objects the objects are allocated on
	// the heap and the array stores the pointers to the objects
	void CScriptArray::Copy(void* dst, void* src)
	{
		memcpy(dst, src, elementSize);
	}

	// internal
	// Swap two elements
	// Even in arrays of objects the objects are allocated on 
	// the heap and the array stores the pointers to the objects.
	void CScriptArray::Swap(void* a, void* b)
	{
		asBYTE tmp[16];
		Copy(tmp, a);
		Copy(a, b);
		Copy(b, tmp);
	}

	// internal
	// Return pointer to array item (object handle or primitive value)
	void* CScriptArray::GetArrayItemPointer(int index)
	{
		return buffer->data + index * elementSize;
	}

	// internal
	// Return pointer to data in buffer (object or primitive)
	void* CScriptArray::GetDataPointer(void* buf)
	{
		if ((subTypeId & asTYPEID_MASK_OBJECT) && !(subTypeId & asTYPEID_OBJHANDLE)) {
			// Real address of object
			return reinterpret_cast<void*>(*(size_t*)buf);
		} else {
			// Primitive is just a raw data
			return buf;
		}
	}


	// Sort ascending
	void CScriptArray::SortAsc()
	{
		Sort(0, GetSize(), true);
	}

	// Sort ascending
	void CScriptArray::SortAsc(asUINT startAt, asUINT count)
	{
		Sort(startAt, count, true);
	}

	// Sort descending
	void CScriptArray::SortDesc()
	{
		Sort(0, GetSize(), false);
	}

	// Sort descending
	void CScriptArray::SortDesc(asUINT startAt, asUINT count)
	{
		Sort(startAt, count, false);
	}

	// Internal
	void CScriptArray::Sort(asUINT startAt, asUINT count, bool asc)
	{
		// Subtype isn't primitive and doesn't have opCmp
		SArrayCache* cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
		if (subTypeId & ~asTYPEID_MASK_SEQNBR) {
			if (cache == nullptr || cache->cmpFunc == 0) {
				asIScriptContext* ctx = asGetActiveContext();
				asITypeInfo* subType = objType->GetEngine()->GetTypeInfoById(subTypeId);

				// Throw an exception
				if (ctx != nullptr) {
					char tmp[512];

					if (cache && cache->cmpFuncReturnCode == asMULTIPLE_FUNCTIONS) {
#if defined(DEATH_TARGET_MSVC)
						sprintf_s(tmp, arraySize(tmp), "Type '%s' has multiple matching opCmp methods", subType->GetName());
#else
						sprintf(tmp, "Type '%s' has multiple matching opCmp methods", subType->GetName());
#endif
					} else {
#if defined(DEATH_TARGET_MSVC)
						sprintf_s(tmp, arraySize(tmp), "Type '%s' does not have a matching opCmp method", subType->GetName());
#else
						sprintf(tmp, "Type '%s' does not have a matching opCmp method", subType->GetName());
#endif
					}
					ctx->SetException(tmp);
				}

				return;
			}
		}

		// No need to sort
		if (count < 2) {
			return;
		}

		int start = startAt;
		int end = startAt + count;

		// Check if we could access invalid item while sorting
		if (start >= (int)buffer->numElements || end > (int)buffer->numElements) {
			asIScriptContext* ctx = asGetActiveContext();

			// Throw an exception
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}

			return;
		}

		if (subTypeId & ~asTYPEID_MASK_SEQNBR) {
			asIScriptContext* cmpContext = 0;
			bool isNested = false;

			// Try to reuse the active context
			cmpContext = asGetActiveContext();
			if (cmpContext != nullptr) {
				if (cmpContext->GetEngine() == objType->GetEngine() && cmpContext->PushState() >= 0) {
					isNested = true;
				} else {
					cmpContext = nullptr;
				}
			}
			if (cmpContext == nullptr) {
				cmpContext = objType->GetEngine()->RequestContext();
			}
			// Do the sorting
			struct {
				bool asc;
				asIScriptContext* cmpContext;
				asIScriptFunction* cmpFunc;
				bool operator()(void* a, void* b) const
				{
					if (!asc) {
						// Swap items
						void* TEMP = a;
						a = b;
						b = TEMP;
					}

					int r = 0;

					// Allow sort to work even if the array contains null handles
					if (a == 0) return true;
					if (b == 0) return false;

					// Execute object opCmp
					if (cmpFunc) {
						// TODO: Add proper error handling
						r = cmpContext->Prepare(cmpFunc); ASSERT(r >= 0);
						r = cmpContext->SetObject(a); ASSERT(r >= 0);
						r = cmpContext->SetArgObject(0, b); ASSERT(r >= 0);
						r = cmpContext->Execute();

						if (r == asEXECUTION_FINISHED) {
							return (int)cmpContext->GetReturnDWord() < 0;
						}
					}

					return false;
				}
			} customLess = { asc, cmpContext, cache ? cache->cmpFunc : 0 };
			std::sort((void**)GetArrayItemPointer(start), (void**)GetArrayItemPointer(end), customLess);

			// Clean up
			if (cmpContext != nullptr) {
				if (isNested) {
					asEContextState state = cmpContext->GetState();
					cmpContext->PopState();
					if (state == asEXECUTION_ABORTED) {
						cmpContext->Abort();
					}
				} else {
					objType->GetEngine()->ReturnContext(cmpContext);
				}
			}
		} else {
			// TODO: Use std::sort for primitive types too

			// Insertion sort
			asBYTE tmp[16];
			for (int i = start + 1; i < end; i++) {
				Copy(tmp, GetArrayItemPointer(i));

				int j = i - 1;

				while (j >= start && Less(GetDataPointer(tmp), At(j), asc)) {
					Copy(GetArrayItemPointer(j + 1), GetArrayItemPointer(j));
					j--;
				}

				Copy(GetArrayItemPointer(j + 1), tmp);
			}
		}
	}

	// Sort with script callback for comparing elements
	void CScriptArray::Sort(asIScriptFunction* func, asUINT startAt, asUINT count)
	{
		// No need to sort
		if (count < 2) {
			return;
		}

		// Check if we could access invalid item while sorting
		asUINT start = startAt;
		asUINT end = asQWORD(startAt) + asQWORD(count) >= (asQWORD(1) << 32) ? 0xFFFFFFFF : startAt + count;
		if (end > buffer->numElements) {
			end = buffer->numElements;
		}
		if (start >= buffer->numElements) {
			asIScriptContext* ctx = asGetActiveContext();

			// Throw an exception
			if (ctx) {
				ctx->SetException("Index out of bounds");
			}
			return;
		}

		asIScriptContext* cmpContext = 0;
		bool isNested = false;

		// Try to reuse the active context
		cmpContext = asGetActiveContext();
		if (cmpContext != nullptr) {
			if (cmpContext->GetEngine() == objType->GetEngine() && cmpContext->PushState() >= 0) {
				isNested = true;
			} else {
				cmpContext = 0;
			}
		}
		if (cmpContext == nullptr) {
			cmpContext = objType->GetEngine()->RequestContext();
		}

		// TODO: Security issue: If the array is accessed from the callback while the sort is going on the result may be unpredictable
		//       For example, the callback resizes the array in the middle of the sort
		//       Possible solution: set a lock flag on the array, and prohibit modifications while the lock flag is set

		// Bubble sort
		// TODO: optimize: Use an efficient sort algorithm
		for (asUINT i = start; i + 1 < end; i++) {
			asUINT best = i;
			for (asUINT j = i + 1; j < end; j++) {
				cmpContext->Prepare(func);
				cmpContext->SetArgAddress(0, At(j));
				cmpContext->SetArgAddress(1, At(best));
				int r = cmpContext->Execute();
				if (r != asEXECUTION_FINISHED) {
					break;
				}
				if (*(bool*)(cmpContext->GetAddressOfReturnValue())) {
					best = j;
				}
			}

			// With Swap we guarantee that the array always sees all references
			// if the GC calls the EnumReferences in the middle of the sorting
			if (best != i) {
				Swap(GetArrayItemPointer(i), GetArrayItemPointer(best));
			}
		}

		if (cmpContext != nullptr) {
			if (isNested) {
				asEContextState state = cmpContext->GetState();
				cmpContext->PopState();
				if (state == asEXECUTION_ABORTED) {
					cmpContext->Abort();
				}
			} else {
				objType->GetEngine()->ReturnContext(cmpContext);
			}
		}
	}

	// internal
	void CScriptArray::CopyBuffer(SArrayBuffer* dst, SArrayBuffer* src)
	{
		asIScriptEngine* engine = objType->GetEngine();
		if (subTypeId & asTYPEID_OBJHANDLE) {
			// Copy the references and increase the reference counters
			if (dst->numElements > 0 && src->numElements > 0) {
				int count = dst->numElements > src->numElements ? src->numElements : dst->numElements;

				void** max = (void**)(dst->data + count * sizeof(void*));
				void** d = (void**)dst->data;
				void** s = (void**)src->data;

				for (; d < max; d++, s++) {
					void* tmp = *d;
					*d = *s;
					if (*d) {
						engine->AddRefScriptObject(*d, objType->GetSubType());
					}
					// Release the old ref after incrementing the new to avoid problem incase it is the same ref
					if (tmp) {
						engine->ReleaseScriptObject(tmp, objType->GetSubType());
					}
				}
			}
		} else {
			if (dst->numElements > 0 && src->numElements > 0) {
				int count = dst->numElements > src->numElements ? src->numElements : dst->numElements;
				if (subTypeId & asTYPEID_MASK_OBJECT) {
					// Call the assignment operator on all of the objects
					void** max = (void**)(dst->data + count * sizeof(void*));
					void** d = (void**)dst->data;
					void** s = (void**)src->data;

					asITypeInfo* subType = objType->GetSubType();
					if (subType->GetFlags() & asOBJ_ASHANDLE) {
						// For objects that should work as handles we must use the opHndlAssign method
						// TODO: Must support alternative syntaxes as well
						// TODO: Move the lookup of the opHndlAssign method to Precache() so it is only done once
						String decl = StringView(subType->GetName()) + "& opHndlAssign(const "_s + StringView(subType->GetName()) + "&in)"_s;
						asIScriptFunction* func = subType->GetMethodByDecl(decl.data());
						if (func) {
							// TODO: Reuse active context if existing
							asIScriptContext* ctx = engine->RequestContext();
							for (; d < max; d++, s++) {
								ctx->Prepare(func);
								ctx->SetObject(*d);
								ctx->SetArgAddress(0, *s);
								// TODO: Handle errors
								ctx->Execute();
							}
							engine->ReturnContext(ctx);
						} else {
							// opHndlAssign doesn't exist, so try ordinary value assign instead
							for (; d < max; d++, s++) {
								engine->AssignScriptObject(*d, *s, subType);
							}
						}
					} else
						for (; d < max; d++, s++) {
							engine->AssignScriptObject(*d, *s, subType);
						}
				} else {
					// Primitives are copied byte for byte
					memcpy(dst->data, src->data, count * elementSize);
				}
			}
		}
	}

	// internal
	// Precache some info
	void CScriptArray::Precache()
	{
		subTypeId = objType->GetSubTypeId();

		// Check if it is an array of objects. Only for these do we need to cache anything
		// Type ids for primitives and enums only has the sequence number part
		if (!(subTypeId & ~asTYPEID_MASK_SEQNBR)) {
			return;
		}

		// The opCmp and opEquals methods are cached because the searching for the
		// methods is quite time consuming if a lot of array objects are created.

		// First check if a cache already exists for this array type
		SArrayCache* cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
		if (cache != nullptr) return;

		// We need to make sure the cache is created only once, even
		// if multiple threads reach the same point at the same time
		asAcquireExclusiveLock();

		// Now that we got the lock, we need to check again to make sure the
		// cache wasn't created while we were waiting for the lock
		cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
		if (cache != nullptr) {
			asReleaseExclusiveLock();
			return;
		}

		// Create the cache
		cache = reinterpret_cast<SArrayCache*>(asAllocMem(sizeof(SArrayCache)));
		if (cache == nullptr) {
			asIScriptContext* ctx = asGetActiveContext();
			if (ctx != nullptr) {
				ctx->SetException("Out of memory");
			}
			asReleaseExclusiveLock();
			return;
		}
		memset(cache, 0, sizeof(SArrayCache));

		// If the sub type is a handle to const, then the methods must be const too
		bool mustBeConst = (subTypeId & asTYPEID_HANDLETOCONST) ? true : false;

		asITypeInfo* subType = objType->GetEngine()->GetTypeInfoById(subTypeId);
		if (subType != nullptr) {
			for (asUINT i = 0; i < subType->GetMethodCount(); i++) {
				asIScriptFunction* func = subType->GetMethodByIndex(i);

				if (func->GetParamCount() == 1 && (!mustBeConst || func->IsReadOnly())) {
					asDWORD flags = 0;
					int returnTypeId = func->GetReturnTypeId(&flags);

					// The method must not return a reference
					if (flags != asTM_NONE) {
						continue;
					}

					// opCmp returns an int and opEquals returns a bool
					bool isCmp = false, isEq = false;
					if (returnTypeId == asTYPEID_INT32 && strcmp(func->GetName(), "opCmp") == 0)
						isCmp = true;
					if (returnTypeId == asTYPEID_BOOL && strcmp(func->GetName(), "opEquals") == 0)
						isEq = true;

					if (!isCmp && !isEq)
						continue;

					// The parameter must either be a reference to the subtype or a handle to the subtype
					int paramTypeId;
					func->GetParam(0, &paramTypeId, &flags);

					if ((paramTypeId & ~(asTYPEID_OBJHANDLE | asTYPEID_HANDLETOCONST)) != (subTypeId & ~(asTYPEID_OBJHANDLE | asTYPEID_HANDLETOCONST)))
						continue;

					if ((flags & asTM_INREF)) {
						if ((paramTypeId & asTYPEID_OBJHANDLE) || (mustBeConst && !(flags & asTM_CONST)))
							continue;
					} else if (paramTypeId & asTYPEID_OBJHANDLE) {
						if (mustBeConst && !(paramTypeId & asTYPEID_HANDLETOCONST))
							continue;
					} else {
						continue;
					}

					if (isCmp) {
						if (cache->cmpFunc || cache->cmpFuncReturnCode) {
							cache->cmpFunc = 0;
							cache->cmpFuncReturnCode = asMULTIPLE_FUNCTIONS;
						} else {
							cache->cmpFunc = func;
						}
					} else if (isEq) {
						if (cache->eqFunc || cache->eqFuncReturnCode) {
							cache->eqFunc = 0;
							cache->eqFuncReturnCode = asMULTIPLE_FUNCTIONS;
						} else {
							cache->eqFunc = func;
						}
					}
				}
			}
		}

		if (cache->eqFunc == 0 && cache->eqFuncReturnCode == 0)
			cache->eqFuncReturnCode = asNO_FUNCTION;
		if (cache->cmpFunc == 0 && cache->cmpFuncReturnCode == 0)
			cache->cmpFuncReturnCode = asNO_FUNCTION;

		// Set the user data only at the end so others that retrieve it will know it is complete
		objType->SetUserData(cache, ARRAY_CACHE);

		asReleaseExclusiveLock();
	}

	// GC behaviour
	void CScriptArray::EnumReferences(asIScriptEngine* engine)
	{
		// TODO: If garbage collection can be done from a separate thread, then this method must be
		//       protected so that it doesn't get lost during the iteration if the array is modified

		// If the array is holding handles, then we need to notify the GC of them
		if (subTypeId & asTYPEID_MASK_OBJECT) {
			void** d = (void**)buffer->data;

			asITypeInfo* subType = engine->GetTypeInfoById(subTypeId);
			if ((subType->GetFlags() & asOBJ_REF)) {
				// For reference types we need to notify the GC of each instance
				for (asUINT n = 0; n < buffer->numElements; n++) {
					if (d[n]) {
						engine->GCEnumCallback(d[n]);
					}
				}
			} else if ((subType->GetFlags() & asOBJ_VALUE) && (subType->GetFlags() & asOBJ_GC)) {
				// For value types we need to forward the enum callback
				// to the object so it can decide what to do
				for (asUINT n = 0; n < buffer->numElements; n++) {
					if (d[n]) {
						engine->ForwardGCEnumReferences(d[n], subType);
					}
				}
			}
		}
	}

	// GC behaviour
	void CScriptArray::ReleaseAllHandles(asIScriptEngine*)
	{
		// Resizing to zero will release everything
		Resize(0);
	}

	void CScriptArray::AddRef() const
	{
		// Clear the GC flag then increase the counter
		gcFlag = false;
		asAtomicInc(refCount);
	}

	void CScriptArray::Release() const
	{
		// Clearing the GC flag then descrease the counter
		gcFlag = false;
		if (asAtomicDec(refCount) == 0) {
			// When reaching 0 no more references to this instance
			// exists and the object should be destroyed
			this->~CScriptArray();
			asFreeMem(const_cast<CScriptArray*>(this));
		}
	}

	// GC behaviour
	int CScriptArray::GetRefCount()
	{
		return refCount;
	}

	// GC behaviour
	void CScriptArray::SetFlag()
	{
		gcFlag = true;
	}

	// GC behaviour
	bool CScriptArray::GetFlag()
	{
		return gcFlag;
	}
}

#endif