| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487 |
- //
- // Copyright (c) 2008-2020 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 "../AngelScript/Addons.h"
- #include <cstring>
- #include <new>
- #include <cstdio>
- // Adapted from Angelscript's scriptarray & scriptstdstring add-ons, but with garbage collection disabled
- namespace Urho3D
- {
- using namespace std;
- // Set the default memory routines
- // Use the angelscript engine's memory routines by default
- static asALLOCFUNC_t userAlloc = asAllocMem;
- static asFREEFUNC_t userFree = asFreeMem;
- // Allows the application to set which memory routines should be used by the array object
- void CScriptArray::SetMemoryFunctions(asALLOCFUNC_t allocFunc, asFREEFUNC_t freeFunc)
- {
- userAlloc = allocFunc;
- userFree = freeFunc;
- }
- static void RegisterScriptArray_Native(asIScriptEngine *engine);
- static void RegisterScriptArray_Generic(asIScriptEngine *engine);
- /// %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 CleanupTypeInfoArrayCache(asITypeInfo *type)
- {
- auto*cache = reinterpret_cast<SArrayCache*>(type->GetUserData(ARRAY_CACHE));
- if (cache)
- {
- cache->~SArrayCache();
- userFree(cache);
- }
- }
- CScriptArray* CScriptArray::Create(asITypeInfo *ot, asUINT length)
- {
- asIScriptContext *ctx = asGetActiveContext();
- // Allocate the memory
- void *mem = userAlloc(sizeof(CScriptArray));
- if( mem == nullptr )
- {
- if( ctx )
- ctx->SetException("Out of memory");
- return nullptr;
- }
- // Initialize the object
- auto*a = new(mem) 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.
- if( ctx && ctx->GetState() == asEXECUTION_EXCEPTION )
- {
- a->Release();
- return nullptr;
- }
- return a;
- }
- CScriptArray* CScriptArray::Create(asITypeInfo *ot, void *initList)
- {
- asIScriptContext *ctx = asGetActiveContext();
- // Allocate the memory
- void *mem = userAlloc(sizeof(CScriptArray));
- if( mem == nullptr )
- {
- if( ctx )
- ctx->SetException("Out of memory");
- return nullptr;
- }
- // Initialize the object
- auto*a = new(mem) CScriptArray(ot, initList);
- // 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.
- if( ctx && ctx->GetState() == asEXECUTION_EXCEPTION )
- {
- a->Release();
- return nullptr;
- }
- return a;
- }
- CScriptArray* CScriptArray::Create(asITypeInfo *ot, asUINT length, void *defVal)
- {
- asIScriptContext *ctx = asGetActiveContext();
- // Allocate the memory
- void *mem = userAlloc(sizeof(CScriptArray));
- if( mem == nullptr )
- {
- if( ctx )
- ctx->SetException("Out of memory");
- return nullptr;
- }
- // Initialize the object
- auto*a = new(mem) 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.
- if( ctx && ctx->GetState() == asEXECUTION_EXCEPTION )
- {
- a->Release();
- return nullptr;
- }
- return a;
- }
- CScriptArray* CScriptArray::Create(asITypeInfo *ot)
- {
- return CScriptArray::Create(ot, 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
- // 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 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;
- }
- 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;
- }
- 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 *ot, void *buf)
- {
- refCount = 1;
- gcFlag = false;
- objType = ot;
- objType->AddRef();
- buffer = nullptr;
- Precache();
- asIScriptEngine *engine = ot->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( (ot->GetSubTypeId() & asTYPEID_MASK_OBJECT) == 0 )
- {
- CreateBuffer(&buffer, length);
- // Copy the values of the primitive type into the internal buffer
- memcpy(At(0), (((asUINT*)buf)+1), length * elementSize);
- }
- else if( ot->GetSubTypeId() & asTYPEID_OBJHANDLE )
- {
- CreateBuffer(&buffer, length);
- // Copy the handles into the internal buffer
- 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( ot->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
- 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);
- auto*srcObj = (asBYTE*)buf;
- srcObj += 4 + n*ot->GetSubType()->GetSize();
- engine->AssignScriptObject(obj, srcObj, ot->GetSubType());
- }
- }
- // Urho3D: garbage collection disabled
- // Notify the GC of the successful creation
- /*
- if( objType->GetFlags() & asOBJ_GC )
- objType->GetEngine()->NotifyGarbageCollectorOfNewObject(this, objType);
- */
- }
- CScriptArray::CScriptArray(asUINT length, asITypeInfo *ot)
- {
- refCount = 1;
- gcFlag = false;
- objType = ot;
- objType->AddRef();
- buffer = nullptr;
- 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 = nullptr;
- Precache();
- elementSize = other.elementSize;
- // Urho3D: garbage collection disabled
- /*
- 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 *ot)
- {
- refCount = 1;
- gcFlag = false;
- objType = ot;
- objType->AddRef();
- buffer = nullptr;
- 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);
- */
- // 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) )
- 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 = nullptr;
- }
- 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;
- if( !CheckMaxSize(maxElements) )
- return;
- // Allocate memory for the buffer
- auto*newBuffer = reinterpret_cast<SArrayBuffer*>(userAlloc(sizeof(SArrayBuffer)-1 + elementSize*maxElements));
- if( newBuffer )
- {
- newBuffer->numElements = buffer->numElements;
- newBuffer->maxElements = maxElements;
- }
- else
- {
- // Out of memory
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- ctx->SetException("Out of memory");
- return;
- }
- // TODO: memcpy assumes the objects in the array doesn't hold pointers to themselves
- // This should really be using the objects copy/move constructor to copy each object
- // to the new location. It would most likely be a hit on the performance though.
- memcpy(newBuffer->data, buffer->data, buffer->numElements*elementSize);
- // Release the old buffer
- userFree(buffer);
- buffer = newBuffer;
- }
- void CScriptArray::Resize(asUINT numElements)
- {
- if( !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
- auto*newBuffer = reinterpret_cast<SArrayBuffer*>(userAlloc(sizeof(SArrayBuffer)-1 + elementSize*(buffer->numElements + delta)));
- 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;
- }
- // TODO: memcpy assumes the objects in the array doesn't hold pointers to themselves
- // This should really be using the objects copy/move constructor to copy each object
- // to the new location. It would most likely be a hit on the performance though.
- 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
- userFree(buffer);
- buffer = newBuffer;
- }
- else if( delta < 0 )
- {
- Destruct(buffer, at, at-delta);
- // TODO: memmove assumes the objects in the array doesn't hold pointers to themselves
- // This should really be using the objects copy/move constructor to copy each object
- // to the new location. It would most likely be a hit on the performance though.
- memmove(buffer->data + at*elementSize, buffer->data + (at-delta)*elementSize, (buffer->numElements - (at-delta))*elementSize);
- buffer->numElements += delta;
- }
- else
- {
- // TODO: memmove assumes the objects in the array doesn't hold pointers to themselves
- // This should really be using the objects copy/move constructor to copy each object
- // to the new location. It would most likely be a hit on the performance though.
- 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;
- }
- // OK
- 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::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 )
- ctx->SetException("Index out of bounds");
- return nullptr;
- }
- 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));
- }
- // internal
- void CScriptArray::CreateBuffer(SArrayBuffer **buf, asUINT numElements)
- {
- *buf = reinterpret_cast<SArrayBuffer*>(userAlloc(sizeof(SArrayBuffer)-1+elementSize*numElements));
- 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
- userFree(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
- auto**max = (void**)(buf->data + end * sizeof(void*));
- auto**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 == nullptr )
- {
- // 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
- auto*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();
- auto**max = (void**)(buf->data + end * sizeof(void*));
- auto**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 == nullptr ) return true;
- if( *(void**)b == nullptr ) 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 = nullptr;
- bool isNested = false;
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- // Try to reuse the active context
- cmpContext = asGetActiveContext();
- if( cmpContext )
- {
- 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 every time. 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;
- auto*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::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;
- }
- bool CScriptArray::Swap(CScriptArray& other)
- {
- if (other.GetArrayObjectType() == GetArrayObjectType())
- {
- Urho3D::Swap(buffer, other.buffer);
- return true;
- }
- return false;
- }
- 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 = nullptr;
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
- if( !cache || (cache->cmpFunc == nullptr && cache->eqFunc == nullptr) )
- {
- asIScriptContext *ctx = asGetActiveContext();
- asITypeInfo* subType = objType->GetEngine()->GetTypeInfoById(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 = nullptr;
- bool isNested = false;
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- // Try to reuse the active context
- cmpContext = asGetActiveContext();
- if( cmpContext )
- {
- 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 every time. 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 )
- {
- 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 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
- auto*cache = reinterpret_cast<SArrayCache*>(objType->GetUserData(ARRAY_CACHE));
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- if( !cache || cache->cmpFunc == nullptr )
- {
- asIScriptContext *ctx = asGetActiveContext();
- asITypeInfo* subType = objType->GetEngine()->GetTypeInfoById(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 = 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;
- }
- asBYTE tmp[16];
- asIScriptContext *cmpContext = nullptr;
- bool isNested = false;
- if( subTypeId & ~asTYPEID_MASK_SEQNBR )
- {
- // Try to reuse the active context
- cmpContext = asGetActiveContext();
- if( cmpContext )
- {
- 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 every time. 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;
- auto**max = (void**)(dst->data + count * sizeof(void*));
- auto**d = (void**)dst->data;
- auto**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
- auto**max = (void**)(dst->data + count * sizeof(void*));
- auto**d = (void**)dst->data;
- auto**s = (void**)src->data;
- asITypeInfo *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
- auto*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 = reinterpret_cast<SArrayCache*>(userAlloc(sizeof(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;
- asITypeInfo *subType = objType->GetEngine()->GetTypeInfoById(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->GetParam(0, ¶mTypeId, &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 = nullptr;
- cache->cmpFuncReturnCode = asMULTIPLE_FUNCTIONS;
- }
- else
- cache->cmpFunc = func;
- }
- else if( isEq )
- {
- if( cache->eqFunc || cache->eqFuncReturnCode )
- {
- cache->eqFunc = nullptr;
- cache->eqFuncReturnCode = asMULTIPLE_FUNCTIONS;
- }
- else
- cache->eqFunc = func;
- }
- }
- }
- }
- if( cache->eqFunc == nullptr && cache->eqFuncReturnCode == 0 )
- cache->eqFuncReturnCode = asNO_FUNCTION;
- if( cache->cmpFunc == nullptr && 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 )
- {
- auto**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 *)
- {
- // 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();
- userFree(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;
- }
- static void ScriptArrayClear(CScriptArray* ptr)
- {
- ptr->Resize(0);
- }
- void RegisterArray(asIScriptEngine* engine)
- {
- 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(CScriptArray::Create, (asITypeInfo*), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_FACTORY, "Array<T>@ f(int&in, uint)", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*, asUINT), CScriptArray*), asCALL_CDECL);
- engine->RegisterObjectBehaviour("Array<T>", asBEHAVE_FACTORY, "Array<T>@ f(int&in, uint, const T &in)", asFUNCTIONPR(CScriptArray::Create, (asITypeInfo*, asUINT, void *), CScriptArray*), asCALL_CDECL);
- 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);
- 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 Reserve(uint)", asMETHOD(CScriptArray, Reserve), 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>", "int FindByRef(const T&in) const", asMETHODPR(CScriptArray, FindByRef, (void*) const, int), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "int FindByRef(uint, const T&in) const", asMETHODPR(CScriptArray, FindByRef, (asUINT, void*) const, int), asCALL_THISCALL);
- engine->RegisterObjectMethod("Array<T>", "bool Swap(Array<T>&inout)", asMETHOD(CScriptArray, Swap), 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>");
- }
- CScriptDictionary *CScriptDictionary::Create(asIScriptEngine *engine)
- {
- // Use the custom memory routine from AngelScript to allow application to better control how much memory is used
- auto*obj = (CScriptDictionary*)asAllocMem(sizeof(CScriptDictionary));
- new(obj) CScriptDictionary(engine);
- return obj;
- }
- CScriptDictionary *CScriptDictionary::Create(asBYTE *buffer)
- {
- // Use the custom memory routine from AngelScript to allow application to better control how much memory is used
- auto*obj = (CScriptDictionary*)asAllocMem(sizeof(CScriptDictionary));
- new(obj) CScriptDictionary(buffer);
- return obj;
- }
- CScriptDictionary::CScriptDictionary(asIScriptEngine *engine)
- {
- // We start with one reference
- refCount = 1;
- gcFlag = false;
- // Keep a reference to the engine for as long as we live
- // We don't increment the reference counter, because the
- // engine will hold a pointer to the object.
- this->engine = engine;
- // Urho3D: garbage collection disabled
- /*
- // Notify the garbage collector of this object
- // TODO: The object type should be cached
- engine->NotifyGarbageCollectorOfNewObject(this, engine->GetObjectTypeByName("Dictionary"));
- */
- }
- CScriptDictionary::CScriptDictionary(asBYTE *buffer)
- {
- // We start with one reference
- refCount = 1;
- gcFlag = false;
- // This constructor will always be called from a script
- // so we can get the engine from the active context
- asIScriptContext *ctx = asGetActiveContext();
- engine = ctx->GetEngine();
- // Urho3D: garbage collection disabled
- /*
- // Notify the garbage collector of this object
- // TODO: The type id should be cached
- engine->NotifyGarbageCollectorOfNewObject(this, engine->GetObjectTypeByName("Dictionary"));
- */
- // Initialize the dictionary from the buffer
- asUINT length = *(asUINT*)buffer;
- buffer += 4;
- while( length-- )
- {
- // Align the buffer pointer on a 4 byte boundary in
- // case previous value was smaller than 4 bytes
- if( asPWORD(buffer) & 0x3 )
- buffer += 4 - (asPWORD(buffer) & 0x3);
- // Get the name value pair from the buffer and insert it in the dictionary
- String name = *(String*)buffer;
- buffer += sizeof(String);
- // Get the type id of the value
- int typeId = *(int*)buffer;
- buffer += sizeof(int);
- // Depending on the type id, the value will inline in the buffer or a pointer
- auto*ref = (void*)buffer;
- if( typeId >= asTYPEID_INT8 && typeId <= asTYPEID_DOUBLE )
- {
- // Convert primitive values to either int64 or double, so we can use the overloaded Set methods
- asINT64 i64;
- double d;
- switch( typeId )
- {
- case asTYPEID_INT8: i64 = *(char*)ref; break;
- case asTYPEID_INT16: i64 = *(short*)ref; break;
- case asTYPEID_INT32: i64 = *(int*)ref; break;
- case asTYPEID_INT64: i64 = *(asINT64*)ref; break;
- case asTYPEID_UINT8: i64 = *(unsigned char*)ref; break;
- case asTYPEID_UINT16: i64 = *(unsigned short*)ref; break;
- case asTYPEID_UINT32: i64 = *(unsigned int*)ref; break;
- case asTYPEID_UINT64: i64 = *(asINT64*)ref; break;
- case asTYPEID_FLOAT: d = *(float*)ref; break;
- case asTYPEID_DOUBLE: d = *(double*)ref; break;
- }
- if( typeId >= asTYPEID_FLOAT )
- Set(name, d);
- else
- Set(name, i64);
- }
- else
- {
- if( (typeId & asTYPEID_MASK_OBJECT) &&
- !(typeId & asTYPEID_OBJHANDLE) &&
- (engine->GetTypeInfoById(typeId)->GetFlags() & asOBJ_REF))
- {
- // Dereference the pointer to get the reference to the actual object
- ref = *(void**)ref;
- }
- Set(name, ref, typeId);
- }
- // Advance the buffer pointer with the size of the value
- if( typeId & asTYPEID_MASK_OBJECT )
- {
- asITypeInfo *ot = engine->GetTypeInfoById(typeId);
- if( ot->GetFlags() & asOBJ_VALUE )
- buffer += ot->GetSize();
- else
- buffer += sizeof(void*);
- }
- else if( typeId == 0 )
- {
- // null pointer
- buffer += sizeof(void*);
- }
- else
- {
- buffer += engine->GetSizeOfPrimitiveType(typeId);
- }
- }
- }
- CScriptDictionary::~CScriptDictionary()
- {
- // Delete all keys and values
- DeleteAll();
- }
- void CScriptDictionary::AddRef() const
- {
- // We need to clear the GC flag
- gcFlag = false;
- asAtomicInc(refCount);
- }
- void CScriptDictionary::Release() const
- {
- // We need to clear the GC flag
- gcFlag = false;
- if( asAtomicDec(refCount) == 0 )
- {
- this->~CScriptDictionary();
- asFreeMem(const_cast<CScriptDictionary*>(this));
- }
- }
- int CScriptDictionary::GetRefCount()
- {
- return refCount;
- }
- void CScriptDictionary::SetGCFlag()
- {
- gcFlag = true;
- }
- bool CScriptDictionary::GetGCFlag()
- {
- return gcFlag;
- }
- void CScriptDictionary::EnumReferences(asIScriptEngine *engine)
- {
- // Call the gc enum callback for each of the objects
- HashMap<String, CScriptDictValue>::Iterator it;
- for( it = dict.Begin(); it != dict.End(); it++ )
- {
- if( it->second_.m_typeId & asTYPEID_MASK_OBJECT )
- engine->GCEnumCallback(it->second_.m_valueObj);
- }
- }
- void CScriptDictionary::ReleaseAllReferences(asIScriptEngine * /*engine*/)
- {
- // We're being told to release all references in
- // order to break circular references for dead objects
- DeleteAll();
- }
- CScriptDictionary &CScriptDictionary::operator =(const CScriptDictionary &other)
- {
- // Clear everything we had before
- DeleteAll();
- // Do a shallow copy of the dictionary
- HashMap<String, CScriptDictValue>::ConstIterator it;
- for( it = other.dict.Begin(); it != other.dict.End(); it++ )
- {
- if( it->second_.m_typeId & asTYPEID_OBJHANDLE )
- Set(it->first_, (void*)&it->second_.m_valueObj, it->second_.m_typeId);
- else if( it->second_.m_typeId & asTYPEID_MASK_OBJECT )
- Set(it->first_, (void*)it->second_.m_valueObj, it->second_.m_typeId);
- else
- Set(it->first_, (void*)&it->second_.m_valueInt, it->second_.m_typeId);
- }
- return *this;
- }
- CScriptDictValue *CScriptDictionary::operator[](const String &key)
- {
- // Return the existing value if it exists, else insert an empty value
- HashMap<String, CScriptDictValue>::Iterator it;
- it = dict.Find(key);
- if( it == dict.End() )
- it = dict.Insert(MakePair(key, CScriptDictValue()));
- return &it->second_;
- }
- const CScriptDictValue *CScriptDictionary::operator[](const String &key) const
- {
- // Return the existing value if it exists
- HashMap<String, CScriptDictValue>::ConstIterator it;
- it = dict.Find(key);
- if( it != dict.End() )
- return &it->second_;
- // Else raise an exception
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- ctx->SetException("Invalid access to non-existing value");
- return nullptr;
- }
- void CScriptDictionary::Set(const String &key, void *value, int typeId)
- {
- HashMap<String, CScriptDictValue>::Iterator it;
- it = dict.Find(key);
- if( it == dict.End() )
- it = dict.Insert(MakePair(key, CScriptDictValue()));
- it->second_.Set(engine, value, typeId);
- }
- // This overloaded method is implemented so that all integer and
- // unsigned integers types will be stored in the dictionary as int64
- // through implicit conversions. This simplifies the management of the
- // numeric types when the script retrieves the stored value using a
- // different type.
- void CScriptDictionary::Set(const String &key, const asINT64 &value)
- {
- Set(key, const_cast<asINT64*>(&value), asTYPEID_INT64);
- }
- // This overloaded method is implemented so that all floating point types
- // will be stored in the dictionary as double through implicit conversions.
- // This simplifies the management of the numeric types when the script
- // retrieves the stored value using a different type.
- void CScriptDictionary::Set(const String &key, const double &value)
- {
- Set(key, const_cast<double*>(&value), asTYPEID_DOUBLE);
- }
- // Returns true if the value was successfully retrieved
- bool CScriptDictionary::Get(const String &key, void *value, int typeId) const
- {
- HashMap<String, CScriptDictValue>::ConstIterator it;
- it = dict.Find(key);
- if( it != dict.End() )
- return it->second_.Get(engine, value, typeId);
- // AngelScript has already initialized the value with a default value,
- // so we don't have to do anything if we don't find the element, or if
- // the element is incompatible with the requested type.
- return false;
- }
- // Returns the type id of the stored value
- int CScriptDictionary::GetTypeId(const String &key) const
- {
- HashMap<String, CScriptDictValue>::ConstIterator it;
- it = dict.Find(key);
- if( it != dict.End() )
- return it->second_.m_typeId;
- return -1;
- }
- bool CScriptDictionary::Get(const String &key, asINT64 &value) const
- {
- return Get(key, &value, asTYPEID_INT64);
- }
- bool CScriptDictionary::Get(const String &key, double &value) const
- {
- return Get(key, &value, asTYPEID_DOUBLE);
- }
- bool CScriptDictionary::Exists(const String &key) const
- {
- HashMap<String, CScriptDictValue>::ConstIterator it;
- it = dict.Find(key);
- if( it != dict.End() )
- return true;
- return false;
- }
- bool CScriptDictionary::IsEmpty() const
- {
- if( dict.Size() == 0 )
- return true;
- return false;
- }
- asUINT CScriptDictionary::GetSize() const
- {
- return asUINT(dict.Size());
- }
- void CScriptDictionary::Delete(const String &key)
- {
- HashMap<String, CScriptDictValue>::Iterator it;
- it = dict.Find(key);
- if( it != dict.End() )
- {
- it->second_.FreeValue(engine);
- dict.Erase(it);
- }
- }
- void CScriptDictionary::DeleteAll()
- {
- HashMap<String, CScriptDictValue>::Iterator it;
- for( it = dict.Begin(); it != dict.End(); it++ )
- it->second_.FreeValue(engine);
- dict.Clear();
- }
- CScriptArray* CScriptDictionary::GetKeys() const
- {
- // TODO: optimize: The string array type should only be determined once.
- // It should be recomputed when registering the dictionary class.
- // Only problem is if multiple engines are used, as they may not
- // share the same type id. Alternatively it can be stored in the
- // user data for the dictionary type.
- asITypeInfo *ot = engine->GetTypeInfoByDecl("Array<String>");
- // Create the array object
- CScriptArray *array = CScriptArray::Create(ot, asUINT(dict.Size()));
- long current = -1;
- HashMap<String, CScriptDictValue>::ConstIterator it;
- for( it = dict.Begin(); it != dict.End(); it++ )
- {
- current++;
- *(String*)array->At(current) = it->first_;
- }
- return array;
- }
- void ScriptDictionaryFactory_Generic(asIScriptGeneric *gen)
- {
- *(CScriptDictionary**)gen->GetAddressOfReturnLocation() = CScriptDictionary::Create(gen->GetEngine());
- }
- void ScriptDictionaryListFactory_Generic(asIScriptGeneric *gen)
- {
- auto*buffer = (asBYTE*)gen->GetArgAddress(0);
- *(CScriptDictionary**)gen->GetAddressOfReturnLocation() = CScriptDictionary::Create(buffer);
- }
- CScriptDictValue::CScriptDictValue() // NOLINT(hicpp-member-init)
- {
- m_valueObj = nullptr;
- m_typeId = 0;
- }
- CScriptDictValue::CScriptDictValue(asIScriptEngine *engine, void *value, int typeId) // NOLINT(hicpp-member-init)
- {
- m_valueObj = nullptr;
- m_typeId = 0;
- Set(engine, value, typeId);
- }
- CScriptDictValue::~CScriptDictValue()
- {
- // Must not hold an object when destroyed, as then the object will never be freed
- assert( (m_typeId & asTYPEID_MASK_OBJECT) == 0 );
- }
- void CScriptDictValue::FreeValue(asIScriptEngine *engine)
- {
- // If it is a handle or a ref counted object, call release
- if( m_typeId & asTYPEID_MASK_OBJECT )
- {
- // Let the engine release the object
- engine->ReleaseScriptObject(m_valueObj, engine->GetTypeInfoById(m_typeId));
- m_valueObj = nullptr;
- m_typeId = 0;
- }
- // For primitives, there's nothing to do
- }
- void CScriptDictValue::Set(asIScriptEngine *engine, void *value, int typeId)
- {
- FreeValue(engine);
- m_typeId = typeId;
- if( typeId & asTYPEID_OBJHANDLE )
- {
- // We're receiving a reference to the handle, so we need to dereference it
- m_valueObj = *(void**)value;
- engine->AddRefScriptObject(m_valueObj, engine->GetTypeInfoById(typeId));
- }
- else if( typeId & asTYPEID_MASK_OBJECT )
- {
- // Create a copy of the object
- m_valueObj = engine->CreateScriptObjectCopy(value, engine->GetTypeInfoById(typeId));
- }
- else
- {
- // Copy the primitive value
- // We receive a pointer to the value.
- int size = engine->GetSizeOfPrimitiveType(typeId);
- memcpy(&m_valueInt, value, size);
- }
- }
- // This overloaded method is implemented so that all integer and
- // unsigned integers types will be stored in the dictionary as int64
- // through implicit conversions. This simplifies the management of the
- // numeric types when the script retrieves the stored value using a
- // different type.
- void CScriptDictValue::Set(asIScriptEngine *engine, const asINT64 &value)
- {
- Set(engine, const_cast<asINT64*>(&value), asTYPEID_INT64);
- }
- // This overloaded method is implemented so that all floating point types
- // will be stored in the dictionary as double through implicit conversions.
- // This simplifies the management of the numeric types when the script
- // retrieves the stored value using a different type.
- void CScriptDictValue::Set(asIScriptEngine *engine, const double &value)
- {
- Set(engine, const_cast<double*>(&value), asTYPEID_DOUBLE);
- }
- bool CScriptDictValue::Get(asIScriptEngine *engine, void *value, int typeId) const
- {
- // Return the value
- if( typeId & asTYPEID_OBJHANDLE )
- {
- // A handle can be retrieved if the stored type is a handle of same or compatible type
- // or if the stored type is an object that implements the interface that the handle refer to.
- void* cast = nullptr;
- if ((m_typeId & asTYPEID_MASK_OBJECT) &&
- engine->RefCastObject(m_valueObj, engine->GetTypeInfoById(m_typeId), engine->GetTypeInfoById(typeId), &cast) >= 0)
- {
- *(void**)value = m_valueObj;
- return true;
- }
- }
- else if( typeId & asTYPEID_MASK_OBJECT )
- {
- // Verify that the copy can be made
- bool isCompatible = false;
- if( m_typeId == typeId )
- isCompatible = true;
- // Copy the object into the given reference
- if( isCompatible )
- {
- engine->AssignScriptObject(value, m_valueObj, engine->GetTypeInfoById(typeId));
- return true;
- }
- }
- else
- {
- if( m_typeId == typeId )
- {
- int size = engine->GetSizeOfPrimitiveType(typeId);
- memcpy(value, &m_valueInt, size);
- return true;
- }
- // We know all numbers are stored as either int64 or double, since we register overloaded functions for those
- if( m_typeId == asTYPEID_INT64 && typeId == asTYPEID_DOUBLE )
- {
- *(double*)value = double(m_valueInt);
- return true;
- }
- else if( m_typeId == asTYPEID_DOUBLE && typeId == asTYPEID_INT64 )
- {
- *(asINT64*)value = asINT64(m_valueFlt);
- return true;
- }
- }
- // It was not possible to retrieve the value using the desired typeId
- return false;
- }
- bool CScriptDictValue::Get(asIScriptEngine *engine, asINT64 &value) const
- {
- return Get(engine, &value, asTYPEID_INT64);
- }
- bool CScriptDictValue::Get(asIScriptEngine *engine, double &value) const
- {
- return Get(engine, &value, asTYPEID_DOUBLE);
- }
- int CScriptDictValue::GetTypeId() const
- {
- return m_typeId;
- }
- static void CScriptDictValue_Construct(void *mem)
- {
- new(mem) CScriptDictValue();
- }
- static void CScriptDictValue_Destruct(CScriptDictValue *obj)
- {
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- {
- asIScriptEngine *engine = ctx->GetEngine();
- obj->FreeValue(engine);
- }
- obj->~CScriptDictValue();
- }
- static CScriptDictValue &CScriptDictValue_opAssign(void *ref, int typeId, CScriptDictValue *obj)
- {
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- {
- asIScriptEngine *engine = ctx->GetEngine();
- obj->Set(engine, ref, typeId);
- }
- return *obj;
- }
- static CScriptDictValue &CScriptDictValue_opAssign(double val, CScriptDictValue *obj)
- {
- return CScriptDictValue_opAssign(&val, asTYPEID_DOUBLE, obj);
- }
- static CScriptDictValue &CScriptDictValue_opAssign(asINT64 val, CScriptDictValue *obj)
- {
- return CScriptDictValue_opAssign(&val, asTYPEID_INT64, obj);
- }
- static void CScriptDictValue_opCast(void *ref, int typeId, CScriptDictValue *obj)
- {
- asIScriptContext *ctx = asGetActiveContext();
- if( ctx )
- {
- asIScriptEngine *engine = ctx->GetEngine();
- obj->Get(engine, ref, typeId);
- }
- }
- static asINT64 CScriptDictValue_opConvInt(CScriptDictValue *obj)
- {
- asINT64 value;
- CScriptDictValue_opCast(&value, asTYPEID_INT64, obj);
- return value;
- }
- static double CScriptDictValue_opConvDouble(CScriptDictValue *obj)
- {
- double value;
- CScriptDictValue_opCast(&value, asTYPEID_DOUBLE, obj);
- return value;
- }
- #ifdef AS_MAX_PORTABILITY
- static void CScriptDictValue_opAssign_gen(asIScriptGeneric* gen)
- {
- *static_cast<CScriptDictValue**>(gen->GetAddressOfReturnLocation()) =
- &CScriptDictValue_opAssign(*static_cast<void**>(gen->GetAddressOfArg(0)),
- gen->GetArgTypeId(0),
- static_cast<CScriptDictValue*>(gen->GetObject()));
- }
- static void CScriptDictValue_opCast_gen(asIScriptGeneric* gen)
- {
- CScriptDictValue_opCast(
- *static_cast<void**>(gen->GetAddressOfArg(0)),
- gen->GetArgTypeId(0),
- static_cast<CScriptDictValue*>(gen->GetObject()));
- }
- static void ScriptDictionarySet_Generic(asIScriptGeneric *gen)
- {
- CScriptDictionary* dict = (CScriptDictionary*)gen->GetObject();
- String* key = *(String**)gen->GetAddressOfArg(0);
- void* ref = *(void**)gen->GetAddressOfArg(1);
- int type_id = gen->GetArgTypeId(1);
- dict->Set(*key, ref, type_id);
- }
- static void ScriptDictionaryGet_Generic(asIScriptGeneric *gen)
- {
- CScriptDictionary* dict = (CScriptDictionary*)gen->GetObject();
- String* key = *(String**)gen->GetAddressOfArg(0);
- void* ref = *(void**)gen->GetAddressOfArg(1);
- int type_id = gen->GetArgTypeId(1);
- *(bool*)gen->GetAddressOfReturnLocation() = dict->Get(*key, ref, type_id);
- }
- #endif
- void RegisterDictionary(asIScriptEngine *engine)
- {
- engine->RegisterObjectBehaviour("DictionaryValue", asBEHAVE_CONSTRUCT, "void f()", asFUNCTION(CScriptDictValue_Construct), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectBehaviour("DictionaryValue", asBEHAVE_DESTRUCT, "void f()", asFUNCTION(CScriptDictValue_Destruct), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opAssign(double)", asFUNCTIONPR(CScriptDictValue_opAssign, (double, CScriptDictValue*), CScriptDictValue &), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opAssign(int64)", asFUNCTIONPR(CScriptDictValue_opAssign, (asINT64, CScriptDictValue*), CScriptDictValue &), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "int64 opConv()", asFUNCTIONPR(CScriptDictValue_opConvInt, (CScriptDictValue*), asINT64), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "double opConv()", asFUNCTIONPR(CScriptDictValue_opConvDouble, (CScriptDictValue*), double), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectType("Dictionary", sizeof(CScriptDictionary), asOBJ_REF);
- engine->RegisterObjectBehaviour("Dictionary", asBEHAVE_FACTORY, "Dictionary@ f()", asFUNCTION(ScriptDictionaryFactory_Generic), asCALL_GENERIC);
- engine->RegisterObjectBehaviour("Dictionary", asBEHAVE_LIST_FACTORY, "Dictionary @f(int &in) {repeat {String, ?}}", asFUNCTION(ScriptDictionaryListFactory_Generic), asCALL_GENERIC);
- engine->RegisterObjectBehaviour("Dictionary", asBEHAVE_ADDREF, "void f()", asMETHOD(CScriptDictionary,AddRef), asCALL_THISCALL);
- engine->RegisterObjectBehaviour("Dictionary", asBEHAVE_RELEASE, "void f()", asMETHOD(CScriptDictionary,Release), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "Dictionary &opAssign(const Dictionary &in)", asMETHODPR(CScriptDictionary, operator=, (const CScriptDictionary &), CScriptDictionary&), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "void Set(const String &in, const int64&in)", asMETHODPR(CScriptDictionary,Set,(const String&,const asINT64&),void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "bool Get(const String &in, int64&out) const", asMETHODPR(CScriptDictionary,Get,(const String&,asINT64&) const,bool), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "void Set(const String &in, const double&in)", asMETHODPR(CScriptDictionary,Set,(const String&,const double&),void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "bool Get(const String &in, double&out) const", asMETHODPR(CScriptDictionary,Get,(const String&,double&) const,bool), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "bool Exists(const String &in) const", asMETHOD(CScriptDictionary,Exists), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "bool get_empty() const", asMETHOD(CScriptDictionary, IsEmpty), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "uint get_length() const", asMETHOD(CScriptDictionary, GetSize), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "void Erase(const String &in)", asMETHOD(CScriptDictionary,Delete), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "void Clear()", asMETHOD(CScriptDictionary,DeleteAll), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "Array<String> @get_keys() const", asMETHOD(CScriptDictionary,GetKeys), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "DictionaryValue &opIndex(const String &in)", asMETHODPR(CScriptDictionary, operator[], (const String &), CScriptDictValue*), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "const DictionaryValue &opIndex(const String &in) const", asMETHODPR(CScriptDictionary, operator[], (const String &) const, const CScriptDictValue*), asCALL_THISCALL);
- #ifndef AS_MAX_PORTABILITY
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opHndlAssign(const ?&in)", asFUNCTIONPR(CScriptDictValue_opAssign, (void *, int, CScriptDictValue*), CScriptDictValue &), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opAssign(const ?&in)", asFUNCTIONPR(CScriptDictValue_opAssign, (void *, int, CScriptDictValue*), CScriptDictValue &), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "void opCast(?&out)", asFUNCTIONPR(CScriptDictValue_opCast, (void *, int, CScriptDictValue*), void), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("DictionaryValue", "void opConv(?&out)", asFUNCTIONPR(CScriptDictValue_opCast, (void *, int, CScriptDictValue*), void), asCALL_CDECL_OBJLAST);
- engine->RegisterObjectMethod("Dictionary", "void Set(const String &in, const ?&in)", asMETHODPR(CScriptDictionary, Set, (const String&, void*, int), void), asCALL_THISCALL);
- engine->RegisterObjectMethod("Dictionary", "bool Get(const String &in, ?&out) const", asMETHODPR(CScriptDictionary, Get, (const String&, void*, int) const, bool), asCALL_THISCALL);
- #else
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opHndlAssign(const ?&in)", asFUNCTION(CScriptDictValue_opAssign_gen), asCALL_GENERIC);
- engine->RegisterObjectMethod("DictionaryValue", "DictionaryValue &opAssign(const ?&in)", asFUNCTION(CScriptDictValue_opAssign_gen), asCALL_GENERIC);
- engine->RegisterObjectMethod("DictionaryValue", "void opCast(?&out)", asFUNCTION(CScriptDictValue_opCast_gen), asCALL_GENERIC);
- engine->RegisterObjectMethod("DictionaryValue", "void opConv(?&out)", asFUNCTION(CScriptDictValue_opCast_gen), asCALL_GENERIC);
- engine->RegisterObjectMethod("Dictionary", "void Set(const String &in, const ?&in)", asFUNCTION(ScriptDictionarySet_Generic), asCALL_GENERIC);
- engine->RegisterObjectMethod("Dictionary", "bool Get(const String &in, ?&out) const", asFUNCTION(ScriptDictionaryGet_Generic), asCALL_GENERIC);
- #endif
- }
- 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 i, String& str)
- {
- if (i >= str.Length())
- {
- asIScriptContext* context = asGetActiveContext();
- if (context)
- context->SetException("Index out of bounds");
- return nullptr;
- }
- 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 ConstructStringDouble(double value, String* ptr)
- {
- new(ptr)String(value);
- }
- static void ConstructStringBool(bool value, String* ptr)
- {
- new(ptr) String(value);
- }
- static void StringSetUTF8FromLatin1(const String& src, String& str)
- {
- str.SetUTF8FromLatin1(src.CString());
- }
- void RegisterString(asIScriptEngine *engine)
- {
- //static const unsigned NPOS = String::NPOS; // workaround for GCC
- //static StringFactory stringFactory;
- //engine->RegisterGlobalProperty("const uint NPOS", (void*)&NPOS);
- //engine->RegisterStringFactory("String", &stringFactory);
-
- // TODO Обертку с контролем размера массива
- //engine->RegisterObjectMethod("String", "uint8 &opIndex(uint)", asFUNCTION(StringCharAt), asCALL_CDECL_OBJLAST);
- //engine->RegisterObjectMethod("String", "const uint8 &opIndex(uint) const", asFUNCTION(StringCharAt), asCALL_CDECL_OBJLAST);
- // TODO у меня int8
- //engine->RegisterObjectMethod("String", "void Replace(uint8, uint8, bool caseSensitive = true)", asMETHODPR(String, Replace, (char, char, 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", "uint Find(uint8, uint start = 0, bool caseSensitive = true) const", asMETHODPR(String, Find, (char, unsigned, bool) const, unsigned), asCALL_THISCALL);
-
- // NPOS починить
- //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", "void SetUTF8FromLatin1(const String& in)", asFUNCTION(StringSetUTF8FromLatin1), asCALL_CDECL_OBJLAST);
-
- // TODO у меня int8
- //engine->RegisterObjectMethod("String", "bool Contains(uint8, bool caseSensitive = true) const", asMETHODPR(String, Contains, (char, bool) const, bool), asCALL_THISCALL);
- }
- const void* StringFactory::GetStringConstant(const char* data, asUINT length)
- {
- assert(strlen(data) == length);
- StringHash hash(data);
- auto iter = map_.Find(hash);
- return reinterpret_cast<const void*>(&(iter == map_.End() ? map_.Insert(MakePair(hash, String(data))) : iter)->second_);
- }
- int StringFactory::ReleaseStringConstant(const void* str)
- {
- // Cache all the strings
- return str ? asSUCCESS : asERROR;
- }
- int StringFactory::GetRawStringData(const void* str, char* data, asUINT* length) const
- {
- if (!str)
- return asERROR;
- auto p = reinterpret_cast<const String*>(str);
- if (length)
- *length = p->Length();
- if (data)
- memcpy(data, p->CString(), p->Length());
- return asSUCCESS;
- }
- // This function is called before ASRegisterGenerated()
- void ASRegisterManualFirst_Addons(asIScriptEngine* engine)
- {
- engine->SetTypeInfoUserDataCleanupCallback(CleanupTypeInfoArrayCache, ARRAY_CACHE);
- engine->RegisterObjectType("Array<class T>", 0, asOBJ_REF | asOBJ_TEMPLATE);
- RegisterArray(engine);
-
- engine->RegisterObjectType("DictionaryValue", sizeof(CScriptDictValue), asOBJ_VALUE | asOBJ_ASHANDLE | asOBJ_APP_CLASS_CD);
- }
- // This function is called after ASRegisterGenerated()
- void ASRegisterManualLast_Addons(asIScriptEngine* engine)
- {
- //RegisterArray(engine);
- RegisterDictionary(engine);
- }
- }
|