#include "tfxparam.h"

#include "tfx.h"

#include "tparamset.h"

#include "stdfx.h"

#include "time.h"

#include "trop.h"

#include "trasterfx.h"

#include "tpixelutils.h"

#include "tparamuiconcept.h"

#include "timage_io.h"

#include "toonz/tdistort.h"

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

namespace {

inline bool myIsEmpty(const TRectD &r) { return r.x0 >= r.x1 || r.y0 >= r.y1; }

}  // namespace

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

class FreeDistortBaseFx : public TStandardRasterFx {

  enum { PERSPECTIVE, BILINEAR };

public:

  FreeDistortBaseFx(bool isCastShadow);

  ~FreeDistortBaseFx();

  bool doGetBBox(double frame, TRectD &bBox,

                 const TRenderSettings &info) override;

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

  void doDryCompute(TRectD &rect, double frame,

                    const TRenderSettings &ri) override;

  void onPortConnected(TFxPort *port);

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

  int getMemoryRequirement(const TRectD &rect, double frame,

                           const TRenderSettings &info) override;

  void transform(double frame, int port, const TRectD &rectOnOutput,

                 const TRenderSettings &infoOnOutput, TRectD &rectOnInput,

                 TRenderSettings &infoOnInput) override;

  void safeTransform(double frame, int port, const TRectD &rectOnOutput,

                     const TRenderSettings &infoOnOutput, TRectD &rectOnInput,

                     TRenderSettings &infoOnInput, TRectD &inBBox);

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

private:

  bool m_isCastShadow;

  TRasterFxPort m_input;

  TIntEnumParamP m_distortType;

  TPointParamP m_p00_a;

  TPointParamP m_p00_b;

  TPointParamP m_p01_a;

  TPointParamP m_p01_b;

  TPointParamP m_p11_a;

  TPointParamP m_p11_b;

  TPointParamP m_p10_a;

  TPointParamP m_p10_b;

  TBoolParamP m_deactivate;

  TPixelParamP m_color;

  TDoubleParamP m_fade;

  TDoubleParamP m_upTransp;

  TDoubleParamP m_downTransp;

  TDoubleParamP m_upBlur;

  TDoubleParamP m_downBlur;

};

class FreeDistortFx final : public FreeDistortBaseFx {

  FX_PLUGIN_DECLARATION(FreeDistortFx)

public:

  FreeDistortFx() : FreeDistortBaseFx(false) {}

};

class CastShadowFx final : public FreeDistortBaseFx {

  FX_PLUGIN_DECLARATION(CastShadowFx)

public:

  CastShadowFx() : FreeDistortBaseFx(true) {}

};

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

