centroid.hpp 5.36 KB
Newer Older
Pierre Kraemer's avatar
Pierre Kraemer committed
1 2 3
/*******************************************************************************
 * CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
 * version 0.1                                                                  *
4
 * Copyright (C) 2009-2011, IGG Team, LSIIT, University of Strasbourg           *
Pierre Kraemer's avatar
Pierre Kraemer committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
 *                                                                              *
 * 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.u-strasbg.fr/                                         *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24
 * Contact information: cgogn@unistra.fr                                        *
 *                                                                              *
 *******************************************************************************/

25 26
#include "Topology/generic/traversorCell.h"
#include "Topology/generic/traversor2.h"
Pierre Kraemer's avatar
Pierre Kraemer committed
27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
#include "Topology/generic/cellmarker.h"

namespace CGoGN
{

namespace Algo
{

namespace Geometry
{

template <typename PFP, typename EMBV, typename EMB>
EMB volumeCentroidGen(typename PFP::MAP& map, Dart d, const EMBV& attributs)
{
	EMB center = AttribOps::zero<EMB,PFP>() ;
	unsigned count = 0 ;

44
//	CellMarkerStore marker(map, VERTEX) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
//
//	std::vector<Dart> visitedVertices ;
//	visitedVertices.reserve(100) ;
//	visitedVertices.push_back(d) ;
//	marker.mark(d) ;
//
//	for(typename std::vector<Dart>::iterator vert = visitedVertices.begin(); vert != visitedVertices.end(); ++vert)
//	{
//		Dart e = *vert;
//		center += attributs[e];	// add to center and mark
//		++count;
//		do	// add all vertex neighbors to the table
//		{
//			Dart ee = map.phi2(e) ;
//			if(!marker.isMarked(ee))
//			{
//				visitedVertices.push_back(ee) ;
//				marker.mark(e) ;
//			}
//			e = map.phi1(ee) ;
//		} while(e != *vert) ;
//	}
	DartMarkerStore mark(map);		// Lock a marker

	std::vector<Dart> visitedFaces;		// Faces that are traversed
70
	visitedFaces.reserve(128);
Pierre Kraemer's avatar
Pierre Kraemer committed
71 72
	visitedFaces.push_back(d);			// Start with the face of d

73
	mark.markOrbit(VERTEX, d) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
74

75
	for(unsigned int i = 0; i < visitedFaces.size(); ++i)
Pierre Kraemer's avatar
Pierre Kraemer committed
76
	{
77
		Dart e = visitedFaces[i] ;
Pierre Kraemer's avatar
Pierre Kraemer committed
78 79 80 81 82 83 84 85 86 87

		center += attributs[e];	// add to center and mark
		++count;

		do	// add all face neighbours to the table
		{
			Dart ee = map.phi2(e) ;
			if(!mark.isMarked(ee)) // not already marked
			{
				visitedFaces.push_back(ee) ;
88
				mark.markOrbit(VERTEX, ee) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
89 90
			}
			e = map.phi1(e) ;
91
		} while(e != visitedFaces[i]) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
92 93 94 95 96 97 98 99 100 101 102
	}

	center /= double(count) ;
	return center ;
}

template <typename PFP, typename EMBV, typename EMB>
EMB faceCentroidGen(typename PFP::MAP& map, Dart d, const EMBV& attributs)
{
	EMB center = AttribOps::zero<EMB,PFP>();
	unsigned int count = 0 ;
103 104
	Traversor2FV<typename PFP::MAP> t(map, d) ;
	for(Dart it = t.begin(); it != t.end(); it = t.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
105 106 107
	{
		center += attributs[it];
		++count ;
108
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
109 110 111 112 113 114 115 116 117
	center /= double(count);
	return center ;
}

template <typename PFP, typename EMBV, typename EMB>
EMB vertexNeighborhoodCentroidGen(typename PFP::MAP& map, Dart d, const EMBV& attributs)
{
	EMB center = AttribOps::zero<EMB,PFP>();
	unsigned int count = 0 ;
118 119
	Traversor2VVaE<typename PFP::MAP> t(map, d) ;
	for(Dart it = t.begin(); it != t.end(); it = t.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
120
	{
121
		center += attributs[it];
Pierre Kraemer's avatar
Pierre Kraemer committed
122
		++count ;
123
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
124 125 126 127 128 129 130
	center /= count ;
	return center ;
}

template <typename PFP>
void computeCentroidVolumes(typename PFP::MAP& map, const typename PFP::TVEC3& position, typename PFP::TVEC3& vol_centroid, const FunctorSelect& select)
{
Pierre Kraemer's avatar
Pierre Kraemer committed
131
	TraversorW<typename PFP::MAP> t(map, select) ;
132
	for(Dart d = t.begin(); d != t.end(); d = t.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
133
		vol_centroid[d] = volumeCentroid<PFP>(map, d, position) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
134 135 136 137 138
}

template <typename PFP>
void computeCentroidFaces(typename PFP::MAP& map, const typename PFP::TVEC3& position, typename PFP::TVEC3& face_centroid, const FunctorSelect& select)
{
Pierre Kraemer's avatar
Pierre Kraemer committed
139
	TraversorF<typename PFP::MAP> t(map, select) ;
140
	for(Dart d = t.begin(); d != t.end(); d = t.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
141
		face_centroid[d] = faceCentroid<PFP>(map, d, position) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
142 143 144 145 146
}

template <typename PFP>
void computeNeighborhoodCentroidVertices(typename PFP::MAP& map, const typename PFP::TVEC3& position, typename PFP::TVEC3& vertex_centroid, const FunctorSelect& select)
{
Pierre Kraemer's avatar
Pierre Kraemer committed
147
	TraversorV<typename PFP::MAP> t(map, select) ;
148
	for(Dart d = t.begin(); d != t.end(); d = t.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
149
		vertex_centroid[d] = vertexNeighborhoodCentroid<PFP>(map, d, position) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
150 151 152 153 154 155 156
}

} // namespace Geometry

} // namespace Algo

} // namespace CGoGN