| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047 |
- // Copyright (c) 2008-2022 the Urho3D project
- // License: MIT
- #include "../Precompiled.h"
- #include "../AngelScript/Script.h"
- #include "../AngelScript/ScriptFile.h"
- #include "../AngelScript/ScriptInstance.h"
- #include "../AngelScript/Addons.h"
- #include "../AngelScript/APITemplates.h"
- #include "../Core/Context.h"
- #include "../Core/Profiler.h"
- #include "../IO/Log.h"
- #include "../IO/MemoryBuffer.h"
- #if defined(URHO3D_PHYSICS) || defined(URHO3D_PHYSICS2D)
- #include "../Physics/PhysicsEvents.h"
- #endif
- #include "../Resource/ResourceCache.h"
- #include "../Resource/ResourceEvents.h"
- #include "../Scene/Scene.h"
- #include "../Scene/SceneEvents.h"
- #include <AngelScript/angelscript.h>
- #include "../DebugNew.h"
- namespace Urho3D
- {
- static const char* methodDeclarations[] = {
- "void Start()",
- "void Stop()",
- "void DelayedStart()",
- "void Update(float)",
- "void PostUpdate(float)",
- "void FixedUpdate(float)",
- "void FixedPostUpdate(float)",
- "void Load(Deserializer&)",
- "void Save(Serializer&)",
- "void ReadNetworkUpdate(Deserializer&)",
- "void WriteNetworkUpdate(Serializer&)",
- "void ApplyAttributes()",
- "void TransformChanged()"
- };
- void CleanupTypeInfoScriptInstance(asITypeInfo *type)
- {
- delete static_cast<Vector<AttributeInfo>*>(type->GetUserData(eAttrMapUserIdx));
- }
- template<typename Op>
- inline void ScriptInstance::executeScript(asIScriptFunction* method, Op func) const {
- Script* scriptSystem = GetSubsystem<Script>();
- asIScriptContext* context = scriptSystem->GetScriptFileContext();
- if (context->Prepare(method) < 0)
- return;
- context->SetObject(scriptObject_);
- func(context);
- scriptSystem->IncScriptNestingLevel();
- context->Execute();
- context->Unprepare();
- scriptSystem->DecScriptNestingLevel();
- }
- ScriptInstance::ScriptInstance(Context* context) :
- Component(context)
- {
- ClearScriptAttributes();
- }
- ScriptInstance::~ScriptInstance()
- {
- ReleaseObject();
- }
- void ScriptInstance::RegisterObject(Context* context)
- {
- context->RegisterFactory<ScriptInstance>(LOGIC_CATEGORY);
- URHO3D_ACCESSOR_ATTRIBUTE("Is Enabled", IsEnabled, SetEnabled, bool, true, AM_DEFAULT);
- URHO3D_MIXED_ACCESSOR_ATTRIBUTE("Delayed Method Calls", GetDelayedCallsAttr, SetDelayedCallsAttr, PODVector<unsigned char>,
- Variant::emptyBuffer, AM_FILE | AM_NOEDIT);
- URHO3D_MIXED_ACCESSOR_ATTRIBUTE("Script File", GetScriptFileAttr, SetScriptFileAttr, ResourceRef,
- ResourceRef(ScriptFile::GetTypeStatic()), AM_DEFAULT);
- URHO3D_ACCESSOR_ATTRIBUTE("Class Name", GetClassName, SetClassName, String, String::EMPTY, AM_DEFAULT);
- URHO3D_MIXED_ACCESSOR_ATTRIBUTE("Script Data", GetScriptDataAttr, SetScriptDataAttr, PODVector<unsigned char>, Variant::emptyBuffer,
- AM_FILE | AM_NOEDIT);
- URHO3D_MIXED_ACCESSOR_ATTRIBUTE("Script Network Data", GetScriptNetworkDataAttr, SetScriptNetworkDataAttr, PODVector<unsigned char>,
- Variant::emptyBuffer, AM_NET | AM_NOEDIT);
- }
- void ScriptInstance::OnSetAttribute(const AttributeInfo& attr, const Variant& src)
- {
- // Update stored script attributes if being manipulated while the actual object doesn't exist
- if (!scriptObject_ && storedAttributes_.Size())
- {
- HashMap<String, Variant>::Iterator i = storedAttributes_.Find(attr.name_);
- if (i != storedAttributes_.End())
- i->second_ = src;
- }
- if (attr.mode_ & (AM_NODEID | AM_COMPONENTID))
- {
- // The component / node to which the ID refers to may not be in the scene yet, and furthermore the ID must go through the
- // SceneResolver first. Delay searching for the object to ApplyAttributes
- auto* attrPtr = const_cast<AttributeInfo*>(&attr);
- idAttributes_[attrPtr] = src.GetUInt();
- }
- else if (attr.type_ == VAR_RESOURCEREF && attr.ptr_)
- {
- Resource*& resourcePtr = *(reinterpret_cast<Resource**>(attr.ptr_));
- // Decrease reference count of the old object if any, then increment the new
- if (resourcePtr)
- resourcePtr->ReleaseRef();
- const ResourceRef& ref = src.GetResourceRef();
- resourcePtr = GetSubsystem<ResourceCache>()->GetResource(ref.type_, ref.name_);
- if (resourcePtr)
- resourcePtr->AddRef();
- }
- else if (attr.type_ == VAR_VARIANTVECTOR && attr.ptr_)
- {
- auto* arr = reinterpret_cast<CScriptArray*>(attr.ptr_);
- if (arr)
- {
- const Vector<Variant>& vector = src.GetVariantVector();
- unsigned size = vector.Size();
- arr->Resize(size);
- for (unsigned i = 0; i < size; i++)
- *(static_cast<Variant*>(arr->At(i))) = vector[i];
- }
- }
- else if (attr.type_ == VAR_STRINGVECTOR && attr.ptr_)
- {
- auto* arr = reinterpret_cast<CScriptArray*>(attr.ptr_);
- if (arr)
- {
- const Vector<String>& vector = src.GetStringVector();
- unsigned size = vector.Size();
- arr->Resize(size);
- for (unsigned i = 0; i < size; i++)
- *(static_cast<String*>(arr->At(i))) = vector[i];
- }
- }
- else
- Serializable::OnSetAttribute(attr, src);
- }
- void ScriptInstance::OnGetAttribute(const AttributeInfo& attr, Variant& dest) const
- {
- auto* attrPtr = const_cast<AttributeInfo*>(&attr);
- // Get ID's for node / component handle attributes
- if (attr.mode_ & (AM_NODEID | AM_COMPONENTID))
- {
- // If a cached ID value has been stored, return it instead of querying from the actual object
- // (the object handle is likely null at that point)
- HashMap<AttributeInfo*, unsigned>::ConstIterator i = idAttributes_.Find(attrPtr);
- if (i != idAttributes_.End())
- dest = i->second_;
- else if (attr.mode_ & AM_NODEID)
- {
- Node* node = *(reinterpret_cast<Node**>(attr.ptr_));
- unsigned nodeID = node ? node->GetID() : 0;
- dest = nodeID;
- }
- else
- {
- Component* component = *(reinterpret_cast<Component**>(attr.ptr_));
- unsigned componentID = component ? component->GetID() : 0;
- dest = componentID;
- }
- }
- else if (attr.type_ == VAR_RESOURCEREF && attr.ptr_)
- {
- Resource* resource = *(reinterpret_cast<Resource**>(attr.ptr_));
- // If resource is non-null get its type and name hash. Otherwise get type from the default value
- dest = GetResourceRef(resource, attr.defaultValue_.GetResourceRef().type_);
- }
- else if (attr.type_ == VAR_VARIANTVECTOR && attr.ptr_)
- {
- auto* arr = reinterpret_cast<CScriptArray*>(attr.ptr_);
- if (arr)
- dest = ArrayToVector<Variant>(arr);
- }
- else if (attr.type_ == VAR_STRINGVECTOR && attr.ptr_)
- {
- auto* arr = reinterpret_cast<CScriptArray*>(attr.ptr_);
- if (arr)
- dest = ArrayToVector<String>(arr);
- } else
- Serializable::OnGetAttribute(attr, dest);
- }
- void ScriptInstance::ApplyAttributes()
- {
- // Apply node / component ID attributes now (find objects from the scene and assign to the object handles)
- for (HashMap<AttributeInfo*, unsigned>::Iterator i = idAttributes_.Begin(); i != idAttributes_.End(); ++i)
- {
- AttributeInfo& attr = *i->first_;
- if (attr.mode_ & AM_NODEID)
- {
- Node*& nodePtr = *(reinterpret_cast<Node**>(attr.ptr_));
- // Decrease reference count of the old object if any, then increment the new
- if (nodePtr)
- nodePtr->ReleaseRef();
- nodePtr = GetScene()->GetNode(i->second_);
- if (nodePtr)
- nodePtr->AddRef();
- }
- else if (attr.mode_ & AM_COMPONENTID)
- {
- Component*& componentPtr = *(reinterpret_cast<Component**>(attr.ptr_));
- if (componentPtr)
- componentPtr->ReleaseRef();
- componentPtr = GetScene()->GetComponent(i->second_);
- if (componentPtr)
- componentPtr->AddRef();
- }
- }
- idAttributes_.Clear();
- if (scriptObject_ && methods_[METHOD_APPLYATTRIBUTES])
- executeScript(methods_[METHOD_APPLYATTRIBUTES], [](asIScriptContext*) {});
- }
- void ScriptInstance::OnSetEnabled()
- {
- UpdateEventSubscription();
- }
- bool ScriptInstance::CreateObject(ScriptFile* scriptFile, const String& className)
- {
- className_ = String::EMPTY; // Do not create object during SetScriptFile()
- SetScriptFile(scriptFile);
- SetClassName(className);
- return scriptObject_ != nullptr;
- }
- void ScriptInstance::SetScriptFile(ScriptFile* scriptFile)
- {
- if (scriptFile == scriptFile_ && scriptObject_)
- return;
- ReleaseObject();
- // Unsubscribe from the reload event of previous script file (if any), then subscribe to the new
- if (scriptFile_)
- {
- UnsubscribeFromEvent(scriptFile_, E_RELOADSTARTED);
- UnsubscribeFromEvent(scriptFile_, E_RELOADFINISHED);
- }
- if (scriptFile)
- {
- SubscribeToEvent(scriptFile, E_RELOADSTARTED, URHO3D_HANDLER(ScriptInstance, HandleScriptFileReload));
- SubscribeToEvent(scriptFile, E_RELOADFINISHED, URHO3D_HANDLER(ScriptInstance, HandleScriptFileReloadFinished));
- }
- scriptFile_ = scriptFile;
- CreateObject();
- MarkNetworkUpdate();
- }
- void ScriptInstance::SetClassName(const String& className)
- {
- if (className == className_ && scriptObject_)
- return;
- ReleaseObject();
- className_ = className;
- CreateObject();
- MarkNetworkUpdate();
- }
- bool ScriptInstance::Execute(const String& declaration, const VariantVector& parameters)
- {
- if (!scriptObject_)
- return false;
- asIScriptFunction* method = scriptFile_->GetMethod(scriptObject_, declaration);
- if (!method)
- {
- URHO3D_LOGERROR("Method " + declaration + " not found in class " + className_);
- return false;
- }
- return scriptFile_->Execute(scriptObject_, method, parameters);
- }
- bool ScriptInstance::Execute(asIScriptFunction* method, const VariantVector& parameters)
- {
- if (!method || !scriptObject_)
- return false;
- return scriptFile_->Execute(scriptObject_, method, parameters);
- }
- void ScriptInstance::DelayedExecute(float delay, bool repeat, const String& declaration, const VariantVector& parameters)
- {
- if (!scriptObject_)
- return;
- DelayedCall call;
- call.period_ = call.delay_ = Max(delay, 0.0f);
- call.repeat_ = repeat;
- call.declaration_ = declaration;
- call.parameters_ = parameters;
- delayedCalls_.Push(call);
- // Make sure we are registered to the scene update event, because delayed calls are executed there
- if (!subscribed_)
- UpdateEventSubscription();
- }
- void ScriptInstance::ClearDelayedExecute(const String& declaration)
- {
- if (declaration.Empty())
- delayedCalls_.Clear();
- else
- {
- for (Vector<DelayedCall>::Iterator i = delayedCalls_.Begin(); i != delayedCalls_.End();)
- {
- if (declaration == i->declaration_)
- i = delayedCalls_.Erase(i);
- else
- ++i;
- }
- }
- }
- void ScriptInstance::AddEventHandler(StringHash eventType, const String& handlerName)
- {
- if (!scriptObject_)
- return;
- String declaration = "void " + handlerName + "(StringHash, VariantMap&)";
- asIScriptFunction* method = scriptFile_->GetMethod(scriptObject_, declaration);
- if (!method)
- {
- // Retry with parameterless signature
- method = scriptFile_->GetMethod(scriptObject_, handlerName);
- if (!method)
- {
- URHO3D_LOGERROR("Event handler method " + handlerName + " not found in " + scriptFile_->GetName());
- return;
- }
- }
- SubscribeToEvent(eventType, URHO3D_HANDLER_USERDATA(ScriptInstance, HandleScriptEvent, (void*)method));
- }
- void ScriptInstance::AddEventHandler(Object* sender, StringHash eventType, const String& handlerName)
- {
- if (!scriptObject_)
- return;
- if (!sender)
- {
- URHO3D_LOGERROR("Null event sender for event " + String(eventType) + ", handler " + handlerName);
- return;
- }
- String declaration = "void " + handlerName + "(StringHash, VariantMap&)";
- asIScriptFunction* method = scriptFile_->GetMethod(scriptObject_, declaration);
- if (!method)
- {
- // Retry with parameterless signature
- method = scriptFile_->GetMethod(scriptObject_, handlerName);
- if (!method)
- {
- URHO3D_LOGERROR("Event handler method " + handlerName + " not found in " + scriptFile_->GetName());
- return;
- }
- }
- SubscribeToEvent(sender, eventType, URHO3D_HANDLER_USERDATA(ScriptInstance, HandleScriptEvent, (void*)method));
- }
- void ScriptInstance::RemoveEventHandler(StringHash eventType)
- {
- UnsubscribeFromEvent(eventType);
- }
- void ScriptInstance::RemoveEventHandler(Object* sender, StringHash eventType)
- {
- UnsubscribeFromEvent(sender, eventType);
- }
- void ScriptInstance::RemoveEventHandlers(Object* sender)
- {
- UnsubscribeFromEvents(sender);
- }
- void ScriptInstance::RemoveEventHandlers()
- {
- UnsubscribeFromAllEventsExcept(PODVector<StringHash>(), true);
- }
- void ScriptInstance::RemoveEventHandlersExcept(const PODVector<StringHash>& exceptions)
- {
- UnsubscribeFromAllEventsExcept(exceptions, true);
- }
- bool ScriptInstance::HasEventHandler(StringHash eventType) const
- {
- return HasSubscribedToEvent(eventType);
- }
- bool ScriptInstance::HasEventHandler(Object* sender, StringHash eventType) const
- {
- return HasSubscribedToEvent(sender, eventType);
- }
- bool ScriptInstance::IsA(const String& className) const
- {
- // Early out for the easiest case where that's what we are
- if (className_ == className)
- return true;
- if (scriptObject_)
- {
- asITypeInfo* myType = scriptObject_->GetObjectType();
- asITypeInfo* searchType = myType->GetModule()->GetTypeInfoByName(className.CString());
- return searchType && (searchType->GetTypeId() & asTYPEID_MASK_OBJECT) != 0 &&
- (myType->DerivesFrom(searchType) || myType->Implements(searchType));
- }
- return false;
- }
- bool ScriptInstance::HasMethod(const String& declaration) const
- {
- if (!scriptFile_ || !scriptObject_)
- return false;
- else
- return scriptFile_->GetMethod(scriptObject_, declaration) != nullptr;
- }
- void ScriptInstance::SetScriptFileAttr(const ResourceRef& value)
- {
- auto* cache = GetSubsystem<ResourceCache>();
- SetScriptFile(cache->GetResource<ScriptFile>(value.name_));
- }
- void ScriptInstance::SetDelayedCallsAttr(const PODVector<unsigned char>& value)
- {
- MemoryBuffer buf(value);
- delayedCalls_.Resize(buf.ReadVLE());
- for (Vector<DelayedCall>::Iterator i = delayedCalls_.Begin(); i != delayedCalls_.End(); ++i)
- {
- i->period_ = buf.ReadFloat();
- i->delay_ = buf.ReadFloat();
- i->repeat_ = buf.ReadBool();
- i->declaration_ = buf.ReadString();
- i->parameters_ = buf.ReadVariantVector();
- }
- if (scriptObject_ && delayedCalls_.Size() && !subscribed_)
- UpdateEventSubscription();
- }
- void ScriptInstance::SetScriptDataAttr(const PODVector<unsigned char>& data)
- {
- if (scriptObject_ && methods_[METHOD_LOAD])
- {
- MemoryBuffer buf(data);
- executeScript(methods_[METHOD_LOAD], [&](asIScriptContext* context) {
- context->SetArgObject(0, &buf);
- });
- }
- }
- void ScriptInstance::SetScriptNetworkDataAttr(const PODVector<unsigned char>& data)
- {
- if (scriptObject_ && methods_[METHOD_READNETWORKUPDATE])
- {
- MemoryBuffer buf(data);
- executeScript(methods_[METHOD_READNETWORKUPDATE], [&](asIScriptContext* context) {
- context->SetArgObject(0, &buf);
- });
- }
- }
- ResourceRef ScriptInstance::GetScriptFileAttr() const
- {
- return GetResourceRef(scriptFile_, ScriptFile::GetTypeStatic());
- }
- PODVector<unsigned char> ScriptInstance::GetDelayedCallsAttr() const
- {
- VectorBuffer buf;
- buf.WriteVLE(delayedCalls_.Size());
- for (Vector<DelayedCall>::ConstIterator i = delayedCalls_.Begin(); i != delayedCalls_.End(); ++i)
- {
- buf.WriteFloat(i->period_);
- buf.WriteFloat(i->delay_);
- buf.WriteBool(i->repeat_);
- buf.WriteString(i->declaration_);
- buf.WriteVariantVector(i->parameters_);
- }
- return buf.GetBuffer();
- }
- PODVector<unsigned char> ScriptInstance::GetScriptDataAttr() const
- {
- if (!scriptObject_ || !methods_[METHOD_SAVE])
- return PODVector<unsigned char>();
- else
- {
- VectorBuffer buf;
- executeScript(methods_[METHOD_SAVE], [&](asIScriptContext* context) {
- context->SetArgObject(0, &buf);
- });
- return buf.GetBuffer();
- }
- }
- PODVector<unsigned char> ScriptInstance::GetScriptNetworkDataAttr() const
- {
- if (!scriptObject_ || !methods_[METHOD_WRITENETWORKUPDATE])
- return PODVector<unsigned char>();
- else
- {
- VectorBuffer buf;
- executeScript(methods_[METHOD_WRITENETWORKUPDATE], [&](asIScriptContext* context) {
- context->SetArgObject(0, &buf);
- });
- return buf.GetBuffer();
- }
- }
- void ScriptInstance::OnSceneSet(Scene* scene)
- {
- if (scene)
- UpdateEventSubscription();
- else
- {
- UnsubscribeFromEvent(E_SCENEUPDATE);
- UnsubscribeFromEvent(E_SCENEPOSTUPDATE);
- #if defined(URHO3D_PHYSICS) || defined(URHO3D_URHO2D)
- UnsubscribeFromEvent(E_PHYSICSPRESTEP);
- UnsubscribeFromEvent(E_PHYSICSPOSTSTEP);
- #endif
- subscribed_ = false;
- subscribedPostFixed_ = false;
- }
- }
- void ScriptInstance::OnMarkedDirty(Node* node)
- {
- // Script functions are not safe from worker threads
- Scene* scene = GetScene();
- if (scene && scene->IsThreadedUpdate())
- {
- scene->DelayedMarkedDirty(this);
- return;
- }
- if (scriptObject_ && methods_[METHOD_TRANSFORMCHANGED])
- executeScript(methods_[METHOD_TRANSFORMCHANGED], [](asIScriptContext*) {});
- }
- void ScriptInstance::CreateObject()
- {
- if (!scriptFile_ || className_.Empty())
- return;
- URHO3D_PROFILE(CreateScriptObject);
- scriptObject_ = scriptFile_->CreateObject(className_);
- if (scriptObject_)
- {
- // Map script object to script instance with userdata
- scriptObject_->SetUserData(this);
- GetScriptMethods();
- GetScriptAttributes();
- UpdateEventSubscription();
- if (methods_[METHOD_START])
- executeScript(methods_[METHOD_START], [](asIScriptContext*) {});
- }
- else
- URHO3D_LOGERROR("Failed to create object of class " + className_ + " from " + scriptFile_->GetName());
- }
- void ScriptInstance::ReleaseObject()
- {
- if (scriptObject_)
- {
- if (methods_[METHOD_STOP])
- executeScript(methods_[METHOD_STOP], [](asIScriptContext*) {});
- PODVector<StringHash> exceptions;
- exceptions.Push(E_RELOADSTARTED);
- exceptions.Push(E_RELOADFINISHED);
- UnsubscribeFromAllEventsExcept(exceptions, false);
- if (node_)
- node_->RemoveListener(this);
- subscribed_ = false;
- subscribedPostFixed_ = false;
- ClearScriptMethods();
- ClearScriptAttributes();
- scriptObject_->SetUserData(nullptr);
- scriptObject_->Release();
- scriptObject_ = nullptr;
- }
- }
- void ScriptInstance::ClearScriptMethods()
- {
- for (auto& method : methods_)
- method = nullptr;
- delayedCalls_.Clear();
- }
- void ScriptInstance::ClearScriptAttributes()
- {
- attributeInfos_ = *context_->GetAttributes(GetTypeStatic());
- idAttributes_.Clear();
- }
- void ScriptInstance::GetScriptMethods()
- {
- for (unsigned i = 0; i < MAX_SCRIPT_METHODS; ++i)
- methods_[i] = scriptFile_->GetMethod(scriptObject_, methodDeclarations[i]);
- }
- void ScriptInstance::GetScriptAttributes()
- {
- attributeInfos_ = *context_->GetAttributes(GetTypeStatic());
- asITypeInfo* pTypeInfo = scriptObject_->GetObjectType();
- Vector<AttributeInfo>* attrTemplate = reinterpret_cast<Vector<AttributeInfo>*>(pTypeInfo->GetUserData(eAttrMapUserIdx));
- if (!attrTemplate)
- {
- attrTemplate = new Vector<AttributeInfo>;
- pTypeInfo->SetUserData(attrTemplate, eAttrMapUserIdx);
- asIScriptEngine* engine = pTypeInfo->GetEngine();
- unsigned numProperties = pTypeInfo->GetPropertyCount();
- for (unsigned i = 0; i < numProperties; ++i)
- {
- const char* name = nullptr;
- int typeId = 0, offset; // AngelScript void typeid
- bool isPrivate = false, isProtected = false, isHandle = false, isEnum = false;
- pTypeInfo->GetProperty(i, &name, &typeId, &isPrivate, &isProtected, &offset);
- // Hide private variables or ones that begin with an underscore
- if (isPrivate || isProtected || name[0] == '_')
- continue;
- isHandle = typeId & asTYPEID_OBJHANDLE;
- String typeName = engine->GetTypeDeclaration(typeId & ~asTYPEID_OBJHANDLE);
- isEnum = (typeId & asTYPEID_MASK_OBJECT) == 0 && typeId > asTYPEID_DOUBLE;
- AttributeInfo info;
- if (isEnum)
- {
- info.type_ = VAR_INT;
- info.enumNames_ = GetSubsystem<Script>()->GetEnumValues(typeId);
- }
- else if (!isHandle)
- {
- switch (typeId)
- {
- case asTYPEID_BOOL:
- info.type_ = VAR_BOOL;
- break;
- case asTYPEID_INT32:
- case asTYPEID_UINT32:
- info.type_ = VAR_INT;
- break;
- case asTYPEID_FLOAT:
- info.type_ = VAR_FLOAT;
- break;
- case asTYPEID_DOUBLE:
- info.type_ = VAR_DOUBLE;
- break;
- case asTYPEID_INT64:
- case asTYPEID_UINT64:
- info.type_ = VAR_INT64;
- break;
- default:
- if (typeName == "Variant[]")
- info.type_ = VAR_VARIANTVECTOR;
- else if (typeName == "String[]")
- info.type_ = VAR_STRINGVECTOR;
- else
- info.type_ = Variant::GetTypeFromName(typeName);
- break;
- }
- }
- else
- {
- // For a handle type, check if it's an Object subclass with a registered factory
- StringHash typeHash(typeName);
- const HashMap<StringHash, SharedPtr<ObjectFactory>>& factories = context_->GetObjectFactories();
- HashMap<StringHash, SharedPtr<ObjectFactory>>::ConstIterator j = factories.Find(typeHash);
- if (j != factories.End())
- {
- // Check base class type. Node & Component are supported as ID attributes, Resource as a resource reference
- const TypeInfo* typeInfo = j->second_->GetTypeInfo();
- if (typeInfo->IsTypeOf<Node>())
- {
- info.mode_ = AM_NODEID;
- info.type_ = VAR_INT;
- }
- else if (typeInfo->IsTypeOf<Component>())
- {
- info.mode_ = AM_COMPONENTID;
- info.type_ = VAR_INT;
- }
- else if (typeInfo->IsTypeOf<Resource>())
- {
- info.type_ = VAR_RESOURCEREF;
- info.defaultValue_ = ResourceRef(typeHash);
- }
- }
- }
- if (info.type_ != VAR_NONE)
- {
- info.mode_ |= AM_FILE;
- info.name_ = name;
- info.ptr_ = (void*)i;
- attrTemplate->Push(info);
- }
- }
- }
- attributeInfos_.Reserve(attributeInfos_.Size() + attrTemplate->Size());
- for (unsigned i = 0; i < attrTemplate->Size(); ++i)
- {
- attributeInfos_.Push(attrTemplate->At(i));
- AttributeInfo& back = attributeInfos_.Back();
- back.ptr_ = scriptObject_->GetAddressOfProperty((size_t)back.ptr_);
- int t = 1;
- }
- }
- void ScriptInstance::StoreScriptAttributes()
- {
- if (!scriptObject_)
- return;
- storedAttributes_.Clear();
- for (unsigned i = 0; i < attributeInfos_.Size(); ++i)
- {
- const AttributeInfo& attr = attributeInfos_[i];
- if (attr.ptr_)
- storedAttributes_[attr.name_] = GetAttribute(i);
- }
- }
- void ScriptInstance::RestoreScriptAttributes()
- {
- if (!scriptObject_)
- return;
- for (unsigned i = 0; i < attributeInfos_.Size(); ++i)
- {
- const AttributeInfo& attr = attributeInfos_[i];
- if (attr.ptr_)
- {
- HashMap<String, Variant>::ConstIterator j = storedAttributes_.Find(attr.name_);
- if (j != storedAttributes_.End())
- SetAttribute(i, j->second_);
- }
- }
- // Clear after restoring once. If the attributes are no longer in use (script code has changed) they are forgotten now
- storedAttributes_.Clear();
- }
- void ScriptInstance::UpdateEventSubscription()
- {
- Scene* scene = GetScene();
- if (!scene)
- {
- URHO3D_LOGWARNING("Node is detached from scene, can not subscribe script object to update events");
- return;
- }
- bool enabled = scriptObject_ && IsEnabledEffective();
- if (enabled)
- {
- if (!subscribed_ && (methods_[METHOD_UPDATE] || methods_[METHOD_DELAYEDSTART] || delayedCalls_.Size()))
- {
- SubscribeToEvent(scene, E_SCENEUPDATE, URHO3D_HANDLER(ScriptInstance, HandleSceneUpdate));
- subscribed_ = true;
- }
- if (!subscribedPostFixed_)
- {
- if (methods_[METHOD_POSTUPDATE])
- SubscribeToEvent(scene, E_SCENEPOSTUPDATE, URHO3D_HANDLER(ScriptInstance, HandleScenePostUpdate));
- #if defined(URHO3D_PHYSICS) || defined(URHO3D_URHO2D)
- if (methods_[METHOD_FIXEDUPDATE] || methods_[METHOD_FIXEDPOSTUPDATE])
- {
- Component* world = GetFixedUpdateSource();
- if (world)
- {
- if (methods_[METHOD_FIXEDUPDATE])
- SubscribeToEvent(world, E_PHYSICSPRESTEP, URHO3D_HANDLER(ScriptInstance, HandlePhysicsPreStep));
- if (methods_[METHOD_FIXEDPOSTUPDATE])
- SubscribeToEvent(world, E_PHYSICSPOSTSTEP, URHO3D_HANDLER(ScriptInstance, HandlePhysicsPostStep));
- }
- else
- URHO3D_LOGERROR("No physics world, can not subscribe script object to fixed update events");
- }
- #endif
- subscribedPostFixed_ = true;
- }
- if (methods_[METHOD_TRANSFORMCHANGED])
- node_->AddListener(this);
- }
- else
- {
- if (subscribed_)
- {
- UnsubscribeFromEvent(scene, E_SCENEUPDATE);
- subscribed_ = false;
- }
- if (subscribedPostFixed_)
- {
- UnsubscribeFromEvent(scene, E_SCENEPOSTUPDATE);
- #if defined(URHO3D_PHYSICS) || defined(URHO3D_URHO2D)
- Component* world = GetFixedUpdateSource();
- if (world)
- {
- UnsubscribeFromEvent(world, E_PHYSICSPRESTEP);
- UnsubscribeFromEvent(world, E_PHYSICSPOSTSTEP);
- }
- #endif
- subscribedPostFixed_ = false;
- }
- if (methods_[METHOD_TRANSFORMCHANGED])
- node_->RemoveListener(this);
- }
- }
- void ScriptInstance::HandleSceneUpdate(StringHash eventType, VariantMap& eventData)
- {
- if (!scriptObject_)
- return;
- using namespace SceneUpdate;
- float timeStep = eventData[P_TIMESTEP].GetFloat();
- // Execute delayed calls
- for (unsigned i = 0; i < delayedCalls_.Size();)
- {
- DelayedCall& call = delayedCalls_[i];
- bool remove = false;
- call.delay_ -= timeStep;
- if (call.delay_ <= 0.0f)
- {
- if (!call.repeat_)
- remove = true;
- else
- call.delay_ += call.period_;
- Execute(call.declaration_, call.parameters_);
- }
- if (remove)
- delayedCalls_.Erase(i);
- else
- ++i;
- }
- // Execute delayed start before first update
- if (methods_[METHOD_DELAYEDSTART])
- {
- executeScript(methods_[METHOD_DELAYEDSTART], [](asIScriptContext*) {});
- methods_[METHOD_DELAYEDSTART] = nullptr; // Only execute once
- }
- if (methods_[METHOD_UPDATE])
- executeScript(methods_[METHOD_UPDATE], [&](asIScriptContext* context) {
- context->SetArgFloat(0, timeStep);
- });
- }
- void ScriptInstance::HandleScenePostUpdate(StringHash eventType, VariantMap& eventData)
- {
- if (!scriptObject_ || !methods_[METHOD_POSTUPDATE])
- return;
- using namespace ScenePostUpdate;
- executeScript(methods_[METHOD_POSTUPDATE], [&](asIScriptContext* context) {
- context->SetArgFloat(0, eventData[P_TIMESTEP].GetFloat());
- });
- }
- #if defined(URHO3D_PHYSICS) || defined(URHO3D_URHO2D)
- void ScriptInstance::HandlePhysicsPreStep(StringHash eventType, VariantMap& eventData)
- {
- if (!scriptObject_)
- return;
- // Execute delayed start before first fixed update if not called yet
- if (methods_[METHOD_DELAYEDSTART])
- {
- executeScript(methods_[METHOD_DELAYEDSTART], [](asIScriptContext*) {});
- methods_[METHOD_DELAYEDSTART] = nullptr; // Only execute once
- }
- if (methods_[METHOD_FIXEDUPDATE]) {
- using namespace PhysicsPreStep;
- executeScript(methods_[METHOD_FIXEDUPDATE], [&](asIScriptContext* context) {
- context->SetArgFloat(0, eventData[P_TIMESTEP].GetFloat());
- });
- }
- }
- void ScriptInstance::HandlePhysicsPostStep(StringHash eventType, VariantMap& eventData)
- {
- if (!scriptObject_ || !methods_[METHOD_FIXEDPOSTUPDATE])
- return;
- using namespace PhysicsPostStep;
- executeScript(methods_[METHOD_FIXEDPOSTUPDATE], [&](asIScriptContext* context) {
- context->SetArgFloat(0, eventData[P_TIMESTEP].GetFloat());
- });
- }
- #endif
- void ScriptInstance::HandleScriptEvent(StringHash eventType, VariantMap& eventData)
- {
- if (!IsEnabledEffective() || !scriptFile_ || !scriptObject_)
- return;
- asIScriptFunction* method = static_cast<asIScriptFunction*>(GetEventHandler()->GetUserData());
- if (method->GetParamCount()) {
- executeScript(method, [&](asIScriptContext* context) {
- context->SetArgObject(0, &eventType);
- context->SetArgObject(1, &eventData);
- });
- } else {
- executeScript(method, [](asIScriptContext* context) {});
- }
- }
- void ScriptInstance::HandleScriptFileReload(StringHash eventType, VariantMap& eventData)
- {
- StoreScriptAttributes();
- ReleaseObject();
- }
- void ScriptInstance::HandleScriptFileReloadFinished(StringHash eventType, VariantMap& eventData)
- {
- if (!className_.Empty())
- {
- CreateObject();
- RestoreScriptAttributes();
- }
- }
- asIScriptContext* GetActiveASContext()
- {
- return asGetActiveContext();
- }
- Context* GetScriptContext()
- {
- asIScriptContext* context = asGetActiveContext();
- if (context)
- return static_cast<Script*>(context->GetEngine()->GetUserData())->GetContext();
- else
- return nullptr;
- }
- ScriptInstance* GetScriptContextInstance()
- {
- asIScriptContext* context = asGetActiveContext();
- asIScriptObject* object = context ? static_cast<asIScriptObject*>(context->GetThisPointer()) : nullptr;
- if (object)
- return static_cast<ScriptInstance*>(object->GetUserData());
- else
- return nullptr;
- }
- Node* GetScriptContextNode()
- {
- ScriptInstance* instance = GetScriptContextInstance();
- return instance ? instance->GetNode() : nullptr;
- }
- Scene* GetScriptContextScene()
- {
- Scene* scene = nullptr;
- Node* node = GetScriptContextNode();
- if (node)
- scene = node->GetScene();
- // If null, try to get the default scene
- if (!scene)
- scene = GetScriptContext()->GetSubsystem<Script>()->GetDefaultScene();
- return scene;
- }
- ScriptEventListener* GetScriptContextEventListener()
- {
- // If the context has an object and that object has user data set, try and get the ScriptInstance, otherwise try and get a ScriptFile.
- asIScriptContext* context = asGetActiveContext();
- if (context)
- {
- auto* object = static_cast<asIScriptObject*>(context->GetThisPointer());
- if (object && object->GetUserData())
- return GetScriptContextInstance();
- else
- return GetScriptContextFile();
- }
- else
- return nullptr;
- }
- Object* GetScriptContextEventListenerObject()
- {
- return dynamic_cast<Object*>(GetScriptContextEventListener());
- }
- }
|