BlockVector.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_BLOCKVECTOR_HPP_
#define GEOS_LINEARALGEBRA_UTILITIES_BLOCKVECTOR_HPP_
 
#include "linearAlgebra/utilities/BlockVectorView.hpp"
 
namespace geos
{
 
template< typename VECTOR >
class BlockVector : public BlockVectorView< VECTOR >
{
public:
 
  using Base = BlockVectorView< VECTOR >;
 
  explicit BlockVector( localIndex const nBlocks )
    : Base( nBlocks ),
    m_vectorStorage( nBlocks )
  {
    setPointers();
  }
 
  explicit BlockVector()
    : BlockVector( 0 )
  {}
 
  BlockVector( BlockVector const & rhs )
    : Base( rhs ),
    m_vectorStorage( rhs.m_vectorStorage )
  {
    setPointers();
  }
 
  BlockVector( BlockVector && rhs )
    : Base( std::move( rhs ) ),
    m_vectorStorage( std::move( rhs.m_vectorStorage ) )
  {
    setPointers();
  }
 
  explicit BlockVector( BlockVectorView< VECTOR > const & rhs )
    : Base( rhs.blockSize() )
  {
    for( localIndex i = 0; i < rhs.blockSize(); ++i )
    {
      m_vectorStorage.emplace_back( rhs.block( i ) );
    }
    setPointers();
  }
 
  BlockVector & operator=( BlockVector const & x )
  {
    m_vectorStorage = x.m_vectorStorage;
    setPointers();
    return *this;
  }
 
  BlockVector & operator=( BlockVector && x ) noexcept
  {
    m_vectorStorage = std::move( x.m_vectorStorage );
    setPointers();
    return *this;
  }
 
  virtual ~BlockVector() override = default;
 
  void resize( localIndex const nBlocks )
  {
    m_vectorStorage.resize( nBlocks );
    setPointers();
  }
 
private:
 
  void setPointers()
  {
    Base::resize( m_vectorStorage.size() );
    for( localIndex i = 0; i < m_vectorStorage.size(); ++i )
    {
      this->setPointer( i, &m_vectorStorage[i] );
    }
  }
 
  array1d< VECTOR > m_vectorStorage;
};
 
} //namespace geos
 
#endif //GEOS_LINEARALGEBRA_UTILITIES_BLOCKVECTOR_HPP_