Skip to content

GitLab

C
CGoGN
Commit 0b944888 authored by Pierre Kraemer's avatar Pierre Kraemer
Browse files
Options

update LinearSolving algo to lambdas

parent 7a132517
Hide whitespace changes
Inline Side-by-side
Showing with 221 additions and 697 deletions
SCHNApps/CMakeLists.txt
View file @ 0b944888
......@@ -2,7 +2,7 @@ cmake_minimum_required(VERSION 2.8)
project(SCHNApps)
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -fPIC")
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wall -fPIC -std=c++11")
SET(CGoGN_ROOT_DIR ${CMAKE_SOURCE_DIR}/.. CACHE STRING "CGoGN root dir")
......
include/Algo/LinearSolving/basic.h
View file @ 0b944888
......@@ -26,9 +26,7 @@
#define __LINEAR_SOLVING_BASIC__
#include "NL/nl.h"
#include "Algo/LinearSolving/variablesSetup.h"
#include "Algo/LinearSolving/matrixSetup.h"
#include "Algo/Topo/basic.h"
#include "Topology/generic/traversor/traversorCell.h"
namespace CGoGN
{
......@@ -36,6 +34,15 @@ namespace CGoGN
namespace LinearSolving
{
template <typename CoeffType>
struct Coeff
{
unsigned int index;
CoeffType value;
Coeff(unsigned int i, CoeffType v) : index(i), value(v)
{}
} ;
/*******************************************************************************
* VARIABLES SETUP
*******************************************************************************/
......@@ -44,29 +51,31 @@ template <typename PFP, typename ATTR_TYPE>
void setupVariables(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const CellMarker<typename PFP::MAP, VERTEX>& fm,
const CellMarker<typename PFP::MAP, VERTEX>& freeMarker,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr)
{
// TraversorV<MAP> t(m);
// for (Dart d = t.begin(); d != t.end(); d = t.next())
// {
// }
FunctorMeshToSolver_Scalar<PFP, ATTR_TYPE> fmts(index, fm, attr) ;
Algo::Topo::foreach_orbit<VERTEX>(m, fmts) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlSetVariable(index[d], attr[d]);
if(!freeMarker.isMarked(d))
nlLockVariable(index[d]);
});
}
template <typename PFP, typename ATTR_TYPE>
void setupVariables(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const CellMarker<typename PFP::MAP, VERTEX>& fm,
const CellMarker<typename PFP::MAP, VERTEX>& freeMarker,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr,
unsigned int coord)
{
FunctorMeshToSolver_Vector<PFP, ATTR_TYPE> fmts(index, fm, attr, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, fmts) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlSetVariable(index[d], (attr[d])[coord]);
if(!freeMarker.isMarked(d))
nlLockVariable(index[d]);
});
}
/*******************************************************************************
......@@ -81,8 +90,16 @@ void addRowsRHS_Equality(
const VertexAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& weight)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorEquality_PerVertexWeight_Scalar<PFP, ATTR_TYPE> feq(index, attr, weight) ;
Algo::Topo::foreach_orbit<VERTEX>(m, feq) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, attr[d]) ;
nlRowParameterd(NL_ROW_SCALING, weight[d]) ;
nlBegin(NL_ROW) ;
nlCoefficient(index[d], 1) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -94,8 +111,16 @@ void addRowsRHS_Equality(
float weight)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorEquality_UniformWeight_Scalar<PFP, ATTR_TYPE> feq(index, attr, weight) ;
Algo::Topo::foreach_orbit<VERTEX>(m, feq) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, attr[d]) ;
nlRowParameterd(NL_ROW_SCALING, weight) ;
nlBegin(NL_ROW) ;
nlCoefficient(index[d], 1) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -108,8 +133,16 @@ void addRowsRHS_Equality(
unsigned int coord)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorEquality_PerVertexWeight_Vector<PFP, ATTR_TYPE> feq(index, attr, weight, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, feq) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attr[d])[coord]) ;
nlRowParameterd(NL_ROW_SCALING, weight[d]) ;
nlBegin(NL_ROW) ;
nlCoefficient(index[d], 1) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -122,8 +155,16 @@ void addRowsRHS_Equality(
unsigned int coord)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorEquality_UniformWeight_Vector<PFP, ATTR_TYPE> feq(index, attr, weight, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, feq) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attr[d])[coord]) ;
nlRowParameterd(NL_ROW_SCALING, weight) ;
nlBegin(NL_ROW) ;
nlCoefficient(index[d], 1) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -134,36 +175,99 @@ void addRowsRHS_Equality(
template <typename PFP>
void addRows_Laplacian_Topo(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index)
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianTopo<PFP> flt(m, index) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flt) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, 0) ;
nlBegin(NL_ROW);
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = 1 ;
aii += aij ;
nlCoefficient(index[m.phi1(it)], aij) ;
}
nlCoefficient(index[d], -aii) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
template <typename PFP, typename ATTR_TYPE>
void addRowsRHS_Laplacian_Topo(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianTopoRHS_Scalar<PFP, ATTR_TYPE> flt(m, index, attr) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flt) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
std::vector<Coeff<typename PFP::REAL> > coeffs ;
coeffs.reserve(12) ;
typename PFP::REAL norm2 = 0 ;
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = 1 ;
aii += aij ;
coeffs.push_back(Coeff<typename PFP::REAL>(index[m.phi1(it)], aij)) ;
norm2 += aij * aij ;
}
coeffs.push_back(Coeff<typename PFP::REAL>(index[d], -aii)) ;
norm2 += aii * aii ;
nlRowParameterd(NL_RIGHT_HAND_SIDE, attr[d] * sqrt(norm2)) ;
nlBegin(NL_ROW);
for(unsigned int i = 0; i < coeffs.size(); ++i)
nlCoefficient(coeffs[i].index, coeffs[i].value) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
template <typename PFP, typename ATTR_TYPE>
void addRowsRHS_Laplacian_Topo(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr,
unsigned int coord)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianTopoRHS_Vector<PFP, ATTR_TYPE> flt(m, index, attr, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flt) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
std::vector<Coeff<typename PFP::REAL> > coeffs ;
coeffs.reserve(12) ;
typename PFP::REAL norm2 = 0 ;
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = 1 ;
aii += aij ;
coeffs.push_back(Coeff<typename PFP::REAL>(index[m.phi1(it)], aij)) ;
norm2 += aij * aij ;
}
coeffs.push_back(Coeff<typename PFP::REAL>(index[d], -aii)) ;
norm2 += aii * aii ;
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attr[d])[coord] * sqrt(norm2)) ;
nlBegin(NL_ROW);
for(unsigned int i = 0; i < coeffs.size(); ++i)
nlCoefficient(coeffs[i].index, coeffs[i].value) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -174,57 +278,108 @@ void addRowsRHS_Laplacian_Topo(
template <typename PFP>
void addRows_Laplacian_Cotan(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const EdgeAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& edgeWeight,
const VertexAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& vertexArea)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianCotan<PFP> flc(m, index, edgeWeight, vertexArea) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flc) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, 0) ;
nlBegin(NL_ROW);
typename PFP::REAL vArea = vertexArea[d] ;
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = edgeWeight[it] / vArea ;
aii += aij ;
nlCoefficient(index[m.phi1(it)], aij) ;
}
nlCoefficient(index[d], -aii) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
template <typename PFP, typename ATTR_TYPE>
void addRowsRHS_Laplacian_Cotan(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const EdgeAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& edgeWeight,
const VertexAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& vertexArea,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianCotanRHS_Scalar<PFP, ATTR_TYPE> flc(m, index, edgeWeight, vertexArea, attr) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flc) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
std::vector<Coeff<typename PFP::REAL> > coeffs ;
coeffs.reserve(12) ;
typename PFP::REAL vArea = vertexArea[d] ;
typename PFP::REAL norm2 = 0 ;
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = edgeWeight[it] / vArea ;
aii += aij ;
coeffs.push_back(Coeff<typename PFP::REAL>(index[m.phi1(it)], aij)) ;
norm2 += aij * aij ;
}
coeffs.push_back(Coeff<typename PFP::REAL>(index[d], -aii)) ;
norm2 += aii * aii ;
nlRowParameterd(NL_RIGHT_HAND_SIDE, attr[d] * sqrt(norm2)) ;
nlBegin(NL_ROW);
for(unsigned int i = 0; i < coeffs.size(); ++i)
nlCoefficient(coeffs[i].index, coeffs[i].value) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
template <typename PFP, typename ATTR_TYPE>
void addRowsRHS_Laplacian_Cotan(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
const EdgeAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& edgeWeight,
const VertexAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& vertexArea,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr,
unsigned int coord)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianCotanRHS_Vector<PFP, ATTR_TYPE> flc(m, index, edgeWeight, vertexArea, attr, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flc) ;
nlDisable(NL_NORMALIZE_ROWS) ;
}
template <typename PFP, typename ATTR_TYPE>
void addRowsRHS_Laplacian_Cotan_NL(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const EdgeAttribute<typename PFP::REAL, typename PFP::MAP::IMPL>& edgeWeight,
const VertexAttribute<typename PFP::REA, typename PFP::MAP::IMPLL>& vertexArea,
const VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr,
unsigned int coord)
{
nlEnable(NL_NORMALIZE_ROWS) ;
FunctorLaplacianCotanRHS_Vector<PFP, ATTR_TYPE> flc(m, index, edgeWeight, vertexArea, attr, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, flc) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
std::vector<Coeff<typename PFP::REAL> > coeffs ;
coeffs.reserve(12) ;
typename PFP::REAL vArea = vertexArea[d] ;
typename PFP::REAL norm2 = 0 ;
typename PFP::REAL aii = 0 ;
Traversor2VE<typename PFP::MAP> t(m, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
typename PFP::REAL aij = edgeWeight[it] / vArea ;
aii += aij ;
coeffs.push_back(Coeff<typename PFP::REAL>(index[m.phi1(it)], aij)) ;
norm2 += aij * aij ;
}
coeffs.push_back(Coeff<typename PFP::REAL>(index[d], -aii)) ;
norm2 += aii * aii ;
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attr[d])[coord] * sqrt(norm2)) ;
nlBegin(NL_ROW);
for(unsigned int i = 0; i < coeffs.size(); ++i)
nlCoefficient(coeffs[i].index, coeffs[i].value) ;
nlEnd(NL_ROW) ;
});
nlDisable(NL_NORMALIZE_ROWS) ;
}
......@@ -235,22 +390,26 @@ void addRowsRHS_Laplacian_Cotan_NL(
template <typename PFP, typename ATTR_TYPE>
void getResult(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr)
{
FunctorSolverToMesh_Scalar<PFP, ATTR_TYPE> fstm(index, attr) ;
Algo::Topo::foreach_orbit<VERTEX>(m, fstm) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
attr[d] = nlGetVariable(index[d]) ;
});
}
template <typename PFP, typename ATTR_TYPE>
void getResult(
typename PFP::MAP& m,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL> index,
const VertexAttribute<unsigned int, typename PFP::MAP::IMPL>& index,
VertexAttribute<ATTR_TYPE, typename PFP::MAP::IMPL>& attr,
unsigned int coord)
{
FunctorSolverToMesh_Vector<PFP, ATTR_TYPE> fstm(index, attr, coord) ;
Algo::Topo::foreach_orbit<VERTEX>(m, fstm) ;
foreach_cell<VERTEX>(m, [&] (Dart d)
{
(attr[d])[coord] = nlGetVariable(index[d]) ;
});
}
} // namespace LinearSolving
......
include/Algo/LinearSolving/matrixSetup.h deleted 100644 → 0
View file @ 7a132517
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps *
* version 0.1 *
* Copyright (C) 2009-2012, 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. *
* *
* Web site: http://cgogn.unistra.fr/ *
* Contact information: cgogn@unistra.fr *
* *
*******************************************************************************/
#ifndef __LINEAR_SOLVING_MATRIX_SETUP__
#define __LINEAR_SOLVING_MATRIX_SETUP__
namespace CGoGN
{
namespace LinearSolving
{
template <typename CoeffType>
struct Coeff
{
unsigned int index;
CoeffType value;
Coeff(unsigned int i, CoeffType v) : index(i), value(v)
{}
} ;
/*******************************************************************************
* EQUALITY MATRIX : right-hand-side SCALAR
*******************************************************************************/
template<typename PFP, typename ATTR_TYPE>
class FunctorEquality_PerVertexWeight_Scalar : public FunctorType
{
typedef typename PFP::MAP::IMPL MAP_IMPL;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attrTable ;
const VertexAttribute<typename PFP::REAL, MAP_IMPL>& weightTable ;
public:
FunctorEquality_PerVertexWeight_Scalar(
const VertexAttribute<unsigned int, MAP_IMPL>& index,
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attr,
const VertexAttribute<typename PFP::REAL, MAP_IMPL>& weight
) : indexTable(index), attrTable(attr), weightTable(weight)
{}
bool operator()(Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, attrTable[d]) ;
nlRowParameterd(NL_ROW_SCALING, weightTable[d]) ;
nlBegin(NL_ROW) ;
nlCoefficient(indexTable[d], 1) ;
nlEnd(NL_ROW) ;
return false ;
}
} ;
template<typename PFP, typename ATTR_TYPE>
class FunctorEquality_UniformWeight_Scalar : public FunctorType
{
typedef typename PFP::MAP::IMPL MAP_IMPL;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attrTable ;
float weight ;
public:
FunctorEquality_UniformWeight_Scalar(
const VertexAttribute<unsigned int, MAP_IMPL>& index,
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attr,
float w
) : indexTable(index), attrTable(attr), weight(w)
{}
bool operator()(Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, attrTable[d]) ;
nlRowParameterd(NL_ROW_SCALING, weight) ;
nlBegin(NL_ROW) ;
nlCoefficient(indexTable[d], 1) ;
nlEnd(NL_ROW) ;
return false ;
}
} ;
/*******************************************************************************
* EQUALITY MATRIX : right-hand-side VECTOR + coordinate
*******************************************************************************/
template<typename PFP, typename ATTR_TYPE>
class FunctorEquality_PerVertexWeight_Vector : public FunctorType
{
typedef typename PFP::MAP::IMPL MAP_IMPL;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attrTable ;
const VertexAttribute<typename PFP::REAL, MAP_IMPL>& weightTable ;
unsigned int coord ;
public:
FunctorEquality_PerVertexWeight_Vector(
const VertexAttribute<unsigned int, MAP_IMPL>& index,
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attr,
const VertexAttribute<typename PFP::REAL, MAP_IMPL>& weight,
unsigned int c
) : indexTable(index), attrTable(attr), weightTable(weight), coord(c)
{}
bool operator()(Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attrTable[d])[coord]) ;
nlRowParameterd(NL_ROW_SCALING, weightTable[d]) ;
nlBegin(NL_ROW) ;
nlCoefficient(indexTable[d], 1) ;
nlEnd(NL_ROW) ;
return false ;
}
} ;
template<typename PFP, typename ATTR_TYPE>
class FunctorEquality_UniformWeight_Vector : public FunctorType
{
typedef typename PFP::MAP::IMPL MAP_IMPL;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attrTable ;
float weight ;
unsigned int coord ;
public:
FunctorEquality_UniformWeight_Vector(
const VertexAttribute<unsigned int, MAP_IMPL>& index,
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attr,
float w,
unsigned int c
) : indexTable(index), attrTable(attr), weight(w), coord(c)
{}
bool operator()(Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, (attrTable[d])[coord]) ;
nlRowParameterd(NL_ROW_SCALING, weight) ;
nlBegin(NL_ROW) ;
nlCoefficient(indexTable[d], 1) ;
nlEnd(NL_ROW) ;
return false ;
}
} ;
/*******************************************************************************
* LAPLACIAN TOPO MATRIX : right-hand-side 0
*******************************************************************************/
template<typename PFP>
class FunctorLaplacianTopo : public FunctorMap<typename PFP::MAP>
{
typedef typename PFP::MAP MAP ;
typedef typename PFP::MAP::IMPL MAP_IMPL;
typedef typename PFP::REAL REAL ;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
public:
FunctorLaplacianTopo(
MAP& m,
const VertexAttribute<unsigned int, MAP_IMPL>& index
) : FunctorMap<MAP>(m), indexTable(index)
{}
bool operator()(Dart d)
{
nlRowParameterd(NL_RIGHT_HAND_SIDE, 0) ;
nlBegin(NL_ROW);
REAL aii = 0 ;
Traversor2VE<MAP> t(this->m_map, d) ;
for(Dart it = t.begin(); it != t.end(); it = t.next())
{
REAL aij = 1 ;
aii += aij ;
nlCoefficient(indexTable[this->m_map.phi1(it)], aij) ;
}
nlCoefficient(indexTable[d], -aii) ;
nlEnd(NL_ROW) ;
return false ;
}
} ;
/*******************************************************************************
* LAPLACIAN TOPO MATRIX : right-hand-side SCALAR
*******************************************************************************/
template<typename PFP, typename ATTR_TYPE>
class FunctorLaplacianTopoRHS_Scalar : public FunctorMap<typename PFP::MAP>
{
typedef typename PFP::MAP MAP ;
typedef typename PFP::MAP::IMPL MAP_IMPL;
typedef typename PFP::REAL REAL ;
protected:
const VertexAttribute<unsigned int, MAP_IMPL>& indexTable ;
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attrTable ;
public:
FunctorLaplacianTopoRHS_Scalar(
MAP& m,
const VertexAttribute<unsigned int, MAP_IMPL>& index,
const VertexAttribute<ATTR_TYPE, MAP_IMPL>& attr
) : FunctorMap<MAP>(m), indexTable(index), attrTable(attr)
{}
bool operator()(Dart d)
{
std::vector<Coeff<REAL> > coeffs ;
coeffs.reserve(12) ;
REAL norm2 = 0 ;
REAL aii = 0 ;
Traversor2VE<MAP> t(this->m_map, d) ;