BansheeEngine/Source/BansheeUtility/Utility/BsDynLibManager.cpp

//********************************** Banshee Engine (www.banshee3d.com) **************************************************//
//**************** Copyright (c) 2016 Marko Pintera ([email protected]). All rights reserved. **********************//
#include "Utility/BsDynLibManager.h"
#include "Utility/BsDynLib.h"

namespace bs
{
	DynLibManager::DynLibManager()
	{
	}

    DynLib* DynLibManager::load(const String& name)
    {
		// Add the extension (.dll, .so, ...) if necessary.
		String filename = name;
		const UINT32 length = (UINT32)filename.length();
		const String extension = String(".") + DynLib::EXTENSION;
		const UINT32 extLength = (UINT32)extension.length();
		if (length <= extLength || filename.substr(length - extLength) != extension)
			filename += extension;

		if(DynLib::PREFIX != nullptr)
			filename = DynLib::PREFIX + filename;

		auto iterFind = mLoadedLibraries.find(filename);
		if (iterFind != mLoadedLibraries.end())
		{
			return iterFind->second;
		}
		else
		{
	        DynLib* newLib = new (bs_alloc<DynLib>()) DynLib(filename);
        	mLoadedLibraries[filename] = newLib;

	        return newLib;
		}
    }

	void DynLibManager::unload(DynLib* lib)
	{
		auto iterFind = mLoadedLibraries.find(lib->getName());
		if (iterFind != mLoadedLibraries.end())
		{
			mLoadedLibraries.erase(iterFind);
		}

		lib->unload();
		bs_delete(lib);
	}

	DynLibManager::~DynLibManager()
    {
        // Unload & delete resources in turn
        for(auto& entry : mLoadedLibraries)
        {
            entry.second->unload();
			bs_delete(entry.second);
        }

        // Empty the list
        mLoadedLibraries.clear();
    }

	DynLibManager& gDynLibManager()
	{
		return DynLibManager::instance();
	}
}