TransposeOperator.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_TRANSPOSEMATRIXOPERATOR_HPP_
#define GEOS_LINEARALGEBRA_TRANSPOSEMATRIXOPERATOR_HPP_
 
#include "linearAlgebra/common/LinearOperator.hpp"
 
namespace geos
{
 
template< typename LAI >
class TransposeOperator : public LinearOperator< typename LAI::ParallelVector >
{
public:
 
  using Base = LinearOperator< typename LAI::ParallelVector >;
 
  using Vector = typename Base::Vector;
 
  using Matrix = typename LAI::ParallelMatrix;
 
  explicit TransposeOperator( Matrix const & mat )
    : Base(),
    m_matrix( mat )
  { }
 
  virtual ~TransposeOperator() override = default;
 
  virtual void apply( Vector const & src, Vector & dst ) const override
  {
    m_matrix.applyTranspose( src, dst );
  }
 
  virtual globalIndex numGlobalRows() const override
  {
    return m_matrix.numGlobalCols();
  }
 
  virtual globalIndex numGlobalCols() const override
  {
    return m_matrix.numGlobalRows();
  }
 
  virtual localIndex numLocalRows() const override
  {
    return m_matrix.numLocalCols();
  }
 
  virtual localIndex numLocalCols() const override
  {
    return m_matrix.numLocalRows();
  }
 
  virtual MPI_Comm comm() const override
  {
    return m_matrix.comm();
  }
 
private:
 
  Matrix const & m_matrix;
};
 
}
 
#endif //GEOS_LINEARALGEBRA_TRANSPOSEMATRIXOPERATOR_HPP_