DistributedFactory.hpp

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
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#pragma once
 
#include "linearAlgebra/DistributedHaloVector.hpp"
#include "linearAlgebra/DistributedMatrix.hpp"
#include "linearAlgebra/DistributedVector.hpp"
#include "pde/common/AbstractDiscretization.hpp"
 
class DistributedFactory
{
public:
 
  DistributedFactory ();
 
  DistributedFactory (CepsUInt globalRows, CepsUInt globalCols,
                      CepsUInt localRows, CepsUInt localCols,
                      const CepsVector<CepsInt>& diagBlockNnz, 
                      const CepsVector<CepsInt>& offdiagBlockNnz);
 
  DistributedFactory (CepsUInt globalRows, CepsUInt globalCols,
                      CepsUInt localRows, CepsUInt localCols,
                      CepsUInt diagBlockNz, CepsUInt offdiagBlockNz);
 
  ~DistributedFactory ();
 
  void
  setGlobalSizes (CepsUInt globalRows, CepsUInt globalCols);
 
  void
  setLocalSizes (CepsUInt localRows, CepsUInt localCols);
 
  void
  setHaloIndices (CepsIndex *haloRows, CepsUInt nHaloRows, CepsIndex *haloCols, CepsUInt nHaloCols);
 
  void
  setNonZeroTemplate (const CepsIndex *d_nnz, const CepsIndex *o_nnz, CepsInt nbLocalRows);
 
  void
  setNonZeroTemplate (CepsIndex d_nz, CepsIndex o_nz);
 
  DVecPtr
  getDistributedVector () const;
 
  DHVecPtr
  getDistributedHaloVector () const;
 
  DMatPtr
  getDistributedMatrix () const;
 
protected:
  // Shared by each procs
  CepsIndex m_globalRows;  
  CepsIndex m_globalCols;  
 
  // Proc specific data
  CepsIndex m_localRows;  
  CepsIndex m_localCols;  
 
  CepsIndex *m_haloRows;   
  CepsIndex  m_nHaloRows;  
  CepsIndex *m_haloCols;   
  CepsIndex  m_nHaloCols;  
 
  CepsInt *m_d_nnz;  
  CepsInt  m_d_nz;   
  CepsInt *m_o_nnz;  
  CepsInt  m_o_nz;   
 
  void
  initialize ();
};