//
// Urho3D Engine
// Copyright (c) 2008-2011 Lasse Öörni
//
// 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.
//

#pragma once

#include "File.h"
#include "Resource.h"

class PackageFile;

/// Container of resources with specific type
struct ResourceGroup
{
    /// Construct with defaults
    ResourceGroup() :
        memoryBudget_(0),
        memoryUse_(0)
    {
    }
    
    /// Memory budget
    unsigned memoryBudget_;
    /// Current memory use
    unsigned memoryUse_;
    /// Resources
    std::map<StringHash, SharedPtr<Resource> > resources_;
};

/// Resource cache subsystem. Loads resources on demand and stores them for later access
class ResourceCache : public Object
{
    OBJECT(ResourceCache);
    
public:
    /// Construct
    ResourceCache(Context* context);
    /// Destruct. Free all resources
    virtual ~ResourceCache();
    
    /// Add a resource load path
    bool AddResourcePath(const std::string& path);
    /// Add a package file for loading resources from
    void AddPackageFile(PackageFile* package, bool addAsFirst = false);
    /// Add a manually created resource. Must be uniquely named
    bool AddManualResource(Resource* resource);
    /// Remove a resource load path
    void RemoveResourcePath(const std::string& path);
    /// Remove a package file. Optionally release the resources loaded from it
    void RemovePackageFile(PackageFile* package, bool ReleaseResources = true, bool forceRelease = false);
    /// Remove a package file by name. Optionally release the resources loaded from it
    void RemovePackageFile(const std::string& fileName, bool ReleaseResources = true, bool forceRelease = false);
    /// Release a resource by name
    void ReleaseResource(ShortStringHash type, const std::string& name, bool force = false);
    /// Release a resource by name hash
    void ReleaseResource(ShortStringHash type, StringHash nameHash, bool force = false); 
    /// Release all resources of a specific type
    void ReleaseResources(ShortStringHash type, bool force = false);
    /// Release resources of a specific type and partial name
    void ReleaseResources(ShortStringHash type, const std::string& partialName, bool force = false);
    /// Release all resources
    void ReleaseAllResources(bool force = false);
    /// Reload a resource. Return false and release it if fails
    bool ReloadResource(Resource* resource);
    /// Set memory budget for a specific resource type, default 0 is unlimited
    void SetMemoryBudget(ShortStringHash type, unsigned budget);
    
    /// Open and return a file from either the resource load paths or from inside a package file. Return null if fails
    SharedPtr<File> GetFile(const std::string& name);
    /// Return a resource by type and name. Load if not loaded yet. Return null if fails
    Resource* GetResource(ShortStringHash type, const std::string& name);
    /// Return a resource by type and name hash. Load if not loaded yet. Return null if fails
    Resource* GetResource(ShortStringHash type, StringHash nameHash);
    /// Return all loaded resources of a specific type
    void GetResources(std::vector<Resource*>& result, ShortStringHash type) const;
    /// Return all loaded resources
    const std::map<ShortStringHash, ResourceGroup>& GetAllResources() const { return resourceGroups_; }
    /// Return added resource load paths
    const std::vector<std::string>& GetResourcePaths() const { return resourcePaths_; }
    /// Return added package files
    const std::vector<SharedPtr<PackageFile> >& GetPackageFiles() const { return packages_; }
    /// Template version of returning a resource by name
    template <class T> T* GetResource(const std::string& name);
    /// Template version of returning a resource by name hash
    template <class T> T* GetResource(StringHash nameHash);
    /// Template version of returning loaded resources of a specific type
    template <class T> void GetResources(std::vector<T*>& result) const;
    /// Return whether a file exists by name
    bool Exists(const std::string& name) const;
    /// Return whether a file exists by name hash
    bool Exists(StringHash nameHash) const;
    /// Return memory budget for a resource type
    unsigned GetMemoryBudget(ShortStringHash type) const;
    /// Return total memory use for a resource type
    unsigned GetMemoryUse(ShortStringHash type) const;
    /// Return total memory use for all resources
    unsigned GetTotalMemoryUse() const;
    /// Return resource name from hash, or empty if not found
    const std::string& GetResourceName(StringHash nameHash) const;
    /// Return either the path itself or its parent, based on which of them has recognized resource subdirectories
    std::string GetPreferredResourcePath(const std::string& path);
    
private:
    /// Find a resource
    const SharedPtr<Resource>& FindResource(ShortStringHash type, StringHash nameHash);
    /// Release resources loaded from a package file
    void ReleasePackageResources(PackageFile* package, bool force = false);
    /// Update a resource group. Recalculate memory use and release resources if over memory budget
    void UpdateResourceGroup(ShortStringHash type);
    /// Store a hash-to-name mapping
    void StoreNameHash(const std::string& name);
    
    /// Resources by type
    std::map<ShortStringHash, ResourceGroup> resourceGroups_;
    /// Resource load paths
    std::vector<std::string> resourcePaths_;
    /// Package files
    std::vector<SharedPtr<PackageFile> > packages_;
    /// Mapping of hashes to filenames
    std::map<StringHash, std::string> hashToName_;
};

template <class T> T* ResourceCache::GetResource(const std::string& name)
{
    ShortStringHash type = T::GetTypeStatic();
    return static_cast<T*>(GetResource(type, name));
}

template <class T> T* ResourceCache::GetResource(StringHash nameHash)
{
    ShortStringHash type = T::GetTypeStatic();
    return static_cast<T*>(GetResource(type, nameHash));
}

template <class T> void ResourceCache::GetResources(std::vector<T*>& result) const
{
    std::vector<Resource*>& resources = reinterpret_cast<std::vector<Resource*>&>(result);
    ShortStringHash type = T::GetTypeStatic();
    GetResources(resources, type);
    
    // Perform conversion of the returned pointers
    for (unsigned i = 0; i < result.size(); ++i)
    {
        Resource* resource = resources[i];
        result[i] = static_cast<T*>(resource);
    }
}

/// Register Resource library subsystems and objects
void RegisterResourceLibrary(Context* context);