Homec4science
Diffusion LibMultiScale (master) Browse

libmultiscale/src/math/band_matrix.hh
1e97b3b33259master

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