FreeDistortBaseFx::FreeDistortBaseFx(bool isCastShadow)

    : m_deactivate(false)

    , m_color(TPixel32::Black)

    , m_fade(0.0)

    , m_upTransp(0.0)

    , m_downTransp(0.0)

    , m_upBlur(0.0)

    , m_downBlur(0.0)

    , m_isCastShadow(isCastShadow)

    , m_distortType(new TIntEnumParam(PERSPECTIVE, "Perspective")) {

  m_upBlur->setMeasureName("fxLength");

  m_downBlur->setMeasureName("fxLength");

  double ext = 400.;

  double inn = 400.;

  m_p00_a    = TPointD(-ext, -ext);

  m_p00_b    = TPointD(-inn, -inn);

  m_p01_a = TPointD(-ext, ext);

  m_p01_b = TPointD(-inn, inn);

  m_p11_a = TPointD(ext, ext);

  m_p11_b = TPointD(inn, inn);

  m_p10_a = TPointD(ext, -ext);

  m_p10_b = TPointD(inn, -inn);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  bindParam(this, "distort_type", m_distortType);

  bindParam(this, "bottom_left_b", m_p00_b);

  bindParam(this, "bottom_right_b", m_p10_b);

  bindParam(this, "top_left_b", m_p01_b);

  bindParam(this, "top_right_b", m_p11_b);

  bindParam(this, "bottom_left_a", m_p00_a);

  bindParam(this, "bottom_right_a", m_p10_a);

  bindParam(this, "top_left_a", m_p01_a);

  bindParam(this, "top_right_a", m_p11_a);

  if (isCastShadow) {

    bindParam(this, "color", m_color);

    bindParam(this, "fade", m_fade);

    bindParam(this, "up_transp", m_upTransp);

    bindParam(this, "down_transp", m_downTransp);

    bindParam(this, "up_blur", m_upBlur);

    bindParam(this, "down_blur", m_downBlur);

    m_color->enableMatte(false);

    m_fade->setValueRange(0.0, 100.0);

    m_upTransp->setValueRange(0.0, 100.0);

    m_downTransp->setValueRange(0.0, 100.0);

    m_upBlur->setValueRange(0.0, (std::numeric_limits<double>::max)());</double>

    m_downBlur->setValueRange(0.0, (std::numeric_limits<double>::max)());</double>

  }

  bindParam(this, "deactivate", m_deactivate);

  addInputPort("source", m_input);

  m_p00_a->getX()->setValueRange(-1000, 1000);

  m_p00_b->getX()->setValueRange(-1000, 1000);

  m_p00_a->getY()->setValueRange(-1000, 1000);

  m_p00_b->getY()->setValueRange(-1000, 1000);

  m_p01_a->getX()->setValueRange(-1000, 1000);

  m_p01_b->getX()->setValueRange(-1000, 1000);

  m_p01_a->getY()->setValueRange(-1000, 1000);

  m_p01_b->getY()->setValueRange(-1000, 1000);

  m_p11_a->getX()->setValueRange(-1000, 1000);

  m_p11_b->getX()->setValueRange(-1000, 1000);

  m_p11_a->getY()->setValueRange(-1000, 1000);

  m_p11_b->getY()->setValueRange(-1000, 1000);

  m_p10_a->getX()->setValueRange(-1000, 1000);

  m_p10_b->getX()->setValueRange(-1000, 1000);

  m_p10_a->getY()->setValueRange(-1000, 1000);

  m_p10_b->getY()->setValueRange(-1000, 1000);

  m_distortType->addItem(BILINEAR, "Bilinear");

}

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

FreeDistortBaseFx::~FreeDistortBaseFx() {}

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

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

  // Blurs cant deal with anisotropic affines, so these are retained above

  return m_upBlur->getValue(frame) == 0 && m_downBlur->getValue(frame) == 0;

}

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

bool FreeDistortBaseFx::doGetBBox(double frame, TRectD &bBox,

                                  const TRenderSettings &info) {

  if (m_input.isConnected()) {

    // We'll treat this fx like a zerary one. The idea is, since

    // the task of taking images and pre-images of rects through

    // these kind of distortions is definitely non-trivial, we'll

    // simply avoid doing so...

    bBox = TConsts::infiniteRectD;

    return true;

    /*

bool ret = m_input->doGetBBox(frame, bBox);

if(m_deactivate->getValue())

return ret;

TPointD p00_a = m_p00_a->getValue(frame);

TPointD p00_b = m_p00_b->getValue(frame);

TPointD p10_a = m_p10_a->getValue(frame);

TPointD p10_b = m_p10_b->getValue(frame);

TPointD p01_a = m_p01_a->getValue(frame);

TPointD p01_b = m_p01_b->getValue(frame);

TPointD p11_a = m_p11_a->getValue(frame);

TPointD p11_b = m_p11_b->getValue(frame);

    if (m_isCastShadow)

{

            std::swap(p00_a, p01_a);

std::swap(p10_a, p11_a);

    }*/

    /*TRectD source;

source.x0 = std::min(p00_b.x, p10_b.x, p01_b.x, p11_b.x);

source.y0 = std::min(p00_b.y, p10_b.y, p01_b.y, p11_b.y);

source.x1 = std::max(p00_b.x, p10_b.x, p01_b.x, p11_b.x);

source.y1 = std::max(p00_b.y, p10_b.y, p01_b.y, p11_b.y);

bBox *= source;

int distortType = m_distortType->getValue();

if(distortType == PERSPECTIVE)

{

PerspectiveDistorter distorter(

p00_b, p10_b, p01_b, p11_b,

p00_a, p10_a, p01_a, p11_a

);

bBox = distorter.map(bBox);

}

else if(distortType == BILINEAR)

{

BilinearDistorter distorter(

p00_b, p10_b, p01_b, p11_b,

p00_a, p10_a, p01_a, p11_a

);

bBox = distorter.map(bBox);

}

else assert(0);*/

    // Further, we will also avoid the assumption that the bbox

    // is defined by that of destination points...

    /*TRectD result;

result.x0 = std::min(p00_a.x, p10_a.x, p01_a.x, p11_a.x);

result.y0 = std::min(p00_a.y, p10_a.y, p01_a.y, p11_a.y);

result.x1 = std::max(p00_a.x, p10_a.x, p01_a.x, p11_a.x);

result.y1 = std::max(p00_a.y, p10_a.y, p01_a.y, p11_a.y);

bBox = result;    //bBox *= result;*/

  } else {

    bBox = TRectD();

    return false;

  }

  if (m_upBlur->getValue(frame) > 0 || m_downBlur->getValue(frame) > 0) {

    int brad = std::max((int)m_upBlur->getValue(frame),

                        (int)m_downBlur->getValue(frame));

    bBox.x0 -= brad;

    bBox.x1 += brad;

    bBox.y0 -= brad;

    bBox.y1 += brad;

  }

  return true;

}

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

