collector.h 15.7 KB
Newer Older
Pierre Kraemer's avatar
Pierre Kraemer committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
* version 0.1                                                                  *
* Copyright (C) 2009, IGG Team, LSIIT, University of Strasbourg                *
*                                                                              *
* This library is free software; you can redistribute it and/or modify it      *
* under the terms of the GNU Lesser General Public License as published by the *
* Free Software Foundation; either version 2.1 of the License, or (at your     *
* option) any later version.                                                   *
*                                                                              *
* This library is distributed in the hope that it will be useful, but WITHOUT  *
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or        *
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License  *
* for more details.                                                            *
*                                                                              *
* You should have received a copy of the GNU Lesser General Public License     *
* along with this library; if not, write to the Free Software Foundation,      *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.           *
*                                                                              *
20
* Web site: http://cgogn.unistra.fr/                                  *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24 25 26 27
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#ifndef __COLLECTOR_H__
#define __COLLECTOR_H__

Sylvain Thery's avatar
Sylvain Thery committed
28
#include "Container/fakeAttribute.h"
Sylvain Thery's avatar
Sylvain Thery committed
29
#include "Geometry/basic.h"
Sylvain Thery's avatar
Sylvain Thery committed
30

Pierre Kraemer's avatar
Pierre Kraemer committed
31 32 33 34 35 36 37 38 39 40
/*****************************************
 * Class hierarchy :
 * Collector (virtual)
 * - Collector_WithinSphere
 * - Collector_OneRing
 ****************************************/

namespace CGoGN
{

41 42 43
namespace Algo
{

Pierre Kraemer's avatar
Pierre Kraemer committed
44 45 46 47
namespace Selection
{

/*********************************************************
48
 * Generic Collector
Pierre Kraemer's avatar
Pierre Kraemer committed
49
 *********************************************************/
50

Pierre Kraemer's avatar
Pierre Kraemer committed
51 52 53 54 55 56 57 58
template <typename PFP>
class Collector
{
protected:
	typedef typename PFP::VEC3 VEC3;
	typedef typename PFP::REAL REAL;

	typename PFP::MAP& map;
Sylvain Thery's avatar
Sylvain Thery committed
59
	unsigned int m_thread;
Pierre Kraemer's avatar
Pierre Kraemer committed
60 61 62

