// TnzCore includes
#include "trop.h"
#include "tpixelutils.h"

// TnzBase includes
#include "tparamset.h"
#include "tdoubleparam.h"
#include "tfxparam.h"
#include "tparamuiconcept.h"

#include "tbasefx.h"
#include "tzeraryfx.h"

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

class ColorCardFx final : public TBaseZeraryFx {
  FX_DECLARATION(ColorCardFx)

  TPixelParamP m_color;

public:
  ColorCardFx() : m_color(TPixel32::Green) {
    bindParam(this, "color", m_color);
    m_color->setDefaultValue(TPixel32::Green);
    setName(L"ColorCardFx");
    enableComputeInFloat(true);
  }

  bool canHandle(const TRenderSettings &info, double frame) override {
    return true;
  }

  bool doGetBBox(double frame, TRectD &bBox,
                 const TRenderSettings &info) override {
    bBox = TConsts::infiniteRectD;
    return true;
  }

  void doCompute(TTile &tile, double frame,
                 const TRenderSettings &ri) override {
    TRaster32P ras32 = tile.getRaster();
    TRaster64P ras64 = tile.getRaster();
    TRasterFP rasF   = tile.getRaster();
    // currently the tile should always be nonlinear
    assert(!tile.getRaster()->isLinear());
    if (!tile.getRaster()->isLinear()) {
      if (ras32)
        ras32->fill(m_color->getPremultipliedValue(frame));
      else if (ras64)
        ras64->fill(toPixel64(m_color->getPremultipliedValue(frame)));
      else if (rasF)
        rasF->fill(toPixelF(m_color->getPremultipliedValue(frame)));
      else
        throw TException("ColorCardFx unsupported pixel type");
    } else {  // linear color space
      if (ras32)
        ras32->fill(toLinear(m_color->getPremultipliedValue(frame),
                             ri.m_colorSpaceGamma));
      else if (ras64)
        ras64->fill(toLinear(toPixel64(m_color->getPremultipliedValue(frame)),
                             ri.m_colorSpaceGamma));
      else if (rasF)
        rasF->fill(toLinear(toPixelF(m_color->getPremultipliedValue(frame)),
                            ri.m_colorSpaceGamma));
      else
        throw TException("ColorCardFx unsupported pixel type");
    }
  }
};

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

class CheckBoardFx final : public TBaseZeraryFx {
  FX_DECLARATION(CheckBoardFx)

  TPixelParamP m_color1, m_color2;
  TDoubleParamP m_size;

public:
  CheckBoardFx()
      : m_color1(TPixel32::Black), m_color2(TPixel32::White), m_size(50) {
    m_size->setMeasureName("fxLength");
    bindParam(this, "color1", m_color1);
    bindParam(this, "color2", m_color2);
    bindParam(this, "size", m_size);
    m_color1->setDefaultValue(TPixel32::Black);
    m_color2->setDefaultValue(TPixel32::White);
    m_size->setValueRange(1, 1000);
    m_size->setDefaultValue(50);
    setName(L"CheckBoardFx");
    enableComputeInFloat(true);
  }

  bool canHandle(const TRenderSettings &info, double frame) override {
    return false;
  }

  bool doGetBBox(double, TRectD &bBox, const TRenderSettings &info) override {
    bBox = TConsts::infiniteRectD;
    return true;
  }

  void doCompute(TTile &tile, double frame,
                 const TRenderSettings &info) override {
    bool isLinear = tile.getRaster()->isLinear();
    // currently the tile should always be nonlinear
    assert(!isLinear);
    const TPixel32 &c1 =
        m_color1->getValue(frame, isLinear, info.m_colorSpaceGamma);
    const TPixel32 &c2 =
        m_color2->getValue(frame, isLinear, info.m_colorSpaceGamma);

    double size = m_size->getValue(frame);

    assert(info.m_shrinkX == info.m_shrinkY);
    size *= info.m_affine.a11 / info.m_shrinkX;

    TDimensionD dim(size, size);
    TRop::checkBoard(tile.getRaster(), c1, c2, dim, tile.m_pos);
  }

  void getParamUIs(TParamUIConcept *&concepts, int &length) override {
    concepts = new TParamUIConcept[length = 1];

    concepts[0].m_type  = TParamUIConcept::SIZE;
    concepts[0].m_label = "Size";
    concepts[0].m_params.push_back(m_size);
  }
};

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

FX_IDENTIFIER(ColorCardFx, "colorCardFx")
FX_IDENTIFIER(CheckBoardFx, "checkBoardFx")