#include "trop.h"

#include "tpixelgr.h"

namespace {

template <typename pixtype=""></typename>

inline void do_invert(TRasterPT<pixtype> ras) {</pixtype>

  int wrap         = ras->getWrap();

  int lx           = ras->getLx();

  PixType *rowIn   = ras->pixels();

  PixType *lastPix = rowIn + wrap * ras->getLy();

  PixType *pixIn   = 0;

  PixType *endPix  = 0;

  while (pixIn < lastPix) {

    pixIn  = rowIn;

    endPix = pixIn + lx;

    while (pixIn < endPix) {

      pixIn->r = pixIn->m - pixIn->r;

      pixIn->g = pixIn->m - pixIn->g;

      pixIn->b = pixIn->m - pixIn->b; /*pixIn->m = pixIn->m;*/

      ++pixIn;

    }

    rowIn += wrap;

  }

}

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

template <typename pixtype=""></typename>

inline void do_invert(TRasterPT<pixtype> ras, bool invRed, bool invGreen,</pixtype>

                      bool invBlue, bool invMatte) {

  int wrap         = ras->getWrap();

  int lx           = ras->getLx();

  PixType *rowIn   = ras->pixels();

  PixType *lastPix = rowIn + wrap * ras->getLy();

  PixType *pixIn   = 0;

  PixType *endPix  = 0;

  while (pixIn < lastPix) {

    pixIn  = rowIn;

    endPix = pixIn + lx;

    while (pixIn < endPix) {

      if (invRed) pixIn->r = pixIn->m - pixIn->r;

      if (invGreen) pixIn->g = pixIn->m - pixIn->g;

      if (invBlue) pixIn->b = pixIn->m - pixIn->b;

      if (invMatte) pixIn->m = ~pixIn->m;

      ++pixIn;

    }

    rowIn += wrap;

  }

}

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

template <>

inline void do_invert<tpixelgr8>(TRasterPT<tpixelgr8> ras) {</tpixelgr8></tpixelgr8>

  int wrap           = ras->getWrap();

  int lx             = ras->getLx();

  TPixelGR8 *rowIn   = ras->pixels();

  TPixelGR8 *lastPix = rowIn + wrap * ras->getLy();

  TPixelGR8 *pixIn   = 0;

  TPixelGR8 *endPix  = 0;

  while (pixIn < lastPix) {

    pixIn  = rowIn;

    endPix = pixIn + lx;

    while (pixIn < endPix) {

      pixIn->value = 255 - pixIn->value;

      ++pixIn;

    }

    rowIn += wrap;

  }

}

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

template <>

inline void do_invert<tpixelf>(TRasterFP ras, bool invRed, bool invGreen,</tpixelf>

                               bool invBlue, bool invMatte) {

  int wrap         = ras->getWrap();

  int lx           = ras->getLx();

  TPixelF *rowIn   = ras->pixels();

  TPixelF *lastPix = rowIn + wrap * ras->getLy();

  TPixelF *pixIn   = 0;

  TPixelF *endPix  = 0;

  while (pixIn < lastPix) {

    pixIn  = rowIn;

    endPix = pixIn + lx;

    while (pixIn < endPix) {

      if (invRed) pixIn->r = pixIn->m - pixIn->r;

      if (invGreen) pixIn->g = pixIn->m - pixIn->g;

      if (invBlue) pixIn->b = pixIn->m - pixIn->b;

      if (invMatte) pixIn->m = 1.f - pixIn->m;

      ++pixIn;

    }

    rowIn += wrap;

  }

}

}  // namespace

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

void TRop::invert(TRasterP ras, bool invRed, bool invGreen, bool invBlue,

                  bool invMatte) {

  if (!invRed && !invGreen && !invBlue) return;

  bool flag = invRed && invGreen && invBlue && !invMatte;

  TRaster32P ras32 = ras;

  TRaster64P ras64 = ras;

  TRasterFP rasF   = ras;

  TRasterGR8P ras8 = ras;

  ras->lock();

  if (ras32) {

    if (flag)

      do_invert<tpixel32>(ras32);</tpixel32>

    else

      do_invert<tpixel32>(ras, invRed, invGreen, invBlue, invMatte);</tpixel32>

  } else if (ras64) {

    if (flag)

      do_invert<tpixel64>(ras64);</tpixel64>

    else

      do_invert<tpixel64>(ras64, invRed, invGreen, invBlue, invMatte);</tpixel64>

  } else if (rasF) {

    if (flag)

      do_invert<tpixelf>(rasF);</tpixelf>

    else

      do_invert<tpixelf>(rasF, invRed, invGreen, invBlue, invMatte);</tpixelf>

  } else if (ras8)

    do_invert<tpixelgr8>(ras8);</tpixelgr8>

  else {

    ras->unlock();

    throw TRopException("unsupported pixel type");

  }

  ras->unlock();

}

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