| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063 |
- //
- // Copyright (c) 2008-2015 the Urho3D project.
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to deal
- // in the Software without restriction, including without limitation the rights
- // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- // copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
- // THE SOFTWARE.
- //
- #include "../Precompiled.h"
- #include "../Core/Context.h"
- #include "../Core/CoreEvents.h"
- #include "../Core/Profiler.h"
- #include "../Core/WorkQueue.h"
- #include "../IO/FileSystem.h"
- #include "../IO/FileWatcher.h"
- #include "../IO/Log.h"
- #include "../IO/PackageFile.h"
- #include "../Resource/BackgroundLoader.h"
- #include "../Resource/Image.h"
- #include "../Resource/JSONFile.h"
- #include "../Resource/PListFile.h"
- #include "../Resource/ResourceCache.h"
- #include "../Resource/ResourceEvents.h"
- #include "../Resource/XMLFile.h"
- #include "../DebugNew.h"
- namespace Atomic
- {
- static const char* checkDirs[] =
- {
- "Fonts",
- "Materials",
- "Models",
- "Music",
- "Objects",
- "Particle",
- "PostProcess",
- "RenderPaths",
- "Scenes",
- "Scripts",
- "Sounds",
- "Shaders",
- "Techniques",
- "Textures",
- "UI",
- 0
- };
- static const SharedPtr<Resource> noResource;
- const char* PAK_EXTENSION = ".pak";
- const char* PAK_FILTER = "*.pak";
- ResourceCache::ResourceCache(Context* context) :
- Object(context),
- autoReloadResources_(false),
- returnFailedResources_(false),
- searchPackagesFirst_(true),
- isRouting_(false),
- finishBackgroundResourcesMs_(5)
- {
- // Register Resource library object factories
- RegisterResourceLibrary(context_);
- // Create resource background loader. Its thread will start on the first background request
- backgroundLoader_ = new BackgroundLoader(this);
- // Subscribe BeginFrame for handling directory watchers and background loaded resource finalization
- SubscribeToEvent(E_BEGINFRAME, HANDLER(ResourceCache, HandleBeginFrame));
- }
- ResourceCache::~ResourceCache()
- {
- // Shut down the background loader first
- backgroundLoader_.Reset();
- }
- bool ResourceCache::AddResourceDir(const String& pathName, unsigned priority)
- {
- MutexLock lock(resourceMutex_);
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- if (!fileSystem || !fileSystem->DirExists(pathName))
- {
- LOGERROR("Could not open directory " + pathName);
- return false;
- }
- // Convert path to absolute
- String fixedPath = SanitateResourceDirName(pathName);
- // Check that the same path does not already exist
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- if (!resourceDirs_[i].Compare(fixedPath, false))
- return true;
- }
- if (priority < resourceDirs_.Size())
- resourceDirs_.Insert(priority, fixedPath);
- else
- resourceDirs_.Push(fixedPath);
- // If resource auto-reloading active, create a file watcher for the directory
- if (autoReloadResources_)
- {
- SharedPtr<FileWatcher> watcher(new FileWatcher(context_));
- watcher->StartWatching(fixedPath, true);
- fileWatchers_.Push(watcher);
- }
- LOGINFO("Added resource path " + fixedPath);
- return true;
- }
- bool ResourceCache::AddPackageFile(PackageFile* package, unsigned priority)
- {
- MutexLock lock(resourceMutex_);
- // Do not add packages that failed to load
- if (!package || !package->GetNumFiles())
- return false;
- if (priority < packages_.Size())
- packages_.Insert(priority, SharedPtr<PackageFile>(package));
- else
- packages_.Push(SharedPtr<PackageFile>(package));
- LOGINFO("Added resource package " + package->GetName());
- return true;
- }
- bool ResourceCache::AddPackageFile(const String& fileName, unsigned priority)
- {
- SharedPtr<PackageFile> package(new PackageFile(context_));
- return package->Open(fileName) && AddPackageFile(package);
- }
- bool ResourceCache::AddManualResource(Resource* resource)
- {
- if (!resource)
- {
- LOGERROR("Null manual resource");
- return false;
- }
- const String& name = resource->GetName();
- if (name.Empty())
- {
- LOGERROR("Manual resource with empty name, can not add");
- return false;
- }
- resource->ResetUseTimer();
- resourceGroups_[resource->GetType()].resources_[resource->GetNameHash()] = resource;
- UpdateResourceGroup(resource->GetType());
- return true;
- }
- void ResourceCache::RemoveResourceDir(const String& pathName)
- {
- MutexLock lock(resourceMutex_);
- String fixedPath = SanitateResourceDirName(pathName);
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- if (!resourceDirs_[i].Compare(fixedPath, false))
- {
- resourceDirs_.Erase(i);
- // Remove the filewatcher with the matching path
- for (unsigned j = 0; j < fileWatchers_.Size(); ++j)
- {
- if (!fileWatchers_[j]->GetPath().Compare(fixedPath, false))
- {
- fileWatchers_.Erase(j);
- break;
- }
- }
- LOGINFO("Removed resource path " + fixedPath);
- return;
- }
- }
- }
- void ResourceCache::RemovePackageFile(PackageFile* package, bool releaseResources, bool forceRelease)
- {
- MutexLock lock(resourceMutex_);
- for (Vector<SharedPtr<PackageFile> >::Iterator i = packages_.Begin(); i != packages_.End(); ++i)
- {
- if (*i == package)
- {
- if (releaseResources)
- ReleasePackageResources(*i, forceRelease);
- LOGINFO("Removed resource package " + (*i)->GetName());
- packages_.Erase(i);
- return;
- }
- }
- }
- void ResourceCache::RemovePackageFile(const String& fileName, bool releaseResources, bool forceRelease)
- {
- MutexLock lock(resourceMutex_);
- // Compare the name and extension only, not the path
- String fileNameNoPath = GetFileNameAndExtension(fileName);
- for (Vector<SharedPtr<PackageFile> >::Iterator i = packages_.Begin(); i != packages_.End(); ++i)
- {
- if (!GetFileNameAndExtension((*i)->GetName()).Compare(fileNameNoPath, false))
- {
- if (releaseResources)
- ReleasePackageResources(*i, forceRelease);
- LOGINFO("Removed resource package " + (*i)->GetName());
- packages_.Erase(i);
- return;
- }
- }
- }
- void ResourceCache::ReleaseResource(StringHash type, const String& name, bool force)
- {
- StringHash nameHash(name);
- const SharedPtr<Resource>& existingRes = FindResource(type, nameHash);
- if (!existingRes)
- return;
- // If other references exist, do not release, unless forced
- if ((existingRes.Refs() == 1 && existingRes.WeakRefs() == 0) || force)
- {
- resourceGroups_[type].resources_.Erase(nameHash);
- UpdateResourceGroup(type);
- }
- }
- void ResourceCache::ReleaseResources(StringHash type, bool force)
- {
- bool released = false;
- HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
- if (i != resourceGroups_.End())
- {
- for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End();)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
- // If other references exist, do not release, unless forced
- if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
- {
- i->second_.resources_.Erase(current);
- released = true;
- }
- }
- }
- if (released)
- UpdateResourceGroup(type);
- }
- void ResourceCache::ReleaseResources(StringHash type, const String& partialName, bool force)
- {
- bool released = false;
- HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
- if (i != resourceGroups_.End())
- {
- for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End();)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
- if (current->second_->GetName().Contains(partialName))
- {
- // If other references exist, do not release, unless forced
- if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
- {
- i->second_.resources_.Erase(current);
- released = true;
- }
- }
- }
- }
- if (released)
- UpdateResourceGroup(type);
- }
- void ResourceCache::ReleaseResources(const String& partialName, bool force)
- {
- // Some resources refer to others, like materials to textures. Release twice to ensure these get released.
- // This is not necessary if forcing release
- unsigned repeat = force ? 1 : 2;
- while (repeat--)
- {
- for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
- {
- bool released = false;
- for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End();)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
- if (current->second_->GetName().Contains(partialName))
- {
- // If other references exist, do not release, unless forced
- if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
- {
- i->second_.resources_.Erase(current);
- released = true;
- }
- }
- }
- if (released)
- UpdateResourceGroup(i->first_);
- }
- }
- }
- void ResourceCache::ReleaseAllResources(bool force)
- {
- unsigned repeat = force ? 1 : 2;
- while (repeat--)
- {
- for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin();
- i != resourceGroups_.End(); ++i)
- {
- bool released = false;
- for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End();)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator current = j++;
- // If other references exist, do not release, unless forced
- if ((current->second_.Refs() == 1 && current->second_.WeakRefs() == 0) || force)
- {
- i->second_.resources_.Erase(current);
- released = true;
- }
- }
- if (released)
- UpdateResourceGroup(i->first_);
- }
- }
- }
- bool ResourceCache::ReloadResource(Resource* resource)
- {
- if (!resource)
- return false;
- resource->SendEvent(E_RELOADSTARTED);
- bool success = false;
- SharedPtr<File> file = GetFile(resource->GetName());
- if (file)
- success = resource->Load(*(file.Get()));
- if (success)
- {
- resource->ResetUseTimer();
- UpdateResourceGroup(resource->GetType());
- resource->SendEvent(E_RELOADFINISHED);
- return true;
- }
- // If reloading failed, do not remove the resource from cache, to allow for a new live edit to
- // attempt loading again
- resource->SendEvent(E_RELOADFAILED);
- return false;
- }
- void ResourceCache::ReloadResourceWithDependencies(const String& fileName)
- {
- StringHash fileNameHash(fileName);
- // If the filename is a resource we keep track of, reload it
- const SharedPtr<Resource>& resource = FindResource(fileNameHash);
- if (resource)
- {
- LOGDEBUG("Reloading changed resource " + fileName);
- ReloadResource(resource);
- }
- // Always perform dependency resource check for resource loaded from XML file as it could be used in inheritance
- if (!resource || GetExtension(resource->GetName()) == ".xml")
- {
- // Check if this is a dependency resource, reload dependents
- HashMap<StringHash, HashSet<StringHash> >::ConstIterator j = dependentResources_.Find(fileNameHash);
- if (j != dependentResources_.End())
- {
- // Reloading a resource may modify the dependency tracking structure. Therefore collect the
- // resources we need to reload first
- Vector<SharedPtr<Resource> > dependents;
- dependents.Reserve(j->second_.Size());
- for (HashSet<StringHash>::ConstIterator k = j->second_.Begin(); k != j->second_.End(); ++k)
- {
- const SharedPtr<Resource>& dependent = FindResource(*k);
- if (dependent)
- dependents.Push(dependent);
- }
- for (unsigned k = 0; k < dependents.Size(); ++k)
- {
- LOGDEBUG("Reloading resource " + dependents[k]->GetName() + " depending on " + fileName);
- ReloadResource(dependents[k]);
- }
- }
- }
- }
- void ResourceCache::SetMemoryBudget(StringHash type, unsigned budget)
- {
- resourceGroups_[type].memoryBudget_ = budget;
- }
- void ResourceCache::SetAutoReloadResources(bool enable)
- {
- if (enable != autoReloadResources_)
- {
- if (enable)
- {
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- SharedPtr<FileWatcher> watcher(new FileWatcher(context_));
- watcher->StartWatching(resourceDirs_[i], true);
- fileWatchers_.Push(watcher);
- }
- }
- else
- fileWatchers_.Clear();
- autoReloadResources_ = enable;
- }
- }
- void ResourceCache::AddResourceRouter(ResourceRouter* router, bool addAsFirst)
- {
- // Check for duplicate
- for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
- {
- if (resourceRouters_[i] == router)
- return;
- }
- if (addAsFirst)
- resourceRouters_.Insert(0, SharedPtr<ResourceRouter>(router));
- else
- resourceRouters_.Push(SharedPtr<ResourceRouter>(router));
- }
- void ResourceCache::RemoveResourceRouter(ResourceRouter* router)
- {
- for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
- {
- if (resourceRouters_[i] == router)
- {
- resourceRouters_.Erase(i);
- return;
- }
- }
- }
- SharedPtr<File> ResourceCache::GetFile(const String& nameIn, bool sendEventOnFailure)
- {
- MutexLock lock(resourceMutex_);
- String name = SanitateResourceName(nameIn);
- if (!isRouting_)
- {
- isRouting_ = true;
- for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
- resourceRouters_[i]->Route(name, RESOURCE_GETFILE);
- isRouting_ = false;
- }
- if (name.Length())
- {
- File* file = 0;
- if (searchPackagesFirst_)
- {
- file = SearchPackages(name);
- if (!file)
- file = SearchResourceDirs(name);
- }
- else
- {
- file = SearchResourceDirs(name);
- if (!file)
- file = SearchPackages(name);
- }
- if (file)
- return SharedPtr<File>(file);
- }
- if (sendEventOnFailure)
- {
- if (resourceRouters_.Size() && name.Empty() && !nameIn.Empty())
- LOGERROR("Resource request " + nameIn + " was blocked");
- else
- LOGERROR("Could not find resource " + name);
- if (Thread::IsMainThread())
- {
- using namespace ResourceNotFound;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_RESOURCENAME] = name.Length() ? name : nameIn;
- SendEvent(E_RESOURCENOTFOUND, eventData);
- }
- }
- return SharedPtr<File>();
- }
- Resource* ResourceCache::GetExistingResource(StringHash type, const String& nameIn)
- {
- String name = SanitateResourceName(nameIn);
- if (!Thread::IsMainThread())
- {
- LOGERROR("Attempted to get resource " + name + " from outside the main thread");
- return 0;
- }
- // If empty name, return null pointer immediately
- if (name.Empty())
- return 0;
- StringHash nameHash(name);
- const SharedPtr<Resource>& existing = FindResource(type, nameHash);
- return existing;
- }
- Resource* ResourceCache::GetResource(StringHash type, const String& nameIn, bool sendEventOnFailure)
- {
- String name = SanitateResourceName(nameIn);
- if (!Thread::IsMainThread())
- {
- LOGERROR("Attempted to get resource " + name + " from outside the main thread");
- return 0;
- }
- // If empty name, return null pointer immediately
- if (name.Empty())
- return 0;
- StringHash nameHash(name);
- // Check if the resource is being background loaded but is now needed immediately
- backgroundLoader_->WaitForResource(type, nameHash);
- const SharedPtr<Resource>& existing = FindResource(type, nameHash);
- if (existing)
- return existing;
- SharedPtr<Resource> resource;
- // Make sure the pointer is non-null and is a Resource subclass
- resource = DynamicCast<Resource>(context_->CreateObject(type));
- if (!resource)
- {
- LOGERROR("Could not load unknown resource type " + String(type));
- if (sendEventOnFailure)
- {
- using namespace UnknownResourceType;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_RESOURCETYPE] = type;
- SendEvent(E_UNKNOWNRESOURCETYPE, eventData);
- }
- return 0;
- }
- // Attempt to load the resource
- SharedPtr<File> file = GetFile(name, sendEventOnFailure);
- if (!file)
- return 0; // Error is already logged
- LOGDEBUG("Loading resource " + name);
- resource->SetName(name);
- if (!resource->Load(*(file.Get())))
- {
- // Error should already been logged by corresponding resource descendant class
- if (sendEventOnFailure)
- {
- using namespace LoadFailed;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_RESOURCENAME] = name;
- SendEvent(E_LOADFAILED, eventData);
- }
- if (!returnFailedResources_)
- return 0;
- }
- // Store to cache
- resource->ResetUseTimer();
- resourceGroups_[type].resources_[nameHash] = resource;
- UpdateResourceGroup(type);
- return resource;
- }
- bool ResourceCache::BackgroundLoadResource(StringHash type, const String& nameIn, bool sendEventOnFailure, Resource* caller)
- {
- // If empty name, fail immediately
- String name = SanitateResourceName(nameIn);
- if (name.Empty())
- return false;
- // First check if already exists as a loaded resource
- StringHash nameHash(name);
- if (FindResource(type, nameHash) != noResource)
- return false;
- return backgroundLoader_->QueueResource(type, name, sendEventOnFailure, caller);
- }
- SharedPtr<Resource> ResourceCache::GetTempResource(StringHash type, const String& nameIn, bool sendEventOnFailure)
- {
- String name = SanitateResourceName(nameIn);
- // If empty name, return null pointer immediately
- if (name.Empty())
- return SharedPtr<Resource>();
- SharedPtr<Resource> resource;
- // Make sure the pointer is non-null and is a Resource subclass
- resource = DynamicCast<Resource>(context_->CreateObject(type));
- if (!resource)
- {
- LOGERROR("Could not load unknown resource type " + String(type));
- if (sendEventOnFailure)
- {
- using namespace UnknownResourceType;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_RESOURCETYPE] = type;
- SendEvent(E_UNKNOWNRESOURCETYPE, eventData);
- }
- return SharedPtr<Resource>();
- }
- // Attempt to load the resource
- SharedPtr<File> file = GetFile(name, sendEventOnFailure);
- if (!file)
- return SharedPtr<Resource>(); // Error is already logged
- LOGDEBUG("Loading temporary resource " + name);
- resource->SetName(file->GetName());
- if (!resource->Load(*(file.Get())))
- {
- // Error should already been logged by corresponding resource descendant class
- if (sendEventOnFailure)
- {
- using namespace LoadFailed;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_RESOURCENAME] = name;
- SendEvent(E_LOADFAILED, eventData);
- }
- return SharedPtr<Resource>();
- }
- return resource;
- }
- unsigned ResourceCache::GetNumBackgroundLoadResources() const
- {
- return backgroundLoader_->GetNumQueuedResources();
- }
- void ResourceCache::GetResources(PODVector<Resource*>& result, StringHash type) const
- {
- result.Clear();
- HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
- if (i != resourceGroups_.End())
- {
- for (HashMap<StringHash, SharedPtr<Resource> >::ConstIterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End(); ++j)
- result.Push(j->second_);
- }
- }
- bool ResourceCache::Exists(const String& nameIn) const
- {
- MutexLock lock(resourceMutex_);
- String name = SanitateResourceName(nameIn);
- if (!isRouting_)
- {
- isRouting_ = true;
- for (unsigned i = 0; i < resourceRouters_.Size(); ++i)
- resourceRouters_[i]->Route(name, RESOURCE_CHECKEXISTS);
- isRouting_ = false;
- }
- if (name.Empty())
- return false;
- for (unsigned i = 0; i < packages_.Size(); ++i)
- {
- if (packages_[i]->Exists(name))
- return true;
- }
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- if (fileSystem->FileExists(resourceDirs_[i] + name))
- return true;
- }
- // Fallback using absolute path
- return fileSystem->FileExists(name);
- }
- unsigned ResourceCache::GetMemoryBudget(StringHash type) const
- {
- HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
- return i != resourceGroups_.End() ? i->second_.memoryBudget_ : 0;
- }
- unsigned ResourceCache::GetMemoryUse(StringHash type) const
- {
- HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Find(type);
- return i != resourceGroups_.End() ? i->second_.memoryUse_ : 0;
- }
- unsigned ResourceCache::GetTotalMemoryUse() const
- {
- unsigned total = 0;
- for (HashMap<StringHash, ResourceGroup>::ConstIterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
- total += i->second_.memoryUse_;
- return total;
- }
- String ResourceCache::GetResourceFileName(const String& name) const
- {
- MutexLock lock(resourceMutex_);
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- if (fileSystem->FileExists(resourceDirs_[i] + name))
- return resourceDirs_[i] + name;
- }
- return String();
- }
- ResourceRouter* ResourceCache::GetResourceRouter(unsigned index) const
- {
- return index < resourceRouters_.Size() ? resourceRouters_[index] : (ResourceRouter*)0;
- }
- String ResourceCache::GetPreferredResourceDir(const String& path) const
- {
- String fixedPath = AddTrailingSlash(path);
- bool pathHasKnownDirs = false;
- bool parentHasKnownDirs = false;
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- for (unsigned i = 0; checkDirs[i] != 0; ++i)
- {
- if (fileSystem->DirExists(fixedPath + checkDirs[i]))
- {
- pathHasKnownDirs = true;
- break;
- }
- }
- if (!pathHasKnownDirs)
- {
- String parentPath = GetParentPath(fixedPath);
- for (unsigned i = 0; checkDirs[i] != 0; ++i)
- {
- if (fileSystem->DirExists(parentPath + checkDirs[i]))
- {
- parentHasKnownDirs = true;
- break;
- }
- }
- // If path does not have known subdirectories, but the parent path has, use the parent instead
- if (parentHasKnownDirs)
- fixedPath = parentPath;
- }
- return fixedPath;
- }
- String ResourceCache::SanitateResourceName(const String& nameIn) const
- {
- // Sanitate unsupported constructs from the resource name
- String name = GetInternalPath(nameIn);
- name.Replace("../", "");
- name.Replace("./", "");
- // If the path refers to one of the resource directories, normalize the resource name
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- if (resourceDirs_.Size())
- {
- String namePath = GetPath(name);
- String exePath = fileSystem->GetProgramDir();
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- String relativeResourcePath = resourceDirs_[i];
- if (relativeResourcePath.StartsWith(exePath))
- relativeResourcePath = relativeResourcePath.Substring(exePath.Length());
- if (namePath.StartsWith(resourceDirs_[i], false))
- namePath = namePath.Substring(resourceDirs_[i].Length());
- else if (namePath.StartsWith(relativeResourcePath, false))
- namePath = namePath.Substring(relativeResourcePath.Length());
- }
- name = namePath + GetFileNameAndExtension(name);
- }
- return name.Trimmed();
- }
- String ResourceCache::SanitateResourceDirName(const String& nameIn) const
- {
- String fixedPath = AddTrailingSlash(nameIn);
- if (!IsAbsolutePath(fixedPath))
- fixedPath = GetSubsystem<FileSystem>()->GetCurrentDir() + fixedPath;
- // Sanitate away /./ construct
- fixedPath.Replace("/./", "/");
- return fixedPath.Trimmed();
- }
- void ResourceCache::StoreResourceDependency(Resource* resource, const String& dependency)
- {
- // If resource reloading is not on, do not create the dependency data structure (saves memory)
- if (!resource || !autoReloadResources_)
- return;
- MutexLock lock(resourceMutex_);
- StringHash nameHash(resource->GetName());
- HashSet<StringHash>& dependents = dependentResources_[dependency];
- dependents.Insert(nameHash);
- }
- void ResourceCache::ResetDependencies(Resource* resource)
- {
- if (!resource || !autoReloadResources_)
- return;
- MutexLock lock(resourceMutex_);
- StringHash nameHash(resource->GetName());
- for (HashMap<StringHash, HashSet<StringHash> >::Iterator i = dependentResources_.Begin(); i != dependentResources_.End();)
- {
- HashSet<StringHash>& dependents = i->second_;
- dependents.Erase(nameHash);
- if (dependents.Empty())
- i = dependentResources_.Erase(i);
- else
- ++i;
- }
- }
- const SharedPtr<Resource>& ResourceCache::FindResource(StringHash type, StringHash nameHash)
- {
- MutexLock lock(resourceMutex_);
- HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
- if (i == resourceGroups_.End())
- return noResource;
- HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Find(nameHash);
- if (j == i->second_.resources_.End())
- return noResource;
- return j->second_;
- }
- const SharedPtr<Resource>& ResourceCache::FindResource(StringHash nameHash)
- {
- MutexLock lock(resourceMutex_);
- for (HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Begin(); i != resourceGroups_.End(); ++i)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Find(nameHash);
- if (j != i->second_.resources_.End())
- return j->second_;
- }
- return noResource;
- }
- void ResourceCache::ReleasePackageResources(PackageFile* package, bool force)
- {
- HashSet<StringHash> affectedGroups;
- const HashMap<String, PackageEntry>& entries = package->GetEntries();
- for (HashMap<String, PackageEntry>::ConstIterator i = entries.Begin(); i != entries.End(); ++i)
- {
- StringHash nameHash(i->first_);
- // We do not know the actual resource type, so search all type containers
- for (HashMap<StringHash, ResourceGroup>::Iterator j = resourceGroups_.Begin(); j != resourceGroups_.End(); ++j)
- {
- HashMap<StringHash, SharedPtr<Resource> >::Iterator k = j->second_.resources_.Find(nameHash);
- if (k != j->second_.resources_.End())
- {
- // If other references exist, do not release, unless forced
- if ((k->second_.Refs() == 1 && k->second_.WeakRefs() == 0) || force)
- {
- j->second_.resources_.Erase(k);
- affectedGroups.Insert(j->first_);
- }
- break;
- }
- }
- }
- for (HashSet<StringHash>::Iterator i = affectedGroups.Begin(); i != affectedGroups.End(); ++i)
- UpdateResourceGroup(*i);
- }
- void ResourceCache::UpdateResourceGroup(StringHash type)
- {
- HashMap<StringHash, ResourceGroup>::Iterator i = resourceGroups_.Find(type);
- if (i == resourceGroups_.End())
- return;
- for (;;)
- {
- unsigned totalSize = 0;
- unsigned oldestTimer = 0;
- HashMap<StringHash, SharedPtr<Resource> >::Iterator oldestResource = i->second_.resources_.End();
- for (HashMap<StringHash, SharedPtr<Resource> >::Iterator j = i->second_.resources_.Begin();
- j != i->second_.resources_.End(); ++j)
- {
- totalSize += j->second_->GetMemoryUse();
- unsigned useTimer = j->second_->GetUseTimer();
- if (useTimer > oldestTimer)
- {
- oldestTimer = useTimer;
- oldestResource = j;
- }
- }
- i->second_.memoryUse_ = totalSize;
- // If memory budget defined and is exceeded, remove the oldest resource and loop again
- // (resources in use always return a zero timer and can not be removed)
- if (i->second_.memoryBudget_ && i->second_.memoryUse_ > i->second_.memoryBudget_ &&
- oldestResource != i->second_.resources_.End())
- {
- LOGDEBUG("Resource group " + oldestResource->second_->GetTypeName() + " over memory budget, releasing resource " +
- oldestResource->second_->GetName());
- i->second_.resources_.Erase(oldestResource);
- }
- else
- break;
- }
- }
- void ResourceCache::HandleBeginFrame(StringHash eventType, VariantMap& eventData)
- {
- for (unsigned i = 0; i < fileWatchers_.Size(); ++i)
- {
- String fileName;
- while (fileWatchers_[i]->GetNextChange(fileName))
- {
- ReloadResourceWithDependencies(fileName);
- // Finally send a general file changed event even if the file was not a tracked resource
- using namespace FileChanged;
- VariantMap& eventData = GetEventDataMap();
- eventData[P_FILENAME] = fileWatchers_[i]->GetPath() + fileName;
- eventData[P_RESOURCENAME] = fileName;
- SendEvent(E_FILECHANGED, eventData);
- }
- }
- // Check for background loaded resources that can be finished
- {
- PROFILE(FinishBackgroundResources);
- backgroundLoader_->FinishResources(finishBackgroundResourcesMs_);
- }
- }
- File* ResourceCache::SearchResourceDirs(const String& nameIn)
- {
- FileSystem* fileSystem = GetSubsystem<FileSystem>();
- for (unsigned i = 0; i < resourceDirs_.Size(); ++i)
- {
- if (fileSystem->FileExists(resourceDirs_[i] + nameIn))
- {
- // Construct the file first with full path, then rename it to not contain the resource path,
- // so that the file's name can be used in further GetFile() calls (for example over the network)
- File* file(new File(context_, resourceDirs_[i] + nameIn));
- file->SetName(nameIn);
- return file;
- }
- }
- // Fallback using absolute path
- if (fileSystem->FileExists(nameIn))
- return new File(context_, nameIn);
- return 0;
- }
- File* ResourceCache::SearchPackages(const String& nameIn)
- {
- for (unsigned i = 0; i < packages_.Size(); ++i)
- {
- if (packages_[i]->Exists(nameIn))
- return new File(context_, packages_[i], nameIn);
- }
- return 0;
- }
- void RegisterResourceLibrary(Context* context)
- {
- Image::RegisterObject(context);
- JSONFile::RegisterObject(context);
- PListFile::RegisterObject(context);
- XMLFile::RegisterObject(context);
- }
- }
|