#include "stdfx.h"
#include "tfxparam.h"
//#include "trop.h"
#include "tdoubleparam.h"
//#include "tnotanimatableparam.h"
#include "tpixelgr.h"
#include "trasterfx.h"

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

namespace {

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

enum Status {
  StatusGood        = 0,
  OutOfTime         = 1 << 1,  // Non utilizzato
  Port0NotConnected = 1 << 2,
  Port1NotConnected = 1 << 3,
  NoPortsConnected  = Port0NotConnected | Port1NotConnected
};
inline Status operator|(const Status &l, const Status &r) {
  return Status(((int)l | (int)r));
}
inline int operator&(const Status &l) { return int(l); }
Status getFxStatus(const TRasterFxPort &port0, const TRasterFxPort &port1) {
  Status status                    = StatusGood;
  if (!port0.isConnected()) status = status | Port0NotConnected;
  if (!port1.isConnected()) status = status | Port1NotConnected;
  return status;
}

/*inline*/
template <typename T, typename Q, typename P>
void doLocalTransparency(TRasterPT<T> out, TRasterPT<T> src, TRasterPT<T> ref,
                         double transp) {
  out->lock();
  ref->lock();
  src->lock();

  T *outRow   = out->pixels();
  T *refRow   = ref->pixels();
  T *srcRow   = src->pixels();
  int outWrap = out->getWrap();
  int refWrap = ref->getWrap();
  int srcWrap = src->getWrap();

  T *outPix  = outRow;
  T *srcPix  = srcRow;
  T *refPix  = refRow;
  T *lastPix = outRow + outWrap * out->getLy();

  const int cropVal = T::maxChannelValue;
  double factor     = transp / (double)cropVal;

  while (outPix < lastPix) {
    T *endPix = outPix + out->getLx();
    while (outPix < endPix) {
      double local_transp = 1 - (Q::from(*refPix).value) * factor;
      if (local_transp > 0.0) {
        int val   = (int)(local_transp * srcPix->r + 0.5);
        outPix->r = (P)((val < cropVal) ? val : cropVal);
        val       = (int)(local_transp * srcPix->g + 0.5);
        outPix->g = (P)((val < cropVal) ? val : cropVal);
        val       = (int)(local_transp * srcPix->b + 0.5);
        outPix->b = (P)((val < cropVal) ? val : cropVal);
        val       = (int)(local_transp * srcPix->m + 0.5);
        outPix->m = (P)((val < cropVal) ? val : cropVal);
      } else {
        outPix->r = outPix->g = outPix->b = outPix->m = 0;
      }
      ++outPix;
      ++refPix;
      ++srcPix;
    }
    srcRow += srcWrap;
    outRow += outWrap;
    refRow += refWrap;
    srcPix = srcRow;
    outPix = outRow;
    refPix = refRow;
  }
  out->unlock();
  ref->unlock();
  src->unlock();
}

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

inline double func(int x, int y, int lx, int ly) {
  return ((1 + sin(3.14 * x / (0.5 * lx))) * (1 + cos(3.14 * y / (0.5 * ly))));
}

void drawCheckboard(TRaster32P &raster) {
  int lx = raster->getLx();
  int ly = raster->getLy();
  // int chessSize = 4;
  int x, y;
  raster->lock();
  for (y = 0; y < ly; ++y) {
    TPixel32 *pix = raster->pixels(y);
    for (x = 0; x < lx; ++x, ++pix) {
      pix->r = pix->g = pix->b =
          troundp(255.0 * func(x, y, lx, ly) / func(lx - 1, ly - 1, lx, ly));
      pix->m = 255;
    }
  }
  raster->unlock();
}
};

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

class LocalTransparencyFx final : public TStandardRasterFx {
  FX_PLUGIN_DECLARATION(LocalTransparencyFx)
protected:
  TRasterFxPort m_src, m_ref;
  TDoubleParamP m_value;

public:
  LocalTransparencyFx() : m_value(100) {
    addInputPort("Source", m_src);
    addInputPort("Reference", m_ref);
    bindParam(this, "value", m_value);
    m_value->setValueRange(0, 100);
  }
  virtual ~LocalTransparencyFx() {}

  bool doGetBBox(double frame, TRectD &bBox,
                 const TRenderSettings &info) override {
    if (m_src.isConnected())
      return m_src->doGetBBox(frame, bBox, info);
    else {
      bBox = TRectD();
      return false;
    }
  }

  void doCompute(TTile &tile, double frame,
                 const TRenderSettings &ri) override {
    if (!checkBeforeCompute(tile, frame, ri)) return;
    TTile srcTile;
    m_src->allocateAndCompute(srcTile, tile.m_pos, tile.getRaster()->getSize(),
                              tile.getRaster(), frame, ri);
    // TTile refTile = tile;
    m_ref->compute(tile, frame, ri);

    // TRaster32P ref32 (refTile.getRaster());
    TRaster32P out32(tile.getRaster());
    TRaster32P src32(srcTile.getRaster());
    if (out32 && src32)
      doLocalTransparency<TPixelRGBM32, TPixelGR8, UCHAR>(
          out32, src32, out32, m_value->getValue(frame) / 100.);
    else {
      // TRaster64P ref64(refTile.getRaster());
      TRaster64P out64(tile.getRaster());
      TRaster64P src64(srcTile.getRaster());
      if (out64 && src64)
        doLocalTransparency<TPixelRGBM64, TPixelGR16, USHORT>(
            out64, src64, out64, m_value->getValue(frame) / 100.);
      else
        throw TException("LocalTransparencyFx: unsupported raster type");
    }
  }

  bool checkBeforeCompute(TTile &tile, double frame,
                          const TRenderSettings &info) {
    Status status = getFxStatus(m_src, m_ref);
    if ((status & NoPortsConnected) == NoPortsConnected) return false;
    if ((status & OutOfTime) == OutOfTime) return false;
    if ((status & Port0NotConnected) == Port0NotConnected) return false;
    if ((status & Port1NotConnected) == Port1NotConnected) {
      m_src->compute(tile, frame, info);
      return false;
    }
    assert(status == StatusGood);
    return true;
  }

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

  int getMemoryRequirement(const TRectD &rect, double frame,
                           const TRenderSettings &info) override {
    return TRasterFx::memorySize(rect, info.m_bpp);
  }
};

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

FX_PLUGIN_IDENTIFIER(LocalTransparencyFx, "localTransparencyFx")