BlockOperator.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2020 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2020 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2020 TotalEnergies
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_LINEARALGEBRA_UTILITIES_BLOCKOPERATOR_HPP_
#define GEOS_LINEARALGEBRA_UTILITIES_BLOCKOPERATOR_HPP_
 
#include "linearAlgebra/utilities/BlockOperatorView.hpp"
 
namespace geos
{
 
template< typename VECTOR, typename OPERATOR >
class BlockOperator : public BlockOperatorView< VECTOR, OPERATOR >
{
public:
 
  using Base = BlockOperatorView< VECTOR, OPERATOR >;
 
  using Vector = typename Base::Vector;
 
  BlockOperator( localIndex const nRows, localIndex const nCols );
 
  BlockOperator( BlockOperator const & rhs );
 
  BlockOperator( BlockOperator && rhs );
 
  virtual ~BlockOperator() override = default;
 
private:
 
  void setPointers();
 
  array2d< OPERATOR > m_operatorStorage;
};
 
template< typename VECTOR, typename OPERATOR >
BlockOperator< VECTOR, OPERATOR >::BlockOperator( localIndex const nRows, localIndex const nCols )
  : Base( nRows, nCols ),
  m_operatorStorage( nRows, nCols )
{
  setPointers();
}
 
template< typename VECTOR, typename OPERATOR >
void BlockOperator< VECTOR, OPERATOR >::setPointers()
{
  GEOS_LAI_ASSERT_EQ( this->numBlockRows(), m_operatorStorage.size( 0 ) );
  GEOS_LAI_ASSERT_EQ( this->numBlockCols(), m_operatorStorage.size( 1 ) );
  for( localIndex i = 0; i < m_operatorStorage.size( 0 ); ++i )
  {
    for( localIndex j = 0; j < m_operatorStorage.size( 1 ); ++j )
    {
      this->setPointer( i, j, &m_operatorStorage( i, j ) );
    }
  }
}
 
template< typename VECTOR, typename OPERATOR >
BlockOperator< VECTOR, OPERATOR >::BlockOperator( BlockOperator const & rhs )
  : Base( rhs ),
  m_operatorStorage( rhs.m_operatorStorage )
{
  setPointers();
}
 
template< typename VECTOR, typename OPERATOR >
BlockOperator< VECTOR, OPERATOR >::BlockOperator( BlockOperator && rhs )
  : Base( std::move( rhs ) ),
  m_operatorStorage( std::move( rhs.m_operatorStorage ) )
{
  setPointers();
}
 
} // namespace geos
 
#endif //GEOS_LINEARALGEBRA_UTILITIES_BLOCKOPERATOR_HPP_