190 lines
5.5 KiB
C++
190 lines
5.5 KiB
C++
|
|
||
|
|
||
|
#include "toonzqt/strokesdata.h"
|
||
|
#include "tthreadmessage.h"
|
||
|
#include "tstroke.h"
|
||
|
#include "tpalette.h"
|
||
|
#include "toonzqt/rasterimagedata.h"
|
||
|
#include "toonz/toonzimageutils.h"
|
||
|
#include "toonz/trasterimageutils.h"
|
||
|
#include "toonz/stage.h"
|
||
|
|
||
|
using namespace std;
|
||
|
|
||
|
//=============================================================================
|
||
|
namespace
|
||
|
{
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
int findStroke(const TVectorImageP &img, TStroke *stroke, const TAffine &aff)
|
||
|
{
|
||
|
TRectD strokeBBox = aff * stroke->getBBox();
|
||
|
int count = img->getStrokeCount();
|
||
|
for (int i = 0; i < count; i++) {
|
||
|
TStroke *s = img->getStroke(i);
|
||
|
TRectD bbox = s->getBBox();
|
||
|
if (tdistance2(bbox.getP00(), strokeBBox.getP00()) +
|
||
|
tdistance2(bbox.getP11(), strokeBBox.getP11()) >
|
||
|
0.001)
|
||
|
continue;
|
||
|
return i;
|
||
|
}
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
TAffine findOffset(const TVectorImageP &srcImg, const TVectorImageP &img)
|
||
|
{
|
||
|
TAffine offset;
|
||
|
|
||
|
TVectorImageP tarImg = img;
|
||
|
if (!tarImg)
|
||
|
return offset;
|
||
|
if (tarImg->getStrokeCount() == 0 || srcImg->getStrokeCount() == 0)
|
||
|
return offset;
|
||
|
bool done = false;
|
||
|
int i;
|
||
|
while (!done)
|
||
|
for (i = 0; i < (int)srcImg->getStrokeCount(); i++) {
|
||
|
TStroke *stroke = srcImg->getStroke(i);
|
||
|
assert(stroke);
|
||
|
if (findStroke(tarImg, stroke, offset) >= 0) {
|
||
|
offset = offset * TTranslation(10, -10);
|
||
|
break;
|
||
|
}
|
||
|
done = true;
|
||
|
}
|
||
|
return offset;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
TStroke getStrokeByRect(TRectD r)
|
||
|
{
|
||
|
TStroke stroke;
|
||
|
if (r.isEmpty())
|
||
|
return stroke;
|
||
|
vector<TThickPoint> points;
|
||
|
points.push_back(r.getP00());
|
||
|
points.push_back((r.getP00() + r.getP01()) * 0.5);
|
||
|
points.push_back(r.getP01());
|
||
|
points.push_back((r.getP01() + r.getP11()) * 0.5);
|
||
|
points.push_back(r.getP11());
|
||
|
points.push_back((r.getP11() + r.getP10()) * 0.5);
|
||
|
points.push_back(r.getP10());
|
||
|
points.push_back((r.getP10() + r.getP00()) * 0.5);
|
||
|
points.push_back(r.getP00());
|
||
|
stroke.reshape(&(points[0]), points.size());
|
||
|
stroke.setSelfLoop(true);
|
||
|
return stroke;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
} // namespace
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
//=============================================================================
|
||
|
// StrokesData
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
void StrokesData::setImage(TVectorImageP image, const std::set<int> &indices)
|
||
|
{
|
||
|
if (!image)
|
||
|
return;
|
||
|
if (indices.empty())
|
||
|
return;
|
||
|
|
||
|
// indices e' un set; splitImage si aspetta un vector
|
||
|
vector<int> indicesV(indices.begin(), indices.end());
|
||
|
QMutexLocker lock(image->getMutex());
|
||
|
m_image = image->splitImage(indicesV, false);
|
||
|
if (m_image->getPalette() == 0) {
|
||
|
// nel caso lo stroke sia un path (e quindi senza palette)
|
||
|
m_image->setPalette(new TPalette());
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
void StrokesData::getImage(TVectorImageP image, std::set<int> &indices, bool insert) const
|
||
|
{
|
||
|
if (!m_image)
|
||
|
return;
|
||
|
|
||
|
TVectorImageP srcImg = m_image;
|
||
|
|
||
|
QMutexLocker lock(image->getMutex());
|
||
|
if (insert) {
|
||
|
TAffine offset = findOffset(srcImg, image);
|
||
|
UINT oldImageSize = image->getStrokeCount();
|
||
|
|
||
|
image->mergeImage(srcImg, offset, false);
|
||
|
UINT newImageSize = image->getStrokeCount();
|
||
|
indices.clear();
|
||
|
for (UINT sI = oldImageSize; sI < newImageSize; sI++)
|
||
|
indices.insert(sI);
|
||
|
} else {
|
||
|
std::vector<int> indicesToInsert(indices.begin(), indices.end());
|
||
|
if (indicesToInsert.empty())
|
||
|
return;
|
||
|
image->insertImage(srcImg, indicesToInsert);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
ToonzImageData *StrokesData::toToonzImageData(const TToonzImageP &imageToPaste) const
|
||
|
{
|
||
|
double dpix, dpiy;
|
||
|
imageToPaste->getDpi(dpix, dpiy);
|
||
|
assert(dpix != 0 && dpiy != 0);
|
||
|
TScale sc(dpix / Stage::inch, dpiy / Stage::inch);
|
||
|
|
||
|
TRectD bbox = sc * m_image->getBBox();
|
||
|
bbox.x0 = tfloor(bbox.x0);
|
||
|
bbox.y0 = tfloor(bbox.y0);
|
||
|
bbox.x1 = tceil(bbox.x1);
|
||
|
bbox.y1 = tceil(bbox.y1);
|
||
|
TDimension size(bbox.getLx(), bbox.getLy());
|
||
|
TToonzImageP app = ToonzImageUtils::vectorToToonzImage(
|
||
|
m_image, sc, m_image->getPalette(), bbox.getP00(), size, 0, true);
|
||
|
|
||
|
vector<TRectD> rects;
|
||
|
vector<TStroke> strokes;
|
||
|
TStroke stroke = getStrokeByRect(bbox);
|
||
|
strokes.push_back(stroke);
|
||
|
ToonzImageData *data = new ToonzImageData();
|
||
|
data->setData(app->getRaster(), m_image->getPalette(), dpix, dpiy, TDimension(), rects, strokes, strokes, TAffine());
|
||
|
return data;
|
||
|
}
|
||
|
|
||
|
//-----------------------------------------------------------------------------
|
||
|
|
||
|
FullColorImageData *StrokesData::toFullColorImageData(const TRasterImageP &imageToPaste) const
|
||
|
{
|
||
|
double dpix, dpiy;
|
||
|
imageToPaste->getDpi(dpix, dpiy);
|
||
|
assert(dpix != 0 && dpiy != 0);
|
||
|
TScale sc(dpix / Stage::inch, dpiy / Stage::inch);
|
||
|
|
||
|
TRectD bbox = sc * m_image->getBBox();
|
||
|
bbox.x0 = tfloor(bbox.x0);
|
||
|
bbox.y0 = tfloor(bbox.y0);
|
||
|
bbox.x1 = tceil(bbox.x1);
|
||
|
bbox.y1 = tceil(bbox.y1);
|
||
|
TDimension size(bbox.getLx(), bbox.getLy());
|
||
|
TRasterImageP app = TRasterImageUtils::vectorToFullColorImage(
|
||
|
m_image, sc, m_image->getPalette(), bbox.getP00(), size, 0, true);
|
||
|
|
||
|
vector<TRectD> rects;
|
||
|
vector<TStroke> strokes;
|
||
|
TStroke stroke = getStrokeByRect(bbox);
|
||
|
strokes.push_back(stroke);
|
||
|
FullColorImageData *data = new FullColorImageData();
|
||
|
data->setData(app->getRaster(), m_image->getPalette(), dpix, dpiy, TDimension(), rects, strokes, strokes, TAffine());
|
||
|
return data;
|
||
|
}
|