	Dart centerDart;

63 64
	bool isInsideCollected;

Pierre Kraemer's avatar
Pierre Kraemer committed
65 66 67 68 69 70
	std::vector<Dart> insideVertices;
	std::vector<Dart> insideEdges;
	std::vector<Dart> insideFaces;
	std::vector<Dart> border;

public:
Sylvain Thery's avatar
Sylvain Thery committed
71
	Collector(typename PFP::MAP& m, unsigned int thread =0);
Pierre Kraemer's avatar
Pierre Kraemer committed
72

Pierre Kraemer's avatar
Pierre Kraemer committed
73 74 75
	inline void init(Dart d)
	{
		centerDart = d;
76
		isInsideCollected = false;
Pierre Kraemer's avatar
Pierre Kraemer committed
77 78 79 80 81
		insideVertices.clear();
		insideEdges.clear();
		insideFaces.clear();
		border.clear();
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
82

Pierre Kraemer's avatar
Pierre Kraemer committed
83 84
	virtual void collectAll(Dart d) = 0;
	virtual void collectBorder(Dart d) = 0;
Pierre Kraemer's avatar
Pierre Kraemer committed
85

Pierre Kraemer's avatar
Pierre Kraemer committed
86 87 88 89
	bool applyOnInsideVertices(FunctorType& f);
	bool applyOnInsideEdges(FunctorType& f);
	bool applyOnInsideFaces(FunctorType& f);
	bool applyOnBorder(FunctorType& f);
90

Pierre Kraemer's avatar
Pierre Kraemer committed
91
	inline void sort()
Pierre Kraemer's avatar
Pierre Kraemer committed
92 93 94 95 96 97 98
	{
		std::sort(insideVertices.begin(), insideVertices.end());
		std::sort(insideEdges.begin(), insideEdges.end());
		std::sort(insideFaces.begin(), insideFaces.end());
		std::sort(border.begin(), border.end());
	}

Pierre Kraemer's avatar
Pierre Kraemer committed
99
	inline typename PFP::MAP& getMap() { return map; }
Pierre Kraemer's avatar
Pierre Kraemer committed
100

Pierre Kraemer's avatar
Pierre Kraemer committed
101
	inline Dart getCenterDart() const { return centerDart; }
Pierre Kraemer's avatar
Pierre Kraemer committed
102

103 104 105
	inline const std::vector<Dart>& getInsideVertices() const { assert(isInsideCollected || !"getInsideVertices: inside cells have not been collected.") ; return insideVertices; }
	inline const std::vector<Dart>& getInsideEdges() const { assert(isInsideCollected || !"getInsideEdges: inside cells have not been collected.") ; return insideEdges; }
	inline const std::vector<Dart>& getInsideFaces() const { assert(isInsideCollected || !"getInsideFaces: inside cells have not been collected.") ; return insideFaces; }
Pierre Kraemer's avatar
Pierre Kraemer committed
106
	inline const std::vector<Dart>& getBorder() const { return border; }
Pierre Kraemer's avatar
Pierre Kraemer committed
107

108 109 110
	inline unsigned int getNbInsideVertices() const { assert(isInsideCollected || !"getNbInsideVertices: inside cells have not been collected.") ; return insideVertices.size(); }
	inline unsigned int getNbInsideEdges() const { assert(isInsideCollected || !"getNbInsideEdges: inside cells have not been collected.") ; return insideEdges.size(); }
	inline unsigned int getNbInsideFaces() const { assert(isInsideCollected || !"getNbInsideFaces: inside cells have not been collected.") ; return insideFaces.size(); }
Pierre Kraemer's avatar
Pierre Kraemer committed
111
	inline unsigned int getNbBorder() const { return border.size(); }
Pierre Kraemer's avatar
Pierre Kraemer committed
112 113 114

	template <typename PPFP>
	friend std::ostream& operator<<(std::ostream &out, const Collector<PPFP>& c);
115 116 117 118 119 120 121

	REAL computeArea () {
		assert(!"Warning: Collector<PFP>::computeArea() should be overloaded in non-virtual derived classes");
		return 0.0;
	}
	void computeNormalCyclesTensor (const EdgeAttribute<REAL>&, typename PFP::MATRIX33&) {assert(!"Warning: Collector<PFP>::computeNormalCyclesTensor() should be overloaded in non-virtual derived classes"); }

Pierre Kraemer's avatar
Pierre Kraemer committed
122 123 124 125 126 127 128 129 130 131
};

/*********************************************************
 * Collector One Ring
 *********************************************************/

/*
 * insideVertices = centerDart
 * insideEdges = star (edges incident to centerDart)
 * insideFaces = triangles incident to centerDart
132
 * border = vertices of 1-ring -> link (set of adjacent vertices)
Pierre Kraemer's avatar
Pierre Kraemer committed
133 134 135 136 137 138
 *        = edges of 1-ring
 */
template <typename PFP>
class Collector_OneRing : public Collector<PFP>
{
public:
Sylvain Thery's avatar
Sylvain Thery committed
139 140
	Collector_OneRing(typename PFP::MAP& m, unsigned int thread=0):
		Collector<PFP>(m, thread) {}
Pierre Kraemer's avatar
Pierre Kraemer committed
141 142
	void collectAll(Dart d);
	void collectBorder(Dart d);
Pierre Kraemer's avatar
Pierre Kraemer committed
143 144 145 146 147 148 149 150 151 152
};

/*********************************************************
 * Collector Within Sphere
 *********************************************************/

/*
 * collect all primitives of the connected component containing "centerDart"
 * within the sphere of radius "radius" and center "position[centerDart]"
 * (hopefully) it defines a 2-manifold (if inserting border-vertices along the border-edges)
Basile Sauvage's avatar
Basile Sauvage committed
153
 * NB : is equivalent to Collector_Vertices with CollectorCriterion_VertexWithinSphere
Pierre Kraemer's avatar
Pierre Kraemer committed
154 155 156 157 158
 */
template <typename PFP>
class Collector_WithinSphere : public Collector<PFP>
{
protected:
159 160 161 162 163
	typedef typename PFP::VEC3 VEC3 ;
	typedef typename PFP::REAL REAL ;

