LuaScriptInstance.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781
  1. //
  2. // Copyright (c) 2008-2014 the Urho3D project.
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to deal
  6. // in the Software without restriction, including without limitation the rights
  7. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. // copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. // THE SOFTWARE.
  21. //
  22. #include "Precompiled.h"
  23. #include "CoreEvents.h"
  24. #include "Context.h"
  25. #include "Log.h"
  26. #include "LuaFile.h"
  27. #include "LuaFunction.h"
  28. #include "LuaScript.h"
  29. #include "LuaScriptInstance.h"
  30. #include "MemoryBuffer.h"
  31. #ifdef URHO3D_PHYSICS
  32. #include "PhysicsEvents.h"
  33. #include "PhysicsWorld.h"
  34. #endif
  35. #include "ResourceCache.h"
  36. #include "Scene.h"
  37. #include "SceneEvents.h"
  38. #include "ProcessUtils.h"
  39. #include "VectorBuffer.h"
  40. extern "C"
  41. {
  42. #include <lua.h>
  43. #include <lualib.h>
  44. #include <lauxlib.h>
  45. }
  46. #include "tolua++.h"
  47. #include "ToluaUtils.h"
  48. #include "DebugNew.h"
  49. namespace Urho3D
  50. {
  51. static const char* scriptObjectMethodNames[] = {
  52. "Start",
  53. "Stop",
  54. "Update",
  55. "PostUpdate",
  56. "FixedUpdate",
  57. "FixedPostUpdate",
  58. "Load",
  59. "Save",
  60. "ReadNetworkUpdate",
  61. "WriteNetworkUpdate",
  62. "ApplyAttributes",
  63. "TransformChanged"
  64. };
  65. LuaScriptInstance::LuaScriptInstance(Context* context) :
  66. Component(context),
  67. scriptObjectRef_(LUA_REFNIL)
  68. {
  69. luaScript_ = GetSubsystem<LuaScript>();
  70. luaState_ = luaScript_->GetState();
  71. attributeInfos_ = *context_->GetAttributes(GetTypeStatic());
  72. }
  73. LuaScriptInstance::~LuaScriptInstance()
  74. {
  75. ReleaseObject();
  76. }
  77. void LuaScriptInstance::RegisterObject(Context* context)
  78. {
  79. context->RegisterFactory<LuaScriptInstance>(LOGIC_CATEGORY);
  80. ACCESSOR_ATTRIBUTE("Is Enabled", IsEnabled, SetEnabled, bool, true, AM_DEFAULT);
  81. MIXED_ACCESSOR_ATTRIBUTE("Script File", GetScriptFileAttr, SetScriptFileAttr, ResourceRef, ResourceRef(LuaFile::GetTypeStatic()), AM_DEFAULT);
  82. ACCESSOR_ATTRIBUTE("Script Object Type", GetScriptObjectType, SetScriptObjectType, String, String::EMPTY, AM_DEFAULT);
  83. MIXED_ACCESSOR_ATTRIBUTE("Script Data", GetScriptDataAttr, SetScriptDataAttr, PODVector<unsigned char>, Variant::emptyBuffer, AM_FILE | AM_NOEDIT);
  84. MIXED_ACCESSOR_ATTRIBUTE("Script Network Data", GetScriptNetworkDataAttr, SetScriptNetworkDataAttr, PODVector<unsigned char>, Variant::emptyBuffer, AM_NET | AM_NOEDIT);
  85. }
  86. void LuaScriptInstance::OnSetAttribute(const AttributeInfo& attr, const Variant& src)
  87. {
  88. if (attr.ptr_ != (void*)0xffffffff)
  89. {
  90. Serializable::OnSetAttribute(attr, src);
  91. return;
  92. }
  93. if (scriptObjectRef_ == LUA_REFNIL)
  94. return;
  95. String name = attr.name_;
  96. unsigned length = name.Length();
  97. if (name.Back() == '_')
  98. length -= 1;
  99. int top = lua_gettop(luaState_);
  100. String functionName = String("Set") + name.Substring(0, 1).ToUpper() + name.Substring(1, length - 1);
  101. WeakPtr<LuaFunction> function = GetScriptObjectFunction(functionName);
  102. // If set function exist
  103. if (function)
  104. {
  105. if (function->BeginCall(this))
  106. {
  107. function->PushVariant(src);
  108. function->EndCall();
  109. }
  110. }
  111. else
  112. {
  113. lua_rawgeti(luaState_, LUA_REGISTRYINDEX, scriptObjectRef_);
  114. lua_pushstring(luaState_, name.CString());
  115. switch (attr.type_)
  116. {
  117. case VAR_BOOL:
  118. lua_pushboolean(luaState_, src.GetBool());
  119. break;
  120. case VAR_FLOAT:
  121. lua_pushnumber(luaState_, src.GetFloat());
  122. break;
  123. case VAR_STRING:
  124. tolua_pushurho3dstring(luaState_, src.GetString());
  125. break;
  126. case VAR_VECTOR2:
  127. {
  128. Vector2* value = new Vector2(src.GetVector2());
  129. tolua_pushusertype(luaState_, value, "Vector2");
  130. tolua_register_gc(luaState_, lua_gettop(luaState_));
  131. }
  132. break;
  133. case VAR_VECTOR3:
  134. {
  135. Vector3* value = new Vector3(src.GetVector3());
  136. tolua_pushusertype(luaState_, value, "Vector3");
  137. tolua_register_gc(luaState_, lua_gettop(luaState_));
  138. }
  139. break;
  140. case VAR_VECTOR4:
  141. {
  142. Vector4* value = new Vector4(src.GetVector4());
  143. tolua_pushusertype(luaState_, value, "Vector4");
  144. tolua_register_gc(luaState_, lua_gettop(luaState_));
  145. }
  146. break;
  147. case VAR_QUATERNION:
  148. {
  149. Quaternion* value = new Quaternion(src.GetQuaternion());
  150. tolua_pushusertype(luaState_, value, "Quaternion");
  151. tolua_register_gc(luaState_, lua_gettop(luaState_));
  152. }
  153. break;
  154. case VAR_COLOR:
  155. {
  156. Color* value = new Color(src.GetColor());
  157. tolua_pushusertype(luaState_, value, "Color");
  158. tolua_register_gc(luaState_, lua_gettop(luaState_));
  159. }
  160. break;
  161. case VAR_INTRECT:
  162. {
  163. IntRect* value = new IntRect(src.GetIntRect());
  164. tolua_pushusertype(luaState_, value, "IntRect");
  165. tolua_register_gc(luaState_, lua_gettop(luaState_));
  166. }
  167. break;
  168. case VAR_INTVECTOR2:
  169. {
  170. IntVector2* value = new IntVector2(src.GetIntVector2());
  171. tolua_pushusertype(luaState_, value, "IntVector2");
  172. tolua_register_gc(luaState_, lua_gettop(luaState_));
  173. }
  174. break;
  175. default:
  176. LOGERROR("Unsupported data type");
  177. lua_settop(luaState_, top);
  178. return;
  179. }
  180. lua_settable(luaState_, -3);
  181. }
  182. lua_settop(luaState_, top);
  183. }
  184. void LuaScriptInstance::OnGetAttribute(const AttributeInfo& attr, Variant& dest) const
  185. {
  186. if (attr.ptr_ != (void*)0xffffffff)
  187. {
  188. Serializable::OnGetAttribute(attr, dest);
  189. return;
  190. }
  191. if (scriptObjectRef_ == LUA_REFNIL)
  192. return;
  193. String name = attr.name_;
  194. unsigned length = name.Length();
  195. if (name.Back() == '_')
  196. length -= 1;
  197. int top = lua_gettop(luaState_);
  198. String functionName = String("Get") + name.Substring(0, 1).ToUpper() + name.Substring(1, length - 1);
  199. WeakPtr<LuaFunction> function = GetScriptObjectFunction(functionName);
  200. // If get function exist
  201. if (function)
  202. {
  203. if (function->BeginCall(this))
  204. function->EndCall(1);
  205. }
  206. else
  207. {
  208. lua_rawgeti(luaState_, LUA_REGISTRYINDEX, scriptObjectRef_);
  209. lua_pushstring(luaState_, name.CString());
  210. lua_gettable(luaState_, -2);
  211. }
  212. switch (attr.type_)
  213. {
  214. case VAR_BOOL:
  215. dest = lua_toboolean(luaState_, -1) != 0;
  216. break;
  217. case VAR_FLOAT:
  218. dest = (float)lua_tonumber(luaState_, -1);
  219. break;
  220. case VAR_STRING:
  221. dest = tolua_tourho3dstring(luaState_, -1, "");
  222. break;
  223. case VAR_VECTOR2:
  224. dest = *((Vector2*)tolua_tousertype(luaState_, -1, 0));
  225. break;
  226. case VAR_VECTOR3:
  227. dest = *((Vector3*)tolua_tousertype(luaState_, -1, 0));
  228. break;
  229. case VAR_VECTOR4:
  230. dest = *((Vector4*)tolua_tousertype(luaState_, -1, 0));
  231. break;
  232. case VAR_QUATERNION:
  233. dest = *((Quaternion*)tolua_tousertype(luaState_, -1, 0));
  234. break;
  235. case VAR_COLOR:
  236. dest = *((Color*)tolua_tousertype(luaState_, -1, 0));
  237. break;
  238. case VAR_INTRECT:
  239. dest = *((IntRect*)tolua_tousertype(luaState_, -1, 0));
  240. break;
  241. case VAR_INTVECTOR2:
  242. dest = *((IntVector2*)tolua_tousertype(luaState_, -1, 0));
  243. break;
  244. default:
  245. LOGERROR("Unsupported data type");
  246. return;
  247. }
  248. lua_settop(luaState_, top);
  249. }
  250. void LuaScriptInstance::ApplyAttributes()
  251. {
  252. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_APPLYATTRIBUTES];
  253. if (function && function->BeginCall(this))
  254. {
  255. function->EndCall();
  256. }
  257. }
  258. void LuaScriptInstance::OnSetEnabled()
  259. {
  260. if (IsEnabledEffective())
  261. SubscribeToScriptMethodEvents();
  262. else
  263. UnsubscribeFromScriptMethodEvents();
  264. }
  265. bool LuaScriptInstance::CreateObject(const String& scriptObjectType)
  266. {
  267. SetScriptFile(0);
  268. SetScriptObjectType(scriptObjectType);
  269. return scriptObjectRef_ != LUA_REFNIL;
  270. }
  271. bool LuaScriptInstance::CreateObject(LuaFile* scriptFile, const String& scriptObjectType)
  272. {
  273. SetScriptFile(scriptFile);
  274. SetScriptObjectType(scriptObjectType);
  275. return scriptObjectRef_ != LUA_REFNIL;
  276. }
  277. void LuaScriptInstance::SetScriptFile(LuaFile* scriptFile)
  278. {
  279. if (scriptFile == scriptFile_)
  280. return;
  281. scriptFile_ = scriptFile;
  282. if (!scriptFile_)
  283. return;
  284. if (!scriptFile_->LoadAndExecute(luaState_))
  285. LOGERROR("Execute Lua file failed: " + scriptFile_->GetName());
  286. }
  287. void LuaScriptInstance::SetScriptObjectType(const String& scriptObjectType)
  288. {
  289. if (scriptObjectType == scriptObjectType_)
  290. return;
  291. ReleaseObject();
  292. WeakPtr<LuaFunction> function = luaScript_->GetFunction("CreateScriptObjectInstance");
  293. if (!function || !function->BeginCall())
  294. return;
  295. function->PushLuaTable(scriptObjectType);
  296. function->PushUserType((void*)this, "LuaScriptInstance");
  297. // Return script object and attribute names
  298. if (!function->EndCall(2))
  299. return;
  300. GetScriptAttributes();
  301. scriptObjectType_ = scriptObjectType;
  302. scriptObjectRef_ = luaL_ref(luaState_, LUA_REGISTRYINDEX);
  303. // Find script object method refs
  304. FindScriptObjectMethodRefs();
  305. }
  306. void LuaScriptInstance::SetScriptDataAttr(const PODVector<unsigned char>& data)
  307. {
  308. if (scriptObjectRef_ == LUA_REFNIL)
  309. return;
  310. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_LOAD];
  311. if (function && function->BeginCall(this))
  312. {
  313. MemoryBuffer buf(data);
  314. function->PushUserType((Deserializer&)buf, "Deserializer");
  315. function->EndCall();
  316. }
  317. }
  318. void LuaScriptInstance::SetScriptNetworkDataAttr(const PODVector<unsigned char>& data)
  319. {
  320. if (scriptObjectRef_ == LUA_REFNIL)
  321. return;
  322. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_READNETWORKUPDATE];
  323. if (function && function->BeginCall(this))
  324. {
  325. MemoryBuffer buf(data);
  326. function->PushUserType((Deserializer&)buf, "Deserializer");
  327. function->EndCall();
  328. }
  329. }
  330. void LuaScriptInstance::ScriptSubscribeToEvent(const String& eventName, int functionIndex)
  331. {
  332. WeakPtr<LuaFunction> function = luaScript_->GetFunction(functionIndex);
  333. if (function)
  334. {
  335. StringHash eventType(eventName);
  336. SubscribeToEvent(eventType, HANDLER(LuaScriptInstance, HandleEvent));
  337. eventTypeToFunctionMap_[eventType] = function;
  338. }
  339. }
  340. void LuaScriptInstance::ScriptSubscribeToEvent(const String& eventName, const String& functionName)
  341. {
  342. String realFunctionName = functionName.Replaced(":", ".");
  343. WeakPtr<LuaFunction> function = luaScript_->GetFunction(realFunctionName);
  344. if (function)
  345. {
  346. StringHash eventType(eventName);
  347. SubscribeToEvent(eventType, HANDLER(LuaScriptInstance, HandleEvent));
  348. eventTypeToFunctionMap_[eventType] = function;
  349. }
  350. }
  351. void LuaScriptInstance::ScriptUnsubscribeFromEvent(const String& eventName)
  352. {
  353. StringHash eventType(eventName);
  354. HashMap<StringHash, WeakPtr<LuaFunction> >::Iterator i = eventTypeToFunctionMap_.Find(eventType);
  355. if (i != eventTypeToFunctionMap_.End())
  356. {
  357. UnsubscribeFromEvent(eventType);
  358. eventTypeToFunctionMap_.Erase(i);
  359. }
  360. }
  361. void LuaScriptInstance::ScriptUnsubscribeFromAllEvents()
  362. {
  363. if (eventTypeToFunctionMap_.Empty())
  364. return;
  365. UnsubscribeFromAllEvents();
  366. eventTypeToFunctionMap_.Clear();
  367. }
  368. void LuaScriptInstance::ScriptSubscribeToEvent(void* sender, const String& eventName, int functionIndex)
  369. {
  370. WeakPtr<LuaFunction> function = luaScript_->GetFunction(functionIndex);
  371. if (function)
  372. {
  373. Object* object = (Object*)sender;
  374. StringHash eventType(eventName);
  375. SubscribeToEvent(object, eventType, HANDLER(LuaScriptInstance, HandleObjectEvent));
  376. objectToEventTypeToFunctionMap_[object][eventType] = function;
  377. }
  378. }
  379. void LuaScriptInstance::ScriptSubscribeToEvent(void* sender, const String& eventName, const String& functionName)
  380. {
  381. String realFunctionName = functionName.Replaced(":", ".");
  382. WeakPtr<LuaFunction> function = luaScript_->GetFunction(realFunctionName);
  383. if (function)
  384. {
  385. Object* object = (Object*)sender;
  386. StringHash eventType(eventName);
  387. SubscribeToEvent(object, eventType, HANDLER(LuaScriptInstance, HandleObjectEvent));
  388. objectToEventTypeToFunctionMap_[object][eventType] = function;
  389. }
  390. }
  391. void LuaScriptInstance::ScriptUnsubscribeFromEvent(void* sender, const String& eventName)
  392. {
  393. StringHash eventType(eventName);
  394. Object* object = (Object*)sender ;
  395. HashMap<StringHash, WeakPtr<LuaFunction> >::Iterator i = objectToEventTypeToFunctionMap_[object].Find(eventType);
  396. if (i != objectToEventTypeToFunctionMap_[object].End())
  397. {
  398. UnsubscribeFromEvent(object, eventType);
  399. objectToEventTypeToFunctionMap_[object].Erase(i);
  400. }
  401. }
  402. void LuaScriptInstance::ScriptUnsubscribeFromEvents(void* sender)
  403. {
  404. Object* object = (Object*)sender;
  405. HashMap<Object*, HashMap<StringHash, WeakPtr<LuaFunction> > >::Iterator it = objectToEventTypeToFunctionMap_.Find(object);
  406. if (it == objectToEventTypeToFunctionMap_.End())
  407. return;
  408. UnsubscribeFromEvents(object);
  409. objectToEventTypeToFunctionMap_.Erase(it);
  410. }
  411. LuaFile* LuaScriptInstance::GetScriptFile() const
  412. {
  413. return scriptFile_;
  414. }
  415. PODVector<unsigned char> LuaScriptInstance::GetScriptDataAttr() const
  416. {
  417. if (scriptObjectRef_ == LUA_REFNIL)
  418. return PODVector<unsigned char>();
  419. VectorBuffer buf;
  420. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_SAVE];
  421. if (function && function->BeginCall(this))
  422. {
  423. function->PushUserType((Serializer&)buf, "Serializer");
  424. function->EndCall();
  425. }
  426. return buf.GetBuffer();
  427. }
  428. PODVector<unsigned char> LuaScriptInstance::GetScriptNetworkDataAttr() const
  429. {
  430. if (scriptObjectRef_ == LUA_REFNIL)
  431. return PODVector<unsigned char>();
  432. VectorBuffer buf;
  433. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_WRITENETWORKUPDATE];
  434. if (function && function->BeginCall(this))
  435. {
  436. function->PushUserType((Serializer&)buf, "Serializer");
  437. function->EndCall();
  438. }
  439. return buf.GetBuffer();
  440. }
  441. void LuaScriptInstance::OnMarkedDirty(Node* node)
  442. {
  443. // Script functions are not safe from worker threads
  444. Scene* scene = GetScene();
  445. if (scene && scene->IsThreadedUpdate())
  446. {
  447. scene->DelayedMarkedDirty(this);
  448. return;
  449. }
  450. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_TRANSFORMCHANGED];
  451. if (function && function->BeginCall(this))
  452. {
  453. function->EndCall();
  454. }
  455. }
  456. void LuaScriptInstance::GetScriptAttributes()
  457. {
  458. // Get all attribute names
  459. Vector<String> names;
  460. if (lua_istable(luaState_, -1))
  461. {
  462. int length = lua_objlen(luaState_, -1);
  463. for (int i = 1; i <= length; ++i)
  464. {
  465. lua_pushinteger(luaState_, i);
  466. lua_gettable(luaState_, -2);
  467. if (!lua_isstring(luaState_, -1))
  468. {
  469. lua_pop(luaState_, 1);
  470. continue;
  471. }
  472. String name = lua_tostring(luaState_, -1);
  473. names.Push(name);
  474. lua_pop(luaState_, 1);
  475. }
  476. }
  477. lua_pop(luaState_, 1);
  478. attributeInfos_ = *context_->GetAttributes(GetTypeStatic());
  479. for (unsigned i = 0; i < names.Size(); ++i)
  480. {
  481. lua_pushstring(luaState_, names[i].CString());
  482. lua_gettable(luaState_, -2);
  483. // Get attribute type
  484. int type = lua_type(luaState_, -1);
  485. AttributeInfo info;
  486. info.mode_ = AM_FILE;
  487. info.name_ = names[i];
  488. info.ptr_ = (void*)0xffffffff;
  489. switch (type)
  490. {
  491. case LUA_TBOOLEAN:
  492. info.type_ = VAR_BOOL;
  493. break;
  494. case LUA_TNUMBER:
  495. info.type_ = VAR_FLOAT;
  496. break;
  497. case LUA_TSTRING:
  498. info.type_ = VAR_STRING;
  499. break;
  500. case LUA_TUSERDATA:
  501. {
  502. String typeName = tolua_typename(luaState_, -1);
  503. lua_pop(luaState_, 1);
  504. if (typeName == "Vector2")
  505. info.type_ = VAR_VECTOR2;
  506. else if (typeName == "Vector3")
  507. info.type_ = VAR_VECTOR3;
  508. else if (typeName == "Vector4")
  509. info.type_ = VAR_VECTOR4;
  510. else if (typeName == "Quaternion")
  511. info.type_ = VAR_QUATERNION;
  512. else if (typeName == "Color")
  513. info.type_ = VAR_COLOR;
  514. else if (typeName == "Intrect")
  515. info.type_ = VAR_INTRECT;
  516. else if (typeName == "Intvector2")
  517. info.type_ = VAR_INTVECTOR2;
  518. }
  519. break;
  520. default:
  521. break;
  522. }
  523. lua_pop(luaState_, 1);
  524. if (info.type_ != VAR_NONE)
  525. attributeInfos_.Push(info);
  526. }
  527. }
  528. void LuaScriptInstance::FindScriptObjectMethodRefs()
  529. {
  530. for (unsigned i = 0; i < MAX_LUA_SCRIPT_OBJECT_METHODS; ++i)
  531. scriptObjectMethods_[i] = GetScriptObjectFunction(scriptObjectMethodNames[i]);
  532. if (IsEnabledEffective())
  533. SubscribeToScriptMethodEvents();
  534. }
  535. void LuaScriptInstance::SubscribeToScriptMethodEvents()
  536. {
  537. Scene* scene = GetScene();
  538. if (scene && scriptObjectMethods_[LSOM_UPDATE])
  539. SubscribeToEvent(scene, E_SCENEUPDATE, HANDLER(LuaScriptInstance, HandleUpdate));
  540. if (scene && scriptObjectMethods_[LSOM_POSTUPDATE])
  541. SubscribeToEvent(scene, E_SCENEPOSTUPDATE, HANDLER(LuaScriptInstance, HandlePostUpdate));
  542. #ifdef URHO3D_PHYSICS
  543. PhysicsWorld* physicsWorld = scene ? scene->GetComponent<PhysicsWorld>() : 0;
  544. if (physicsWorld && scriptObjectMethods_[LSOM_FIXEDUPDATE])
  545. SubscribeToEvent(physicsWorld, E_PHYSICSPRESTEP, HANDLER(LuaScriptInstance, HandleFixedUpdate));
  546. if (physicsWorld && scriptObjectMethods_[LSOM_FIXEDPOSTUPDATE])
  547. SubscribeToEvent(physicsWorld, E_PHYSICSPOSTSTEP, HANDLER(LuaScriptInstance, HandlePostFixedUpdate));
  548. #endif
  549. if (node_ && scriptObjectMethods_[LSOM_TRANSFORMCHANGED])
  550. node_->AddListener(this);
  551. }
  552. void LuaScriptInstance::UnsubscribeFromScriptMethodEvents()
  553. {
  554. Scene* scene = GetScene();
  555. if (scene && scriptObjectMethods_[LSOM_UPDATE])
  556. UnsubscribeFromEvent(scene, E_SCENEUPDATE);
  557. if (scene && scriptObjectMethods_[LSOM_POSTUPDATE])
  558. UnsubscribeFromEvent(scene, E_SCENEPOSTUPDATE);
  559. #ifdef URHO3D_PHYSICS
  560. PhysicsWorld* physicsWorld = scene ? scene->GetComponent<PhysicsWorld>() : 0;
  561. if (physicsWorld && scriptObjectMethods_[LSOM_FIXEDUPDATE])
  562. UnsubscribeFromEvent(physicsWorld, E_PHYSICSPRESTEP);
  563. if (physicsWorld && scriptObjectMethods_[LSOM_FIXEDPOSTUPDATE])
  564. UnsubscribeFromEvent(physicsWorld, E_PHYSICSPOSTSTEP);
  565. #endif
  566. if (node_ && scriptObjectMethods_[LSOM_TRANSFORMCHANGED])
  567. node_->RemoveListener(this);
  568. }
  569. void LuaScriptInstance::HandleUpdate(StringHash eventType, VariantMap& eventData)
  570. {
  571. using namespace Update;
  572. float timeStep = eventData[P_TIMESTEP].GetFloat();
  573. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_UPDATE];
  574. if (function && function->BeginCall(this))
  575. {
  576. function->PushFloat(timeStep);
  577. function->EndCall();
  578. }
  579. }
  580. void LuaScriptInstance::HandlePostUpdate(StringHash eventType, VariantMap& eventData)
  581. {
  582. using namespace PostUpdate;
  583. float timeStep = eventData[P_TIMESTEP].GetFloat();
  584. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_POSTUPDATE];
  585. if (function && function->BeginCall(this))
  586. {
  587. function->PushFloat(timeStep);
  588. function->EndCall();
  589. }
  590. }
  591. #ifdef URHO3D_PHYSICS
  592. void LuaScriptInstance::HandleFixedUpdate(StringHash eventType, VariantMap& eventData)
  593. {
  594. using namespace PhysicsPreStep;
  595. float timeStep = eventData[P_TIMESTEP].GetFloat();
  596. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_FIXEDUPDATE];
  597. if (function && function->BeginCall(this))
  598. {
  599. function->PushFloat(timeStep);
  600. function->EndCall();
  601. }
  602. }
  603. void LuaScriptInstance::HandlePostFixedUpdate(StringHash eventType, VariantMap& eventData)
  604. {
  605. using namespace PhysicsPostStep;
  606. float timeStep = eventData[P_TIMESTEP].GetFloat();
  607. WeakPtr<LuaFunction> function = scriptObjectMethods_[LSOM_FIXEDPOSTUPDATE];
  608. if (function && function->BeginCall(this))
  609. {
  610. function->PushFloat(timeStep);
  611. function->EndCall();
  612. }
  613. }
  614. #endif
  615. void LuaScriptInstance::HandleEvent(StringHash eventType, VariantMap& eventData)
  616. {
  617. WeakPtr<LuaFunction> function = eventTypeToFunctionMap_[eventType];
  618. if (function && function->BeginCall(this))
  619. {
  620. function->PushUserType(eventType, "StringHash");
  621. function->PushUserType(eventData, "VariantMap");
  622. function->EndCall();
  623. }
  624. }
  625. void LuaScriptInstance::HandleObjectEvent(StringHash eventType, VariantMap& eventData)
  626. {
  627. Object* object = GetEventSender();
  628. WeakPtr<LuaFunction> function = objectToEventTypeToFunctionMap_[object][eventType];
  629. if (function && function->BeginCall(this))
  630. {
  631. function->PushUserType(eventType, "StringHash");
  632. function->PushUserType(eventData, "VariantMap");
  633. function->EndCall();
  634. }
  635. }
  636. void LuaScriptInstance::ReleaseObject()
  637. {
  638. if (scriptObjectRef_ == LUA_REFNIL)
  639. return;
  640. attributeInfos_ = *context_->GetAttributes(GetTypeStatic());
  641. if (IsEnabledEffective())
  642. UnsubscribeFromScriptMethodEvents();
  643. // Unref script object
  644. luaL_unref(luaState_, LUA_REGISTRYINDEX, scriptObjectRef_);
  645. scriptObjectRef_ = LUA_REFNIL;
  646. WeakPtr<LuaFunction> function = luaScript_->GetFunction("DestroyScriptObjectInstance");
  647. if (function && function->BeginCall())
  648. {
  649. function->PushUserType((void*)this, "LuaScriptInstance");
  650. function->EndCall();
  651. }
  652. }
  653. WeakPtr<LuaFunction> LuaScriptInstance::GetScriptObjectFunction(const String& functionName) const
  654. {
  655. return luaScript_->GetFunction(scriptObjectType_ + "." + functionName, true);
  656. }
  657. void LuaScriptInstance::SetScriptFileAttr(const ResourceRef& value)
  658. {
  659. ResourceCache* cache = GetSubsystem<ResourceCache>();
  660. SetScriptFile(cache->GetResource<LuaFile>(value.name_));
  661. }
  662. ResourceRef LuaScriptInstance::GetScriptFileAttr() const
  663. {
  664. return GetResourceRef(scriptFile_, LuaFile::GetTypeStatic());
  665. }
  666. }