void FreeDistortBaseFx::transform(double frame, int port,

                                  const TRectD &rectOnOutput,

                                  const TRenderSettings &infoOnOutput,

                                  TRectD &rectOnInput,

                                  TRenderSettings &infoOnInput) {

  TPointD p00_b = m_p00_b->getValue(frame);

  TPointD p10_b = m_p10_b->getValue(frame);

  TPointD p01_b = m_p01_b->getValue(frame);

  TPointD p11_b = m_p11_b->getValue(frame);

  TPointD p00_a = m_p00_a->getValue(frame);

  TPointD p10_a = m_p10_a->getValue(frame);

  TPointD p01_a = m_p01_a->getValue(frame);

  TPointD p11_a = m_p11_a->getValue(frame);

  if (m_isCastShadow) {

    // Shadows are mirrored

    std::swap(p00_a, p01_a);

    std::swap(p10_a, p11_a);

  }

  // Build the input affine

  infoOnInput = infoOnOutput;

  double scale = 0;

  scale        = std::max(scale, norm(p10_a - p00_a) / norm(p10_b - p00_b));

  scale        = std::max(scale, norm(p01_a - p00_a) / norm(p01_b - p00_b));

  scale        = std::max(scale, norm(p11_a - p10_a) / norm(p11_b - p10_b));

  scale        = std::max(scale, norm(p11_a - p01_a) / norm(p11_b - p01_b));

  scale *= sqrt(fabs(infoOnInput.m_affine.det()));

  if (infoOnOutput.m_isSwatch) {

    // Swatch viewers work out a lite version of the distort: only contractions

    // are passed below in the schematic - so that input memory load is always

    // at minimum.

    // This is allowed in order to make fx adjusts the less frustrating as

    // possible.

    if (scale > 1) scale = 1;

  }

  infoOnInput.m_affine = TScale(scale);

  // Build rectOnInput

  p00_a = infoOnOutput.m_affine * p00_a;

  p10_a = infoOnOutput.m_affine * p10_a;

  p01_a = infoOnOutput.m_affine * p01_a;

  p11_a = infoOnOutput.m_affine * p11_a;

  p00_b = infoOnInput.m_affine * p00_b;

  p10_b = infoOnInput.m_affine * p10_b;

  p01_b = infoOnInput.m_affine * p01_b;

  p11_b = infoOnInput.m_affine * p11_b;

  if (m_distortType->getValue() == PERSPECTIVE) {

    PerspectiveDistorter pd(p00_b, p10_b, p01_b, p11_b, p00_a, p10_a, p01_a,

                            p11_a);

    rectOnInput = ((TQuadDistorter *)&pd)->invMap(rectOnOutput);

  } else {

    BilinearDistorter bd(p00_b, p10_b, p01_b, p11_b, p00_a, p10_a, p01_a,

                         p11_a);

    rectOnInput = ((TQuadDistorter *)&bd)->invMap(rectOnOutput);

  }

  if (rectOnInput.x0 != TConsts::infiniteRectD.x0)

    rectOnInput.x0 = tfloor(rectOnInput.x0);

  if (rectOnInput.y0 != TConsts::infiniteRectD.y0)

    rectOnInput.y0 = tfloor(rectOnInput.y0);

  if (rectOnInput.x1 != TConsts::infiniteRectD.x1)

    rectOnInput.x1 = tceil(rectOnInput.x1);

  if (rectOnInput.y1 != TConsts::infiniteRectD.y1)

    rectOnInput.y1 = tceil(rectOnInput.y1);

}

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