	const VertexAttribute<VEC3>& position;
	REAL radius;
164

Pierre Kraemer's avatar
Pierre Kraemer committed
165
public:
166
	Collector_WithinSphere(typename PFP::MAP& m, const VertexAttribute<VEC3>& p, REAL r = 0, unsigned int thread=0) :
Sylvain Thery's avatar
Sylvain Thery committed
167
		Collector<PFP>(m, thread),
Pierre Kraemer's avatar
Pierre Kraemer committed
168
		position(p),
169
		radius(r)
Pierre Kraemer's avatar
Pierre Kraemer committed
170
	{}
171 172 173
	inline void setRadius(REAL r) { radius = r; }
	inline REAL getRadius() const { return radius; }
	inline const VertexAttribute<VEC3>& getPosition() const { return position; }
Pierre Kraemer's avatar
Pierre Kraemer committed
174 175 176 177

	void collectAll(Dart d);
	void collectBorder(Dart d);

178 179
	REAL computeArea();
	void computeNormalCyclesTensor (const EdgeAttribute<REAL>&, typename PFP::MATRIX33&);
Pierre Kraemer's avatar
Pierre Kraemer committed
180 181
};

182
/*********************************************************
183
 * Collector Normal Angle (Vertices)
184 185 186 187 188 189
 *********************************************************/

/*
 * collect all primitives of the connected component containing "centerDart"
 * the angle between the included vertices normal vectors and the central normal vector
 * stays under a given threshold
Basile Sauvage's avatar
Basile Sauvage committed
190
 * NB : is equivalent to Collector_Vertices with CollectorCriterion_VertexNormalAngle
191 192 193 194 195 196 197 198 199 200 201 202
 */
template <typename PFP>
class Collector_NormalAngle : public Collector<PFP>
{
protected:
	const VertexAttribute<typename PFP::VEC3>& normal ;
	typename PFP::REAL angleThreshold ;

public:
	Collector_NormalAngle(
		typename PFP::MAP& m,
		const VertexAttribute<typename PFP::VEC3>& n,
Sylvain Thery's avatar
Sylvain Thery committed
203 204
		typename PFP::REAL a,
		unsigned int thread=0
205
	) :	Collector<PFP>(m,thread), normal(n), angleThreshold(a)
206 207 208 209 210 211 212 213 214
	{}
	inline void setAngleThreshold(typename PFP::REAL a) { angleThreshold = a; }
	inline typename PFP::REAL getAngleThreshold() const { return angleThreshold; }
	inline const VertexAttribute<typename PFP::VEC3>& getNormal() const { return normal ; }

	void collectAll(Dart d) ;
	void collectBorder(Dart d) ;
};

Basile Sauvage's avatar
Basile Sauvage committed
215
/*********************************************************
Basile Sauvage's avatar
Basile Sauvage committed
216
 * Collector Criterions
Basile Sauvage's avatar
Basile Sauvage committed
217 218 219 220 221
 *********************************************************/
class CollectorCriterion
{
public :
	CollectorCriterion() {};
Sauvage's avatar
Sauvage committed
222
	virtual ~CollectorCriterion() {} ;
Basile Sauvage's avatar
Basile Sauvage committed
223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247
	virtual void init(Dart center) = 0;
	virtual bool isInside(Dart d) = 0;

};

template <typename PFP>
class CollectorCriterion_VertexNormalAngle : public CollectorCriterion
{ // tests if the angle between vertex normals is below some threshold
private :
	typedef typename PFP::VEC3 VEC3;
	typedef typename PFP::REAL REAL;

	const VertexAttribute<VEC3> & vertexNormals;
	REAL threshold;
	VEC3 centerNormal;
public :
	CollectorCriterion_VertexNormalAngle(const VertexAttribute<VEC3> & n, REAL th) :
		vertexNormals(n), threshold(th), centerNormal(0) {}

	void init (Dart center) {centerNormal = vertexNormals[center];}
	bool isInside (Dart d) {
		return ( Geom::angle(centerNormal, vertexNormals[d]) < threshold);
	}
};

Basile Sauvage's avatar
Basile Sauvage committed
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267
template <typename PFP>
class CollectorCriterion_TriangleNormalAngle : public CollectorCriterion
{ // tests if the angle between vertex normals is below some threshold
private :
	typedef typename PFP::VEC3 VEC3;
	typedef typename PFP::REAL REAL;

	const FaceAttribute<VEC3> & faceNormals;
	REAL threshold;
	VEC3 centerNormal;
public :
	CollectorCriterion_TriangleNormalAngle(const FaceAttribute<VEC3> & n, REAL th) :
		faceNormals(n), threshold(th), centerNormal(0) {}

