340 lines
10 KiB
C++
340 lines
10 KiB
C++
|
|
||
|
|
||
|
#include "toonzqt/rasterimagedata.h"
|
||
|
#include "toonzqt/strokesdata.h"
|
||
|
#include "tpaletteutil.h"
|
||
|
#include "trasterimage.h"
|
||
|
#include "toonz/stage.h"
|
||
|
#include "toonz/tcenterlinevectorizer.h"
|
||
|
#include "toonz/ttileset.h"
|
||
|
#include "toonz/toonzimageutils.h"
|
||
|
#include "toonz/toonzscene.h"
|
||
|
#include "toonz/sceneproperties.h"
|
||
|
#include "toonz/fullcolorpalette.h"
|
||
|
|
||
|
namespace
|
||
|
{
|
||
|
TVectorImageP vectorize(const TImageP &source, const TRectD &rect, const VectorizerConfiguration &config, TAffine transform)
|
||
|
{
|
||
|
VectorizerCore vc;
|
||
|
TVectorImageP vi = vc.vectorize(source.getPointer(), config, source->getPalette());
|
||
|
assert(vi);
|
||
|
vi->setPalette(source->getPalette());
|
||
|
|
||
|
double dpiX, dpiY;
|
||
|
TToonzImageP ti(source);
|
||
|
TRasterImageP ri(source);
|
||
|
if (ti)
|
||
|
ti->getDpi(dpiX, dpiY);
|
||
|
else if (ri)
|
||
|
ri->getDpi(dpiX, dpiY);
|
||
|
else
|
||
|
return vi;
|
||
|
|
||
|
TScale sc(dpiX / Stage::inch, dpiY / Stage::inch);
|
||
|
TTranslation tr(rect.getP00());
|
||
|
|
||
|
int i;
|
||
|
for (i = 0; i < (int)vi->getStrokeCount(); i++) {
|
||
|
TStroke *stroke = vi->getStroke(i);
|
||
|
stroke->transform((sc.inv() * tr) * transform, true);
|
||
|
}
|
||
|
return vi;
|
||
|
}
|
||
|
|
||
|
} //namspace
|
||
|
|
||
|
//===================================================================
|
||
|
// RasterImageData
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
RasterImageData::RasterImageData()
|
||
|
: m_dpiX(0), m_dpiY(0), m_transformation(), m_dim()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
RasterImageData::~RasterImageData()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//===================================================================
|
||
|
// ToonzImageData
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
ToonzImageData::ToonzImageData()
|
||
|
: RasterImageData(), m_copiedRaster(0), m_palette(new TPalette())
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
ToonzImageData::ToonzImageData(const ToonzImageData &src)
|
||
|
: m_copiedRaster(src.m_copiedRaster), m_palette(src.m_palette), m_usedStyles(src.m_usedStyles)
|
||
|
{
|
||
|
m_dpiX = src.m_dpiX;
|
||
|
m_dpiY = src.m_dpiY;
|
||
|
m_rects = src.m_rects;
|
||
|
m_strokes = src.m_strokes;
|
||
|
m_transformation = src.m_transformation;
|
||
|
m_originalStrokes = src.m_originalStrokes;
|
||
|
m_dim = src.m_dim;
|
||
|
assert(m_palette);
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
ToonzImageData::~ToonzImageData()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void ToonzImageData::setData(const TRasterP &copiedRaster, const TPaletteP &palette, double dpiX, double dpiY, const TDimension &dim,
|
||
|
const vector<TRectD> &rects, const vector<TStroke> &strokes,
|
||
|
const vector<TStroke> &originalStrokes,
|
||
|
const TAffine &transformation)
|
||
|
{
|
||
|
m_copiedRaster = copiedRaster;
|
||
|
m_palette = palette;
|
||
|
m_dpiX = dpiX;
|
||
|
m_dpiY = dpiY;
|
||
|
m_rects = rects;
|
||
|
m_strokes = strokes;
|
||
|
m_transformation = transformation;
|
||
|
m_originalStrokes = originalStrokes;
|
||
|
m_dim = dim;
|
||
|
|
||
|
/*-- 使用されているStyleの一覧を作る --*/
|
||
|
TToonzImageP ti(m_copiedRaster, m_copiedRaster->getBounds());
|
||
|
ToonzImageUtils::getUsedStyles(m_usedStyles, ti);
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void ToonzImageData::getData(TRasterP &copiedRaster, double &dpiX, double &dpiY,
|
||
|
vector<TRectD> &rects, vector<TStroke> &strokes, vector<TStroke> &originalStrokes,
|
||
|
TAffine &transformation, TPalette *targetPalette) const
|
||
|
{
|
||
|
if (!m_copiedRaster || (m_rects.empty() && m_strokes.empty()))
|
||
|
return;
|
||
|
copiedRaster = m_copiedRaster->clone();
|
||
|
dpiX = m_dpiX;
|
||
|
dpiY = m_dpiY;
|
||
|
assert(m_palette);
|
||
|
int i;
|
||
|
for (i = 0; i < (int)m_rects.size(); i++)
|
||
|
rects.push_back(m_rects[i]);
|
||
|
for (i = 0; i < (int)m_strokes.size(); i++)
|
||
|
strokes.push_back(m_strokes[i]);
|
||
|
for (i = 0; i < (int)m_originalStrokes.size(); i++)
|
||
|
originalStrokes.push_back(m_originalStrokes[i]);
|
||
|
|
||
|
transformation = m_transformation;
|
||
|
TRasterCM32P cmRas = copiedRaster;
|
||
|
|
||
|
if (!targetPalette)
|
||
|
targetPalette = new TPalette();
|
||
|
|
||
|
if (!cmRas)
|
||
|
return;
|
||
|
std::set<int> usedStyles(m_usedStyles);
|
||
|
TToonzImageP ti(cmRas, cmRas->getBounds());
|
||
|
if (usedStyles.size() == 0)
|
||
|
ToonzImageUtils::getUsedStyles(usedStyles, ti);
|
||
|
std::map<int, int> indexTable;
|
||
|
mergePalette(targetPalette, indexTable, m_palette, usedStyles);
|
||
|
ToonzImageUtils::scrambleStyles(ti, indexTable);
|
||
|
ti->setPalette(m_palette.getPointer());
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
StrokesData *ToonzImageData::toStrokesData(ToonzScene *scene) const
|
||
|
{
|
||
|
assert(scene);
|
||
|
TRectD rect;
|
||
|
if (!m_rects.empty())
|
||
|
rect = m_rects[0];
|
||
|
else if (!m_strokes.empty())
|
||
|
rect = m_strokes[0].getBBox();
|
||
|
unsigned int i;
|
||
|
for (i = 0; i < m_rects.size(); i++)
|
||
|
rect += m_rects[i];
|
||
|
for (i = 0; i < m_strokes.size(); i++)
|
||
|
rect += m_strokes[i].getBBox();
|
||
|
TToonzImageP image(m_copiedRaster, m_copiedRaster->getBounds());
|
||
|
image->setPalette(m_palette.getPointer());
|
||
|
image->setDpi(m_dpiX, m_dpiY);
|
||
|
|
||
|
const VectorizerParameters &vParams = *scene->getProperties()->getVectorizerParameters();
|
||
|
|
||
|
CenterlineConfiguration cConf = vParams.getCenterlineConfiguration(0.0);
|
||
|
NewOutlineConfiguration oConf = vParams.getOutlineConfiguration(0.0);
|
||
|
|
||
|
const VectorizerConfiguration &config = vParams.m_isOutline ? static_cast<const VectorizerConfiguration &>(oConf) : static_cast<const VectorizerConfiguration &>(cConf);
|
||
|
|
||
|
TVectorImageP vi = vectorize(image, rect, config, m_transformation);
|
||
|
|
||
|
StrokesData *sd = new StrokesData();
|
||
|
|
||
|
std::set<int> indexes;
|
||
|
for (i = 0; i < vi->getStrokeCount(); i++)
|
||
|
indexes.insert(i);
|
||
|
|
||
|
sd->setImage(vi, indexes);
|
||
|
return sd;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
int ToonzImageData::getMemorySize() const
|
||
|
{
|
||
|
int i, size = 0;
|
||
|
for (i = 0; i < (int)m_strokes.size(); i++)
|
||
|
size += m_strokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
|
||
|
for (i = 0; i < (int)m_originalStrokes.size(); i++)
|
||
|
size += m_originalStrokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
|
||
|
return size + sizeof(*(m_copiedRaster.getPointer())) +
|
||
|
sizeof(*(m_palette.getPointer())) +
|
||
|
sizeof(*this);
|
||
|
}
|
||
|
|
||
|
//===================================================================
|
||
|
// FullColorImageData
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
FullColorImageData::FullColorImageData()
|
||
|
: RasterImageData(), m_copiedRaster(0), m_palette(new TPalette())
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
FullColorImageData::FullColorImageData(const FullColorImageData &src)
|
||
|
: m_copiedRaster(src.m_copiedRaster), m_palette(src.m_palette)
|
||
|
{
|
||
|
m_dpiX = src.m_dpiX;
|
||
|
m_dpiY = src.m_dpiY;
|
||
|
m_rects = src.m_rects;
|
||
|
m_strokes = src.m_strokes;
|
||
|
m_transformation = src.m_transformation;
|
||
|
m_originalStrokes = src.m_originalStrokes;
|
||
|
m_dim = src.m_dim;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
FullColorImageData::~FullColorImageData()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void FullColorImageData::setData(const TRasterP &copiedRaster, const TPaletteP &palette, double dpiX, double dpiY, const TDimension &dim,
|
||
|
const vector<TRectD> &rects, const vector<TStroke> &strokes,
|
||
|
const vector<TStroke> &originalStrokes,
|
||
|
const TAffine &transformation)
|
||
|
{
|
||
|
m_copiedRaster = copiedRaster;
|
||
|
m_palette = palette;
|
||
|
m_dpiX = dpiX;
|
||
|
m_dpiY = dpiY;
|
||
|
m_rects = rects;
|
||
|
m_strokes = strokes;
|
||
|
m_originalStrokes = originalStrokes;
|
||
|
m_transformation = transformation;
|
||
|
m_dim = dim;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
void FullColorImageData::getData(TRasterP &copiedRaster, double &dpiX, double &dpiY,
|
||
|
vector<TRectD> &rects, vector<TStroke> &strokes, vector<TStroke> &originalStrokes,
|
||
|
TAffine &transformation, TPalette *targetPalette) const
|
||
|
{
|
||
|
if (!m_copiedRaster || (m_rects.empty() && m_strokes.empty()))
|
||
|
return;
|
||
|
copiedRaster = m_copiedRaster->clone();
|
||
|
dpiX = m_dpiX;
|
||
|
dpiY = m_dpiY;
|
||
|
int i;
|
||
|
for (i = 0; i < (int)m_rects.size(); i++)
|
||
|
rects.push_back(m_rects[i]);
|
||
|
for (i = 0; i < (int)m_strokes.size(); i++)
|
||
|
strokes.push_back(m_strokes[i]);
|
||
|
for (i = 0; i < (int)m_originalStrokes.size(); i++)
|
||
|
originalStrokes.push_back(m_originalStrokes[i]);
|
||
|
transformation = m_transformation;
|
||
|
|
||
|
TRasterP ras = copiedRaster;
|
||
|
if (!ras)
|
||
|
return;
|
||
|
if (!m_palette)
|
||
|
return;
|
||
|
if (!targetPalette)
|
||
|
targetPalette = new TPalette();
|
||
|
std::set<int> usedStyles;
|
||
|
TRasterImageP ri(ras);
|
||
|
|
||
|
for (i = 0; i < m_palette->getPageCount(); i++) {
|
||
|
TPalette::Page *page = m_palette->getPage(i);
|
||
|
int j;
|
||
|
for (j = 0; j < page->getStyleCount(); j++)
|
||
|
usedStyles.insert(page->getStyleId(j));
|
||
|
}
|
||
|
std::map<int, int> indexTable;
|
||
|
mergePalette(targetPalette, indexTable, m_palette, usedStyles);
|
||
|
ri->setPalette(m_palette.getPointer());
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
StrokesData *FullColorImageData::toStrokesData(ToonzScene *scene) const
|
||
|
{
|
||
|
assert(scene);
|
||
|
TRectD rect;
|
||
|
if (!m_rects.empty())
|
||
|
rect = m_rects[0];
|
||
|
else if (!m_strokes.empty())
|
||
|
rect = m_strokes[0].getBBox();
|
||
|
unsigned int i;
|
||
|
for (i = 0; i < m_rects.size(); i++)
|
||
|
rect += m_rects[i];
|
||
|
for (i = 0; i < m_strokes.size(); i++)
|
||
|
rect += m_strokes[i].getBBox();
|
||
|
TRasterImageP image(m_copiedRaster);
|
||
|
image->setPalette(FullColorPalette::instance()->getPalette(scene));
|
||
|
image->setDpi(m_dpiX, m_dpiY);
|
||
|
|
||
|
const VectorizerParameters *vParams = scene->getProperties()->getVectorizerParameters();
|
||
|
assert(vParams);
|
||
|
|
||
|
std::auto_ptr<VectorizerConfiguration> config(vParams->getCurrentConfiguration(0.0));
|
||
|
TVectorImageP vi = vectorize(image, rect, *config, m_transformation);
|
||
|
|
||
|
StrokesData *sd = new StrokesData();
|
||
|
|
||
|
std::set<int> indexes;
|
||
|
for (i = 0; i < vi->getStrokeCount(); i++)
|
||
|
indexes.insert(i);
|
||
|
|
||
|
sd->setImage(vi, indexes);
|
||
|
return sd;
|
||
|
}
|
||
|
|
||
|
//-------------------------------------------------------------------
|
||
|
|
||
|
int FullColorImageData::getMemorySize() const
|
||
|
{
|
||
|
int i, size = 0;
|
||
|
for (i = 0; i < (int)m_strokes.size(); i++)
|
||
|
size += m_strokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
|
||
|
for (i = 0; i < (int)m_originalStrokes.size(); i++)
|
||
|
size += m_originalStrokes[i].getControlPointCount() * sizeof(TThickPoint) + 100;
|
||
|
return size + sizeof(*(m_copiedRaster.getPointer())) +
|
||
|
sizeof(*(m_palette.getPointer())) +
|
||
|
sizeof(*this);
|
||
|
}
|