LuaScriptInstance.cpp 23 KB

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