tahoma2d/toonz/sources/common/tvrender/tpaletteutil.cpp

#include "tpaletteutil.h"
#include "tcolorstyles.h"

void mergePalette(const TPaletteP &targetPalette, std::map<int, int> &indexTable,
				  const TPaletteP &sourcePalette, const std::set<int> &sourceIndices)
{
	assert(targetPalette);
	assert(targetPalette->getPageCount() > 0);

	indexTable[0] = 0;
	std::set<int>::const_iterator styleIt = sourceIndices.begin();
	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) {
			std::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;
}