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 Total, S.A
 * Copyright (c) 2019-     GEOSX Contributors
 * All rights reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */

#ifndef GEOSX_LINEARALGEBRA_UTILITIES_BLOCKVECTOR_HPP_
#define GEOSX_LINEARALGEBRA_UTILITIES_BLOCKVECTOR_HPP_

#include "linearAlgebra/utilities/BlockVectorView.hpp"

namespace geosx
{

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 geosx

#endif //GEOSX_LINEARALGEBRA_UTILITIES_BLOCKVECTOR_HPP_