DistributedMatrix.cpp

Go to the documentation of this file.

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   This file is part of CEPS.
 
   CEPS is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   CEPS 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 General Public License for more details.
 
   You should have received a copy of the GNU General Public License
   along with CEPS (see file LICENSE at root of project).
   If not, see <https://www.gnu.org/licenses/>.
 
 
   Copyright 2019-2024 Inria, Universite de Bordeaux
 
   Authors, in alphabetical order:
 
     Pierre-Elliott BECUE, Florian CARO, Yves COUDIERE(*), Andjela DAVIDOVIC, 
     Charlie DOUANLA-LONTSI, Marc FUENTES, Mehdi JUHOOR, Michael LEGUEBE(*), 
     Pauline MIGERDITICHAN, Valentin PANNETIER(*), Nejib ZEMZEMI.
     * : currently active authors
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#include "linearAlgebra/DistributedMatrix.hpp"
 
// Depending on the compilation options, use
// one implementation. Default is PETSc
#ifdef CEPS_USE_PETSC
  #include "linearAlgebra/PETScMatrix.cpp"
#endif  // CEPS_USE_PETSC
 
void DistributedMatrix::
addSubMatrix (
  const CepsMathDynamic2D           &subMat,
  const CepsVector<CepsGlobalIndex> &indicesI,
  const CepsVector<CepsGlobalIndex> &indicesJ
)
{
  // Eigen matrices are column oriented
  #ifdef CEPS_USE_PETSC
    MatSetOption(m_A, MAT_ROW_ORIENTED, PETSC_FALSE);
  #endif
  CepsInt N = indicesI.size();
  CepsInt M = indicesJ.size();
  #ifdef CEPS_DEBUG_ENABLED
  for (CepsUInt i=0; i<subMat.rows(); i++)
  for (CepsUInt j=0; j<subMat.cols(); j++)
    ceps::checkNanOrInf(subMat(i,j),
      CepsString("while inserting coefficient in matrix at position (") +
      ceps::toString(indicesI[i]) + "," + ceps::toString(indicesJ[j]) + ")"
    );
  #endif
  addValues(subMat.data(),N,M,indicesI.data(),indicesJ.data());
  return;
}
 
void DistributedMatrix::
getSubMatrix (
  const CepsVector<CepsGlobalIndex> &indicesI,
  const CepsVector<CepsGlobalIndex> &indicesJ,
  CepsMathDynamic2D                 &subMat
)
{
// Eigen matrices are column oriented
#ifdef CEPS_USE_PETSC
  MatSetOption (m_A, MAT_ROW_ORIENTED, PETSC_FALSE);
#endif
  CepsInt N = indicesI.size ();
  subMat.resize (N, N);
  getValues (subMat.data (), N, N, indicesI.data (), indicesJ.data ());
  return;
}
 
const PetscMatrix & DistributedMatrix::
getMatrix () const
{
  return m_A;
}