#include "tfxparam.h"

#include "stdfx.h"

#include "tspectrumparam.h"

#include "tpixelutils.h"

#include "tparamuiconcept.h"

namespace {

inline void pixelConvert(TPixel32 &dst, const TPixel32 &src) { dst = src; }

inline void pixelConvert(TPixel64 &dst, const TPixel32 &src) {

  dst = toPixel64(src);

}

}  // namespace

class FourPointsGradientFx : public TStandardZeraryFx {

  FX_PLUGIN_DECLARATION(FourPointsGradientFx)

  TSpectrumParamP m_colors;

  TPointParamP m_point1;

  TPointParamP m_point2;

  TPointParamP m_point3;

  TPointParamP m_point4;

  TPixelParamP m_color1;

  TPixelParamP m_color2;

  TPixelParamP m_color3;

  TPixelParamP m_color4;

public:

  FourPointsGradientFx()

      : m_point1(TPointD(200.0, 200.0))

      , m_point2(TPointD(-200.0, 200.0))

      , m_point3(TPointD(-200.0, -200.0))

      , m_point4(TPointD(200.0, -200.0))

      , m_color1(TPixel32::Red)

      , m_color2(TPixel32::Green)

      , m_color3(TPixel32::Blue)

      , m_color4(TPixel32::Yellow) {

    m_point1->getX()->setMeasureName("fxLength");

    m_point1->getY()->setMeasureName("fxLength");

    m_point2->getX()->setMeasureName("fxLength");

    m_point2->getY()->setMeasureName("fxLength");

    m_point3->getX()->setMeasureName("fxLength");

    m_point3->getY()->setMeasureName("fxLength");

    m_point4->getX()->setMeasureName("fxLength");

    m_point4->getY()->setMeasureName("fxLength");

    /*

bindParam(this,"point1", m_point1);

bindParam(this,"point2", m_point2);

bindParam(this,"point3", m_point3);

bindParam(this,"point4", m_point4);

bindParam(this,"color_1", m_color1);

bindParam(this,"color_2", m_color2);

bindParam(this,"color_3", m_color3);

bindParam(this,"color_4", m_color4);

*/

    bindParam(this, "Point_1", m_point1);

    bindParam(this, "Color_1", m_color1);

    bindParam(this, "Point_2", m_point2);

    bindParam(this, "Color_2", m_color2);

    bindParam(this, "Point_3", m_point3);

    bindParam(this, "Color_3", m_color3);

    bindParam(this, "Point_4", m_point4);

    bindParam(this, "Color_4", m_color4);

  }

  ~FourPointsGradientFx(){};

  bool doGetBBox(double, TRectD &bbox, const TRenderSettings &info) override {

    bbox = TConsts::infiniteRectD;

    return true;

  };

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

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

  void getParamUIs(TParamUIConcept *&concepts, int &length) override {

    concepts = new TParamUIConcept[length = 4];

    concepts[0].m_type  = TParamUIConcept::POINT;

    concepts[0].m_label = "Point 1";

    concepts[0].m_params.push_back(m_point1);

    concepts[1].m_type  = TParamUIConcept::POINT;

    concepts[1].m_label = "Point 2";

    concepts[1].m_params.push_back(m_point2);

    concepts[2].m_type  = TParamUIConcept::POINT;

    concepts[2].m_label = "Point 3";

    concepts[2].m_params.push_back(m_point3);

    concepts[3].m_type  = TParamUIConcept::POINT;

    concepts[3].m_label = "Point 4";

    concepts[3].m_params.push_back(m_point4);

  }

};

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

