| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652 |
- //
- // Copyright (c) 2008-2013 the Urho3D project.
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to deal
- // in the Software without restriction, including without limitation the rights
- // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- // copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
- // THE SOFTWARE.
- //
- #include "Precompiled.h"
- #include "Addons.h"
- #include <cstring>
- #include <new>
- #include <stdio.h>
- // Adapted from Angelscript's scriptarray & scriptstdstring add-ons, but with garbage collection disabled
- namespace Urho3D
- {
- using namespace std;
- /// %Script array buffer.
- struct SArrayBuffer
- {
- asDWORD maxElements;
- asDWORD numElements;
- asBYTE data[1];
- };
- /// %Script array cache.
- 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.
- const asPWORD ARRAY_CACHE = 1000;
- static void CleanupObjectTypeArrayCache(asIObjectType *type)
- {
- SArrayCache *cache = reinterpret_cast<SArrayCache*>(type->GetUserData(ARRAY_CACHE));
- if( cache )
- delete cache;
- }
- static CScriptArray* ScriptArrayFactory2(asIObjectType *ot, asUINT length)
- {
- CScriptArray *a = new CScriptArray(length, ot);
- // It's possible the constructor raised a script exception, in which case we
- // need to free the memory and return null instead, else we get a memory leak.
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx && ctx->GetState() == asEXECUTION_EXCEPTION )
- {
- a->Release();
- return 0;
- }
- return a;
- }
- static CScriptArray* ScriptArrayFactoryDefVal(asIObjectType *ot, asUINT length, void *defVal)
- {
- CScriptArray *a = new CScriptArray(length, defVal, ot);
- // It's possible the constructor raised a script exception, in which case we
- // need to free the memory and return null instead, else we get a memory leak.
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx && ctx->GetState() == asEXECUTION_EXCEPTION )
- {
- a->Release();
- return 0;
- }
- return a;
- }
- static CScriptArray* ScriptArrayFactory(asIObjectType *ot)
- {
- return ScriptArrayFactory2(ot, 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 instanciated 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(asIObjectType *ot, bool &dontGarbageCollect)
- {
- // Urho3D: nothing is garbage collected
- dontGarbageCollect = true;
- // Make sure the subtype can be instanciated with a default factory/constructor,
- // otherwise we won't be able to instanciate the elements.
- int typeId = ot->GetSubTypeId();
- if( typeId == asTYPEID_VOID )
- return false;
- if( (typeId & asTYPEID_MASK_OBJECT) && !(typeId & asTYPEID_OBJHANDLE) )
- {
- asIObjectType *subtype = ot->GetEngine()->GetObjectTypeById(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
- 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( !ot->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
- 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;
- }
- // The type is ok
- return true;
- }
- 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(asUINT length, asIObjectType *ot)
- {
- refCount = 1;
- gcFlag = false;
- objType = ot;
- 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);
- // Urho3D: garbage collection disabled
- /*
- // 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, asIObjectType *ot)
- {
- refCount = 1;
- gcFlag = false;
- objType = ot;
- 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 == 0 ) return;
- if( (subTypeId & ~asTYPEID_MASK_SEQNBR) && !(subTypeId & asTYPEID_OBJHANDLE) )
- objType->GetEngine()->AssignScriptObject(ptr, value, objType->GetSubType());
- else if( subTypeId & asTYPEID_OBJHANDLE )
- {
- void *tmp = *(void**)ptr;
- *(void**)ptr = *(void**)value;
- objType->GetEngine()->AddRefScriptObject(*(void**)value, objType->GetSubType());
- if( tmp )
- 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 )
- {
- DeleteBuffer(buffer);
- buffer = 0;
- }
- if( objType ) 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;
- // Allocate memory for the buffer
- SArrayBuffer *newBuffer;
- #if defined(__S3E__) // Marmalade doesn't understand (nothrow)
- newBuffer = (SArrayBuffer*)new asBYTE[sizeof(SArrayBuffer)-1 + elementSize*maxElements];
- #else
- newBuffer = (SArrayBuffer*)new (nothrow) asBYTE[sizeof(SArrayBuffer)-1 + elementSize*maxElements];
- #endif
- if( newBuffer )
- {
- newBuffer->numElements = buffer->numElements;
- newBuffer->maxElements = maxElements;
- }
- else
- {
- // Out of memory
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- ctx->SetException("Out of memory");
- return;
- }
- memcpy(newBuffer->data, buffer->data, buffer->numElements*elementSize);
- // Release the old buffer
- delete[] (asBYTE*)buffer;
- buffer = newBuffer;
- }
- void CScriptArray::Resize(asUINT numElements)
- {
- if( numElements & 0x80000000 )
- {
- CheckMaxSize(numElements);
- return;
- }
- Resize((int)numElements - (int)buffer->numElements, (asUINT)-1);
- }
- // 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;
- #if defined(__S3E__) // Marmalade doesn't understand (nothrow)
- newBuffer = (SArrayBuffer*)new asBYTE[sizeof(SArrayBuffer)-1 + elementSize*(buffer->numElements + delta)];
- #else
- newBuffer = (SArrayBuffer*)new (nothrow) asBYTE[sizeof(SArrayBuffer)-1 + elementSize*(buffer->numElements + delta)];
- #endif
- if( newBuffer )
- {
- newBuffer->numElements = buffer->numElements + delta;
- newBuffer->maxElements = newBuffer->numElements;
- }
- else
- {
- // Out of memory
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- ctx->SetException("Out of memory");
- return;
- }
- 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);
- if( subTypeId & asTYPEID_MASK_OBJECT )
- Construct(newBuffer, at, at+delta);
- // Release the old buffer
- delete[] (asBYTE*)buffer;
- buffer = newBuffer;
- }
- else if( delta < 0 )
- {
- Destruct(buffer, at, at-delta);
- memmove(buffer->data + at*elementSize, buffer->data + (at-delta)*elementSize, (buffer->numElements - (at-delta))*elementSize);
- buffer->numElements += delta;
- }
- else
- {
- 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( subTypeId & asTYPEID_MASK_OBJECT )
- maxSize /= sizeof(void*);
- else if( elementSize > 0 )
- maxSize /= elementSize;
- if( numElements > maxSize )
- {
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- {
- // Set a script exception
- ctx->SetException("Too large array size");
- }
- return false;
- }
- // OK
- return true;
- }
- asIObjectType *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::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 == 0 || 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 0;
- }
- if( (subTypeId & asTYPEID_MASK_OBJECT) && !(subTypeId & asTYPEID_OBJHANDLE) )
- return (void*)((size_t*)buffer->data)[index];
- else
- return buffer->data + elementSize*index;
- }
- void *CScriptArray::At(asUINT index)
- {
- return const_cast<void*>(const_cast<const CScriptArray *>(this)->At(index));
- }
- // internal
- void CScriptArray::CreateBuffer(SArrayBuffer **buf, asUINT numElements)
- {
- if( subTypeId & asTYPEID_MASK_OBJECT )
- {
- #if defined(__S3E__) // Marmalade doesn't understand (nothrow)
- *buf = (SArrayBuffer*)new asBYTE[sizeof(SArrayBuffer)-1+sizeof(void*)*numElements];
- #else
- *buf = (SArrayBuffer*)new (nothrow) asBYTE[sizeof(SArrayBuffer)-1+sizeof(void*)*numElements];
- #endif
- }
- else
- {
- #if defined(__S3E__)
- *buf = (SArrayBuffer*)new asBYTE[sizeof(SArrayBuffer)-1+elementSize*numElements];
- #else
- *buf = (SArrayBuffer*)new (nothrow) asBYTE[sizeof(SArrayBuffer)-1+elementSize*numElements];
- #endif
- }
- if( *buf )
- {
- (*buf)->numElements = numElements;
- (*buf)->maxElements = numElements;
- Construct(*buf, 0, numElements);
- }
- else
- {
- // Oops, out of memory
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- ctx->SetException("Out of memory");
- }
- }
- // internal
- void CScriptArray::DeleteBuffer(SArrayBuffer *buf)
- {
- Destruct(buf, 0, buf->numElements);
- // Free the buffer
- delete[] (asBYTE*)buf;
- }
- // internal
- void CScriptArray::Construct(SArrayBuffer *buf, asUINT start, asUINT end)
- {
- if( subTypeId & asTYPEID_OBJHANDLE )
- {
- // Set all object handles to null
- void *d = (void*)(buf->data + start * sizeof(void*));
- memset(d, 0, (end-start)*sizeof(void*));
- }
- else if( subTypeId & asTYPEID_MASK_OBJECT )
- {
- void **max = (void**)(buf->data + end * sizeof(void*));
- void **d = (void**)(buf->data + start * sizeof(void*));
- asIScriptEngine *engine = objType->GetEngine();
- asIObjectType *subType = objType->GetSubType();
- for( ; d < max; d++ )
- *d = (void*)engine->CreateScriptObject(subType);
- }
- }
- // 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, asIScriptContext *ctx, SArrayCache *cache)
- {
- 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(signed char);
- case asTYPEID_UINT8: return COMPARE(unsigned char);
- case asTYPEID_INT16: return COMPARE(signed short);
- case asTYPEID_UINT16: return COMPARE(unsigned short);
- case asTYPEID_INT32: return COMPARE(signed int);
- case asTYPEID_UINT32: return COMPARE(unsigned int);
- 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
- }
- }
- else
- {
- int r = 0;
- if( subTypeId & asTYPEID_OBJHANDLE )
- {
- // Allow sort to work even if the array contains null handles
- if( *(void**)a == 0 ) return true;
- if( *(void**)b == 0 ) return false;
- }
- // Execute object opCmp
- if( cache && cache->cmpFunc )
- {
- // 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;
- }
- 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 )
- {
- if( cmpContext->PushState() >= 0 )
- isNested = true;
- else
- cmpContext = 0;
- }
- if( cmpContext == 0 )
- {
- // 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 )
- 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(signed char);
- case asTYPEID_UINT8: return COMPARE(unsigned char);
- case asTYPEID_INT16: return COMPARE(signed short);
- case asTYPEID_UINT16: return COMPARE(unsigned short);
- case asTYPEID_INT32: return COMPARE(signed int);
- case asTYPEID_UINT32: return COMPARE(unsigned int);
- case asTYPEID_FLOAT: return COMPARE(float);
- case asTYPEID_DOUBLE: return COMPARE(double);
- default: return COMPARE(signed int); // All enums fall here
- #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 && cache->eqFunc )
- {
- // 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 && cache->cmpFunc )
- {
- // 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::Find(void *value) const
- {
- return Find(0, value);
- }
- int CScriptArray::Find(asUINT index, 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 || (cache->cmpFunc == 0 && cache->eqFunc == 0) )
- {
- asIScriptContext *ctx = asGetActiveContext();
- asIObjectType* subType = objType->GetEngine()->GetObjectTypeById(subTypeId);
- // Throw an exception
- if( ctx )
- {
- char tmp[512];
- if( cache && cache->eqFuncReturnCode == asMULTIPLE_FUNCTIONS )
- #if defined(_MSC_VER) && _MSC_VER >= 1500 && !defined(__S3E__)
- sprintf_s(tmp, 512, "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(_MSC_VER) && _MSC_VER >= 1500 && !defined(__S3E__)
- sprintf_s(tmp, 512, "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 )
- {
- if( cmpContext->PushState() >= 0 )
- isNested = true;
- else
- cmpContext = 0;
- }
- if( cmpContext == 0 )
- {
- // 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();
- if( index < size )
- {
- for( asUINT i = index; i < size; i++ )
- {
- // value passed by reference
- if( Equals(At(i), (value), cmpContext, cache) )
- {
- ret = (int)i;
- break;
- }
- }
- }
- if( cmpContext )
- 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
- void CScriptArray::Copy(void *dst, void *src)
- {
- memcpy(dst, src, elementSize);
- }
- // 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 *buffer)
- {
- if ((subTypeId & asTYPEID_MASK_OBJECT) && !(subTypeId & asTYPEID_OBJHANDLE) )
- {
- // Real address of object
- return reinterpret_cast<void*>(*(size_t*)buffer);
- }
- else
- {
- // Primitive is just a raw data
- return buffer;
- }
- }
- // Sort ascending
- void CScriptArray::SortAsc()
- {
- Sort(0, GetSize(), true);
- }
- // Sort ascending
- void CScriptArray::SortAsc(asUINT index, asUINT count)
- {
- Sort(index, count, true);
- }
- // Sort descending
- void CScriptArray::SortDesc()
- {
- Sort(0, GetSize(), false);
- }
- // Sort descending
- void CScriptArray::SortDesc(asUINT index, asUINT count)
- {
- Sort(index, count, false);
- }
- // internal
- void CScriptArray::Sort(asUINT index, 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 || cache->cmpFunc == 0 )
- {
- asIScriptContext *ctx = asGetActiveContext();
- asIObjectType* subType = objType->GetEngine()->GetObjectTypeById(subTypeId);
- // Throw an exception
- if( ctx )
- {
- char tmp[512];
- if( cache && cache->cmpFuncReturnCode == asMULTIPLE_FUNCTIONS )
- #if defined(_MSC_VER) && _MSC_VER >= 1500 && !defined(__S3E__)
- sprintf_s(tmp, 512, "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(_MSC_VER) && _MSC_VER >= 1500 && !defined(__S3E__)
- sprintf_s(tmp, 512, "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 = index;
- int end = index + 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;
- }
- asBYTE tmp[16];
- asIScriptContext *cmpContext = 0;
- bool isNested = false;
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- // Try to reuse the active context
- cmpContext = asGetActiveContext();
- if( cmpContext )
- {
- if( cmpContext->PushState() >= 0 )
- isNested = true;
- else
- cmpContext = 0;
- }
- if( cmpContext == 0 )
- {
- // 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();
- }
- }
- // Insertion sort
- 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, cmpContext, cache) )
- {
- Copy(GetArrayItemPointer(j + 1), GetArrayItemPointer(j));
- j--;
- }
- Copy(GetArrayItemPointer(j + 1), tmp);
- }
- if( cmpContext )
- if( isNested )
- {
- asEContextState state = cmpContext->GetState();
- cmpContext->PopState();
- if( state == asEXECUTION_ABORTED )
- cmpContext->Abort();
- }
- else
- cmpContext->Release();
- }
- // 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;
- asIObjectType *subType = objType->GetSubType();
- 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 ) 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 )
- {
- asReleaseExclusiveLock();
- return;
- }
- // Create the cache
- cache = new SArrayCache();
- 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;
- asIObjectType *subType = objType->GetEngine()->GetObjectTypeById(subTypeId);
- if( subType )
- {
- 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->GetParamTypeId(0, &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)
- {
- // 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;
- for( asUINT n = 0; n < buffer->numElements; n++ )
- {
- if( d[n] )
- engine->GCEnumCallback(d[n]);
- }
- }
- }
- // GC behaviour
- void CScriptArray::ReleaseAllHandles(asIScriptEngine *engine)
- {
- // 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
- delete this;
- }
- }
- // GC behaviour
- int CScriptArray::GetRefCount()
- {
- return refCount;
- }
- // GC behaviour
- void CScriptArray::SetFlag()
- {
- gcFlag = true;
- }
- // GC behaviour
- bool CScriptArray::GetFlag()
- {
- return gcFlag;
- }
- static void ScriptArrayClear(CScriptArray* ptr)
- {
- ptr->Resize(0);
- }
- void RegisterArray(asIScriptEngine* engine)
- {
- // Register the object type user data clean up
- engine->SetObjectTypeUserDataCleanupCallback(CleanupObjectTypeArrayCache, ARRAY_CACHE);
- engine->RegisterObjectType("Array<class T>", 0, asOBJ_REF | asOBJ_TEMPLATE);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_TEMPLATE_CALLBACK, "bool f(int&in, bool&out)", asFUNCTION(ScriptArrayTemplateCallback), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_FACTORY, "Array<T>@ f(int& in)", asFUNCTIONPR(ScriptArrayFactory, (asIObjectType*), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_FACTORY, "Array<T>@ f(int& in, uint)", asFUNCTIONPR(ScriptArrayFactory2, (asIObjectType*, asUINT), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_FACTORY, "Array<T>@ f(int& in, uint, const T& in)", asFUNCTIONPR(ScriptArrayFactoryDefVal, (asIObjectType*, asUINT, void *), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_LIST_FACTORY, "Array<T>@ f(int& in, uint)", asFUNCTIONPR(ScriptArrayFactory2, (asIObjectType*, asUINT), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_ADDREF, "void f()", asMETHOD(CScriptArray,AddRef), asCALL_THISCALL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_RELEASE, "void f()", asMETHOD(CScriptArray,Release), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "T& opIndex(uint)", asMETHODPR(CScriptArray, At, (unsigned), void*), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "const T& opIndex(uint) const", asMETHODPR(CScriptArray, At, (unsigned), void*), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "Array<T>& opAssign(const Array<T>& in)", asMETHOD(CScriptArray, operator=), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Insert(uint, const T& in)", asMETHOD(CScriptArray, InsertAt), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Erase(uint)", asMETHOD(CScriptArray, RemoveAt), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Push(const T& in)", asMETHOD(CScriptArray, InsertLast), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Pop()", asMETHOD(CScriptArray, RemoveLast), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Resize(uint)", asMETHODPR(CScriptArray, Resize, (asUINT), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Clear()", asFUNCTION(ScriptArrayClear), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("Array<T>", "void Sort()", asMETHODPR(CScriptArray, SortAsc, (), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Sort(uint, uint)", asMETHODPR(CScriptArray, SortAsc, (asUINT, asUINT), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void SortReverse()", asMETHODPR(CScriptArray, SortDesc, (), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void SortReverse(uint, uint)", asMETHODPR(CScriptArray, SortDesc, (asUINT, asUINT), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void Reverse()", asMETHOD(CScriptArray, Reverse), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "int Find(const T&in) const", asMETHODPR(CScriptArray, Find, (void*) const, int), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "int Find(uint, const T&in) const", asMETHODPR(CScriptArray, Find, (asUINT, void*) const, int), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "bool opEquals(const Array<T>&in) const", asMETHOD(CScriptArray, operator==), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "uint get_length() const", asMETHOD(CScriptArray, GetSize), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "void set_length(uint)", asMETHODPR(CScriptArray, Resize, (asUINT), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "bool get_empty() const", asMETHOD(CScriptArray, IsEmpty), asCALL_THISCALL);
- engine->RegisterDefaultArrayType("Array<T>");
- }
- static String StringFactory(asUINT length, const char* s)
- {
- return String(s, length);
- }
- static void ConstructString(String* ptr)
- {
- new(ptr) String();
- }
- static void ConstructStringCopy(const String& str, String* ptr)
- {
- new(ptr) String(str);
- }
- static void DestructString(String* ptr)
- {
- ptr->~String();
- }
- static char* StringCharAt(unsigned int i, String& str)
- {
- if (i >= str.Length())
- {
- asIScriptContext* context = asGetActiveContext();
- if (context)
- context->SetException("Index out of bounds");
- return 0;
- }
- return &str[i];
- }
- static int StringCmp(const String& lhs, const String& rhs)
- {
- int cmp = 0;
- if (lhs < rhs)
- cmp = -1;
- else if (lhs > rhs)
- cmp = 1;
- return cmp;
- }
- void StringResize(unsigned newSize, String& str)
- {
- unsigned oldSize = str.Length();
- str.Resize(newSize);
- for (unsigned i = oldSize; i < newSize; ++i)
- str[i] = ' ';
- }
- static void ConstructStringInt(int value, String* ptr)
- {
- new(ptr) String(value);
- }
- static void ConstructStringUInt(unsigned value, String* ptr)
- {
- new(ptr) String(value);
- }
- static void ConstructStringFloat(float value, String* ptr)
- {
- new(ptr) String(value);
- }
- static void ConstructStringBool(bool value, String* ptr)
- {
- new(ptr) String(value);
- }
- static String& StringAssignInt(int value, String& str)
- {
- str = String(value);
- return str;
- }
- static String& StringAddAssignInt(int value, String& str)
- {
- str += String(value);
- return str;
- }
- static String StringAddInt(int value, const String& str)
- {
- return str + String(value);
- }
- static String StringAddIntReverse(int value, const String& str)
- {
- return String(value) + str;
- }
- static String& StringAssignUInt(unsigned value, String& str)
- {
- str = String(value);
- return str;
- }
- static String& StringAddAssignUInt(unsigned value, String& str)
- {
- str += String(value);
- return str;
- }
- static String StringAddUInt(unsigned value, const String& str)
- {
- return str + String(value);
- }
- static String StringAddUIntReverse(unsigned value, const String& str)
- {
- return String(value) + str;
- }
- static String& StringAssignFloat(float value, String& str)
- {
- str = String(value);
- return str;
- }
- static String& StringAddAssignFloat(float value, String& str)
- {
- str += String(value);
- return str;
- }
- static String StringAddFloat(float value, const String& str)
- {
- return str + String(value);
- }
- static String StringAddFloatReverse(float value, const String& str)
- {
- return String(value) + str;
- }
- static String& StringAssignBool(bool value, String& str)
- {
- str = String(value);
- return str;
- }
- static String& StringAddAssignBool(bool value, String& str)
- {
- str += String(value);
- return str;
- }
- static String StringAddBool(bool value, const String& str)
- {
- return str + String(value);
- }
- static String StringAddBoolReverse(bool value, const String& str)
- {
- return String(value) + str;
- }
- static void StringSetUTF8FromLatin1(const String& src, String& str)
- {
- str.SetUTF8FromLatin1(src.CString());
- }
- void RegisterString(asIScriptEngine *engine)
- {
- engine->RegisterObjectType("String", sizeof(String), asOBJ_VALUE | asOBJ_APP_CLASS_CDAK);
- engine->RegisterStringFactory("String", asFUNCTION(StringFactory), asCALL_CDECL);
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f()", asFUNCTION(ConstructString), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f(const String&in)", asFUNCTION(ConstructStringCopy), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("String", asBEHAVE_DESTRUCT, "void f()", asFUNCTION(DestructString), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAssign(const String&in)", asMETHODPR(String, operator =, (const String&), String&), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String& opAddAssign(const String&in)", asMETHODPR(String, operator +=, (const String&), String&), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool opEquals(const String&in) const", asMETHODPR(String, operator ==, (const String&) const, bool), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "int opCmp(const String&in) const", asFUNCTION(StringCmp), asCALL_CDECL_OBJFIRST);
- engine->RegisterObjectMethod("String", "String opAdd(const String&in) const", asMETHODPR(String, operator +, (const String&) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint8 &opIndex(uint)", asFUNCTION(StringCharAt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "const uint8 &opIndex(uint) const", asFUNCTION(StringCharAt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "void Replace(uint8, uint8, bool caseSensitive = true)", asMETHODPR(String, Replace, (char, char, bool), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void Replace(const String&in, const String&in, bool caseSensitive = true)", asMETHODPR(String, Replace, (const String&, const String&, bool), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String Replaced(uint8, uint8, bool caseSensitive = true) const", asMETHODPR(String, Replaced, (char, char, bool) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String Replaced(const String&in, const String&in, bool caseSensitive = true) const", asMETHODPR(String, Replaced, (const String&, const String&, bool) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void Resize(uint)", asFUNCTION(StringResize), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "uint Find(const String&in, uint start = 0, bool caseSensitive = true) const", asMETHODPR(String, Find, (const String&, unsigned, bool) const, unsigned), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint Find(uint8, uint start = 0, bool caseSensitive = true) const", asMETHODPR(String, Find, (char, unsigned, bool) const, unsigned), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint FindLast(const String&in, uint start = 0xffffffff, bool caseSensitive = true) const", asMETHODPR(String, FindLast, (const String&, unsigned, bool) const, unsigned), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint FindLast(uint8, uint start = 0xffffffff, bool caseSensitive = true) const", asMETHODPR(String, FindLast, (char, unsigned, bool) const, unsigned), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool StartsWith(const String&in, bool caseSensitive = true) const", asMETHOD(String, StartsWith), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool EndsWith(const String&in, bool caseSensitive = true) const", asMETHOD(String, EndsWith), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String Substring(uint) const", asMETHODPR(String, Substring, (unsigned) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String Substring(uint, uint) const", asMETHODPR(String, Substring, (unsigned, unsigned) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String ToUpper() const", asMETHOD(String, ToUpper), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String ToLower() const", asMETHOD(String, ToLower), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String Trimmed() const", asMETHOD(String, Trimmed), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void SetUTF8FromLatin1(const String& in)", asFUNCTION(StringSetUTF8FromLatin1), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "uint get_utf8Length() const", asMETHOD(String, LengthUTF8), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint ByteOffsetUTF8(uint) const", asMETHOD(String, ByteOffsetUTF8), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint NextUTF8Char(uint&) const", asMETHOD(String, NextUTF8Char), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint AtUTF8(uint) const", asMETHOD(String, AtUTF8), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void ReplaceUTF8(uint, uint)", asMETHOD(String, ReplaceUTF8), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void AppendUTF8(uint)", asMETHOD(String, AppendUTF8), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String SubstringUTF8(uint) const", asMETHODPR(String, SubstringUTF8, (unsigned) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "String SubstringUTF8(uint, uint) const", asMETHODPR(String, SubstringUTF8, (unsigned, unsigned) const, String), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "uint get_length() const", asMETHOD(String, Length), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool get_empty() const", asMETHOD(String, Empty), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "int Compare(const String&in, bool caseSensitive = true) const", asMETHODPR(String, Compare, (const String&, bool) const, int), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool Contains(const String&in) const", asMETHODPR(String, Contains, (const String&) const, bool), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "bool Contains(uint8) const", asMETHODPR(String, Contains, (char) const, bool), asCALL_THISCALL);
- engine->RegisterObjectMethod("String", "void Clear()", asMETHOD(String, Clear), asCALL_THISCALL);
-
- // Register automatic conversion functions for convenience
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f(int)", asFUNCTION(ConstructStringInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f(uint)", asFUNCTION(ConstructStringUInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f(float)", asFUNCTION(ConstructStringFloat), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("String", asBEHAVE_CONSTRUCT, "void f(bool)", asFUNCTION(ConstructStringBool), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAssign(int)", asFUNCTION(StringAssignInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAddAssign(int)", asFUNCTION(StringAddAssignInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd(int) const", asFUNCTION(StringAddInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd_r(int) const", asFUNCTION(StringAddIntReverse), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAssign(uint)", asFUNCTION(StringAssignUInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAddAssign(uint)", asFUNCTION(StringAddAssignUInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd(uint) const", asFUNCTION(StringAddUInt), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd_r(uint) const", asFUNCTION(StringAddUIntReverse), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAssign(float)", asFUNCTION(StringAssignFloat), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAddAssign(float)", asFUNCTION(StringAddAssignFloat), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd(float) const", asFUNCTION(StringAddFloat), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd_r(float) const", asFUNCTION(StringAddFloatReverse), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAssign(bool)", asFUNCTION(StringAssignBool), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String& opAddAssign(bool)", asFUNCTION(StringAddAssignBool), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd(bool) const", asFUNCTION(StringAddBool), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("String", "String opAdd_r(bool) const", asFUNCTION(StringAddBoolReverse), asCALL_CDECL_OBJLAST);
- }
- }
|