void FreeDistortBaseFx::safeTransform(double frame, int port,

                                      const TRectD &rectOnOutput,

                                      const TRenderSettings &infoOnOutput,

                                      TRectD &rectOnInput,

                                      TRenderSettings &infoOnInput,

                                      TRectD &inBBox) {

  assert(port == 0 && m_input.isConnected());

  if (m_deactivate->getValue()) {

    infoOnInput = infoOnOutput;

    rectOnInput = rectOnOutput;

    m_input->getBBox(frame, inBBox, infoOnInput);

    return;

  }

  // Avoid the singular matrix cases

  if (fabs(infoOnOutput.m_affine.det()) < 1.e-3) {

    infoOnInput = infoOnOutput;

    rectOnInput.empty();

    m_input->getBBox(frame, inBBox, infoOnInput);

    return;

  }

  transform(frame, port, rectOnOutput, infoOnOutput, rectOnInput, infoOnInput);

  m_input->getBBox(frame, inBBox, infoOnInput);

  // In case inBBox is infinite, such as with gradients, also intersect with the

  // input quadrilateral's bbox

  if (inBBox == TConsts::infiniteRectD) {

    TPointD affP00_b(infoOnInput.m_affine * m_p00_b->getValue(frame));

    TPointD affP10_b(infoOnInput.m_affine * m_p10_b->getValue(frame));

    TPointD affP01_b(infoOnInput.m_affine * m_p01_b->getValue(frame));

    TPointD affP11_b(infoOnInput.m_affine * m_p11_b->getValue(frame));

    TRectD source;

    source.x0 = std::min({affP00_b.x, affP10_b.x, affP01_b.x, affP11_b.x});

    source.y0 = std::min({affP00_b.y, affP10_b.y, affP01_b.y, affP11_b.y});

    source.x1 = std::max({affP00_b.x, affP10_b.x, affP01_b.x, affP11_b.x});

    source.y1 = std::max({affP00_b.y, affP10_b.y, affP01_b.y, affP11_b.y});

    rectOnInput *= source;

  }

}

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

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

  concepts = new TParamUIConcept[length = 6];

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

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

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

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

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

  concepts[0].m_label = " Src";

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

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

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

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

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

  concepts[1].m_label = " Dst";

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

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

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

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

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

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

  concepts[4].m_type = TParamUIConcept::VECTOR;

  concepts[4].m_params.push_back(m_p01_b);

  concepts[4].m_params.push_back(m_p01_a);

  concepts[5].m_type = TParamUIConcept::VECTOR;

  concepts[5].m_params.push_back(m_p11_b);

  concepts[5].m_params.push_back(m_p11_a);

}

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

namespace {

template <typename pix=""></typename>

void doFade(TRasterPT<pix> &ras, PIX col, double intensity) {</pix>

  double maxChannelValueD = PIX::maxChannelValue;

  int j;

  ras->lock();

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

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

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

    while (pix < endPix) {

      if (pix->m > 0) {

        double factor = pix->m / maxChannelValueD;

        int val;

        val    = troundp(pix->r + intensity * (col.r * factor - pix->r));

        pix->r = (val > PIX::maxChannelValue) ? PIX::maxChannelValue : val;

        val    = troundp(pix->g + intensity * (col.g * factor - pix->g));

        pix->g = (val > PIX::maxChannelValue) ? PIX::maxChannelValue : val;

        val    = troundp(pix->b + intensity * (col.b * factor - pix->b));

        pix->b = (val > PIX::maxChannelValue) ? PIX::maxChannelValue : val;

      }

      ++pix;

    }

  }