void doFourPointsGradient(const TRasterPT<pixel> &ras, TPointD tilepos,</pixel>

                          TPointD pos1, TPointD pos2, TPointD pos3,

                          TPointD pos4, TPixel32 ccol1, TPixel32 ccol2,

                          TPixel32 ccol3, TPixel32 ccol4) {

  PIXEL col1, col2, col3, col4;

  pixelConvert(col1, ccol1);

  pixelConvert(col2, ccol2);

  pixelConvert(col3, ccol3);

  pixelConvert(col4, ccol4);

  int j;

  ras->lock();

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

    TPointD pos = tilepos;

    pos.y += j;

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

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

    while (pix < endPix) {

      double d1 = sqrt((pos1.x - pos.x) * (pos1.x - pos.x) +

                       (pos1.y - pos.y) * (pos1.y - pos.y));

      if (!d1) {

        *pix = col1;

        pos.x += 1;

        ++pix;

        continue;

      }

      double d2 = sqrt((pos2.x - pos.x) * (pos2.x - pos.x) +

                       (pos2.y - pos.y) * (pos2.y - pos.y));

      if (!d2) {

        *pix = col2;

        pos.x += 1;

        ++pix;

        continue;

      }

      double d3 = sqrt((pos3.x - pos.x) * (pos3.x - pos.x) +

                       (pos3.y - pos.y) * (pos3.y - pos.y));

      if (!d3) {

        *pix = col3;

        pos.x += 1;

        ++pix;

        continue;

      }

      double d4 = sqrt((pos4.x - pos.x) * (pos4.x - pos.x) +

                       (pos4.y - pos.y) * (pos4.y - pos.y));

      if (!d4) {

        *pix = col4;

        pos.x += 1;

        ++pix;

        continue;

      }

      double dtotal = 1 / d1 + 1 / d2 + 1 / d3 + 1 / d4;

      pix->r        = (CHANNEL_TYPE)(

          ((1 / d1) / dtotal) * col1.r + ((1 / d2) / dtotal) * col2.r +

          ((1 / d3) / dtotal) * col3.r + ((1 / d4) / dtotal) * col4.r);

      pix->g = (CHANNEL_TYPE)(

          ((1 / d1) / dtotal) * col1.g + ((1 / d2) / dtotal) * col2.g +

          ((1 / d3) / dtotal) * col3.g + ((1 / d4) / dtotal) * col4.g);

      pix->b = (CHANNEL_TYPE)(

          ((1 / d1) / dtotal) * col1.b + ((1 / d2) / dtotal) * col2.b +

          ((1 / d3) / dtotal) * col3.b + ((1 / d4) / dtotal) * col4.b);

      pix->m = (CHANNEL_TYPE)(

          ((1 / d1) / dtotal) * col1.m + ((1 / d2) / dtotal) * col2.m +

          ((1 / d3) / dtotal) * col3.m + ((1 / d4) / dtotal) * col4.m);

      pos.x += 1.0;

      ++pix;

    }

  }

  ras->unlock();

}

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

                                     const TRenderSettings &ri) {

  TPointD pos1  = m_point1->getValue(frame) * (1.0 / ri.m_shrinkX);

  TPointD pos2  = m_point2->getValue(frame) * (1.0 / ri.m_shrinkX);

  TPointD pos3  = m_point3->getValue(frame) * (1.0 / ri.m_shrinkX);

  TPointD pos4  = m_point4->getValue(frame) * (1.0 / ri.m_shrinkX);

  pos1          = ri.m_affine * pos1;

  pos2          = ri.m_affine * pos2;

  pos3          = ri.m_affine * pos3;

  pos4          = ri.m_affine * pos4;

  TPixel32 col1 = m_color1->getPremultipliedValue(frame);

  TPixel32 col2 = m_color2->getPremultipliedValue(frame);

  TPixel32 col3 = m_color3->getPremultipliedValue(frame);

  TPixel32 col4 = m_color4->getPremultipliedValue(frame);

  TPointD pos         = tile.m_pos;

  TRaster32P raster32 = tile.getRaster();

  if (raster32)

    doFourPointsGradient<tpixel32, uchar="">(raster32, pos, pos1, pos2, pos3, pos4,</tpixel32,>

                                          col1, col2, col3, col4);

  else {

    TRaster64P raster64 = tile.getRaster();

    if (raster64)

      doFourPointsGradient<tpixel64, ushort="">(raster64, pos, pos1, pos2, pos3,</tpixel64,>

                                             pos4, col1, col2, col3, col4);

    else

      throw TException("Brightness&Contrast: unsupported Pixel Type");

  }

}

FX_PLUGIN_IDENTIFIER(FourPointsGradientFx, "fourPointsGradientFx");