tahoma2d/toonz/sources/common/trop/tinvert.cpp
2016-06-15 15:43:10 +09:00

113 lines
2.9 KiB
C++

#include "trop.h"
#include "tpixelgr.h"
namespace {
template <typename PixType>
inline void do_invert(TRasterPT<PixType> ras) {
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>
inline void do_invert(TRasterPT<PixType> ras, bool invRed, bool invGreen,
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) {
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;
}
}
}
//------------------------------------------------------------------------------
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;
ras->lock();
if (ras32)
if (flag)
do_invert<TPixel32>(ras32);
else
do_invert<TPixel32>(ras, invRed, invGreen, invBlue, invMatte);
else {
TRaster64P ras64 = ras;
if (ras64)
if (flag)
do_invert<TPixel64>(ras64);
else
do_invert<TPixel64>(ras64, invRed, invGreen, invBlue, invMatte);
else {
TRasterGR8P ras8 = ras;
if (ras8)
do_invert<TPixelGR8>(ras8);
else {
ras->unlock();
throw TRopException("unsupported pixel type");
}
}
}
ras->unlock();
}
//------------------------------------------------------------------------------