	void init (Dart center) {centerNormal = faceNormals[center];}
	bool isInside (Dart d) {
		return ( Geom::angle(centerNormal, faceNormals[d]) < threshold);
	}
};

Basile Sauvage's avatar
Basile Sauvage committed
268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
template <typename PFP>
class CollectorCriterion_VertexWithinSphere : public CollectorCriterion
{ // tests if the distance between vertices is below some threshold
private :
	typedef typename PFP::VEC3 VEC3;
	typedef typename PFP::REAL REAL;

	const VertexAttribute<VEC3> & vertexPositions;
	REAL threshold;
	VEC3 centerPosition;
public :
	CollectorCriterion_VertexWithinSphere(const VertexAttribute<VEC3> & p, REAL th) :
		vertexPositions(p), threshold(th), centerPosition(0) {}

	void init (Dart center) {centerPosition = vertexPositions[center];}
	bool isInside (Dart d) {
		return (vertexPositions[d] - centerPosition).norm() < threshold ;
	}
};


Basile Sauvage's avatar
Basile Sauvage committed
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
/*********************************************************
 * Collector Vertices
 *********************************************************/

/*
 * collect all vertices of the connected component containing "centerDart"
 * within a distance to centerDart defined by the CollectorCriterion
 * (hopefully) it defines a 2-manifold (if inserting border-vertices along the border-edges)
 */
template <typename PFP>
class Collector_Vertices : public Collector<PFP>
{
protected:
	CollectorCriterion & crit;

public:
	Collector_Vertices(typename PFP::MAP& m, CollectorCriterion& c, unsigned int thread=0) :
		Collector<PFP>(m, thread),
		crit(c)
	{}

	void collectAll(Dart d);
	void collectBorder(Dart d);
};


315 316 317 318 319 320 321 322
/*********************************************************
 * Collector Normal Angle (Triangles)
 *********************************************************/

/*
 * collect all primitives of the connected component containing "centerDart"
 * the angle between the included triangles normal vectors and the central normal vector
 * stays under a given threshold
Basile Sauvage's avatar
Basile Sauvage committed
323
 * NB : is equivalent to Collector_Triangles with CollectorCriterion_TriangleNormalAngle
324 325 326 327 328 329 330 331 332 333 334 335
 */
template <typename PFP>
class Collector_NormalAngle_Triangles : public Collector<PFP>
{
protected:
	const FaceAttribute<typename PFP::VEC3>& normal ;
	typename PFP::REAL angleThreshold ;

public:
	Collector_NormalAngle_Triangles(
		typename PFP::MAP& m,
		const FaceAttribute<typename PFP::VEC3>& n,
336 337 338
		typename PFP::REAL a,
		unsigned int thread=0
	) :	Collector<PFP>(m,thread), normal(n), angleThreshold(a)
339 340 341 342 343 344 345 346 347
	{}
	inline void setAngleThreshold(typename PFP::REAL a) { angleThreshold = a; }
	inline typename PFP::REAL getAngleThreshold() const { return angleThreshold; }
	inline const VertexAttribute<typename PFP::VEC3>& getNormal() const { return normal ; }

	void collectAll(Dart d) ;
	void collectBorder(Dart d) ;
};

Basile Sauvage's avatar
Basile Sauvage committed
348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
/*********************************************************
 * Collector Triangles
 *********************************************************/

/*
 * collect all triangles of the connected component containing "centerDart"
 * within a distance to centerDart defined by the CollectorCriterion
 */
template <typename PFP>
class Collector_Triangles : public Collector<PFP>
{
protected:
	CollectorCriterion & crit;

public:
	Collector_Triangles(typename PFP::MAP& m, CollectorCriterion& c, unsigned int thread=0) :
		Collector<PFP>(m,thread), crit(c)
	{}

