439 lines
13 KiB
C++
439 lines
13 KiB
C++
|
|
||
|
|
||
|
#include "toonz/trasterimageutils.h"
|
||
|
#include "toonz/ttileset.h"
|
||
|
#include "tstroke.h"
|
||
|
#include "tofflinegl.h"
|
||
|
#include "tpalette.h"
|
||
|
#include "tvectorrenderdata.h"
|
||
|
#include "ttzpimagefx.h"
|
||
|
#include "tvectorgl.h"
|
||
|
#include "trop.h"
|
||
|
#include "tregion.h"
|
||
|
#include "tcolorstyles.h"
|
||
|
|
||
|
#include <QPainter>
|
||
|
#include <QColor>
|
||
|
#include <QImage>
|
||
|
|
||
|
//--------------------------------------------------------------------------
|
||
|
|
||
|
namespace
|
||
|
{
|
||
|
QImage rasterToQImage(const TRasterP &ras, bool premultiplied, bool mirrored)
|
||
|
{
|
||
|
if (TRaster32P ras32 = ras) {
|
||
|
QImage image(ras->getRawData(), ras->getLx(), ras->getLy(),
|
||
|
premultiplied ? QImage::Format_ARGB32_Premultiplied : QImage::Format_ARGB32);
|
||
|
if (mirrored)
|
||
|
return image.mirrored();
|
||
|
return image;
|
||
|
} else if (TRasterGR8P ras8 = ras) {
|
||
|
QImage image(ras->getRawData(), ras->getLx(), ras->getLy(), ras->getWrap(), QImage::Format_Indexed8);
|
||
|
static QVector<QRgb> colorTable;
|
||
|
if (colorTable.size() == 0) {
|
||
|
int i;
|
||
|
for (i = 0; i < 256; i++)
|
||
|
colorTable.append(QColor(i, i, i).rgb());
|
||
|
}
|
||
|
image.setColorTable(colorTable);
|
||
|
if (mirrored)
|
||
|
return image.mirrored();
|
||
|
return image;
|
||
|
}
|
||
|
return QImage();
|
||
|
}
|
||
|
|
||
|
//--------------------------------------------------------------------------
|
||
|
|
||
|
void rasterizeWholeStroke(TOfflineGL *&gl, TStroke *stroke, TPalette *palette, bool doAnialias)
|
||
|
{
|
||
|
TRectD bbox = stroke->getBBox();
|
||
|
TRect rect = convert(bbox).enlarge(1);
|
||
|
gl = new TOfflineGL(rect.getSize());
|
||
|
gl->makeCurrent();
|
||
|
gl->clear(TPixel32(0, 0, 0, 0));
|
||
|
glEnable(GL_ALPHA_TEST);
|
||
|
glAlphaFunc(GL_GREATER, 0);
|
||
|
|
||
|
TPaletteP plt = palette->clone();
|
||
|
int styleId = stroke->getStyle();
|
||
|
TColorStyleP style = plt->getStyle(styleId);
|
||
|
TTranslation affine(-convert(rect.getP00()));
|
||
|
TVectorRenderData rd(affine, gl->getBounds(), plt.getPointer(), 0, true, true);
|
||
|
if (doAnialias)
|
||
|
tglDraw(rd, stroke);
|
||
|
else {
|
||
|
TStrokeProp *prop = stroke->getProp();
|
||
|
if (prop)
|
||
|
prop->getMutex()->lock();
|
||
|
|
||
|
if (!style->isStrokeStyle() || style->isEnabled() == false) {
|
||
|
if (prop)
|
||
|
prop->getMutex()->unlock();
|
||
|
|
||
|
prop = 0;
|
||
|
} else {
|
||
|
if (!prop || style.getPointer() != prop->getColorStyle()) {
|
||
|
if (prop)
|
||
|
prop->getMutex()->unlock();
|
||
|
|
||
|
stroke->setProp(style->makeStrokeProp(stroke));
|
||
|
prop = stroke->getProp();
|
||
|
if (prop)
|
||
|
prop->getMutex()->lock();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!prop)
|
||
|
return;
|
||
|
prop->getMutex()->lock();
|
||
|
glPushAttrib(GL_ALL_ATTRIB_BITS);
|
||
|
prop->draw(rd);
|
||
|
glPopAttrib();
|
||
|
}
|
||
|
glDisable(GL_ALPHA_TEST);
|
||
|
glFinish();
|
||
|
}
|
||
|
|
||
|
//--------------------------------------------------------------------------
|
||
|
|
||
|
TRect fastAddInkStroke(const TRasterImageP &ri, TStroke *stroke, TRectD clip, double opacity, bool doAntialiasing)
|
||
|
{
|
||
|
TOfflineGL *gl = 0;
|
||
|
TRectD bbox = stroke->getBBox();
|
||
|
TRect sBBox = convert(bbox).enlarge(1);
|
||
|
TRect rectRender = sBBox * ri->getRaster()->getBounds();
|
||
|
|
||
|
if (!rectRender.isEmpty()) {
|
||
|
if (opacity < 1.0) {
|
||
|
int styleId = stroke->getStyle();
|
||
|
TPalette *plt = ri->getPalette();
|
||
|
TPixel32 color = plt->getStyle(styleId)->getMainColor();
|
||
|
color.m = 255 * opacity;
|
||
|
TPaletteP newPlt(plt);
|
||
|
newPlt->getStyle(styleId)->setMainColor(color);
|
||
|
rasterizeWholeStroke(gl, stroke, newPlt.getPointer(), doAntialiasing);
|
||
|
} else
|
||
|
rasterizeWholeStroke(gl, stroke, ri->getPalette(), doAntialiasing);
|
||
|
TRect tmp = rectRender - sBBox.getP00();
|
||
|
TRaster32P glRas = gl->getRaster()->extract(tmp);
|
||
|
TRop::over(ri->getRaster(), glRas, rectRender.getP00());
|
||
|
delete gl;
|
||
|
}
|
||
|
return rectRender;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
TRect rasterizeRegion(TOfflineGL *&gl, TRect rasBounds, TRegion *region, TPalette *palette, TRectD clip)
|
||
|
{
|
||
|
TRectD regionBBox = region->getBBox();
|
||
|
if (!clip.isEmpty())
|
||
|
regionBBox = regionBBox * clip;
|
||
|
|
||
|
TRect rect = convert(regionBBox) * rasBounds;
|
||
|
if (!rect.isEmpty()) {
|
||
|
gl = new TOfflineGL(rect.getSize());
|
||
|
gl->makeCurrent();
|
||
|
gl->clear(TPixel32::Transparent);
|
||
|
glPushAttrib(GL_ALL_ATTRIB_BITS);
|
||
|
glEnable(GL_ALPHA_TEST);
|
||
|
glAlphaFunc(GL_GREATER, 0);
|
||
|
|
||
|
TTranslation affine(-convert(rect.getP00()));
|
||
|
TVectorRenderData rd(affine, gl->getBounds(), palette, 0, true, true);
|
||
|
|
||
|
tglDraw(rd, region);
|
||
|
|
||
|
glDisable(GL_ALPHA_TEST);
|
||
|
glPopAttrib();
|
||
|
|
||
|
glFinish();
|
||
|
}
|
||
|
return rect;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void fastAddPaintRegion(const TRasterImageP &ri, TRegion *region,
|
||
|
int newPaintId, int maxStyleId, TRectD clip = TRectD())
|
||
|
{
|
||
|
TRaster32P ras = ri->getRaster();
|
||
|
TOfflineGL *gl;
|
||
|
TRect rect = rasterizeRegion(gl, ras->getBounds(), region, ri->getPalette(), clip);
|
||
|
if (rect.isEmpty())
|
||
|
return;
|
||
|
|
||
|
TRaster32P glRas = gl->getRaster();
|
||
|
assert(TPixelCM32::getMaxTone() == 255);
|
||
|
TRop::over(ri->getRaster(), glRas, rect.getP00());
|
||
|
delete gl;
|
||
|
|
||
|
TRegion *subregion;
|
||
|
UINT i = 0;
|
||
|
for (; i < region->getSubregionCount(); ++i) {
|
||
|
subregion = region->getSubregion(i);
|
||
|
fastAddPaintRegion(ri, subregion, tmin(maxStyleId, subregion->getStyle()), maxStyleId);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//==========================================================================
|
||
|
|
||
|
TRect TRasterImageUtils::addStroke(const TRasterImageP &ri, TStroke *stroke, TRectD clip,
|
||
|
double opacity, bool doAntialiasing)
|
||
|
{
|
||
|
TStroke *s = new TStroke(*stroke);
|
||
|
TPoint riCenter = ri->getRaster()->getCenter();
|
||
|
s->transform(TTranslation(riCenter.x, riCenter.y));
|
||
|
TRect rect = fastAddInkStroke(ri, s, clip, opacity, doAntialiasing);
|
||
|
rect -= riCenter;
|
||
|
delete s;
|
||
|
return rect;
|
||
|
}
|
||
|
|
||
|
//==========================================================================
|
||
|
|
||
|
TRect TRasterImageUtils::convertWorldToRaster(const TRectD &area, const TRasterImageP ri)
|
||
|
{
|
||
|
if (area.isEmpty())
|
||
|
return TRect();
|
||
|
if (!ri || !ri->getRaster())
|
||
|
return TRect(tfloor(area.x0), tfloor(area.y0), tfloor(area.x1) - 1, tfloor(area.y1) - 1);
|
||
|
TRasterP ras = ri->getRaster();
|
||
|
TRectD rect(area + ras->getCenterD());
|
||
|
return TRect(tfloor(rect.x0), tfloor(rect.y0), tceil(rect.x1) - 1, tceil(rect.y1) - 1);
|
||
|
}
|
||
|
|
||
|
//==========================================================================
|
||
|
|
||
|
TRectD TRasterImageUtils::convertRasterToWorld(const TRect &area, const TRasterImageP ri)
|
||
|
{
|
||
|
if (area.isEmpty())
|
||
|
return TRectD();
|
||
|
|
||
|
TRectD rect(area.x0, area.y0, area.x1 + 1, area.y1 + 1);
|
||
|
if (ri && ri->getRaster())
|
||
|
rect = rect - ri->getRaster()->getCenterD();
|
||
|
return rect;
|
||
|
}
|
||
|
|
||
|
//==========================================================================
|
||
|
|
||
|
// DA RIFARE
|
||
|
// e' lenta da far schifo
|
||
|
|
||
|
//!Converts a TVectorImage into a TRasterImage. The input vector image
|
||
|
//!is transformed through the passed affine \b aff, and put into a
|
||
|
//!TRasterImage strictly covering the bounding box of the transformed
|
||
|
//!vector image. The output image has its lower-left position in the
|
||
|
//!world reference specified by the \b pos parameter, which is granted to
|
||
|
//!be an integer displacement of the passed value. Additional parameters
|
||
|
//!include an integer \b enlarge by which the output image is enlarged with
|
||
|
//!respect to the transformed image's bbox, and the bool \b transformThickness
|
||
|
//!to specify whether the transformation should involve strokes' thickensses
|
||
|
//!or not.
|
||
|
TRasterImageP TRasterImageUtils::vectorToFullColorImage(
|
||
|
const TVectorImageP &vimage, const TAffine &aff, TPalette *palette,
|
||
|
const TPointD &outputPos, const TDimension &outputSize,
|
||
|
const vector<TRasterFxRenderDataP> *fxs, bool transformThickness)
|
||
|
{
|
||
|
if (!vimage || !palette)
|
||
|
return 0;
|
||
|
|
||
|
//Transform the vector image through aff
|
||
|
TVectorImageP vi = vimage->clone();
|
||
|
vi->transform(aff, transformThickness);
|
||
|
|
||
|
//Allocate the output ToonzImage
|
||
|
TRaster32P raster(outputSize.lx, outputSize.ly);
|
||
|
raster->clear();
|
||
|
TRasterImageP ri(raster);
|
||
|
ri->setPalette(palette->clone());
|
||
|
|
||
|
//Shift outputPos to the origin
|
||
|
vi->transform(TTranslation(-outputPos));
|
||
|
|
||
|
int strokeCount = vi->getStrokeCount();
|
||
|
vector<int> strokeIndex(strokeCount);
|
||
|
vector<TStroke *> strokes(strokeCount);
|
||
|
int i;
|
||
|
for (i = 0; i < strokeCount; ++i) {
|
||
|
strokeIndex[i] = i;
|
||
|
strokes[i] = vi->getStroke(i);
|
||
|
}
|
||
|
vi->notifyChangedStrokes(strokeIndex, strokes);
|
||
|
|
||
|
int maxStyleId = palette->getStyleCount() - 1;
|
||
|
for (i = 0; i < (int)vi->getRegionCount(); ++i) {
|
||
|
TRegion *region = vi->getRegion(i);
|
||
|
fastAddPaintRegion(ri, region, tmin(maxStyleId, region->getStyle()), maxStyleId);
|
||
|
}
|
||
|
|
||
|
set<int> colors;
|
||
|
if (fxs) {
|
||
|
for (i = 0; i < (int)fxs->size(); i++) {
|
||
|
SandorFxRenderData *sandorData = dynamic_cast<SandorFxRenderData *>((*fxs)[i].getPointer());
|
||
|
if (sandorData && sandorData->m_type == BlendTz) {
|
||
|
string indexes = toString(sandorData->m_blendParams.m_colorIndex);
|
||
|
vector<string> items;
|
||
|
parseIndexes(indexes, items);
|
||
|
PaletteFilterFxRenderData paletteFilterData;
|
||
|
insertIndexes(items, &paletteFilterData);
|
||
|
colors = paletteFilterData.m_colors;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < strokeCount; ++i) {
|
||
|
TStroke *stroke = vi->getStroke(i);
|
||
|
|
||
|
bool visible = false;
|
||
|
int styleId = stroke->getStyle();
|
||
|
TColorStyleP style = palette->getStyle(styleId);
|
||
|
assert(style);
|
||
|
int colorCount = style->getColorParamCount();
|
||
|
if (colorCount == 0)
|
||
|
visible = true;
|
||
|
else {
|
||
|
visible = false;
|
||
|
for (int j = 0; j < style->getColorParamCount() && !visible; j++) {
|
||
|
TPixel32 color = style->getColorParamValue(j);
|
||
|
if (color.m != 0)
|
||
|
visible = true;
|
||
|
}
|
||
|
}
|
||
|
if (visible)
|
||
|
fastAddInkStroke(ri, stroke, TRectD(), 1, true);
|
||
|
}
|
||
|
return ri;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
TRect TRasterImageUtils::eraseRect(const TRasterImageP &ri, const TRectD &area)
|
||
|
{
|
||
|
TRasterP ras = ri->getRaster();
|
||
|
TRect rect = convertWorldToRaster(area, ri) * ras->getBounds();
|
||
|
if (rect.isEmpty())
|
||
|
return rect;
|
||
|
ras->lock();
|
||
|
TRasterP workRas = ras->extract(rect);
|
||
|
if (workRas->getPixelSize() == 4)
|
||
|
workRas->clear();
|
||
|
else {
|
||
|
TRasterGR8P rasGR8(workRas);
|
||
|
if (rasGR8)
|
||
|
rasGR8->fill(TPixelGR8::White);
|
||
|
}
|
||
|
ras->unlock();
|
||
|
return rect;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
vector<TRect> TRasterImageUtils::paste(const TRasterImageP &ri, const TTileSetFullColor *tileSet)
|
||
|
{
|
||
|
vector<TRect> rects;
|
||
|
TRasterP raster = ri->getRaster();
|
||
|
for (int i = 0; i < tileSet->getTileCount(); i++) {
|
||
|
const TTileSetFullColor::Tile *tile = tileSet->getTile(i);
|
||
|
TRasterP ras;
|
||
|
tile->getRaster(ras);
|
||
|
assert(ras);
|
||
|
raster->copy(ras, tile->m_rasterBounds.getP00());
|
||
|
rects.push_back(tile->m_rasterBounds);
|
||
|
}
|
||
|
return rects;
|
||
|
}
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void TRasterImageUtils::addSceneNumbering(const TRasterImageP &ri, int globalIndex,
|
||
|
const wstring &sceneName, int sceneIndex)
|
||
|
{
|
||
|
if (!ri)
|
||
|
return;
|
||
|
TRasterP raster = ri->getRaster();
|
||
|
int lx = raster->getLx(), ly = raster->getLy();
|
||
|
QColor greyOverlay(100, 100, 100, 140);
|
||
|
QImage image = rasterToQImage(raster, true, false);
|
||
|
QPainter p(&image);
|
||
|
QFont numberingFont = QFont();
|
||
|
numberingFont.setPixelSize(ly * 0.04);
|
||
|
numberingFont.setBold(true);
|
||
|
p.setFont(numberingFont);
|
||
|
QMatrix matrix;
|
||
|
p.setMatrix(matrix.translate(0, ly).scale(1, -1), true);
|
||
|
QFontMetrics fm = p.fontMetrics();
|
||
|
int fontHeight = fm.height();
|
||
|
int offset = fontHeight * 0.2;
|
||
|
|
||
|
//write the scenename and the scene frame
|
||
|
QString sceneFrame = QString::number(sceneIndex);
|
||
|
while (sceneFrame.size() < 4)
|
||
|
sceneFrame.push_front("0");
|
||
|
QString sceneNumberingString = QString::fromStdWString(sceneName) + ": " + sceneFrame;
|
||
|
|
||
|
int sceneNumberingWidth = fm.width(sceneNumberingString);
|
||
|
p.setPen(Qt::NoPen);
|
||
|
p.setBrush(QColor(255, 255, 255, 255));
|
||
|
p.drawRect(offset, ly - offset - fontHeight, sceneNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setBrush(greyOverlay);
|
||
|
p.drawRect(offset, ly - offset - fontHeight, sceneNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setPen(Qt::white);
|
||
|
p.drawText(2 * offset, ly - 2 * offset, sceneNumberingString);
|
||
|
|
||
|
//write the global frame
|
||
|
QString globalFrame = QString::number(globalIndex);
|
||
|
while (globalFrame.size() < 4)
|
||
|
globalFrame.push_front("0");
|
||
|
|
||
|
int gloablNumberingWidth = fm.width(globalFrame);
|
||
|
p.setPen(Qt::NoPen);
|
||
|
p.setBrush(QColor(255, 255, 255, 255));
|
||
|
p.drawRect(lx - 3 * offset - gloablNumberingWidth, ly - offset - fontHeight, gloablNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setBrush(greyOverlay);
|
||
|
p.drawRect(lx - 3 * offset - gloablNumberingWidth, ly - offset - fontHeight, gloablNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setPen(Qt::white);
|
||
|
p.drawText(lx - 2 * offset - gloablNumberingWidth, ly - 2 * offset, globalFrame);
|
||
|
p.end();
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void TRasterImageUtils::addGlobalNumbering(const TRasterImageP &ri, const wstring &sceneName, int globalIndex)
|
||
|
{
|
||
|
if (!ri)
|
||
|
return;
|
||
|
TRasterP raster = ri->getRaster();
|
||
|
int lx = raster->getLx(), ly = raster->getLy();
|
||
|
QColor greyOverlay(100, 100, 100, 140);
|
||
|
QImage image = rasterToQImage(raster, true, false);
|
||
|
QPainter p(&image);
|
||
|
QFont numberingFont = QFont();
|
||
|
numberingFont.setPixelSize(ly * 0.04);
|
||
|
numberingFont.setBold(true);
|
||
|
p.setFont(numberingFont);
|
||
|
QMatrix matrix;
|
||
|
p.setMatrix(matrix.translate(0, ly).scale(1, -1), true);
|
||
|
QFontMetrics fm = p.fontMetrics();
|
||
|
int fontHeight = fm.height();
|
||
|
int offset = fontHeight * 0.2;
|
||
|
QString globalFrame = QString::number(globalIndex);
|
||
|
while (globalFrame.size() < 4)
|
||
|
globalFrame.push_front("0");
|
||
|
QString globalNumberingString = QString::fromStdWString(sceneName) + ": " + globalFrame;
|
||
|
|
||
|
int globalNumberingWidth = fm.width(globalNumberingString);
|
||
|
p.setPen(Qt::NoPen);
|
||
|
p.setBrush(QColor(255, 255, 255, 255));
|
||
|
p.drawRect(offset, ly - offset - fontHeight, globalNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setBrush(greyOverlay);
|
||
|
p.drawRect(offset, ly - offset - fontHeight, globalNumberingWidth + offset * 2, fontHeight);
|
||
|
p.setPen(Qt::white);
|
||
|
p.drawText(2 * offset, ly - 2 * offset, globalNumberingString);
|
||
|
p.end();
|
||
|
}
|