#ifndef _BUILD_IMAGEFILESCACHE_H_
#define _BUILD_IMAGEFILESCACHE_H_

#include "Image.h"
#include "TemporaryCS.h"
#include <map>
#include "../fontengine/ApplicationFonts.h"
#include "../raster/Metafile/MetaFile.h"
#include "../common/File.h"

class CCacheImage
{
private:
	Aggplus::CImage m_oImage;
	LONG m_lRef;

public:
    CCacheImage(CApplicationFonts* pFonts) : m_oImage()
	{
		m_lRef = 1;		
	}

    CCacheImage(CApplicationFonts* pFonts, const std::wstring& strFile)
	{
        if (NULL == pFonts)
        {
            m_oImage.Create(strFile);
        }
        else
        {
            MetaFile::CMetaFile oMetafile(pFonts);
            bool bIsMetafile = oMetafile.LoadFromFile(strFile.c_str());
            if (!bIsMetafile)
            {
                m_oImage.Create(strFile);
            }
            else
            {
                std::wstring sTempFile = NSFile::CFileBinary::CreateTempFileWithUniqueName(NSFile::CFileBinary::GetTempPath(), L"AscMetafile_");
                oMetafile.ConvertToRaster(sTempFile.c_str(), 4, 1000, -1);
                m_oImage.Create(sTempFile);

                NSFile::CFileBinary::Remove(sTempFile);
            }
        }

		m_lRef = 1;
	}

	LONG AddRef()
	{
		++m_lRef;
		return m_lRef;
	}
	LONG Release()
	{
		--m_lRef;

		if (0 == m_lRef)
		{
			delete this;
			return 0;
		}

		return m_lRef;
	}
	
	Aggplus::CImage* GetImage()
	{
		return &m_oImage;
	}
};

class CImageFilesCache
{
private:
	std::map<std::wstring, CCacheImage*> m_mapImages;
	LONG m_lMaxCount;

	LONG m_lRef;

    CApplicationFonts* m_pApplicationFonts;

	NSCriticalSection::CRITICAL_SECTION m_oCS;

public:
    CImageFilesCache(CApplicationFonts* pFonts = NULL)
	{
        m_pApplicationFonts = pFonts;
		m_lMaxCount = 10;
		m_lRef = 1;

		m_oCS.InitializeCriticalSection();
	}

	~CImageFilesCache()
	{
		Clear();
		m_oCS.DeleteCriticalSection();
	}

	void Clear()
	{
		CTemporaryCS oCS(&m_oCS);

		for (std::map<std::wstring,CCacheImage*>::iterator it = m_mapImages.begin(); it != m_mapImages.end(); ++it)
		{
			it->second->Release();
		}

		m_mapImages.clear();
	}

	CCacheImage* Lock(const std::wstring& strFile)
	{
		CTemporaryCS oCS(&m_oCS);

		std::map<std::wstring,CCacheImage*>::iterator it = m_mapImages.find(strFile);
		if (it != m_mapImages.end())
		{
			CCacheImage* pImage = it->second;
			pImage->AddRef();
			return pImage;
		}

		int nCount = (int)m_mapImages.size();
		if (nCount >= m_lMaxCount)
		{
			int nNeedDelete = nCount - m_lMaxCount;

            while (nNeedDelete > 0)
            {
                std::map<std::wstring,CCacheImage*>::iterator it2 = m_mapImages.begin();
                if (it2 != m_mapImages.end())
                {
                    it2->second->Release();
                    m_mapImages.erase(it2);
                }
                --nNeedDelete;
            }
        }

        CCacheImage* pImage = new CCacheImage(m_pApplicationFonts, strFile);
		m_mapImages[strFile] = pImage;

		pImage->AddRef();
		return pImage;
	}
	
	LONG AddRef()
	{
		++m_lRef;
		return m_lRef;
	}
	LONG Release()
	{
		m_oCS.Enter();
		--m_lRef;

		if (0 == m_lRef)
		{
			m_oCS.Leave();
			delete this;
			return 0;
		}

		m_oCS.Leave();
		return m_lRef;
	}
    
    void SetApplicationFonts(CApplicationFonts* pApplicationFonts)
    {
        m_pApplicationFonts = pApplicationFonts;
    }
};

#endif // _BUILD_IMAGEFILESCACHE_H_