380 lines
14 KiB
C++
380 lines
14 KiB
C++
|
// Begin License:
|
||
|
// Copyright (C) 2006-2008 Tobias Sargeant (tobias.sargeant@gmail.com).
|
||
|
// All rights reserved.
|
||
|
//
|
||
|
// This file is part of the Carve CSG Library (http://carve-csg.com/)
|
||
|
//
|
||
|
// This file may be used under the terms of the GNU General Public
|
||
|
// License version 2.0 as published by the Free Software Foundation
|
||
|
// and appearing in the file LICENSE.GPL2 included in the packaging of
|
||
|
// this file.
|
||
|
//
|
||
|
// This file is provided "AS IS" with NO WARRANTY OF ANY KIND,
|
||
|
// INCLUDING THE WARRANTIES OF DESIGN, MERCHANTABILITY AND FITNESS FOR
|
||
|
// A PARTICULAR PURPOSE.
|
||
|
// End:
|
||
|
|
||
|
|
||
|
#if defined(HAVE_CONFIG_H)
|
||
|
# include <carve_config.h>
|
||
|
#endif
|
||
|
|
||
|
#include <carve/csg.hpp>
|
||
|
#include <iostream>
|
||
|
#include "intersect_debug.hpp"
|
||
|
|
||
|
typedef carve::poly::Polyhedron poly_t;
|
||
|
|
||
|
#if defined(CARVE_DEBUG_WRITE_PLY_DATA)
|
||
|
void writePLY(std::string &out_file, const carve::poly::Polyhedron *poly, bool ascii);
|
||
|
#endif
|
||
|
|
||
|
|
||
|
namespace carve {
|
||
|
namespace csg {
|
||
|
namespace {
|
||
|
|
||
|
class BaseCollector : public CSG::Collector {
|
||
|
BaseCollector();
|
||
|
BaseCollector(const BaseCollector &);
|
||
|
BaseCollector &operator=(const BaseCollector &);
|
||
|
|
||
|
protected:
|
||
|
struct face_data_t {
|
||
|
poly_t::face_t *face;
|
||
|
const poly_t::face_t *orig_face;
|
||
|
bool flipped;
|
||
|
face_data_t(poly_t::face_t *_face,
|
||
|
const poly_t::face_t *_orig_face,
|
||
|
bool _flipped) : face(_face), orig_face(_orig_face), flipped(_flipped) {
|
||
|
};
|
||
|
};
|
||
|
|
||
|
std::list<face_data_t> faces;
|
||
|
|
||
|
const poly_t *src_a;
|
||
|
const poly_t *src_b;
|
||
|
|
||
|
BaseCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : CSG::Collector(), src_a(_src_a), src_b(_src_b) {
|
||
|
}
|
||
|
|
||
|
virtual ~BaseCollector() {
|
||
|
}
|
||
|
|
||
|
void FWD(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
std::vector<poly_t::face_t *> new_faces;
|
||
|
new_faces.reserve(1);
|
||
|
new_faces.push_back(orig_face->create(vertices, false));
|
||
|
hooks.processOutputFace(new_faces, orig_face, false);
|
||
|
for (size_t i = 0; i < new_faces.size(); ++i) {
|
||
|
faces.push_back(face_data_t(new_faces[i], orig_face, false));
|
||
|
}
|
||
|
|
||
|
#if defined(CARVE_DEBUG) && defined(DEBUG_PRINT_RESULT_FACES)
|
||
|
std::cerr << "+" << ENUM(face_class) << " ";
|
||
|
for (unsigned i = 0; i < vertices.size(); ++i) std::cerr << " " << vertices[i] << ":" << *vertices[i];
|
||
|
std::cerr << std::endl;
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
void REV(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
normal = -normal;
|
||
|
std::vector<poly_t::face_t *> new_faces;
|
||
|
new_faces.reserve(1);
|
||
|
new_faces.push_back(orig_face->create(vertices, true));
|
||
|
hooks.processOutputFace(new_faces, orig_face, true);
|
||
|
for (size_t i = 0; i < new_faces.size(); ++i) {
|
||
|
faces.push_back(face_data_t(new_faces[i], orig_face, true));
|
||
|
}
|
||
|
|
||
|
#if defined(CARVE_DEBUG) && defined(DEBUG_PRINT_RESULT_FACES)
|
||
|
std::cerr << "-" << ENUM(face_class) << " ";
|
||
|
for (unsigned i = 0; i < vertices.size(); ++i) std::cerr << " " << vertices[i] << ":" << *vertices[i];
|
||
|
std::cerr << std::endl;
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) =0;
|
||
|
|
||
|
virtual void collect(FaceLoopGroup *grp, CSG::Hooks &hooks) {
|
||
|
std::list<ClassificationInfo> &cinfo = (grp->classification);
|
||
|
if (cinfo.size() == 0) {
|
||
|
std::cerr << "WARNING! group " << grp << " has no classification info!" << std::endl;
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
FaceClass fc = FACE_UNCLASSIFIED;
|
||
|
unsigned fc_bits = 0;
|
||
|
|
||
|
for (std::list<ClassificationInfo>::const_iterator i = grp->classification.begin(), e = grp->classification.end(); i != e; ++i) {
|
||
|
if ((*i).intersected_manifold < 0) {
|
||
|
// classifier only returns global info
|
||
|
fc_bits = class_to_class_bit((*i).classification);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if ((*i).intersectedManifoldIsClosed()) {
|
||
|
if ((*i).classification == FACE_UNCLASSIFIED) continue;
|
||
|
fc_bits |= class_to_class_bit((*i).classification);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fc = class_bit_to_class(fc_bits);
|
||
|
|
||
|
// handle the complex cases where a group is classified differently with respect to two or more closed manifolds.
|
||
|
if (fc == FACE_UNCLASSIFIED) {
|
||
|
unsigned inout_bits = fc_bits & FACE_NOT_ON_BIT;
|
||
|
unsigned on_bits = fc_bits & FACE_ON_BIT;
|
||
|
|
||
|
// both in and out. indicates an invalid manifold embedding.
|
||
|
if (inout_bits == (FACE_IN_BIT | FACE_OUT_BIT)) goto out;
|
||
|
|
||
|
// on, both orientations. could be caused by two manifolds touching at a face.
|
||
|
if (on_bits == (FACE_ON_ORIENT_IN_BIT | FACE_ON_ORIENT_OUT_BIT)) goto out;
|
||
|
|
||
|
// in or out, but also on (with orientation). the on classification takes precedence.
|
||
|
fc = class_bit_to_class(on_bits);
|
||
|
}
|
||
|
|
||
|
out:
|
||
|
|
||
|
if (fc == FACE_UNCLASSIFIED) {
|
||
|
std::cerr << "group " << grp << " is unclassified!" << std::endl;
|
||
|
|
||
|
#if defined(CARVE_DEBUG_WRITE_PLY_DATA)
|
||
|
static int uc_count = 0;
|
||
|
|
||
|
std::vector<poly_t::face_t> faces;
|
||
|
|
||
|
for (FaceLoop *f = grp->face_loops.head; f; f = f->next) {
|
||
|
poly_t::face_t *temp = f->orig_face->create(f->vertices, false);
|
||
|
faces.push_back(*temp);
|
||
|
delete temp;
|
||
|
}
|
||
|
|
||
|
std::vector<poly_t::vertex_t> vertices;
|
||
|
carve::csg::VVMap vmap;
|
||
|
|
||
|
poly_t::collectFaceVertices(faces, vertices, vmap);
|
||
|
|
||
|
poly_t *p = new poly_t(faces, vertices);
|
||
|
|
||
|
std::ostringstream filename;
|
||
|
filename << "classifier_fail_" << ++uc_count << ".ply";
|
||
|
std::string out(filename.str().c_str());
|
||
|
::writePLY(out, p, false);
|
||
|
|
||
|
delete p;
|
||
|
#endif
|
||
|
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
bool is_poly_a = cinfo.front().intersected_poly == src_b;
|
||
|
|
||
|
#if defined(CARVE_DEBUG)
|
||
|
bool is_poly_b = cinfo.front().intersected_poly == src_a;
|
||
|
std::cerr << "collect:: " << ENUM(fc) << " grp: " << grp << " (" << grp->face_loops.size() << " faces) is_poly_a?:" << is_poly_a << " is_poly_b?:" << is_poly_b << " against:" << cinfo.front().intersected_poly << std::endl;;
|
||
|
#endif
|
||
|
|
||
|
for (FaceLoop *f = grp->face_loops.head; f; f = f->next) {
|
||
|
collect(f->orig_face, f->vertices, f->orig_face->plane_eqn.N, is_poly_a, fc, hooks);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
virtual poly_t *done(CSG::Hooks &hooks) {
|
||
|
std::vector<poly_t::face_t> f;
|
||
|
f.reserve(faces.size());
|
||
|
for (std::list<face_data_t>::iterator i = faces.begin(); i != faces.end(); ++i) {
|
||
|
f.push_back(poly_t::face_t());
|
||
|
std::swap(f.back(), *(*i).face);
|
||
|
delete (*i).face;
|
||
|
(*i).face = &f.back();
|
||
|
}
|
||
|
|
||
|
std::vector<poly_t::vertex_t> vertices;
|
||
|
carve::csg::VVMap vmap;
|
||
|
|
||
|
poly_t::collectFaceVertices(f, vertices, vmap);
|
||
|
|
||
|
poly_t *p = new poly_t(f, vertices);
|
||
|
|
||
|
if (hooks.hasHook(carve::csg::CSG::Hooks::RESULT_FACE_HOOK)) {
|
||
|
for (std::list<face_data_t>::iterator i = faces.begin(); i != faces.end(); ++i) {
|
||
|
hooks.resultFace((*i).face, (*i).orig_face, (*i).flipped);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return p;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class AllCollector : public BaseCollector {
|
||
|
public:
|
||
|
AllCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~AllCollector() {
|
||
|
}
|
||
|
virtual void collect(FaceLoopGroup *grp, CSG::Hooks &hooks) {
|
||
|
for (FaceLoop *f = grp->face_loops.head; f; f = f->next) {
|
||
|
FWD(f->orig_face, f->vertices, f->orig_face->plane_eqn.N, f->orig_face->owner == src_a, FACE_OUT, hooks);
|
||
|
}
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class UnionCollector : public BaseCollector {
|
||
|
public:
|
||
|
UnionCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~UnionCollector() {
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
if (face_class == FACE_OUT || (poly_a && face_class == FACE_ON_ORIENT_OUT)) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class IntersectionCollector : public BaseCollector {
|
||
|
public:
|
||
|
IntersectionCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~IntersectionCollector() {
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
if (face_class == FACE_IN || (poly_a && face_class == FACE_ON_ORIENT_OUT)) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class SymmetricDifferenceCollector : public BaseCollector {
|
||
|
public:
|
||
|
SymmetricDifferenceCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~SymmetricDifferenceCollector() {
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
if (face_class == FACE_OUT) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
} else if (face_class == FACE_IN) {
|
||
|
REV(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class AMinusBCollector : public BaseCollector {
|
||
|
public:
|
||
|
AMinusBCollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~AMinusBCollector() {
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
if ((face_class == FACE_OUT || face_class == FACE_ON_ORIENT_IN) && poly_a) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
} else if (face_class == FACE_IN && !poly_a) {
|
||
|
REV(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
class BMinusACollector : public BaseCollector {
|
||
|
public:
|
||
|
BMinusACollector(const poly_t *_src_a,
|
||
|
const poly_t *_src_b) : BaseCollector(_src_a, _src_b) {
|
||
|
}
|
||
|
virtual ~BMinusACollector() {
|
||
|
}
|
||
|
virtual void collect(const poly_t::face_t *orig_face,
|
||
|
const std::vector<const poly_t::vertex_t *> &vertices,
|
||
|
carve::geom3d::Vector normal,
|
||
|
bool poly_a,
|
||
|
FaceClass face_class,
|
||
|
CSG::Hooks &hooks) {
|
||
|
if ((face_class == FACE_OUT || face_class == FACE_ON_ORIENT_IN) && !poly_a) {
|
||
|
FWD(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
} else if (face_class == FACE_IN && poly_a) {
|
||
|
REV(orig_face, vertices, normal, poly_a, face_class, hooks);
|
||
|
}
|
||
|
}
|
||
|
};
|
||
|
|
||
|
}
|
||
|
|
||
|
CSG::Collector *makeCollector(CSG::OP op,
|
||
|
const poly_t *poly_a,
|
||
|
const poly_t *poly_b) {
|
||
|
switch (op) {
|
||
|
case CSG::UNION: return new UnionCollector(poly_a, poly_b);
|
||
|
case CSG::INTERSECTION: return new IntersectionCollector(poly_a, poly_b);
|
||
|
case CSG::A_MINUS_B: return new AMinusBCollector(poly_a, poly_b);
|
||
|
case CSG::B_MINUS_A: return new BMinusACollector(poly_a, poly_b);
|
||
|
case CSG::SYMMETRIC_DIFFERENCE: return new SymmetricDifferenceCollector(poly_a, poly_b);
|
||
|
case CSG::ALL: return new AllCollector(poly_a, poly_b);
|
||
|
}
|
||
|
return NULL;
|
||
|
}
|
||
|
}
|
||
|
}
|