473 lines
14 KiB
C++
473 lines
14 KiB
C++
|
|
|
|
// TnzCore includes
|
|
#include "trop.h"
|
|
#include "trop_borders.h"
|
|
#include "tpixelutils.h"
|
|
|
|
// tcg includes
|
|
#include "tcg_wrap.h"
|
|
|
|
#include "tcg/tcg_point.h"
|
|
#include "tcg/tcg_cyclic.h"
|
|
#include "tcg/tcg_containers_reader.h"
|
|
|
|
#define INCLUDE_HPP
|
|
#include "tcg/tcg_triangulate.h"
|
|
#undef INCLUDE_HPP
|
|
|
|
// TnzExt includes
|
|
#define INCLUDE_HPP
|
|
#include "../common/trop/raster_edge_evaluator.h"
|
|
#undef INCLUDE_HPP
|
|
|
|
#include "ext/meshbuilder.h"
|
|
|
|
//**************************************************************************************
|
|
// Local namespace
|
|
//**************************************************************************************
|
|
|
|
namespace
|
|
{
|
|
|
|
struct PolygonVertex {
|
|
double m_pos[3];
|
|
int m_idx;
|
|
|
|
PolygonVertex(const TPoint &p)
|
|
: m_idx(-1) { m_pos[0] = p.x, m_pos[1] = p.y, m_pos[2] = 0.0; }
|
|
};
|
|
|
|
} // namespace
|
|
|
|
//**************************************************************************************
|
|
// tcg stuff
|
|
//**************************************************************************************
|
|
|
|
namespace tcg
|
|
{
|
|
|
|
template <>
|
|
struct traits<TTextureMeshP> {
|
|
typedef TTextureMeshP *pointer_type;
|
|
typedef TTextureMeshP *&reference_type;
|
|
typedef TTextureMesh pointed_type;
|
|
};
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
template <>
|
|
struct point_traits<PolygonVertex> {
|
|
typedef PolygonVertex point_type;
|
|
typedef double value_type;
|
|
typedef double float_type;
|
|
|
|
inline static value_type x(const point_type &p) { return p.m_pos[0]; }
|
|
inline static value_type y(const point_type &p) { return p.m_pos[1]; }
|
|
};
|
|
|
|
} // namespace tcg
|
|
|
|
//**************************************************************************************
|
|
// MeshBuilder Locals
|
|
//**************************************************************************************
|
|
|
|
namespace
|
|
{
|
|
|
|
//======================================================================================
|
|
// Thresholding stuff
|
|
//======================================================================================
|
|
|
|
template <typename Pix>
|
|
void thresholdRaster(const TRasterPT<Pix> &ras, TRasterGR8P &out, const Pix &transp)
|
|
{
|
|
int lx = ras->getLx(), y, ly = ras->getLy();
|
|
for (y = 0; y < ly; ++y) {
|
|
Pix *pix, *line = ras->pixels(y), *lineEnd = line + lx;
|
|
TPixelGR8 *gr, *grLine = out->pixels(y);
|
|
|
|
for (pix = line, gr = grLine; pix != lineEnd; ++pix, ++gr)
|
|
gr->value = (pix->m && *pix != transp) ? 0 : 255;
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
template <typename Pix>
|
|
void thresholdRasterGr(const TRasterPT<Pix> &ras, TRasterGR8P &out, const Pix &transp)
|
|
{
|
|
int lx = ras->getLx(), y, ly = ras->getLy();
|
|
for (y = 0; y < ly; ++y) {
|
|
Pix *pix, *line = ras->pixels(y), *lineEnd = line + lx;
|
|
TPixelGR8 *gr, *grLine = out->pixels(y);
|
|
|
|
for (pix = line, gr = grLine; pix != lineEnd; ++pix, ++gr)
|
|
gr->value = (*pix != transp) ? 0 : 255;
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
void thresholdRasterCM32(const TRasterCM32P &ras, TRasterGR8P &out)
|
|
{
|
|
int lx = ras->getLx(), y, ly = ras->getLy();
|
|
for (y = 0; y < ly; ++y) {
|
|
TPixelCM32 *pix, *line = ras->pixels(y), *lineEnd = line + lx;
|
|
TPixelGR8 *gr, *grLine = out->pixels(y);
|
|
|
|
for (pix = line, gr = grLine; pix != lineEnd; ++pix, ++gr)
|
|
gr->value = (pix->isPurePaint() && !pix->getPaint()) ? 255 : 0;
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
TRasterGR8P thresholdRaster(const TRasterP &ras, const MeshBuilderOptions &opts)
|
|
{
|
|
TRasterGR8P binaryRas(ras->getLx(), ras->getLy());
|
|
|
|
TRasterCM32P rasCM(ras);
|
|
if (rasCM)
|
|
thresholdRasterCM32(rasCM, binaryRas);
|
|
else
|
|
switch (ras->getPixelSize()) {
|
|
case 1: {
|
|
TRasterGR8P rasGR8(ras);
|
|
thresholdRasterGr(rasGR8, binaryRas, TPixelGR8::from(toPixel32(opts.m_transparentColor)));
|
|
break;
|
|
}
|
|
|
|
case 2:
|
|
{
|
|
TRasterGR16P rasGR16(ras);
|
|
thresholdRasterGr(rasGR16, binaryRas, TPixelGR16::from(opts.m_transparentColor));
|
|
break;
|
|
}
|
|
|
|
case 4:
|
|
{
|
|
TRaster32P ras32(ras);
|
|
thresholdRaster(ras32, binaryRas, toPixel32(opts.m_transparentColor));
|
|
break;
|
|
}
|
|
|
|
case 8:
|
|
{
|
|
TRaster64P ras64(ras);
|
|
thresholdRaster(ras64, binaryRas, opts.m_transparentColor);
|
|
break;
|
|
}
|
|
|
|
default:
|
|
assert(false);
|
|
}
|
|
|
|
// Build an enlarged ras to preserve borders. 5 pixels should be fine.
|
|
TRasterGR8P result(ras->getLx(), ras->getLy());
|
|
TRop::blur(result, binaryRas, opts.m_margin, 0, 0);
|
|
|
|
thresholdRasterGr(result, result, TPixelGR8::White);
|
|
|
|
return result;
|
|
}
|
|
|
|
//======================================================================================
|
|
// Borders Extraction stuff
|
|
//======================================================================================
|
|
|
|
using namespace TRop::borders;
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
template <typename T>
|
|
inline void delete_(T t) { delete t; }
|
|
|
|
template <typename T>
|
|
struct Vector : public std::vector<T> {
|
|
|
|
Vector() : std::vector<T>() {}
|
|
~Vector() { std::for_each(this->begin(), this->end(), delete_<T>); }
|
|
};
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
typedef std::vector<TPoint> RasterBorder;
|
|
typedef std::vector<PolygonVertex> Polygon;
|
|
typedef Vector<Polygon *> Family;
|
|
typedef Vector<Family *> Tribe;
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
struct PolygonReader {
|
|
Polygon *m_polygon;
|
|
|
|
public:
|
|
typedef tcg::cyclic_iterator<RasterBorder::iterator> cyclic_iter;
|
|
|
|
public:
|
|
PolygonReader() : m_polygon(0) {}
|
|
|
|
void openContainer(const cyclic_iter &ct)
|
|
{
|
|
m_polygon = new Polygon;
|
|
m_polygon->push_back(PolygonVertex(*ct));
|
|
}
|
|
|
|
void addElement(const cyclic_iter &ct) { m_polygon->push_back(PolygonVertex(*ct)); }
|
|
|
|
void closeContainer() {}
|
|
};
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
class BordersReader : public ImageMeshesReaderT<TPixelGR8>
|
|
{
|
|
public:
|
|
Vector<RasterBorder *> m_borders;
|
|
RasterBorder *m_current;
|
|
|
|
public:
|
|
BordersReader()
|
|
: ImageMeshesReaderT<TPixelGR8>(PixelSelector<TPixelGR8>(false)), m_current(0) {}
|
|
|
|
void openFace(ImageMesh *mesh, int faceIdx, const TPixelGR8 &color)
|
|
{
|
|
ImageMeshesReader::openFace(mesh, faceIdx); // defines imageIndex
|
|
|
|
if (mesh) {
|
|
ImageMesh::face_type &fc = mesh->face(faceIdx);
|
|
fc.imageIndex() = (color.value) ? 0 : 1; // redefines iI
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
void openEdge(const raster_edge_iterator &it)
|
|
{
|
|
m_current = new RasterBorder;
|
|
m_current->push_back(it.pos());
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
void addVertex(const raster_edge_iterator &it)
|
|
{
|
|
m_current->push_back(it.pos());
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
void closeEdge(ImageMesh *mesh, int edgeIdx)
|
|
{
|
|
ImageMesh::edge_type &ed = mesh->edge(edgeIdx);
|
|
ed.imageIndex() = m_borders.size();
|
|
|
|
m_borders.push_back(m_current);
|
|
m_current = 0;
|
|
|
|
ImageMeshesReader::closeEdge(mesh, edgeIdx);
|
|
}
|
|
};
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
Polygon *reduceBorder(RasterBorder *border)
|
|
{
|
|
typedef RasterBorder::iterator iter;
|
|
typedef tcg::cyclic_iterator<iter> cyclic_iter;
|
|
|
|
iter b(border->begin()), e(border->end());
|
|
|
|
cyclic_iter cBegin(b, b, e - 1, 0),
|
|
cEnd(b + 1, b, e - 1, 1);
|
|
|
|
RasterEdgeEvaluator<cyclic_iter> eval(
|
|
cBegin - 1, cEnd + 1, 2.0, (std::numeric_limits<double>::max)());
|
|
|
|
PolygonReader reader;
|
|
tcg::sequence_ops::minimalPath(cBegin, cEnd, eval, reader);
|
|
|
|
return reader.m_polygon;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
void reduceBorders(Tribe *tribe, const ImageMeshesReader &reader, const Vector<RasterBorder *> &borders,
|
|
int meshIdx, const ImageMesh::face_type &fc)
|
|
{
|
|
// Traverse the image structure. Each time a black face is encountered, add its associated family to the
|
|
// resulting tribe.
|
|
|
|
const tcg::list<ImageMeshP> &meshes = reader.meshes();
|
|
|
|
if (meshIdx >= 0 && fc.imageIndex()) {
|
|
// Build a family. Start by extracting the face's contour
|
|
Family *family = new Family;
|
|
tribe->push_back(family);
|
|
|
|
const ImageMeshP &mesh = meshes[meshIdx];
|
|
|
|
Polygon *outerBorder = reduceBorder(borders[mesh->edge(0).imageIndex()]);
|
|
family->push_back(outerBorder);
|
|
|
|
// Then, extract the contours of every sub-mesh
|
|
int m, mCount = fc.meshesCount();
|
|
for (m = 0; m < mCount; ++m) {
|
|
int mIdx = fc.mesh(m);
|
|
const ImageMeshP &childMesh = meshes[mIdx];
|
|
|
|
Polygon *innerBorder = reduceBorder(borders[childMesh->edge(0).imageIndex()]);
|
|
family->push_back(innerBorder);
|
|
|
|
reduceBorders(tribe, reader, borders, mIdx, childMesh->face(0));
|
|
}
|
|
}
|
|
|
|
// Recursive on the face's sub-mesh faces
|
|
int m, mCount = fc.meshesCount();
|
|
for (m = 0; m < mCount; ++m) {
|
|
int mIdx = fc.mesh(m);
|
|
reduceBorders(tribe, reader, borders, mIdx, meshes[mIdx]->face(0));
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------
|
|
|
|
Tribe *extractBorders(const TRasterGR8P &ras)
|
|
{
|
|
Tribe *result = new Tribe;
|
|
|
|
BordersReader reader;
|
|
TRop::borders::readMeshes(ras, reader);
|
|
|
|
reduceBorders(result, reader, reader.m_borders, -1, reader.outerFace());
|
|
|
|
return result;
|
|
}
|
|
|
|
//======================================================================================
|
|
// Maximum Edge Length stuff
|
|
//======================================================================================
|
|
|
|
double buildMinimumEdgeLength(Tribe *tribe, int targetMaxVerticesCount)
|
|
{
|
|
// Calculate the tribe's total area
|
|
double area = 0.0;
|
|
|
|
Tribe::iterator ft, fEnd(tribe->end());
|
|
for (ft = tribe->begin(); ft != fEnd; ++ft) {
|
|
Family *fam = *ft;
|
|
|
|
// Add area corresponding to the external border
|
|
area += fabs(tcg::polyline_ops::area(fam->front()->begin(), fam->front()->end()));
|
|
|
|
// Then, subtract all the internal ones
|
|
Family::iterator pt, pEnd(fam->end());
|
|
for (pt = ++fam->begin(); pt != pEnd; ++pt)
|
|
area -= fabs(tcg::polyline_ops::area((*pt)->begin(), (*pt)->end()));
|
|
}
|
|
|
|
// Given the area, find the approximate edge length corresponding to the required
|
|
// vertices count
|
|
|
|
// The idea is: given a polygon, its uniform triangular mesh - made of, say, lots of equilateral
|
|
// triangles, has vertices count proportional to ( polygon area / sq(mean mesh edge length) ).
|
|
|
|
// Under this assumption, take an equilater triangle as our polygon, and assume that its built
|
|
// mesh is the regular mesh obtained by subdividing it multiple times, Sierpinsky-like.
|
|
|
|
// Since it can easily be sheared to a corresponding half-square, it can be SEEN that in
|
|
// this case its vertices count is EXACTLY (l / e_length)^2 / 2, l being the large
|
|
// triangle's edge. By extension, if A is the area of the triangle, then we have:
|
|
|
|
// A = l^2 * sqrt(3/2) / 2; => l = sqrt(2 * K * A), K = 1.0 / sqrt(3/2);
|
|
// V = K * A / e_length^2; => e_length = sqrt(KA / V);
|
|
|
|
// And we just extend this from 'triangle' to 'polygon'.
|
|
|
|
return sqrt(area / (sqrt(1.5) * targetMaxVerticesCount));
|
|
}
|
|
|
|
//======================================================================================
|
|
// Mesh Refinement stuff
|
|
//======================================================================================
|
|
|
|
void refineMeshes(const TMeshImageP &mi, const MeshBuilderOptions &options)
|
|
{
|
|
std::vector<TTextureMeshP> &meshes = mi->meshes();
|
|
|
|
// Refine every mesh to achieve a target mesh density dependent on the image size
|
|
/*TRectD bbox(mi->getBBox());
|
|
double targetLength = sqrt(bbox.getLx() * bbox.getLy()) * relativeMeshDensity;*/
|
|
|
|
double targetLength = options.m_targetEdgeLength;
|
|
|
|
int m, mCount = meshes.size();
|
|
for (m = 0; m < mCount; ++m) {
|
|
const TTextureMeshP &mesh = meshes[m];
|
|
|
|
tcg::TriMeshStuff::DefaultEvaluator<TTextureMesh> eval(0.0, (std::numeric_limits<double>::max)());
|
|
|
|
// First, perform edge swaps alone. This is useful since results from gluTriangulate
|
|
// tend to be unbalanced to vertical - this is a good correction.
|
|
tcg::refineMesh(*mesh, eval);
|
|
|
|
// Now, launch a full-scale, finishing simplification
|
|
eval.m_collapseValue = targetLength * 0.6;
|
|
eval.m_splitValue = targetLength * 1.4;
|
|
tcg::refineMesh(*mesh, eval, 20000); //Max 10000 iterations - to avoid loops
|
|
|
|
// Since we stopped at a max number of iterations, separate collapses from splits
|
|
// and simplify until the procedure stops.
|
|
eval.m_collapseValue = 0.0;
|
|
eval.m_splitValue = 1.4 * targetLength;
|
|
tcg::refineMesh(*mesh, eval);
|
|
|
|
eval.m_splitValue = (std::numeric_limits<double>::max)();
|
|
eval.m_collapseValue = targetLength * 0.6;
|
|
tcg::refineMesh(*mesh, eval);
|
|
|
|
// Perform 1000 final iterations with uniform split and collapses
|
|
eval.m_splitValue = targetLength * 1.4;
|
|
tcg::refineMesh(*mesh, eval, 1000);
|
|
|
|
// Finally, squeeze the mesh to ensure that VEF containers are tight
|
|
mesh->squeeze();
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
//**************************************************************************************
|
|
// Mesh Builder function
|
|
//**************************************************************************************
|
|
|
|
TMeshImageP buildMesh(const TRasterP &ras, const MeshBuilderOptions &options)
|
|
{
|
|
// Convert the input image to a binary raster
|
|
TRasterGR8P binaryRas = thresholdRaster(ras, options);
|
|
|
|
// Extract the image borders
|
|
Tribe *tribe = extractBorders(binaryRas);
|
|
|
|
// Calculate maximum edge length
|
|
double minEdgeLength = buildMinimumEdgeLength(tribe, options.m_targetMaxVerticesCount);
|
|
|
|
MeshBuilderOptions opts(options);
|
|
opts.m_targetEdgeLength = std::max(opts.m_targetEdgeLength, minEdgeLength);
|
|
|
|
// Perform tessellation
|
|
TMeshImageP meshImage(new TMeshImage);
|
|
std::vector<TTextureMeshP> &meshes = meshImage->meshes();
|
|
|
|
tcg::sequential_reader<std::vector<TTextureMeshP>> reader(&meshes);
|
|
tcg::gluTriangulate(tribe->begin(), tribe->end(), reader);
|
|
|
|
delete tribe;
|
|
|
|
// Perform meshes refinement
|
|
refineMeshes(meshImage, opts);
|
|
|
|
return meshImage;
|
|
}
|