/*
 * (c) Copyright Ascensio System SIA 2010-2018
 *
 * This program is a free software product. You can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License (AGPL)
 * version 3 as published by the Free Software Foundation. In accordance with
 * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
 * that Ascensio System SIA expressly excludes the warranty of non-infringement
 * of any third-party rights.
 *
 * This program is distributed WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For
 * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia,
 * EU, LV-1021.
 *
 * The  interactive user interfaces in modified source and object code versions
 * of the Program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU AGPL version 3.
 *
 * Pursuant to Section 7(b) of the License you must retain the original Product
 * logo when distributing the program. Pursuant to Section 7(e) we decline to
 * grant you any rights under trademark law for use of our trademarks.
 *
 * All the Product's GUI elements, including illustrations and icon sets, as
 * well as technical writing content are licensed under the terms of the
 * Creative Commons Attribution-ShareAlike 4.0 International. See the License
 * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 */
#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_