	void collectAll(Dart d) ;
	void collectBorder(Dart d) ;
};

Basile Sauvage's avatar
Basile Sauvage committed
371
/*********************************************************
372
 * Collector Dijkstra_Vertices
Basile Sauvage's avatar
Basile Sauvage committed
373 374 375 376 377
 *********************************************************/

/*
 * collect all primitives of the connected component containing "centerDart"
 * within a distance < maxDist (the shortest path follows edges)
378
 * the edge length is specified in edge_cost attribute
Basile Sauvage's avatar
Basile Sauvage committed
379
 */
380

Basile Sauvage's avatar
Basile Sauvage committed
381
template <typename PFP>
382
class Collector_Dijkstra_Vertices : public Collector<PFP>
Basile Sauvage's avatar
Basile Sauvage committed
383 384
{
protected:
385
	const EdgeAttribute<typename PFP::REAL>& edge_cost;
Basile Sauvage's avatar
Basile Sauvage committed
386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
	typename PFP::REAL maxDist;

	typedef struct
	{
		typename std::multimap<float,Dart>::iterator it ;
		bool valid ;
		static std::string CGoGNnameOfType() { return "DijkstraVertexInfo" ; }
	} DijkstraVertexInfo ;
	typedef NoMathIOAttribute<DijkstraVertexInfo> VertexInfo ;

	VertexAttribute<VertexInfo> vertexInfo ;

	std::multimap<float,Dart> front ;

public:
401
	Collector_Dijkstra_Vertices(typename PFP::MAP& m, const EdgeAttribute<typename PFP::REAL>& c, typename PFP::REAL d = 0, unsigned int thread=0) :
402
		Collector<PFP>(m,thread),
403
		edge_cost(c),
Basile Sauvage's avatar
Basile Sauvage committed
404 405 406 407
		maxDist(d)
	{
		vertexInfo = m.template addAttribute<VertexInfo, VERTEX>("vertexInfo");
	}
408
	~Collector_Dijkstra_Vertices(){
Basile Sauvage's avatar
Basile Sauvage committed
409 410 411 412 413 414 415 416 417
		this->map.removeAttribute(vertexInfo);
	}
	inline void init (Dart d) {Collector<PFP>::init(d); front.clear();}
	inline void setMaxDistance(typename PFP::REAL d) { maxDist = d; }
	inline typename PFP::REAL getMaxDist() const { return maxDist; }

	void collectAll(Dart d);
	void collectBorder(Dart d);
};
418 419


Basile Sauvage's avatar
Basile Sauvage committed
420
/*********************************************************
421
 * Collector Dijkstra
Basile Sauvage's avatar
Basile Sauvage committed
422 423 424 425 426 427 428
 *********************************************************/

/*
 * collect all primitives of the connected component containing "centerDart"
 * within a distance < maxDist (the shortest path follows edges)
 */
template <typename PFP>
429
class Collector_Dijkstra : public Collector<PFP>
Basile Sauvage's avatar
Basile Sauvage committed
430 431
{
protected:
432
	const VertexAttribute<typename PFP::VEC3>& position;
Basile Sauvage's avatar
Basile Sauvage committed
433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
	typename PFP::REAL maxDist;

	typedef struct
	{
		typename std::multimap<float,Dart>::iterator it ;
		bool valid ;
		static std::string CGoGNnameOfType() { return "DijkstraVertexInfo" ; }
	} DijkstraVertexInfo ;
	typedef NoMathIOAttribute<DijkstraVertexInfo> VertexInfo ;

	VertexAttribute<VertexInfo> vertexInfo ;

	std::multimap<float,Dart> front ;

public:
448
	Collector_Dijkstra(typename PFP::MAP& m, const VertexAttribute<typename PFP::VEC3>& p, typename PFP::REAL d = 0, unsigned int thread=0) :
Basile Sauvage's avatar
Basile Sauvage committed
449
		Collector<PFP>(m,thread),
450
		position(p),
Basile Sauvage's avatar
Basile Sauvage committed
451 452 453 454
		maxDist(d)
	{
		vertexInfo = m.template addAttribute<VertexInfo, VERTEX>("vertexInfo");
	}
455
	~Collector_Dijkstra(){
Basile Sauvage's avatar
Basile Sauvage committed
456 457 458 459 460
		this->map.removeAttribute(vertexInfo);
	}
	inline void init (Dart d) {Collector<PFP>::init(d); front.clear();}
	inline void setMaxDistance(typename PFP::REAL d) { maxDist = d; }
	inline typename PFP::REAL getMaxDist() const { return maxDist; }
461
	inline const VertexAttribute<typename PFP::VEC3>& getPosition() const { return position; }
Basile Sauvage's avatar
Basile Sauvage committed
462 463 464

	void collectAll(Dart d);
	void collectBorder(Dart d);
465 466 467
private :
	inline float edgeLength (Dart d);
//	inline Dart oppositeVertex (Dart d);
Basile Sauvage's avatar
Basile Sauvage committed
468 469 470
};


471

Pierre Kraemer's avatar
Pierre Kraemer committed
472 473
} // namespace Selection

474 475
} // namespace Algo

Pierre Kraemer's avatar
Pierre Kraemer committed
476 477 478 479 480
} // namespace CGoGN

#include "Algo/Selection/collector.hpp"

#endif