#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 != 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;

}