#include "tpalette.h"

#include "toonz/cleanupcolorstyles.h"

#include "toonz/targetcolors.h"

#include "tgl.h"

#include "cleanuppalette.h"

//===================================================================

TPalette *createStandardCleanupPalette() {

  TPalette *palette    = new TPalette();

  TPalette::Page *page = palette->getPage(0);

  page->removeStyle(1);  // tolgo il black che c'e' gia' nella palette di

                         // default

  TBlackCleanupStyle *black = new TBlackCleanupStyle();

  palette->setStyle(1, black);

  page->addStyle(1);

  // page->addStyle(palette->addStyle(new TColorCleanupStyle(TPixel32::Red)));

  // page->addStyle(palette->addStyle(new TColorCleanupStyle(TPixel32::Blue)));

  black->setName(L"color_1");

  palette->addRef();

  palette->setIsCleanupPalette(true);

  return palette;

}

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

TPalette *createToonzPalette(TPalette *cleanupPalette) {

  assert(cleanupPalette);

  assert(cleanupPalette->isCleanupPalette());

  TPalette *palette = new TPalette();

  for (int i = 0; i < cleanupPalette->getPage(0)->getStyleCount(); i++) {

    int styleId = cleanupPalette->getPage(0)->getStyleId(i);

    TCleanupStyle *cs =

        dynamic_cast<tcleanupstyle *="">(cleanupPalette->getStyle(styleId));</tcleanupstyle>

    if (!cs) continue;

    TPixel32 color = cs->getMainColor();

    while (palette->getStyleCount() < styleId) palette->addStyle(TPixel32::Red);

    if (styleId == palette->getStyleCount())

      palette->addStyle(color);

    else

      palette->setStyle(styleId, color);

    if (styleId > 1)  // 0 e 1 sono gia' nella pagina

      palette->getPage(0)->addStyle(styleId);

    assert(0 <= styleId && styleId < palette->getStyleCount());

    if (cs->getFlags()) palette->getStyle(styleId)->setFlags(cs->getFlags());

  }

  return palette;

}

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

TPalette *createToonzPalette(TPalette *cleanupPalette, int colorParamIndex) {

  assert(cleanupPalette);

  assert(cleanupPalette->isCleanupPalette());

  TPalette *palette = new TPalette();

  for (int i = 0; i < cleanupPalette->getPage(0)->getStyleCount(); i++) {

    int styleId = cleanupPalette->getPage(0)->getStyleId(i);

    TCleanupStyle *cs =

        dynamic_cast<tcleanupstyle *="">(cleanupPalette->getStyle(styleId));</tcleanupstyle>

    if (!cs) continue;

    TPixel32 color = cs->getColorParamValue(colorParamIndex);

    while (palette->getStyleCount() < styleId) palette->addStyle(TPixel32::Red);

    if (styleId == palette->getStyleCount())

      palette->addStyle(color);

    else

      palette->setStyle(styleId, color);

    if (styleId > 1)  // 0 e 1 sono gia' nella pagina

      palette->getPage(0)->addStyle(styleId);

    assert(0 <= styleId && styleId < palette->getStyleCount());

    if (cs->getFlags()) palette->getStyle(styleId)->setFlags(cs->getFlags());

  }

  return palette;

}

//====================================================================================

void TargetColors::update(TPalette *palette, bool noAntialias) {

  m_colors.clear();

  TargetColor transparent(TPixel32(255, 255, 255, 0) /*TPixel32::Transparent*/,

                         0,  // BackgroundStyle,

                         0, 0, 0, 0);

  m_colors.push_back(transparent);

  for (int i = 0; i < palette->getPage(0)->getStyleCount(); i++) {

    int styleId     = palette->getPage(0)->getStyleId(i);

    TColorStyle *cs = palette->getStyle(styleId);

    if (!cs) continue;

    if (TBlackCleanupStyle *blackStyle =

            dynamic_cast<tblackcleanupstyle *="">(cs)) {</tblackcleanupstyle>

      TargetColor tc(

          blackStyle->getMainColor(), styleId, (int)blackStyle->getBrightness(),

          noAntialias ? 100 : (int)blackStyle->getContrast(),

          blackStyle->getColorThreshold(), blackStyle->getWhiteThreshold());

      m_colors.push_back(tc);

    } else if (TColorCleanupStyle *colorStyle =

                   dynamic_cast<tcolorcleanupstyle *="">(cs)) {</tcolorcleanupstyle>

      TargetColor tc(colorStyle->getMainColor(), styleId,

                     (int)colorStyle->getBrightness(),

                     noAntialias ? 100 : (int)colorStyle->getContrast(),

                     colorStyle->getHRange(), colorStyle->getLineWidth());

      m_colors.push_back(tc);

    }

  }

}

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

void convertToCleanupPalette(TPalette *palette) {

  if (palette->isCleanupPalette()) return;

  for (int i = 1; i < palette->getStyleCount(); i++) {

    TColorStyle *cs  = palette->getStyle(i);

    TPixel32 color   = cs->getMainColor();

    TColorStyle *ccs = 0;

    if (i == 1)

      ccs = new TBlackCleanupStyle();

    else

      ccs = new TColorCleanupStyle(color);

    palette->setStyle(i, ccs);

  }

  palette->setIsCleanupPalette(true);

}

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

void convertToLevelPalette(TPalette *palette) {

  if (!palette->isCleanupPalette()) return;

  for (int i = 1; i < palette->getStyleCount(); i++) {

    TPixel32 color = palette->getStyle(i)->getMainColor();

    palette->setStyle(i, color);

  }

  palette->setIsCleanupPalette(false);

}