  ras->unlock();

}

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

template <typename pix=""></typename>

void doTransparency(TRasterPT<pix> &ras, double downTransp, double upTransp,</pix>

                    double y0, double y1) {

  assert(y0 <= 0.0 && y1 >= ras->getLy() - 1);

  double den = y1 - y0;

  int j;

  ras->lock();

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

    double transp = (1 - downTransp) + (downTransp - upTransp) * (j - y0) / den;

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

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

    while (pix < endPix) {

      if (pix->m > 0) {

        pix->r = troundp(pix->r * transp);

        pix->g = troundp(pix->g * transp);

        pix->b = troundp(pix->b * transp);

        pix->m = troundp(pix->m * transp);

      }

      ++pix;

    }

  }

  ras->unlock();

}

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

template <typename pix=""></typename>

inline void filterPixel(PIX *bufin, int wrap, PIX *bufout, double blur,

                        int coord, int maxVal) {

  int rad, k, index;

  double coeff, weight;

  double valr = 0.0, valg = 0.0, valb = 0.0, valm = 0.0;

  rad   = tceil(blur);

  coeff = 1.0 / rad;

  valr = bufin[0].r;

  valg = bufin[0].g;

  valb = bufin[0].b;

  valm = bufin[0].m;

  for (k = 1, index = wrap, weight = (1 - coeff); k < rad;

       k++, index += wrap, weight -= coeff) {

    if (k + coord < maxVal) {

      if (-k + coord >= 0) {

        valr += (bufin[index].r + bufin[-index].r) * weight;

        valg += (bufin[index].g + bufin[-index].g) * weight;

        valb += (bufin[index].b + bufin[-index].b) * weight;

        valm += (bufin[index].m + bufin[-index].m) * weight;

      } else {

        valr += bufin[index].r * weight;

        valg += bufin[index].g * weight;

        valb += bufin[index].b * weight;

        valm += bufin[index].m * weight;

      }

    } else if (-k + coord >= 0) {

      if (k + coord < maxVal) {

        valr += (bufin[index].r + bufin[-index].r) * weight;

        valg += (bufin[index].g + bufin[-index].g) * weight;

        valb += (bufin[index].b + bufin[-index].b) * weight;

        valm += (bufin[index].m + bufin[-index].m) * weight;

      } else {

        valr += bufin[-index].r * weight;

        valg += bufin[-index].g * weight;

        valb += bufin[-index].b * weight;

        valm += bufin[-index].m * weight;

      }

    }

  }

  valr *= coeff;

  valg *= coeff;

  valb *= coeff;

  valm *= coeff;

  bufout->r = troundp(valr);

  bufout->g = troundp(valg);

  bufout->b = troundp(valb);

  bufout->m = troundp(valm);

}

/*----------------------------------------------------------------------------*/

template <typename pix=""></typename>

