#include "stdfx.h"

#include "tfxparam.h"

#include "tspectrumparam.h"

class MultiToneFx final : public TStandardRasterFx {

  FX_PLUGIN_DECLARATION(MultiToneFx)

  TRasterFxPort m_input;

  TSpectrumParamP m_colors;

public:

  MultiToneFx() {

    std::vector<tspectrum::colorkey> colors = {</tspectrum::colorkey>

        TSpectrum::ColorKey(0, TPixel32::White),

        TSpectrum::ColorKey(0.5, TPixel32::Yellow),

        TSpectrum::ColorKey(1, TPixel32::Red)};

    m_colors = TSpectrumParamP(colors);

    bool ret = m_colors->isKeyframe(0);

    bindParam(this, "colors", m_colors);

    ret = getParams()->getParam(0)->isKeyframe(0);

    addInputPort("Source", m_input);

  }

  ~MultiToneFx(){};

  bool doGetBBox(double frame, TRectD &bBox,

                 const TRenderSettings &info) override {

    if (m_input.isConnected())

      return m_input->doGetBBox(frame, bBox, info);

    else {

      bBox = TRectD();

      return false;

    }

  }

  void doCompute(TTile &tile, double frame, const TRenderSettings &) override;

  bool canHandle(const TRenderSettings &info, double frame) override {

    return true;

  }

};

template <typename channel_type="" pixel,="" pixelgray,="" typename=""></typename>

void doMultiTone(const TRasterPT<pixel> &ras,</pixel>

                 const TSpectrumT<pixel> &spectrum) {</pixel>

  double aux = (double)PIXEL::maxChannelValue;

  int j;

  ras->lock();

  for (j = 0; j < ras->getLy(); j++) {

    PIXEL *pix    = ras->pixels(j);

    PIXEL *endPix = pix + ras->getLx();

    while (pix < endPix) {

      if (pix->m) {

        CHANNEL_TYPE reference;

        double s;

        reference = PIXELGRAY::from(*pix).value;

        s         = reference / aux;

        if (pix->m == aux) {

          *pix = spectrum.getPremultipliedValue(s);

        } else {

          PIXEL tmp = spectrum.getPremultipliedValue(s);

          s         = pix->m / aux;

          tmp.r     = (CHANNEL_TYPE)(tmp.r * s);

          tmp.g     = (CHANNEL_TYPE)(tmp.g * s);

          tmp.b     = (CHANNEL_TYPE)(tmp.b * s);

          tmp.m     = (CHANNEL_TYPE)(tmp.m * s);

          *pix      = tmp;

        }

      }

      pix++;

    }

  }

  ras->unlock();

}

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

void MultiToneFx::doCompute(TTile &tile, double frame,

                            const TRenderSettings &ri) {

  if (!m_input.isConnected()) return;

  m_input->compute(tile, frame, ri);

  if (TRaster32P raster32 = tile.getRaster())

    doMultiTone<tpixel32, tpixelgr8,="" uchar="">(raster32,</tpixel32,>

                                            m_colors->getValue(frame));

  else if (TRaster64P raster64 = tile.getRaster())

    doMultiTone<tpixel64, tpixelgr16,="" ushort="">(raster64,</tpixel64,>

                                              m_colors->getValue64(frame));

  else

    throw TException("MultiToneFx: unsupported Pixel Type");

}

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

FX_PLUGIN_IDENTIFIER(MultiToneFx, "multiToneFx")