MultiscalePreconditioner.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2019 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2019 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2019 Total, S.A
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All right reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_LINEARALGEBRA_MULTISCALE_MULTISCALEPRECONDITIONER_HPP_
#define GEOS_LINEARALGEBRA_MULTISCALE_MULTISCALEPRECONDITIONER_HPP_
 
#include "linearAlgebra/common/PreconditionerBase.hpp"
#include "linearAlgebra/utilities/LinearSolverParameters.hpp"
#include "mesh/MeshLevel.hpp"
 
#include <memory>
 
namespace geos
{
 
namespace multiscale
{
template< typename LAI >
class LevelBuilderBase;
}
 
template< typename LAI >
class MultiscalePreconditioner : public PreconditionerBase< LAI >
{
public:
 
  using Base = PreconditionerBase< LAI >;
 
  using Vector = typename Base::Vector;
 
  using Matrix = typename Base::Matrix;
 
  using Operator = LinearOperator< Vector >;
 
  MultiscalePreconditioner( LinearSolverParameters params, DomainPartition & domain );
 
  ~MultiscalePreconditioner();
 
  virtual void setup( Matrix const & mat ) override;
 
  virtual void apply( Vector const & src, Vector & dst ) const override;
 
  virtual void clear() override;
 
private:
 
  void createLevels( Matrix const & mat );
 
  void computeLevel( integer const level ) const;
 
  void printLevelInfo() const;
 
  void logMessage( integer const minLevel, string const & msg ) const;
 
  struct LevelData
  {
    std::unique_ptr< multiscale::LevelBuilderBase< LAI > > builder;
    Matrix const * matrix; 
    mutable Vector rhs;    
    mutable Vector sol;    
    mutable Vector tmp;    
  };
 
  LinearSolverParameters m_params;
 
  DomainPartition & m_domain;
 
  std::vector< LevelData > m_levels;
 
  std::unique_ptr< PreconditionerBase< LAI > > m_coarse_solver;
 
  bool m_initialized{ false };
};
 
} // namespace geos
 
#endif //GEOS_LINEARALGEBRA_MULTISCALE_MULTISCALEPRECONDITIONER_HPP_