NormalOperator.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2016-2024 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2024 TotalEnergies
 * Copyright (c) 2018-2024 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2023-2024 Chevron
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_LINEARALGEBRA_NORMALOPERATOR_HPP_
#define GEOS_LINEARALGEBRA_NORMALOPERATOR_HPP_
 
#include "common/LinearOperator.hpp"
 
namespace geos
{
 
template< typename LAI >
class NormalOperator : public LinearOperator< typename LAI::ParallelVector >
{
public:
 
  using Base = LinearOperator< typename LAI::ParallelVector >;
 
  using Vector = typename Base::Vector;
 
  using Matrix = typename LAI::ParallelMatrix;
 
  explicit NormalOperator( Matrix const & mat )
    : m_matrix( mat )
  {}
 
  virtual ~NormalOperator() override = default;
 
  void apply( Vector const & src, Vector & dst ) const override
  {
    m_matrix.gemv( 1.0, src, 0.0, dst, false );
    m_matrix.gemv( 1.0, dst, 0.0, dst, true );
  }
 
  globalIndex numGlobalRows() const override
  {
    return m_matrix.numGlobalCols();
  }
 
  globalIndex numGlobalCols() const override
  {
    return m_matrix.numGlobalCols();
  }
 
  localIndex numLocalRows() const override
  {
    return m_matrix.numLocalCols();
  }
 
  localIndex numLocalCols() const override
  {
    return m_matrix.numLocalCols();
  }
 
  MPI_Comm comm() const override
  {
    return m_matrix.comm();
  }
 
private:
 
  Matrix const & m_matrix;
};
 
} // namespace geos
 
#endif //GEOS_LINEARALGEBRA_NORMALOPERATOR_HPP_