ScriptFile.cpp 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848
  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 "ArrayPtr.h"
  24. #include "Context.h"
  25. #include "CoreEvents.h"
  26. #include "FileSystem.h"
  27. #include "Log.h"
  28. #include "Profiler.h"
  29. #include "ResourceCache.h"
  30. #include "Script.h"
  31. #include "ScriptFile.h"
  32. #include "ScriptInstance.h"
  33. #include <angelscript.h>
  34. #include <cstring>
  35. #include "DebugNew.h"
  36. namespace Urho3D
  37. {
  38. /// Helper class for saving AngelScript bytecode.
  39. class ByteCodeSerializer : public asIBinaryStream
  40. {
  41. public:
  42. /// Construct.
  43. ByteCodeSerializer(Serializer& dest) :
  44. dest_(dest)
  45. {
  46. }
  47. /// Read from stream (no-op).
  48. virtual void Read(void* ptr, asUINT size)
  49. {
  50. // No-op, can not read from a Serializer
  51. }
  52. /// Write to stream.
  53. virtual void Write(const void* ptr, asUINT size)
  54. {
  55. dest_.Write(ptr, size);
  56. }
  57. private:
  58. /// Destination stream.
  59. Serializer& dest_;
  60. };
  61. /// Helper class for loading AngelScript bytecode.
  62. class ByteCodeDeserializer : public asIBinaryStream
  63. {
  64. public:
  65. /// Construct.
  66. ByteCodeDeserializer(Deserializer& source) :
  67. source_(source)
  68. {
  69. }
  70. /// Read from stream.
  71. virtual void Read(void* ptr, asUINT size)
  72. {
  73. source_.Read(ptr, size);
  74. }
  75. /// Write to stream (no-op).
  76. virtual void Write(const void* ptr, asUINT size)
  77. {
  78. }
  79. private:
  80. /// Source stream.
  81. Deserializer& source_;
  82. };
  83. ScriptFile::ScriptFile(Context* context) :
  84. Resource(context),
  85. script_(GetSubsystem<Script>()),
  86. scriptModule_(0),
  87. compiled_(false),
  88. subscribed_(false)
  89. {
  90. }
  91. ScriptFile::~ScriptFile()
  92. {
  93. ReleaseModule();
  94. }
  95. void ScriptFile::RegisterObject(Context* context)
  96. {
  97. context->RegisterFactory<ScriptFile>();
  98. }
  99. bool ScriptFile::Load(Deserializer& source)
  100. {
  101. PROFILE(LoadScript);
  102. ReleaseModule();
  103. // Create the module. Discard previous module if there was one
  104. asIScriptEngine* engine = script_->GetScriptEngine();
  105. scriptModule_ = engine->GetModule(GetName().CString(), asGM_ALWAYS_CREATE);
  106. if (!scriptModule_)
  107. {
  108. LOGERROR("Failed to create script module " + GetName());
  109. return false;
  110. }
  111. // Check if this file is precompiled bytecode
  112. if (source.ReadFileID() == "ASBC")
  113. {
  114. ByteCodeDeserializer deserializer = ByteCodeDeserializer(source);
  115. if (scriptModule_->LoadByteCode(&deserializer) >= 0)
  116. {
  117. LOGINFO("Loaded script module " + GetName() + " from bytecode");
  118. compiled_ = true;
  119. // Map script module to script resource with userdata
  120. scriptModule_->SetUserData(this);
  121. return true;
  122. }
  123. else
  124. return false;
  125. }
  126. else
  127. source.Seek(0);
  128. // Not bytecode: add the initial section and check for includes
  129. if (!AddScriptSection(engine, source))
  130. return false;
  131. // Compile
  132. int result = scriptModule_->Build();
  133. if (result < 0)
  134. {
  135. LOGERROR("Failed to compile script module " + GetName());
  136. return false;
  137. }
  138. LOGINFO("Compiled script module " + GetName());
  139. compiled_ = true;
  140. // Map script module to script resource with userdata
  141. scriptModule_->SetUserData(this);
  142. return true;
  143. }
  144. void ScriptFile::AddEventHandler(StringHash eventType, const String& handlerName)
  145. {
  146. if (!compiled_)
  147. return;
  148. AddEventHandlerInternal(0, eventType, handlerName);
  149. }
  150. void ScriptFile::AddEventHandler(Object* sender, StringHash eventType, const String& handlerName)
  151. {
  152. if (!compiled_)
  153. return;
  154. if (!sender)
  155. {
  156. LOGERROR("Null event sender for event " + String(eventType) + ", handler " + handlerName);
  157. return;
  158. }
  159. AddEventHandlerInternal(sender, eventType, handlerName);
  160. }
  161. void ScriptFile::RemoveEventHandler(StringHash eventType)
  162. {
  163. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  164. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  165. if (i != eventInvokers_.End())
  166. {
  167. i->second_->UnsubscribeFromEvent(eventType);
  168. // If no longer have any subscribed events, remove the event invoker object
  169. if (!i->second_->HasEventHandlers())
  170. eventInvokers_.Erase(i);
  171. }
  172. }
  173. void ScriptFile::RemoveEventHandler(Object* sender, StringHash eventType)
  174. {
  175. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  176. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  177. if (i != eventInvokers_.End())
  178. {
  179. i->second_->UnsubscribeFromEvent(sender, eventType);
  180. if (!i->second_->HasEventHandlers())
  181. eventInvokers_.Erase(i);
  182. }
  183. }
  184. void ScriptFile::RemoveEventHandlers(Object* sender)
  185. {
  186. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  187. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  188. if (i != eventInvokers_.End())
  189. {
  190. i->second_->UnsubscribeFromEvents(sender);
  191. if (!i->second_->HasEventHandlers())
  192. eventInvokers_.Erase(i);
  193. }
  194. }
  195. void ScriptFile::RemoveEventHandlers()
  196. {
  197. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  198. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  199. if (i != eventInvokers_.End())
  200. {
  201. i->second_->UnsubscribeFromAllEvents();
  202. if (!i->second_->HasEventHandlers())
  203. eventInvokers_.Erase(i);
  204. }
  205. }
  206. void ScriptFile::RemoveEventHandlersExcept(const PODVector<StringHash>& exceptions)
  207. {
  208. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  209. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  210. if (i != eventInvokers_.End())
  211. {
  212. i->second_->UnsubscribeFromAllEventsExcept(exceptions, true);
  213. if (!i->second_->HasEventHandlers())
  214. eventInvokers_.Erase(i);
  215. }
  216. }
  217. bool ScriptFile::Execute(const String& declaration, const VariantVector& parameters, bool unprepare)
  218. {
  219. asIScriptFunction* function = GetFunction(declaration);
  220. if (!function)
  221. {
  222. LOGERROR("Function " + declaration + " not found in " + GetName());
  223. return false;
  224. }
  225. return Execute(function, parameters, unprepare);
  226. }
  227. bool ScriptFile::Execute(asIScriptFunction* function, const VariantVector& parameters, bool unprepare)
  228. {
  229. PROFILE(ExecuteFunction);
  230. if (!compiled_ || !function)
  231. return false;
  232. // It is possible that executing the function causes us to unload. Therefore do not rely on member variables
  233. // However, we are not prepared for the whole script system getting destroyed during execution (should never happen)
  234. Script* scriptSystem = script_;
  235. asIScriptContext* context = scriptSystem->GetScriptFileContext();
  236. if (context->Prepare(function) < 0)
  237. return false;
  238. SetParameters(context, function, parameters);
  239. scriptSystem->IncScriptNestingLevel();
  240. bool success = context->Execute() >= 0;
  241. if (unprepare)
  242. context->Unprepare();
  243. scriptSystem->DecScriptNestingLevel();
  244. return success;
  245. }
  246. bool ScriptFile::Execute(asIScriptObject* object, const String& declaration, const VariantVector& parameters, bool unprepare)
  247. {
  248. asIScriptFunction* method = GetMethod(object, declaration);
  249. if (!method)
  250. {
  251. LOGERROR("Method " + declaration + " not found in " + GetName());
  252. return false;
  253. }
  254. return Execute(object, method, parameters, unprepare);
  255. }
  256. bool ScriptFile::Execute(asIScriptObject* object, asIScriptFunction* method, const VariantVector& parameters, bool unprepare)
  257. {
  258. PROFILE(ExecuteMethod);
  259. if (!compiled_ || !object || !method)
  260. return false;
  261. // It is possible that executing the method causes us to unload. Therefore do not rely on member variables
  262. // However, we are not prepared for the whole script system getting destroyed during execution (should never happen)
  263. Script* scriptSystem = script_;
  264. asIScriptContext* context = scriptSystem->GetScriptFileContext();
  265. if (context->Prepare(method) < 0)
  266. return false;
  267. context->SetObject(object);
  268. SetParameters(context, method, parameters);
  269. scriptSystem->IncScriptNestingLevel();
  270. bool success = context->Execute() >= 0;
  271. if (unprepare)
  272. context->Unprepare();
  273. scriptSystem->DecScriptNestingLevel();
  274. return success;
  275. }
  276. void ScriptFile::DelayedExecute(float delay, bool repeat, const String& declaration, const VariantVector& parameters)
  277. {
  278. DelayedCall call;
  279. call.period_ = call.delay_ = Max(delay, 0.0f);
  280. call.repeat_ = repeat;
  281. call.declaration_ = declaration;
  282. call.parameters_ = parameters;
  283. delayedCalls_.Push(call);
  284. // Make sure we are registered to the application update event, because delayed calls are executed there
  285. if (!subscribed_)
  286. {
  287. SubscribeToEvent(E_UPDATE, HANDLER(ScriptFile, HandleUpdate));
  288. subscribed_ = true;
  289. }
  290. }
  291. void ScriptFile::ClearDelayedExecute(const String& declaration)
  292. {
  293. if (declaration.Empty())
  294. delayedCalls_.Clear();
  295. else
  296. {
  297. for (Vector<DelayedCall>::Iterator i = delayedCalls_.Begin(); i != delayedCalls_.End();)
  298. {
  299. if (declaration == i->declaration_)
  300. i = delayedCalls_.Erase(i);
  301. else
  302. ++i;
  303. }
  304. }
  305. }
  306. asIScriptObject* ScriptFile::CreateObject(const String& className)
  307. {
  308. PROFILE(CreateObject);
  309. if (!compiled_)
  310. return 0;
  311. asIScriptContext* context = script_->GetScriptFileContext();
  312. asIScriptEngine* engine = script_->GetScriptEngine();
  313. asIObjectType *type = engine->GetObjectTypeById(scriptModule_->GetTypeIdByDecl(className.CString()));
  314. if (!type)
  315. return 0;
  316. // Ensure that the type implements the "ScriptObject" interface, so it can be returned to script properly
  317. bool found = false;
  318. HashMap<asIObjectType*, bool>::ConstIterator i = validClasses_.Find(type);
  319. if (i != validClasses_.End())
  320. found = i->second_;
  321. else
  322. {
  323. unsigned numInterfaces = type->GetInterfaceCount();
  324. for (unsigned j = 0; j < numInterfaces; ++j)
  325. {
  326. asIObjectType* interfaceType = type->GetInterface(j);
  327. if (!strcmp(interfaceType->GetName(), "ScriptObject"))
  328. {
  329. found = true;
  330. break;
  331. }
  332. }
  333. validClasses_[type] = found;
  334. }
  335. if (!found)
  336. {
  337. LOGERROR("Script class " + className + " does not implement the ScriptObject interface");
  338. return 0;
  339. }
  340. // Get the factory function id from the object type
  341. String factoryName = className + "@ " + className + "()";
  342. asIScriptFunction* factory = type->GetFactoryByDecl(factoryName.CString());
  343. if (!factory || context->Prepare(factory) < 0 || context->Execute() < 0)
  344. return 0;
  345. asIScriptObject* obj = *(static_cast<asIScriptObject**>(context->GetAddressOfReturnValue()));
  346. if (obj)
  347. obj->AddRef();
  348. return obj;
  349. }
  350. bool ScriptFile::SaveByteCode(Serializer& dest)
  351. {
  352. if (compiled_)
  353. {
  354. dest.WriteFileID("ASBC");
  355. ByteCodeSerializer serializer = ByteCodeSerializer(dest);
  356. return scriptModule_->SaveByteCode(&serializer, true) >= 0;
  357. }
  358. else
  359. return false;
  360. }
  361. asIScriptFunction* ScriptFile::GetFunction(const String& declaration)
  362. {
  363. if (!compiled_)
  364. return 0;
  365. HashMap<String, asIScriptFunction*>::ConstIterator i = functions_.Find(declaration);
  366. if (i != functions_.End())
  367. return i->second_;
  368. asIScriptFunction* function = scriptModule_->GetFunctionByDecl(declaration.CString());
  369. functions_[declaration] = function;
  370. return function;
  371. }
  372. asIScriptFunction* ScriptFile::GetMethod(asIScriptObject* object, const String& declaration)
  373. {
  374. if (!compiled_ || !object)
  375. return 0;
  376. asIObjectType* type = object->GetObjectType();
  377. if (!type)
  378. return 0;
  379. HashMap<asIObjectType*, HashMap<String, asIScriptFunction*> >::ConstIterator i = methods_.Find(type);
  380. if (i != methods_.End())
  381. {
  382. HashMap<String, asIScriptFunction*>::ConstIterator j = i->second_.Find(declaration);
  383. if (j != i->second_.End())
  384. return j->second_;
  385. }
  386. asIScriptFunction* function = type->GetMethodByDecl(declaration.CString());
  387. methods_[type][declaration] = function;
  388. return function;
  389. }
  390. void ScriptFile::CleanupEventInvoker(asIScriptObject* object)
  391. {
  392. eventInvokers_.Erase(object);
  393. }
  394. void ScriptFile::AddEventHandlerInternal(Object* sender, StringHash eventType, const String& handlerName)
  395. {
  396. String declaration = "void " + handlerName + "(StringHash, VariantMap&)";
  397. asIScriptFunction* function = 0;
  398. asIScriptObject* receiver = static_cast<asIScriptObject*>(asGetActiveContext()->GetThisPointer());
  399. if (receiver)
  400. function = GetMethod(receiver, declaration);
  401. else
  402. function = GetFunction(declaration);
  403. if (!function)
  404. {
  405. declaration = "void " + handlerName + "()";
  406. if (receiver)
  407. function = GetMethod(receiver, declaration);
  408. else
  409. function = GetFunction(declaration);
  410. if (!function)
  411. {
  412. LOGERROR("Event handler function " + handlerName + " not found in " + GetName());
  413. return;
  414. }
  415. }
  416. HashMap<asIScriptObject*, SharedPtr<ScriptEventInvoker> >::Iterator i = eventInvokers_.Find(receiver);
  417. // Remove previous handler in case an object pointer gets reused
  418. if (i != eventInvokers_.End() && !i->second_->IsObjectAlive())
  419. {
  420. eventInvokers_.Erase(i);
  421. i = eventInvokers_.End();
  422. }
  423. if (i == eventInvokers_.End())
  424. i = eventInvokers_.Insert(MakePair(receiver, SharedPtr<ScriptEventInvoker>(new ScriptEventInvoker(this, receiver))));
  425. if (!sender)
  426. {
  427. i->second_->SubscribeToEvent(eventType, new EventHandlerImpl<ScriptEventInvoker>
  428. (i->second_, &ScriptEventInvoker::HandleScriptEvent, (void*)function));
  429. }
  430. else
  431. {
  432. i->second_->SubscribeToEvent(sender, eventType, new EventHandlerImpl<ScriptEventInvoker>
  433. (i->second_, &ScriptEventInvoker::HandleScriptEvent, (void*)function));
  434. }
  435. }
  436. bool ScriptFile::AddScriptSection(asIScriptEngine* engine, Deserializer& source)
  437. {
  438. ResourceCache* cache = GetSubsystem<ResourceCache>();
  439. unsigned dataSize = source.GetSize();
  440. SharedArrayPtr<char> buffer(new char[dataSize]);
  441. source.Read((void*)buffer.Get(), dataSize);
  442. // Pre-parse for includes
  443. // Adapted from Angelscript's scriptbuilder add-on
  444. Vector<String> includeFiles;
  445. unsigned pos = 0;
  446. while(pos < dataSize)
  447. {
  448. int len;
  449. asETokenClass t = engine->ParseToken(&buffer[pos], dataSize - pos, &len);
  450. if (t == asTC_COMMENT || t == asTC_WHITESPACE)
  451. {
  452. pos += len;
  453. continue;
  454. }
  455. // Is this a preprocessor directive?
  456. if (buffer[pos] == '#')
  457. {
  458. int start = pos++;
  459. asETokenClass t = engine->ParseToken(&buffer[pos], dataSize - pos, &len);
  460. if (t == asTC_IDENTIFIER)
  461. {
  462. String token(&buffer[pos], len);
  463. if (token == "include")
  464. {
  465. pos += len;
  466. t = engine->ParseToken(&buffer[pos], dataSize - pos, &len);
  467. if (t == asTC_WHITESPACE)
  468. {
  469. pos += len;
  470. t = engine->ParseToken(&buffer[pos], dataSize - pos, &len);
  471. }
  472. if (t == asTC_VALUE && len > 2 && buffer[pos] == '"')
  473. {
  474. // Get the include file
  475. String includeFile(&buffer[pos+1], len - 2);
  476. pos += len;
  477. // If the file is not found as it is, add the path of current file but only if it is found there
  478. if (!cache->Exists(includeFile))
  479. {
  480. String prefixedIncludeFile = GetPath(GetName()) + includeFile;
  481. if (cache->Exists(prefixedIncludeFile))
  482. includeFile = prefixedIncludeFile;
  483. }
  484. String includeFileLower = includeFile.ToLower();
  485. // If not included yet, store it for later processing
  486. if (!includeFiles_.Contains(includeFileLower))
  487. {
  488. includeFiles_.Insert(includeFileLower);
  489. includeFiles.Push(includeFile);
  490. }
  491. // Overwrite the include directive with space characters to avoid compiler error
  492. memset(&buffer[start], ' ', pos - start);
  493. }
  494. }
  495. }
  496. }
  497. // Don't search includes within statement blocks or between tokens in statements
  498. else
  499. {
  500. int len;
  501. // Skip until ; or { whichever comes first
  502. while (pos < dataSize && buffer[pos] != ';' && buffer[pos] != '{')
  503. {
  504. engine->ParseToken(&buffer[pos], 0, &len);
  505. pos += len;
  506. }
  507. // Skip entire statement block
  508. if (pos < dataSize && buffer[pos] == '{')
  509. {
  510. ++pos;
  511. // Find the end of the statement block
  512. int level = 1;
  513. while (level > 0 && pos < dataSize)
  514. {
  515. asETokenClass t = engine->ParseToken(&buffer[pos], 0, &len);
  516. if (t == asTC_KEYWORD)
  517. {
  518. if (buffer[pos] == '{')
  519. ++level;
  520. else if(buffer[pos] == '}')
  521. --level;
  522. }
  523. pos += len;
  524. }
  525. }
  526. else
  527. ++pos;
  528. }
  529. }
  530. // Process includes first
  531. for (unsigned i = 0; i < includeFiles.Size(); ++i)
  532. {
  533. cache->StoreResourceDependency(this, includeFiles[i]);
  534. SharedPtr<File> file = cache->GetFile(includeFiles[i]);
  535. if (file)
  536. {
  537. if (!AddScriptSection(engine, *file))
  538. return false;
  539. }
  540. else
  541. {
  542. LOGERROR("Could not process all the include directives in " + GetName() + ": missing " + includeFiles[i]);
  543. return false;
  544. }
  545. }
  546. // Then add this section
  547. if (scriptModule_->AddScriptSection(source.GetName().CString(), (const char*)buffer.Get(), dataSize) < 0)
  548. {
  549. LOGERROR("Failed to add script section " + source.GetName());
  550. return false;
  551. }
  552. SetMemoryUse(GetMemoryUse() + dataSize);
  553. return true;
  554. }
  555. void ScriptFile::SetParameters(asIScriptContext* context, asIScriptFunction* function, const VariantVector& parameters)
  556. {
  557. unsigned paramCount = function->GetParamCount();
  558. for (unsigned i = 0; i < parameters.Size() && i < paramCount; ++i)
  559. {
  560. int paramType = function->GetParamTypeId(i);
  561. switch (paramType)
  562. {
  563. case asTYPEID_BOOL:
  564. context->SetArgByte(i, (unsigned char)parameters[i].GetBool());
  565. break;
  566. case asTYPEID_INT8:
  567. case asTYPEID_UINT8:
  568. context->SetArgByte(i, parameters[i].GetInt());
  569. break;
  570. case asTYPEID_INT16:
  571. case asTYPEID_UINT16:
  572. context->SetArgWord(i, parameters[i].GetInt());
  573. break;
  574. case asTYPEID_INT32:
  575. case asTYPEID_UINT32:
  576. context->SetArgDWord(i, parameters[i].GetInt());
  577. break;
  578. case asTYPEID_FLOAT:
  579. context->SetArgFloat(i, parameters[i].GetFloat());
  580. break;
  581. default:
  582. if (paramType & asTYPEID_APPOBJECT)
  583. {
  584. switch (parameters[i].GetType())
  585. {
  586. case VAR_VECTOR2:
  587. context->SetArgObject(i, (void*)&parameters[i].GetVector2());
  588. break;
  589. case VAR_VECTOR3:
  590. context->SetArgObject(i, (void*)&parameters[i].GetVector3());
  591. break;
  592. case VAR_VECTOR4:
  593. context->SetArgObject(i, (void*)&parameters[i].GetVector4());
  594. break;
  595. case VAR_QUATERNION:
  596. context->SetArgObject(i, (void*)&parameters[i].GetQuaternion());
  597. break;
  598. case VAR_STRING:
  599. context->SetArgObject(i, (void*)&parameters[i].GetString());
  600. break;
  601. case VAR_VOIDPTR:
  602. context->SetArgObject(i, parameters[i].GetVoidPtr());
  603. break;
  604. case VAR_PTR:
  605. context->SetArgObject(i, (void*)parameters[i].GetPtr());
  606. break;
  607. default:
  608. break;
  609. }
  610. }
  611. break;
  612. }
  613. }
  614. }
  615. void ScriptFile::ReleaseModule()
  616. {
  617. if (scriptModule_)
  618. {
  619. script_->ClearObjectTypeCache();
  620. // Clear search caches and event handlers
  621. includeFiles_.Clear();
  622. validClasses_.Clear();
  623. functions_.Clear();
  624. methods_.Clear();
  625. delayedCalls_.Clear();
  626. eventInvokers_.Clear();
  627. // Remove the module
  628. scriptModule_->SetUserData(0);
  629. asIScriptEngine* engine = script_->GetScriptEngine();
  630. engine->DiscardModule(GetName().CString());
  631. scriptModule_ = 0;
  632. compiled_ = false;
  633. SetMemoryUse(0);
  634. ResourceCache* cache = GetSubsystem<ResourceCache>();
  635. cache->ResetDependencies(this);
  636. }
  637. }
  638. void ScriptFile::HandleUpdate(StringHash eventType, VariantMap& eventData)
  639. {
  640. if (!compiled_)
  641. return;
  642. using namespace Update;
  643. float timeStep = eventData[P_TIMESTEP].GetFloat();
  644. // Execute delayed calls
  645. for (unsigned i = 0; i < delayedCalls_.Size();)
  646. {
  647. DelayedCall& call = delayedCalls_[i];
  648. bool remove = false;
  649. call.delay_ -= timeStep;
  650. if (call.delay_ <= 0.0f)
  651. {
  652. if (!call.repeat_)
  653. remove = true;
  654. else
  655. call.delay_ += call.period_;
  656. Execute(call.declaration_, call.parameters_);
  657. }
  658. if (remove)
  659. delayedCalls_.Erase(i);
  660. else
  661. ++i;
  662. }
  663. }
  664. ScriptEventInvoker::ScriptEventInvoker(ScriptFile* file, asIScriptObject* object) :
  665. Object(file->GetContext()),
  666. file_(file),
  667. sharedBool_(0),
  668. object_(object)
  669. {
  670. if (object_)
  671. {
  672. sharedBool_ = object_->GetEngine()->GetWeakRefFlagOfScriptObject(object_, object_->GetObjectType());
  673. if (sharedBool_)
  674. sharedBool_->AddRef();
  675. }
  676. }
  677. ScriptEventInvoker::~ScriptEventInvoker()
  678. {
  679. if (sharedBool_)
  680. sharedBool_->Release();
  681. sharedBool_ = 0;
  682. object_ = 0;
  683. }
  684. bool ScriptEventInvoker::IsObjectAlive() const
  685. {
  686. if (sharedBool_)
  687. {
  688. // Return inverse as Get returns true when an asIScriptObject is dead.
  689. return !sharedBool_->Get();
  690. }
  691. return true;
  692. }
  693. void ScriptEventInvoker::HandleScriptEvent(StringHash eventType, VariantMap& eventData)
  694. {
  695. if (!file_->IsCompiled())
  696. return;
  697. asIScriptFunction* method = static_cast<asIScriptFunction*>(GetEventHandler()->GetUserData());
  698. if (object_ && !IsObjectAlive())
  699. {
  700. file_->CleanupEventInvoker(object_);
  701. return;
  702. }
  703. VariantVector parameters;
  704. if (method->GetParamCount() > 0)
  705. {
  706. parameters.Push(Variant((void*)&eventType));
  707. parameters.Push(Variant((void*)&eventData));
  708. }
  709. if (object_)
  710. file_->Execute(object_, method, parameters);
  711. else
  712. file_->Execute(method, parameters);
  713. }
  714. ScriptFile* GetScriptContextFile()
  715. {
  716. asIScriptContext* context = asGetActiveContext();
  717. asIScriptFunction* function = context ? context->GetFunction() : 0;
  718. asIScriptModule* module = function ? function->GetEngine()->GetModule(function->GetModuleName()) : 0;
  719. if (module)
  720. return static_cast<ScriptFile*>(module->GetUserData());
  721. else
  722. return 0;
  723. }
  724. }