#include "tpaletteutil.h"

#include "tcolorstyles.h"

void mergePalette(const TPaletteP &targetPalette, std::map<int, int=""> &indexTable,</int,>

				  const TPaletteP &sourcePalette, const std::set<int> &sourceIndices)</int>

{

	assert(targetPalette);

	assert(targetPalette->getPageCount() > 0);

	indexTable[0] = 0;

	std::set<int>::const_iterator styleIt = sourceIndices.begin();</int>

	for (styleIt; styleIt != sourceIndices.end(); ++styleIt) {

		int srcStyleId = *styleIt;

		if (srcStyleId == 0)

			continue;

		assert(0 <= srcStyleId && srcStyleId < sourcePalette->getStyleCount());

		TColorStyle *srcStyle = sourcePalette->getStyle(srcStyleId);

		assert(srcStyle);

		TPalette::Page *page = sourcePalette->getStylePage(srcStyleId);

		if (page) {

			wstring pageName = page->getName();

			int j = 0;

			for (j = 0; j < targetPalette->getPageCount(); j++) {

				if (targetPalette->getPage(j)->getName() != pageName)

					continue;

				break;

			}

			if (j < targetPalette->getPageCount())

				page = targetPalette->getPage(j);

			else

				page = targetPalette->addPage(pageName);

		} else

			page = targetPalette->getPage(0);

		int tarStyleId = 0;

		int i = 0;

		for (i = 0; i < targetPalette->getStyleCount(); i++)

			if (*srcStyle == *targetPalette->getStyle(i) && i == srcStyleId)

				break;

		if (i < targetPalette->getStyleCount())

			tarStyleId = i;

		else {

			TColorStyle *dstStyle = srcStyle->clone();

			tarStyleId = targetPalette->addStyle(dstStyle);

			page->addStyle(tarStyleId);

			targetPalette->setDirtyFlag(true);

		}

		assert(indexTable.count(srcStyleId) == 0);

		indexTable[srcStyleId] = tarStyleId;

	}

}

//------------------------------------------------------------------------

// replace palette and lacking amount of styles will be copied from the other one

// return value will be true if the style amount is changed after the operation

bool mergePalette_Overlap(

	const TPaletteP &dstPalette,

	const TPaletteP &copiedPalette,

	bool keepOriginalPalette)

{

	if (!dstPalette || !copiedPalette)

		return false;

	bool styleAdded = false;

	int dstStyleCount = dstPalette->getStyleCount();

	int copiedStyleCount = copiedPalette->getStyleCount();

	//keep original

	if (keepOriginalPalette) {

		// do nothing if the style amount of the dst is equal or larger than the copied

		if (dstStyleCount >= copiedStyleCount) {

			return false;

		}

		// if the style amount of the dst is less than the copied

		else {

			//for lacking amount of styles

			for (int i = dstStyleCount; i < copiedStyleCount; i++) {

				//get the page index of the copied style

				TPalette::Page *tmpPage = copiedPalette->getStylePage(i);

				//clone copied style

				TColorStyle *tmpStyle = copiedPalette->getStyle(i)->clone();

				//add it to the dst

				int id = dstPalette->addStyle(tmpStyle);

				//add to the page if it is not deleted in the copied

				if (tmpPage)

					dstPalette->getPage(0)->addStyle(id);

			}

			styleAdded = true;

		}

	}

	//replace

	else {

		//if the style amount of the dst is larger than the copied

		if (dstStyleCount > copiedStyleCount) {

			TPalette *tmpPalette = copiedPalette->clone();

			//for lacking amount of styles

			for (int i = copiedStyleCount; i < dstStyleCount; i++) {

				TColorStyle *tmpStyle = dstPalette->getStyle(i)->clone();

				int id = tmpPalette->addStyle(tmpStyle);

				TPalette::Page *tmpPage = dstPalette->getStylePage(i);

				if (!tmpPage)

					continue;

				std::wstring pageName = tmpPage->getName();

				//create new page with the same name if needed

				int p;

				for (p = 0; p < tmpPalette->getPageCount(); p++) {

					if (tmpPalette->getPage(p)->getName() == pageName)

						break;

				}

				if (p != tmpPalette->getPageCount())

					tmpPalette->getPage(p)->addStyle(id);

				else {

					tmpPalette->addPage(pageName)->addStyle(id);

				}

			}

			dstPalette->assign(tmpPalette);

			styleAdded = false;

		}

		// if the style amount of the dst is equal or less than the copied

		else {

			dstPalette->assign(copiedPalette.getPointer());

			styleAdded = (dstStyleCount < copiedStyleCount);

		}

	}

	dstPalette->setDirtyFlag(true);

	return styleAdded;

}