Homec4science
Diffusion LibMultiScale (features/reshape_compute_python) Browse

libmultiscale/src/math/band_matrix.hh
b05d5f2aba78features/reshape_comput…

Last Change
Raw File

band_matrix.hh
View Options

/**
* @file band_matrix.hh
*
* @author Guillaume Anciaux <guillaume.anciaux@epfl.ch>
*
* @date Fri Jul 11 15:47:44 2014
*
* @brief This is a wrapper to a matrix with a band storage
*
* @section LICENSE
*
* Copyright INRIA and CEA
*
* The LibMultiScale is a C++ parallel framework for the multiscale
* coupling methods dedicated to material simulations. This framework
* provides an API which makes it possible to program coupled simulations
* and integration of already existing codes.
*
* This Project was initiated in a collaboration between INRIA Futurs Bordeaux
* within ScAlApplix team and CEA/DPTA Ile de France.
* The project is now continued at the Ecole Polytechnique Fédérale de Lausanne
* within the LSMS/ENAC laboratory.
*
* This software is governed by the CeCILL-C license under French law and
* abiding by the rules of distribution of free software. You can use,
* modify and/ or redistribute the software under the terms of the CeCILL-C
* license as circulated by CEA, CNRS and INRIA at the following URL
* "http://www.cecill.info".
*
* As a counterpart to the access to the source code and rights to copy,
* modify and redistribute granted by the license, users are provided only
* with a limited warranty and the software's author, the holder of the
* economic rights, and the successive licensors have only limited
* liability.
*
* In this respect, the user's attention is drawn to the risks associated
* with loading, using, modifying and/or developing or reproducing the
* software by the user in light of its specific status of free software,
* that may mean that it is complicated to manipulate, and that also
* therefore means that it is reserved for developers and experienced
* professionals having in-depth computer knowledge. Users are therefore
* encouraged to load and test the software's suitability as regards their
* requirements in conditions enabling the security of their systems and/or
* data to be ensured and, more generally, to use and operate it in the
* same conditions as regards security.
*
* The fact that you are presently reading this means that you have had
* knowledge of the CeCILL-C license and that you accept its terms.
*
*/
#ifndef __LIBMULTISCALE_BAND_MATRIX_HH__
#define __LIBMULTISCALE_BAND_MATRIX_HH__
/* -------------------------------------------------------------------------- */
#include "matrix.hh"
#include "string.h"
/* -------------------------------------------------------------------------- */
extern "C" {
libmultiscale::UInt * factoLUBand(libmultiscale::Real * A,
libmultiscale::UInt kl,
libmultiscale::UInt ku,
libmultiscale::UInt n);
libmultiscale::UInt solveBandMatrix(libmultiscale::Real * AB,
libmultiscale::Real * B,
libmultiscale::UInt kl,
libmultiscale::UInt ku,
libmultiscale::UInt n,
libmultiscale::UInt * pivot);
}
/* -------------------------------------------------------------------------- */
__BEGIN_LIBMULTISCALE__
/** Class that implements a band matrix as a higher layer of Lapack */
class BandeMatrix : public MyMatrix<BandeMatrix>
{
/* ------------------------------------------------------------------------ */
/* Constructors/Destructors */
/* ------------------------------------------------------------------------ */
public:
BandeMatrix(UInt taille,UInt kl,UInt ku):MyMatrix<BandeMatrix>(taille,taille){
this->kl = kl;
this->ku = ku;
dim = 2*kl+ku+1;
matrix = new Real[taille*dim];
zero();
nulle=0;
};
~BandeMatrix(){};
/* ------------------------------------------------------------------------ */
/* Methods */
/* ------------------------------------------------------------------------ */
Real& operator()(UInt i,UInt j){
UInt k = j-i;
// diagonale superieure
if (k>ku || k < -kl){
nulle = 0;
return nulle;
}
return matrix[kl+ku-k+j*dim];
}
UInt factoLU(){
pivot = factoLUBand(matrix,kl,ku,_m);
if (pivot==0)
return UINT_MAX;
else
return 0;
}
UInt solve(Array * r){
solveBandMatrix(matrix,r->getArray(),kl,ku,_m,pivot);
return 0;
}
UInt zero(){
memset(matrix,0,_m*dim*sizeof(Real)); // mise a zero
return 0;
}
UInt printCompressed(const char * f){
LM_TOIMPLEMENT;
// std::ofstream file(f);
// for (UInt i=0;i< dim;++i)
// for (UInt j=0;j< _m;++j){
// file << i+1 << "\t" << j+1 << "\t"
// << std::setprecision(15)
// << std::scientific << matrix[i+dim*j] << std::endl;
// }
// return 0;
}
/* ------------------------------------------------------------------------ */
/* Class Members */
/* ------------------------------------------------------------------------ */
private:
UInt kl,ku,dim;
Real nulle;
};
/* -------------------------------------------------------------------------- */
__END_LIBMULTISCALE__
#endif /* __LIBMULTISCALE_BAND_MATRIX_HH__ */
c4science · Help