collector.h 5.83 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 28 29 30 31 32 33 34 35 36 37
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#ifndef __COLLECTOR_H__
#define __COLLECTOR_H__

/*****************************************
 * Class hierarchy :
 * Collector (virtual)
 * - Collector_WithinSphere
 * - Collector_OneRing
 ****************************************/

namespace CGoGN
{

38 39 40
namespace Algo
{

Pierre Kraemer's avatar
Pierre Kraemer committed
41 42 43 44
namespace Selection
{

/*********************************************************
45
 * Generic Collector
Pierre Kraemer's avatar
Pierre Kraemer committed
46
 *********************************************************/
47

Pierre Kraemer's avatar
Pierre Kraemer committed
48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
template <typename PFP>
class Collector
{
protected:
	typedef typename PFP::VEC3 VEC3;
	typedef typename PFP::REAL REAL;

	typename PFP::MAP& map;

	Dart centerDart;

	std::vector<Dart> insideVertices;
	std::vector<Dart> insideEdges;
	std::vector<Dart> insideFaces;
	std::vector<Dart> border;

public:
Pierre Kraemer's avatar
Pierre Kraemer committed
65
	Collector(typename PFP::MAP& m);
Pierre Kraemer's avatar
Pierre Kraemer committed
66

Pierre Kraemer's avatar
Pierre Kraemer committed
67 68 69 70 71 72 73 74
	inline void init(Dart d)
	{
		centerDart = d;
		insideVertices.clear();
		insideEdges.clear();
		insideFaces.clear();
		border.clear();
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
75

Pierre Kraemer's avatar
Pierre Kraemer committed
76 77
	virtual void collectAll(Dart d) = 0;
	virtual void collectBorder(Dart d) = 0;
Pierre Kraemer's avatar
Pierre Kraemer committed
78

Pierre Kraemer's avatar
Pierre Kraemer committed
79 80 81 82
	bool applyOnInsideVertices(FunctorType& f);
	bool applyOnInsideEdges(FunctorType& f);
	bool applyOnInsideFaces(FunctorType& f);
	bool applyOnBorder(FunctorType& f);
83

Pierre Kraemer's avatar
Pierre Kraemer committed
84
	inline void sort()
Pierre Kraemer's avatar
Pierre Kraemer committed
85 86 87 88 89 90 91
	{
		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
92
	inline typename PFP::MAP& getMap() { return map; }
Pierre Kraemer's avatar
Pierre Kraemer committed
93

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

Pierre Kraemer's avatar
Pierre Kraemer committed
96 97 98 99
	inline const std::vector<Dart>& getInsideVertices() const { return insideVertices; }
	inline const std::vector<Dart>& getInsideEdges() const { return insideEdges; }
	inline const std::vector<Dart>& getInsideFaces() const { return insideFaces; }
	inline const std::vector<Dart>& getBorder() const { return border; }
Pierre Kraemer's avatar
Pierre Kraemer committed
100

Pierre Kraemer's avatar
Pierre Kraemer committed
101 102 103 104
	inline unsigned int getNbInsideVertices() const { return insideVertices.size(); }
	inline unsigned int getNbInsideEdges() const { return insideEdges.size(); }
	inline unsigned int getNbInsideFaces() const { return insideFaces.size(); }
	inline unsigned int getNbBorder() const { return border.size(); }
Pierre Kraemer's avatar
Pierre Kraemer committed
105 106 107 108 109 110 111 112 113 114 115 116 117

	template <typename PPFP>
	friend std::ostream& operator<<(std::ostream &out, const Collector<PPFP>& c);
};

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

/*
 * insideVertices = centerDart
 * insideEdges = star (edges incident to centerDart)
 * insideFaces = triangles incident to centerDart
118
 * border = vertices of 1-ring -> link (set of adjacent vertices)
Pierre Kraemer's avatar
Pierre Kraemer committed
119 120 121 122 123 124
 *        = edges of 1-ring
 */
template <typename PFP>
class Collector_OneRing : public Collector<PFP>
{
public:
Pierre Kraemer's avatar
Pierre Kraemer committed
125 126
	Collector_OneRing(typename PFP::MAP& m) :
		Collector<PFP>(m)
Pierre Kraemer's avatar
Pierre Kraemer committed
127
	{}
Pierre Kraemer's avatar
Pierre Kraemer committed
128 129
	void collectAll(Dart d);
	void collectBorder(Dart d);
Pierre Kraemer's avatar
Pierre Kraemer committed
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
};

/*********************************************************
 * 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)
 */
template <typename PFP>
class Collector_WithinSphere : public Collector<PFP>
{
protected:
145
	const AttributeHandler<typename PFP::VEC3, VERTEX>& position;
Pierre Kraemer's avatar
Pierre Kraemer committed
146
	typename PFP::REAL radius;
Pierre Kraemer's avatar
Pierre Kraemer committed
147
	typename PFP::REAL area;
148

Pierre Kraemer's avatar
Pierre Kraemer committed
149
public:
150
	Collector_WithinSphere(typename PFP::MAP& m, const AttributeHandler<typename PFP::VEC3, VERTEX>& p, typename PFP::REAL r = 0) :
Pierre Kraemer's avatar
Pierre Kraemer committed
151 152 153 154
		Collector<PFP>(m),
		position(p),
		radius(r),
		area(0)
Pierre Kraemer's avatar
Pierre Kraemer committed
155
	{}
Pierre Kraemer's avatar
Pierre Kraemer committed
156 157
	inline void setRadius(typename PFP::REAL r) { radius = r; }
	inline typename PFP::REAL getRadius() const { return radius; }
158
	inline const AttributeHandler<typename PFP::VEC3, VERTEX>& getPosition() const { return position; }
Pierre Kraemer's avatar
Pierre Kraemer committed
159 160 161 162

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

Pierre Kraemer's avatar
Pierre Kraemer committed
163
	void computeArea();
Pierre Kraemer's avatar
Pierre Kraemer committed
164
	inline typename PFP::REAL getArea() const { return area; }
Pierre Kraemer's avatar
Pierre Kraemer committed
165 166 167 168
};

} // namespace Selection

169 170
} // namespace Algo

Pierre Kraemer's avatar
Pierre Kraemer committed
171 172 173 174 175
} // namespace CGoGN

#include "Algo/Selection/collector.hpp"

#endif