// TnzCore includes

#include "trop.h"

// TnzBase includes

#include "tfxparam.h"

#include "tparamset.h"

// TnzStdfx includes

#include "stdfx.h"

//****************************************************************************

//    ErodeDilate Fx

//****************************************************************************

class ErodeDilateFx : public TStandardRasterFx {

  FX_PLUGIN_DECLARATION(ErodeDilateFx)

  TRasterFxPort m_input;

  TIntEnumParamP m_type;

  TDoubleParamP m_radius;

public:

  ErodeDilateFx() : m_type(new TIntEnumParam(0, "Square")), m_radius(0.0) {

    addInputPort("Source", m_input);

    bindParam(this, "type", m_type);

    m_type->addItem(1, "Circular");

    m_radius->setMeasureName("fxLength");

    bindParam(this, "radius", m_radius);

  }

  bool doGetBBox(double frame, TRectD &bBox,

                 const TRenderSettings &info) override;

  void doDryCompute(TRectD &rect, double frame,

                    const TRenderSettings &ri) override;

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

  int getMemoryRequirement(const TRectD &rect, double frame,

                           const TRenderSettings &info) override;

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

    return isAlmostIsotropic(info.m_affine);

  }

};

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

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

                              const TRenderSettings &info) {

  // Remember: info.m_affine MUST NOT BE CONSIDERED in doGetBBox's

  // implementation

  // Retrieve the input bbox without applied affines.

  if (!m_input.getFx()) return false;

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

  bBox = bBox.enlarge(tceil(m_radius->getValue(frame)));

  return ret;

}

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

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

                              const TRenderSettings &info) {

  if (!m_input.isConnected()) return;

  double radius = m_radius->getValue(frame) * sqrt(info.m_affine.det());

  bool dilate   = (radius >= 0.0);

  TRop::ErodilateMaskType type = TRop::ErodilateMaskType(m_type->getValue());

  if (dilate) {

    // Quite easy, this time - just compute the input tile by forwarding the

    // request

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

    // Then, apply TRop::erodilate directly with the scale-adjusted radius

    const TRasterP &ras = tile.getRaster();

    TRop::erodilate(ras, ras, radius, type);

  } else {

    // In the erosion case, we need to know the input enlarged by the radius

    // value

    int radI = tceil(fabs(radius));

    const TDimension &tileSize = tile.getRaster()->getSize();

    const TRectD &enlargedRect =

        TRectD(tile.m_pos, TDimensionD(tileSize.lx, tileSize.ly)).enlarge(radI);

    TDimension enlargedSize(tround(enlargedRect.getLx()),

                            tround(enlargedRect.getLy()));

    TTile inTile;

    m_input->allocateAndCompute(inTile, enlargedRect.getP00(), enlargedSize,

                                tile.getRaster(), frame, info);

    const TRasterP &inRas = inTile.getRaster();

    TRop::erodilate(inRas, inRas, radius, type);

    tile.getRaster()->copy(inRas, -TPoint(radI, radI));

  }

}

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

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

                                 const TRenderSettings &info) {

  if (!m_input.isConnected()) return;

  double radius = m_radius->getValue(frame) * sqrt(info.m_affine.det());

  bool dilate   = (radius >= 0.0);

  if (dilate)

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

  else {

    int radI = tceil(fabs(radius));

    TRectD enlargedRect(rect.enlarge(radI));

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

  }

}

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

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

                                        const TRenderSettings &info) {

  return (m_type->getValue() == 0) ? TRasterFx::memorySize(rect, 8)

                                   :              // One additional greymaps

             2 * TRasterFx::memorySize(rect, 8);  // Two additional greymaps

}

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

FX_PLUGIN_IDENTIFIER(ErodeDilateFx, "erodeDilateFx");