void doBlur(TRasterPT<pix> &r, double blur0, double blur1, double transp0,</pix>

            double transp1, double y0, double y1) {

  int lx, ly, wrap;

  int i, j, brad, blur_max;

  PIX *bufin, *bufout, *pixin, *pixout, *row;

  double cur_blur, actual_blur, blur_incr, cur_transp, actual_transp,

      transp_incr;

  lx       = r->getLx();

  ly       = r->getLy();

  wrap     = r->getWrap();

  blur_max = (int)std::max(blur0, blur1);

  brad     = tceil(blur_max);

  // assert(y0 <= brad && y1 >= r->getLy()-1 - 2 * brad + y0);

  double den = y1 - y0;

  blur_incr   = (blur1 - blur0) / den;

  transp_incr = (transp0 - transp1) / den;

  row = new PIX[std::max(lx, ly)];

  r->lock();

  bufin = r->pixels(0);

  for (i = 0, cur_blur = blur0 + (blur1 - blur0) * (i - y0) / den,

      actual_blur = std::max(cur_blur, 0.0);

       i < ly; i++, bufin += wrap, cur_blur += blur_incr,

      actual_blur = std::max(cur_blur, 0.0)) {

    pixin = bufin;

    for (j = 0; j < lx; j++, pixin++) {

      if (cur_blur > 1.0)

        filterPixel(pixin, 1, row + j, actual_blur, j, lx);

      else

        row[j] = *pixin;

    }

    for (j = 0; j < lx; j++) bufin[j] = row[j];

  }

  bufin  = r->pixels(0);

  bufout = r->pixels(0);

  for (i = 0; i < lx; i++, bufin++, bufout++) {

    pixin  = bufin;

    pixout = bufout;

    for (j = 0, cur_blur = blur0 + (blur1 - blur0) * (j - y0) / den,

        actual_blur = std::max(cur_blur, 0.0);

         j < ly; j++, pixin += wrap, cur_blur += blur_incr,

        actual_blur = std::max(cur_blur, 0.0)) {

      if (cur_blur > 1.0)

        filterPixel(pixin, wrap, row + j, actual_blur, j, ly);

      else

        row[j] = *pixin;

    }

    if (transp0 > 0 || transp1 > 0)

      for (j            = 0,

          cur_transp    = 1 - transp0 + (transp0 - transp1) * (j - y0) / den,

          actual_transp = std::max(cur_transp, 0.0);

           j < ly; j++, pixout += wrap, cur_transp += transp_incr,

          actual_transp = std::max(cur_transp, 0.0)) {

        pixout->r = troundp(row[j].r * actual_transp);

        pixout->g = troundp(row[j].g * actual_transp);

        pixout->b = troundp(row[j].b * actual_transp);

        pixout->m = troundp(row[j].m * actual_transp);

      }

    else

      for (j = 0; j < ly; j++, pixout += wrap) {

        pixout->r = row[j].r;

        pixout->g = row[j].g;

        pixout->b = row[j].b;

        pixout->m = row[j].m;

      }

  }

  r->unlock();

  delete[] row;

}

}  // namespace

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

