functors.h 3.83 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-2012, 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.unistra.fr/                                           *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24 25 26 27 28
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#ifndef __FILTERING_FUNCTORS_H__
#define __FILTERING_FUNCTORS_H__

#include "Topology/generic/functor.h"
29
#include "Algo/Geometry/intersection.h"
Pierre Kraemer's avatar
Pierre Kraemer committed
30 31 32 33 34 35 36

namespace CGoGN
{

namespace Algo
{

37 38 39
namespace Surface
{

Pierre Kraemer's avatar
Pierre Kraemer committed
40 41 42
namespace Filtering
{

Pierre Kraemer's avatar
Pierre Kraemer committed
43
template <typename ATTR_TYPE>
44
class FunctorAverage : public virtual FunctorType
Pierre Kraemer's avatar
Pierre Kraemer committed
45
{
Pierre Kraemer's avatar
Pierre Kraemer committed
46 47
	typedef typename ATTR_TYPE::DATA_TYPE T ;

Pierre Kraemer's avatar
Pierre Kraemer committed
48
protected:
Pierre Kraemer's avatar
Pierre Kraemer committed
49
	const ATTR_TYPE& attr ;
Pierre Kraemer's avatar
Pierre Kraemer committed
50 51
	T sum ;
	unsigned int count ;
52

Pierre Kraemer's avatar
Pierre Kraemer committed
53
public:
Pierre Kraemer's avatar
Pierre Kraemer committed
54
	FunctorAverage(const ATTR_TYPE& a) : FunctorType(), attr(a), sum(0), count(0)
Pierre Kraemer's avatar
Pierre Kraemer committed
55 56 57 58 59 60 61 62 63
	{}
	bool operator()(Dart d)
	{
		sum += attr[d] ;
		++count ;
		return false ;
	}
	inline void reset() { sum = T(0) ; count = 0 ; }
	inline T getSum() { return sum ; }
64 65 66 67 68 69 70
	inline unsigned int getCount() { return count ; }
	inline T getAverage() { return sum / typename T::DATA_TYPE(count) ; }
} ;

template <typename PFP, typename T>
class FunctorAverageOnSphereBorder : public FunctorMap<typename PFP::MAP>
{
Pierre Kraemer's avatar
Pierre Kraemer committed
71
	typedef typename PFP::MAP::IMPL MAP_IMPL;
72
	typedef typename PFP::VEC3 VEC3;
73

74
protected:
Pierre Kraemer's avatar
Pierre Kraemer committed
75 76
	const VertexAttribute<T, MAP_IMPL>& attr ;
	const VertexAttribute<VEC3, MAP_IMPL>& position ;
77 78 79 80
	VEC3 center;
	typename PFP::REAL radius;
	T sum ;
	unsigned int count ;
81

82
public:
Pierre Kraemer's avatar
Pierre Kraemer committed
83
	FunctorAverageOnSphereBorder(typename PFP::MAP& map, const VertexAttribute<T, MAP_IMPL>& a, const VertexAttribute<VEC3, MAP_IMPL>& p) :
84
		FunctorMap<typename PFP::MAP>(map), attr(a), position(p), sum(0), count(0)
Pierre Kraemer's avatar
Pierre Kraemer committed
85 86 87 88
	{
		center = VEC3(0);
		radius = 0;
	}
89 90 91
	bool operator()(Dart d)
	{
		typename PFP::REAL alpha = 0;
Thery Sylvain's avatar
Thery Sylvain committed
92
		Geometry::intersectionSphereEdge<PFP>(this->m_map, center, radius, d, position, alpha);
93
		sum += (1 - alpha) * attr[d] + alpha * attr[this->m_map.phi1(d)] ;
94 95 96 97 98 99
		++count ;
		return false ;
	}
	inline void reset(VEC3& c, typename PFP::REAL r) { center = c; radius = r; sum = T(0) ; count = 0 ; }
	inline T getSum() { return sum ; }
	inline unsigned int getCount() { return count ; }
Pierre Kraemer's avatar
Pierre Kraemer committed
100 101 102 103 104
	inline T getAverage() { return sum / typename T::DATA_TYPE(count) ; }
} ;

} // namespace Filtering

Pierre Kraemer's avatar
Pierre Kraemer committed
105
} // namespace Surface
106

Pierre Kraemer's avatar
Pierre Kraemer committed
107 108 109 110 111
} // namespace Algo

} // namespace CGoGN

#endif