ResourceCache.cpp 33 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063
  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 "../Precompiled.h"
  23. #include "../Core/Context.h"
  24. #include "../Core/CoreEvents.h"
  25. #include "../Core/Profiler.h"
  26. #include "../Core/WorkQueue.h"
  27. #include "../IO/FileSystem.h"
  28. #include "../IO/FileWatcher.h"
  29. #include "../IO/Log.h"
  30. #include "../IO/PackageFile.h"
  31. #include "../Resource/BackgroundLoader.h"
  32. #include "../Resource/Image.h"
  33. #include "../Resource/JSONFile.h"
  34. #include "../Resource/PListFile.h"
  35. #include "../Resource/ResourceCache.h"
  36. #include "../Resource/ResourceEvents.h"
  37. #include "../Resource/XMLFile.h"
  38. #include "../DebugNew.h"
  39. namespace Atomic
  40. {
  41. static const char* checkDirs[] =
  42. {
  43. "Fonts",
  44. "Materials",
  45. "Models",
  46. "Music",
  47. "Objects",
  48. "Particle",
  49. "PostProcess",
  50. "RenderPaths",
  51. "Scenes",
  52. "Scripts",
  53. "Sounds",
  54. "Shaders",
  55. "Techniques",
  56. "Textures",
  57. "UI",
  58. 0
  59. };
  60. static const SharedPtr<Resource> noResource;
  61. const char* PAK_EXTENSION = ".pak";
  62. const char* PAK_FILTER = "*.pak";
  63. ResourceCache::ResourceCache(Context* context) :
  64. Object(context),
  65. autoReloadResources_(false),
  66. returnFailedResources_(false),
  67. searchPackagesFirst_(true),
  68. isRouting_(false),
  69. finishBackgroundResourcesMs_(5)
  70. {
  71. // Register Resource library object factories
  72. RegisterResourceLibrary(context_);
  73. // Create resource background loader. Its thread will start on the first background request
  74. backgroundLoader_ = new BackgroundLoader(this);
  75. // Subscribe BeginFrame for handling directory watchers and background loaded resource finalization
  76. SubscribeToEvent(E_BEGINFRAME, HANDLER(ResourceCache, HandleBeginFrame));
  77. }
  78. ResourceCache::~ResourceCache()
  79. {
  80. // Shut down the background loader first
  81. backgroundLoader_.Reset();
  82. }
  83. bool ResourceCache::AddResourceDir(const String& pathName, unsigned priority)
  84. {
  85. MutexLock lock(resourceMutex_);
  86. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  87. if (!fileSystem || !fileSystem->DirExists(pathName))
  88. {
  89. LOGERROR("Could not open directory " + pathName);
  90. return false;
  91. }
  92. // Convert path to absolute
  93. String fixedPath = SanitateResourceDirName(pathName);
  94. // Check that the same path does not already exist
  95. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  96. {
  97. if (!resourceDirs_[i].Compare(fixedPath, false))
  98. return true;
  99. }
  100. if (priority < resourceDirs_.Size())
  101. resourceDirs_.Insert(priority, fixedPath);
  102. else
  103. resourceDirs_.Push(fixedPath);
  104. // If resource auto-reloading active, create a file watcher for the directory
  105. if (autoReloadResources_)
  106. {
  107. SharedPtr<FileWatcher> watcher(new FileWatcher(context_));
  108. watcher->StartWatching(fixedPath, true);
  109. fileWatchers_.Push(watcher);
  110. }
  111. LOGINFO("Added resource path " + fixedPath);
  112. return true;
  113. }
  114. bool ResourceCache::AddPackageFile(PackageFile* package, unsigned priority)
  115. {
  116. MutexLock lock(resourceMutex_);
  117. // Do not add packages that failed to load
  118. if (!package || !package->GetNumFiles())
  119. return false;
  120. if (priority < packages_.Size())
  121. packages_.Insert(priority, SharedPtr<PackageFile>(package));
  122. else
  123. packages_.Push(SharedPtr<PackageFile>(package));
  124. LOGINFO("Added resource package " + package->GetName());
  125. return true;
  126. }
  127. bool ResourceCache::AddPackageFile(const String& fileName, unsigned priority)
  128. {
  129. SharedPtr<PackageFile> package(new PackageFile(context_));
  130. return package->Open(fileName) && AddPackageFile(package);
  131. }
  132. bool ResourceCache::AddManualResource(Resource* resource)
  133. {
  134. if (!resource)
  135. {
  136. LOGERROR("Null manual resource");
  137. return false;
  138. }
  139. const String& name = resource->GetName();
  140. if (name.Empty())
  141. {
  142. LOGERROR("Manual resource with empty name, can not add");
  143. return false;
  144. }
  145. resource->ResetUseTimer();
  146. resourceGroups_[resource->GetType()].resources_[resource->GetNameHash()] = resource;
  147. UpdateResourceGroup(resource->GetType());
  148. return true;
  149. }
  150. void ResourceCache::RemoveResourceDir(const String& pathName)
  151. {
  152. MutexLock lock(resourceMutex_);
  153. String fixedPath = SanitateResourceDirName(pathName);
  154. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  155. {
  156. if (!resourceDirs_[i].Compare(fixedPath, false))
  157. {
  158. resourceDirs_.Erase(i);
  159. // Remove the filewatcher with the matching path
  160. for (unsigned j = 0; j < fileWatchers_.Size(); ++j)
  161. {
  162. if (!fileWatchers_[j]->GetPath().Compare(fixedPath, false))
  163. {
  164. fileWatchers_.Erase(j);
  165. break;
  166. }
  167. }
  168. LOGINFO("Removed resource path " + fixedPath);
  169. return;
  170. }
  171. }
  172. }
  173. void ResourceCache::RemovePackageFile(PackageFile* package, bool releaseResources, bool forceRelease)
  174. {
  175. MutexLock lock(resourceMutex_);
  176. for (Vector<SharedPtr<PackageFile> >::Iterator i = packages_.Begin(); i != packages_.End(); ++i)
  177. {
  178. if (*i == package)
  179. {
  180. if (releaseResources)
  181. ReleasePackageResources(*i, forceRelease);
  182. LOGINFO("Removed resource package " + (*i)->GetName());
  183. packages_.Erase(i);
  184. return;
  185. }
  186. }
  187. }
  188. void ResourceCache::RemovePackageFile(const String& fileName, bool releaseResources, bool forceRelease)
  189. {
  190. MutexLock lock(resourceMutex_);
  191. // Compare the name and extension only, not the path
  192. String fileNameNoPath = GetFileNameAndExtension(fileName);
  193. for (Vector<SharedPtr<PackageFile> >::Iterator i = packages_.Begin(); i != packages_.End(); ++i)
  194. {
  195. if (!GetFileNameAndExtension((*i)->GetName()).Compare(fileNameNoPath, false))
  196. {
  197. if (releaseResources)
  198. ReleasePackageResources(*i, forceRelease);
  199. LOGINFO("Removed resource package " + (*i)->GetName());
  200. packages_.Erase(i);
  201. return;
  202. }
  203. }
  204. }
  205. void ResourceCache::ReleaseResource(StringHash type, const String& name, bool force)
  206. {
  207. StringHash nameHash(name);
  208. const SharedPtr<Resource>& existingRes = FindResource(type, nameHash);
  209. if (!existingRes)
  210. return;
  211. // If other references exist, do not release, unless forced
  212. if ((existingRes.Refs() == 1 && existingRes.WeakRefs() == 0) || force)
  213. {
  214. resourceGroups_[type].resources_.Erase(nameHash);
  215. UpdateResourceGroup(type);
  216. }
  217. }
  218. void ResourceCache::ReleaseResources(StringHash type, bool force)
  219. {
  220. bool released = false;
  221. HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
  222. if (i != resourceGroups_.End())
  223. {
  224. for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
  225. j != i->second_.resources_.End();)
  226. {
  227. HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
  228. // If other references exist, do not release, unless forced
  229. if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
  230. {
  231. i->second_.resources_.Erase(current);
  232. released = true;
  233. }
  234. }
  235. }
  236. if (released)
  237. UpdateResourceGroup(type);
  238. }
  239. void ResourceCache::ReleaseResources(StringHash type, const String& partialName, bool force)
  240. {
  241. bool released = false;
  242. HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
  243. if (i != resourceGroups_.End())
  244. {
  245. for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
  246. j != i->second_.resources_.End();)
  247. {
  248. HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
  249. if (current->second_->GetName().Contains(partialName))
  250. {
  251. // If other references exist, do not release, unless forced
  252. if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
  253. {
  254. i->second_.resources_.Erase(current);
  255. released = true;
  256. }
  257. }
  258. }
  259. }
  260. if (released)
  261. UpdateResourceGroup(type);
  262. }
  263. void ResourceCache::ReleaseResources(const String& partialName, bool force)
  264. {
  265. // Some resources refer to others, like materials to textures. Release twice to ensure these get released.
  266. // This is not necessary if forcing release
  267. unsigned repeat = force ? 1 : 2;
  268. while (repeat--)
  269. {
  270. for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
  271. {
  272. bool released = false;
  273. for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
  274. j != i->second_.resources_.End();)
  275. {
  276. HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
  277. if (current->second_->GetName().Contains(partialName))
  278. {
  279. // If other references exist, do not release, unless forced
  280. if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
  281. {
  282. i->second_.resources_.Erase(current);
  283. released = true;
  284. }
  285. }
  286. }
  287. if (released)
  288. UpdateResourceGroup(i->first_);
  289. }
  290. }
  291. }
  292. void ResourceCache::ReleaseAllResources(bool force)
  293. {
  294. unsigned repeat = force ? 1 : 2;
  295. while (repeat--)
  296. {
  297. for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin();
  298. i != resourceGroups_.End(); ++i)
  299. {
  300. bool released = false;
  301. for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
  302. j != i->second_.resources_.End();)
  303. {
  304. HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
  305. // If other references exist, do not release, unless forced
  306. if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
  307. {
  308. i->second_.resources_.Erase(current);
  309. released = true;
  310. }
  311. }
  312. if (released)
  313. UpdateResourceGroup(i->first_);
  314. }
  315. }
  316. }
  317. bool ResourceCache::ReloadResource(Resource* resource)
  318. {
  319. if (!resource)
  320. return false;
  321. resource->SendEvent(E_RELOADSTARTED);
  322. bool success = false;
  323. SharedPtr<File> file = GetFile(resource->GetName());
  324. if (file)
  325. success = resource->Load(*(file.Get()));
  326. if (success)
  327. {
  328. resource->ResetUseTimer();
  329. UpdateResourceGroup(resource->GetType());
  330. resource->SendEvent(E_RELOADFINISHED);
  331. return true;
  332. }
  333. // If reloading failed, do not remove the resource from cache, to allow for a new live edit to
  334. // attempt loading again
  335. resource->SendEvent(E_RELOADFAILED);
  336. return false;
  337. }
  338. void ResourceCache::ReloadResourceWithDependencies(const String& fileName)
  339. {
  340. StringHash fileNameHash(fileName);
  341. // If the filename is a resource we keep track of, reload it
  342. const SharedPtr<Resource>& resource = FindResource(fileNameHash);
  343. if (resource)
  344. {
  345. LOGDEBUG("Reloading changed resource " + fileName);
  346. ReloadResource(resource);
  347. }
  348. // Always perform dependency resource check for resource loaded from XML file as it could be used in inheritance
  349. if (!resource || GetExtension(resource->GetName()) == ".xml")
  350. {
  351. // Check if this is a dependency resource, reload dependents
  352. HashMap<StringHash, HashSet<StringHash> >::ConstIterator j = dependentResources_.Find(fileNameHash);
  353. if (j != dependentResources_.End())
  354. {
  355. // Reloading a resource may modify the dependency tracking structure. Therefore collect the
  356. // resources we need to reload first
  357. Vector<SharedPtr<Resource> > dependents;
  358. dependents.Reserve(j->second_.Size());
  359. for (HashSet<StringHash>::ConstIterator k = j->second_.Begin(); k != j->second_.End(); ++k)
  360. {
  361. const SharedPtr<Resource>& dependent = FindResource(*k);
  362. if (dependent)
  363. dependents.Push(dependent);
  364. }
  365. for (unsigned k = 0; k < dependents.Size(); ++k)
  366. {
  367. LOGDEBUG("Reloading resource " + dependents[k]->GetName() + " depending on " + fileName);
  368. ReloadResource(dependents[k]);
  369. }
  370. }
  371. }
  372. }
  373. void ResourceCache::SetMemoryBudget(StringHash type, unsigned budget)
  374. {
  375. resourceGroups_[type].memoryBudget_ = budget;
  376. }
  377. void ResourceCache::SetAutoReloadResources(bool enable)
  378. {
  379. if (enable != autoReloadResources_)
  380. {
  381. if (enable)
  382. {
  383. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  384. {
  385. SharedPtr<FileWatcher> watcher(new FileWatcher(context_));
  386. watcher->StartWatching(resourceDirs_[i], true);
  387. fileWatchers_.Push(watcher);
  388. }
  389. }
  390. else
  391. fileWatchers_.Clear();
  392. autoReloadResources_ = enable;
  393. }
  394. }
  395. void ResourceCache::AddResourceRouter(ResourceRouter* router, bool addAsFirst)
  396. {
  397. // Check for duplicate
  398. for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
  399. {
  400. if (resourceRouters_[i] == router)
  401. return;
  402. }
  403. if (addAsFirst)
  404. resourceRouters_.Insert(0, SharedPtr<ResourceRouter>(router));
  405. else
  406. resourceRouters_.Push(SharedPtr<ResourceRouter>(router));
  407. }
  408. void ResourceCache::RemoveResourceRouter(ResourceRouter* router)
  409. {
  410. for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
  411. {
  412. if (resourceRouters_[i] == router)
  413. {
  414. resourceRouters_.Erase(i);
  415. return;
  416. }
  417. }
  418. }
  419. SharedPtr<File> ResourceCache::GetFile(const String& nameIn, bool sendEventOnFailure)
  420. {
  421. MutexLock lock(resourceMutex_);
  422. String name = SanitateResourceName(nameIn);
  423. if (!isRouting_)
  424. {
  425. isRouting_ = true;
  426. for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
  427. resourceRouters_[i]->Route(name, RESOURCE_GETFILE);
  428. isRouting_ = false;
  429. }
  430. if (name.Length())
  431. {
  432. File* file = 0;
  433. if (searchPackagesFirst_)
  434. {
  435. file = SearchPackages(name);
  436. if (!file)
  437. file = SearchResourceDirs(name);
  438. }
  439. else
  440. {
  441. file = SearchResourceDirs(name);
  442. if (!file)
  443. file = SearchPackages(name);
  444. }
  445. if (file)
  446. return SharedPtr<File>(file);
  447. }
  448. if (sendEventOnFailure)
  449. {
  450. if (resourceRouters_.Size() && name.Empty() && !nameIn.Empty())
  451. LOGERROR("Resource request " + nameIn + " was blocked");
  452. else
  453. LOGERROR("Could not find resource " + name);
  454. if (Thread::IsMainThread())
  455. {
  456. using namespace ResourceNotFound;
  457. VariantMap& eventData = GetEventDataMap();
  458. eventData[P_RESOURCENAME] = name.Length() ? name : nameIn;
  459. SendEvent(E_RESOURCENOTFOUND, eventData);
  460. }
  461. }
  462. return SharedPtr<File>();
  463. }
  464. Resource* ResourceCache::GetExistingResource(StringHash type, const String& nameIn)
  465. {
  466. String name = SanitateResourceName(nameIn);
  467. if (!Thread::IsMainThread())
  468. {
  469. LOGERROR("Attempted to get resource " + name + " from outside the main thread");
  470. return 0;
  471. }
  472. // If empty name, return null pointer immediately
  473. if (name.Empty())
  474. return 0;
  475. StringHash nameHash(name);
  476. const SharedPtr<Resource>& existing = FindResource(type, nameHash);
  477. return existing;
  478. }
  479. Resource* ResourceCache::GetResource(StringHash type, const String& nameIn, bool sendEventOnFailure)
  480. {
  481. String name = SanitateResourceName(nameIn);
  482. if (!Thread::IsMainThread())
  483. {
  484. LOGERROR("Attempted to get resource " + name + " from outside the main thread");
  485. return 0;
  486. }
  487. // If empty name, return null pointer immediately
  488. if (name.Empty())
  489. return 0;
  490. StringHash nameHash(name);
  491. // Check if the resource is being background loaded but is now needed immediately
  492. backgroundLoader_->WaitForResource(type, nameHash);
  493. const SharedPtr<Resource>& existing = FindResource(type, nameHash);
  494. if (existing)
  495. return existing;
  496. SharedPtr<Resource> resource;
  497. // Make sure the pointer is non-null and is a Resource subclass
  498. resource = DynamicCast<Resource>(context_->CreateObject(type));
  499. if (!resource)
  500. {
  501. LOGERROR("Could not load unknown resource type " + String(type));
  502. if (sendEventOnFailure)
  503. {
  504. using namespace UnknownResourceType;
  505. VariantMap& eventData = GetEventDataMap();
  506. eventData[P_RESOURCETYPE] = type;
  507. SendEvent(E_UNKNOWNRESOURCETYPE, eventData);
  508. }
  509. return 0;
  510. }
  511. // Attempt to load the resource
  512. SharedPtr<File> file = GetFile(name, sendEventOnFailure);
  513. if (!file)
  514. return 0; // Error is already logged
  515. LOGDEBUG("Loading resource " + name);
  516. resource->SetName(name);
  517. if (!resource->Load(*(file.Get())))
  518. {
  519. // Error should already been logged by corresponding resource descendant class
  520. if (sendEventOnFailure)
  521. {
  522. using namespace LoadFailed;
  523. VariantMap& eventData = GetEventDataMap();
  524. eventData[P_RESOURCENAME] = name;
  525. SendEvent(E_LOADFAILED, eventData);
  526. }
  527. if (!returnFailedResources_)
  528. return 0;
  529. }
  530. // Store to cache
  531. resource->ResetUseTimer();
  532. resourceGroups_[type].resources_[nameHash] = resource;
  533. UpdateResourceGroup(type);
  534. return resource;
  535. }
  536. bool ResourceCache::BackgroundLoadResource(StringHash type, const String& nameIn, bool sendEventOnFailure, Resource* caller)
  537. {
  538. // If empty name, fail immediately
  539. String name = SanitateResourceName(nameIn);
  540. if (name.Empty())
  541. return false;
  542. // First check if already exists as a loaded resource
  543. StringHash nameHash(name);
  544. if (FindResource(type, nameHash) != noResource)
  545. return false;
  546. return backgroundLoader_->QueueResource(type, name, sendEventOnFailure, caller);
  547. }
  548. SharedPtr<Resource> ResourceCache::GetTempResource(StringHash type, const String& nameIn, bool sendEventOnFailure)
  549. {
  550. String name = SanitateResourceName(nameIn);
  551. // If empty name, return null pointer immediately
  552. if (name.Empty())
  553. return SharedPtr<Resource>();
  554. SharedPtr<Resource> resource;
  555. // Make sure the pointer is non-null and is a Resource subclass
  556. resource = DynamicCast<Resource>(context_->CreateObject(type));
  557. if (!resource)
  558. {
  559. LOGERROR("Could not load unknown resource type " + String(type));
  560. if (sendEventOnFailure)
  561. {
  562. using namespace UnknownResourceType;
  563. VariantMap& eventData = GetEventDataMap();
  564. eventData[P_RESOURCETYPE] = type;
  565. SendEvent(E_UNKNOWNRESOURCETYPE, eventData);
  566. }
  567. return SharedPtr<Resource>();
  568. }
  569. // Attempt to load the resource
  570. SharedPtr<File> file = GetFile(name, sendEventOnFailure);
  571. if (!file)
  572. return SharedPtr<Resource>(); // Error is already logged
  573. LOGDEBUG("Loading temporary resource " + name);
  574. resource->SetName(file->GetName());
  575. if (!resource->Load(*(file.Get())))
  576. {
  577. // Error should already been logged by corresponding resource descendant class
  578. if (sendEventOnFailure)
  579. {
  580. using namespace LoadFailed;
  581. VariantMap& eventData = GetEventDataMap();
  582. eventData[P_RESOURCENAME] = name;
  583. SendEvent(E_LOADFAILED, eventData);
  584. }
  585. return SharedPtr<Resource>();
  586. }
  587. return resource;
  588. }
  589. unsigned ResourceCache::GetNumBackgroundLoadResources() const
  590. {
  591. return backgroundLoader_->GetNumQueuedResources();
  592. }
  593. void ResourceCache::GetResources(PODVector<Resource*>& result, StringHash type) const
  594. {
  595. result.Clear();
  596. HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
  597. if (i != resourceGroups_.End())
  598. {
  599. for (HashMap<StringHash, SharedPtr<Resource> >::ConstIterator j = i->second_.resources_.Begin();
  600. j != i->second_.resources_.End(); ++j)
  601. result.Push(j->second_);
  602. }
  603. }
  604. bool ResourceCache::Exists(const String& nameIn) const
  605. {
  606. MutexLock lock(resourceMutex_);
  607. String name = SanitateResourceName(nameIn);
  608. if (!isRouting_)
  609. {
  610. isRouting_ = true;
  611. for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
  612. resourceRouters_[i]->Route(name, RESOURCE_CHECKEXISTS);
  613. isRouting_ = false;
  614. }
  615. if (name.Empty())
  616. return false;
  617. for (unsigned i = 0; i < packages_.Size(); ++i)
  618. {
  619. if (packages_[i]->Exists(name))
  620. return true;
  621. }
  622. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  623. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  624. {
  625. if (fileSystem->FileExists(resourceDirs_[i] + name))
  626. return true;
  627. }
  628. // Fallback using absolute path
  629. return fileSystem->FileExists(name);
  630. }
  631. unsigned ResourceCache::GetMemoryBudget(StringHash type) const
  632. {
  633. HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
  634. return i != resourceGroups_.End() ? i->second_.memoryBudget_ : 0;
  635. }
  636. unsigned ResourceCache::GetMemoryUse(StringHash type) const
  637. {
  638. HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
  639. return i != resourceGroups_.End() ? i->second_.memoryUse_ : 0;
  640. }
  641. unsigned ResourceCache::GetTotalMemoryUse() const
  642. {
  643. unsigned total = 0;
  644. for (HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
  645. total += i->second_.memoryUse_;
  646. return total;
  647. }
  648. String ResourceCache::GetResourceFileName(const String& name) const
  649. {
  650. MutexLock lock(resourceMutex_);
  651. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  652. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  653. {
  654. if (fileSystem->FileExists(resourceDirs_[i] + name))
  655. return resourceDirs_[i] + name;
  656. }
  657. return String();
  658. }
  659. ResourceRouter* ResourceCache::GetResourceRouter(unsigned index) const
  660. {
  661. return index < resourceRouters_.Size() ? resourceRouters_[index] : (ResourceRouter*)0;
  662. }
  663. String ResourceCache::GetPreferredResourceDir(const String& path) const
  664. {
  665. String fixedPath = AddTrailingSlash(path);
  666. bool pathHasKnownDirs = false;
  667. bool parentHasKnownDirs = false;
  668. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  669. for (unsigned i = 0; checkDirs[i] != 0; ++i)
  670. {
  671. if (fileSystem->DirExists(fixedPath + checkDirs[i]))
  672. {
  673. pathHasKnownDirs = true;
  674. break;
  675. }
  676. }
  677. if (!pathHasKnownDirs)
  678. {
  679. String parentPath = GetParentPath(fixedPath);
  680. for (unsigned i = 0; checkDirs[i] != 0; ++i)
  681. {
  682. if (fileSystem->DirExists(parentPath + checkDirs[i]))
  683. {
  684. parentHasKnownDirs = true;
  685. break;
  686. }
  687. }
  688. // If path does not have known subdirectories, but the parent path has, use the parent instead
  689. if (parentHasKnownDirs)
  690. fixedPath = parentPath;
  691. }
  692. return fixedPath;
  693. }
  694. String ResourceCache::SanitateResourceName(const String& nameIn) const
  695. {
  696. // Sanitate unsupported constructs from the resource name
  697. String name = GetInternalPath(nameIn);
  698. name.Replace("../", "");
  699. name.Replace("./", "");
  700. // If the path refers to one of the resource directories, normalize the resource name
  701. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  702. if (resourceDirs_.Size())
  703. {
  704. String namePath = GetPath(name);
  705. String exePath = fileSystem->GetProgramDir();
  706. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  707. {
  708. String relativeResourcePath = resourceDirs_[i];
  709. if (relativeResourcePath.StartsWith(exePath))
  710. relativeResourcePath = relativeResourcePath.Substring(exePath.Length());
  711. if (namePath.StartsWith(resourceDirs_[i], false))
  712. namePath = namePath.Substring(resourceDirs_[i].Length());
  713. else if (namePath.StartsWith(relativeResourcePath, false))
  714. namePath = namePath.Substring(relativeResourcePath.Length());
  715. }
  716. name = namePath + GetFileNameAndExtension(name);
  717. }
  718. return name.Trimmed();
  719. }
  720. String ResourceCache::SanitateResourceDirName(const String& nameIn) const
  721. {
  722. String fixedPath = AddTrailingSlash(nameIn);
  723. if (!IsAbsolutePath(fixedPath))
  724. fixedPath = GetSubsystem<FileSystem>()->GetCurrentDir() + fixedPath;
  725. // Sanitate away /./ construct
  726. fixedPath.Replace("/./", "/");
  727. return fixedPath.Trimmed();
  728. }
  729. void ResourceCache::StoreResourceDependency(Resource* resource, const String& dependency)
  730. {
  731. // If resource reloading is not on, do not create the dependency data structure (saves memory)
  732. if (!resource || !autoReloadResources_)
  733. return;
  734. MutexLock lock(resourceMutex_);
  735. StringHash nameHash(resource->GetName());
  736. HashSet<StringHash>& dependents = dependentResources_[dependency];
  737. dependents.Insert(nameHash);
  738. }
  739. void ResourceCache::ResetDependencies(Resource* resource)
  740. {
  741. if (!resource || !autoReloadResources_)
  742. return;
  743. MutexLock lock(resourceMutex_);
  744. StringHash nameHash(resource->GetName());
  745. for (HashMap<StringHash, HashSet<StringHash> >::Iterator i = dependentResources_.Begin(); i != dependentResources_.End();)
  746. {
  747. HashSet<StringHash>& dependents = i->second_;
  748. dependents.Erase(nameHash);
  749. if (dependents.Empty())
  750. i = dependentResources_.Erase(i);
  751. else
  752. ++i;
  753. }
  754. }
  755. const SharedPtr<Resource>& ResourceCache::FindResource(StringHash type, StringHash nameHash)
  756. {
  757. MutexLock lock(resourceMutex_);
  758. HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
  759. if (i == resourceGroups_.End())
  760. return noResource;
  761. HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Find(nameHash);
  762. if (j == i->second_.resources_.End())
  763. return noResource;
  764. return j->second_;
  765. }
  766. const SharedPtr<Resource>& ResourceCache::FindResource(StringHash nameHash)
  767. {
  768. MutexLock lock(resourceMutex_);
  769. for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
  770. {
  771. HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Find(nameHash);
  772. if (j != i->second_.resources_.End())
  773. return j->second_;
  774. }
  775. return noResource;
  776. }
  777. void ResourceCache::ReleasePackageResources(PackageFile* package, bool force)
  778. {
  779. HashSet<StringHash> affectedGroups;
  780. const HashMap<String, PackageEntry>& entries = package->GetEntries();
  781. for (HashMap<String, PackageEntry>::ConstIterator i = entries.Begin(); i != entries.End(); ++i)
  782. {
  783. StringHash nameHash(i->first_);
  784. // We do not know the actual resource type, so search all type containers
  785. for (HashMap<StringHash, ResourceGroup>::Iterator j = resourceGroups_.Begin(); j != resourceGroups_.End(); ++j)
  786. {
  787. HashMap<StringHash, SharedPtr<Resource> >::Iterator k = j->second_.resources_.Find(nameHash);
  788. if (k != j->second_.resources_.End())
  789. {
  790. // If other references exist, do not release, unless forced
  791. if ((k->second_.Refs() == 1 && k->second_.WeakRefs() == 0) || force)
  792. {
  793. j->second_.resources_.Erase(k);
  794. affectedGroups.Insert(j->first_);
  795. }
  796. break;
  797. }
  798. }
  799. }
  800. for (HashSet<StringHash>::Iterator i = affectedGroups.Begin(); i != affectedGroups.End(); ++i)
  801. UpdateResourceGroup(*i);
  802. }
  803. void ResourceCache::UpdateResourceGroup(StringHash type)
  804. {
  805. HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
  806. if (i == resourceGroups_.End())
  807. return;
  808. for (;;)
  809. {
  810. unsigned totalSize = 0;
  811. unsigned oldestTimer = 0;
  812. HashMap<StringHash, SharedPtr<Resource> >::Iterator oldestResource = i->second_.resources_.End();
  813. for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
  814. j != i->second_.resources_.End(); ++j)
  815. {
  816. totalSize += j->second_->GetMemoryUse();
  817. unsigned useTimer = j->second_->GetUseTimer();
  818. if (useTimer > oldestTimer)
  819. {
  820. oldestTimer = useTimer;
  821. oldestResource = j;
  822. }
  823. }
  824. i->second_.memoryUse_ = totalSize;
  825. // If memory budget defined and is exceeded, remove the oldest resource and loop again
  826. // (resources in use always return a zero timer and can not be removed)
  827. if (i->second_.memoryBudget_ && i->second_.memoryUse_ > i->second_.memoryBudget_ &&
  828. oldestResource != i->second_.resources_.End())
  829. {
  830. LOGDEBUG("Resource group " + oldestResource->second_->GetTypeName() + " over memory budget, releasing resource " +
  831. oldestResource->second_->GetName());
  832. i->second_.resources_.Erase(oldestResource);
  833. }
  834. else
  835. break;
  836. }
  837. }
  838. void ResourceCache::HandleBeginFrame(StringHash eventType, VariantMap& eventData)
  839. {
  840. for (unsigned i = 0; i < fileWatchers_.Size(); ++i)
  841. {
  842. String fileName;
  843. while (fileWatchers_[i]->GetNextChange(fileName))
  844. {
  845. ReloadResourceWithDependencies(fileName);
  846. // Finally send a general file changed event even if the file was not a tracked resource
  847. using namespace FileChanged;
  848. VariantMap& eventData = GetEventDataMap();
  849. eventData[P_FILENAME] = fileWatchers_[i]->GetPath() + fileName;
  850. eventData[P_RESOURCENAME] = fileName;
  851. SendEvent(E_FILECHANGED, eventData);
  852. }
  853. }
  854. // Check for background loaded resources that can be finished
  855. {
  856. PROFILE(FinishBackgroundResources);
  857. backgroundLoader_->FinishResources(finishBackgroundResourcesMs_);
  858. }
  859. }
  860. File* ResourceCache::SearchResourceDirs(const String& nameIn)
  861. {
  862. FileSystem* fileSystem = GetSubsystem<FileSystem>();
  863. for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
  864. {
  865. if (fileSystem->FileExists(resourceDirs_[i] + nameIn))
  866. {
  867. // Construct the file first with full path, then rename it to not contain the resource path,
  868. // so that the file's name can be used in further GetFile() calls (for example over the network)
  869. File* file(new File(context_, resourceDirs_[i] + nameIn));
  870. file->SetName(nameIn);
  871. return file;
  872. }
  873. }
  874. // Fallback using absolute path
  875. if (fileSystem->FileExists(nameIn))
  876. return new File(context_, nameIn);
  877. return 0;
  878. }
  879. File* ResourceCache::SearchPackages(const String& nameIn)
  880. {
  881. for (unsigned i = 0; i < packages_.Size(); ++i)
  882. {
  883. if (packages_[i]->Exists(nameIn))
  884. return new File(context_, packages_[i], nameIn);
  885. }
  886. return 0;
  887. }
  888. void RegisterResourceLibrary(Context* context)
  889. {
  890. Image::RegisterObject(context);
  891. JSONFile::RegisterObject(context);
  892. PListFile::RegisterObject(context);
  893. XMLFile::RegisterObject(context);
  894. }
  895. }