void FreeDistortBaseFx::doDryCompute(TRectD &rect, double frame,

                                     const TRenderSettings &info) {

  if (!m_input.isConnected()) return;

  if (m_deactivate->getValue()) {

    m_input->dryCompute(rect, frame, info);

    return;

  }

  TRectD rectOnInput;

  TRenderSettings infoOnInput;

  TRectD inBBox;

  safeTransform(frame, 0, rect, info, rectOnInput, infoOnInput, inBBox);

  rectOnInput *= inBBox;

  if (!myIsEmpty(rectOnInput))

    m_input->dryCompute(rectOnInput, frame, infoOnInput);

}

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

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

                                  const TRenderSettings &ri) {

  if (!m_input.isConnected()) return;

  // Upon deactivation, this fx does nothing.

  if (m_deactivate->getValue()) {

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

    return;

  }

  // Get the source quad

  TPointD p00_b = m_p00_b->getValue(frame);

  TPointD p10_b = m_p10_b->getValue(frame);

  TPointD p01_b = m_p01_b->getValue(frame);

  TPointD p11_b = m_p11_b->getValue(frame);

  // Get destination quad

  TPointD p00_a = m_p00_a->getValue(frame);

  TPointD p10_a = m_p10_a->getValue(frame);

  TPointD p01_a = m_p01_a->getValue(frame);

  TPointD p11_a = m_p11_a->getValue(frame);

  if (m_isCastShadow) {

    // Shadows are mirrored

    std::swap(p00_a, p01_a);

    std::swap(p10_a, p11_a);

  }

  // Get requested tile's geometry

  TRasterP tileRas(tile.getRaster());

  TRectD tileRect(convert(tileRas->getBounds()) + tile.m_pos);

  // Call transform to get the minimal rectOnInput

  TRectD inRect;

  TRenderSettings riNew;

  TRectD inBBox;

  safeTransform(frame, 0, tileRect, ri, inRect, riNew, inBBox);

  // Intersect with the bbox

  inRect *= inBBox;

  if (myIsEmpty(inRect)) return;

  double scale = ri.m_affine.a11;

  double downBlur = m_downBlur->getValue(frame) * scale;

  double upBlur   = m_upBlur->getValue(frame) * scale;

  int brad        = tceil(std::max(downBlur, upBlur));

  inRect = inRect.enlarge(brad);

  TDimension inRectSize(tceil(inRect.getLx()), tceil(inRect.getLy()));

  TTile inTile;

  m_input->allocateAndCompute(inTile, inRect.getP00(), inRectSize, tileRas,

                              frame, riNew);

  TPointD inTilePosRi = inTile.m_pos;

  // Update quads by the scale factors

  p00_b = riNew.m_affine * p00_b;

  p10_b = riNew.m_affine * p10_b;

  p01_b = riNew.m_affine * p01_b;

  p11_b = riNew.m_affine * p11_b;

  p00_a = ri.m_affine * p00_a;

  p10_a = ri.m_affine * p10_a;

  p01_a = ri.m_affine * p01_a;

  p11_a = ri.m_affine * p11_a;

  PerspectiveDistorter perpDistorter(p00_b - inTile.m_pos, p10_b - inTile.m_pos,

                                     p01_b - inTile.m_pos, p11_b - inTile.m_pos,

                                     p00_a, p10_a, p01_a, p11_a);

  BilinearDistorter bilDistorter(p00_b - inTile.m_pos, p10_b - inTile.m_pos,

                                 p01_b - inTile.m_pos, p11_b - inTile.m_pos,

                                 p00_a, p10_a, p01_a, p11_a);

  TQuadDistorter *distorter;

  if (m_distortType->getValue() == PERSPECTIVE)

    distorter = &perpDistorter;

  else if (m_distortType->getValue() == BILINEAR)

    distorter = &bilDistorter;

  else

    assert(0);

  if (m_isCastShadow) {

    TRaster32P ras32 = inTile.getRaster();

    TRaster64P ras64 = inTile.getRaster();

    if (ras32) {

      if (m_fade->getValue(frame) > 0)

        doFade(ras32, m_color->getValue(frame),

               m_fade->getValue(frame) / 100.0);

      if (brad > 0)

        doBlur(ras32, upBlur, downBlur, m_upTransp->getValue(frame) / 100.0,

               m_downTransp->getValue(frame) / 100.0,

               inBBox.y0 - inTile.m_pos.y, inBBox.y1 - inTile.m_pos.y);

      else if (m_upTransp->getValue(frame) > 0 ||

               m_downTransp->getValue(frame) > 0)

        doTransparency(ras32, m_upTransp->getValue(frame) / 100.0,

                       m_downTransp->getValue(frame) / 100.0,

                       inBBox.y0 - inTile.m_pos.y, inBBox.y1 - inTile.m_pos.y);

    } else if (ras64) {

      if (m_fade->getValue(frame) > 0)

        doFade(ras64, toPixel64(m_color->getValue(frame)),

               m_fade->getValue(frame) / 100.0);

      if (brad > 0)

        doBlur(ras64, upBlur, downBlur, m_upTransp->getValue(frame) / 100.0,

               m_downTransp->getValue(frame) / 100.0,

               inBBox.y0 - inTile.m_pos.y, inBBox.y1 - inTile.m_pos.y);

      else if (m_upTransp->getValue(frame) > 0 ||

               m_downTransp->getValue(frame) > 0)

        doTransparency(ras64, m_upTransp->getValue(frame) / 100.0,

                       m_downTransp->getValue(frame) / 100.0,

                       inBBox.y0 - inTile.m_pos.y, inBBox.y1 - inTile.m_pos.y);

    } else

      assert(false);

  }

  distort(tileRas, inTile.getRaster(), *distorter, convert(tile.m_pos),

          TRop::Bilinear);

}

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

int FreeDistortBaseFx::getMemoryRequirement(const TRectD &rect, double frame,

                                            const TRenderSettings &info) {

  if (!m_input.isConnected()) return 0;

  TRectD inRect;

  TRenderSettings riNew;

  TRectD inBBox;

  safeTransform(frame, 0, rect, info, inRect, riNew, inBBox);

  inRect *= inBBox;

  return TRasterFx::memorySize(inRect, riNew.m_bpp);

}

FX_PLUGIN_IDENTIFIER(FreeDistortFx, "freeDistortFx");

FX_PLUGIN_IDENTIFIER(CastShadowFx, "castShadowFx");

/*----------------------------------------------------------------------------*/