DofManager.hpp

Go to the documentation of this file.

/*
 * ------------------------------------------------------------------------------------------------------------
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 * Copyright (c) 2018-2019 Lawrence Livermore National Security LLC
 * Copyright (c) 2018-2019 The Board of Trustees of the Leland Stanford Junior University
 * Copyright (c) 2018-2019 Total, S.A
 * Copyright (c) 2019-     GEOS/GEOSX Contributors
 * All right reserved
 *
 * See top level LICENSE, COPYRIGHT, CONTRIBUTORS, NOTICE, and ACKNOWLEDGEMENTS files for details.
 * ------------------------------------------------------------------------------------------------------------
 */
 
#ifndef GEOS_LINEARALGEBRA_MULTISCALE_DOFMANAGER_HPP
#define GEOS_LINEARALGEBRA_MULTISCALE_DOFMANAGER_HPP
 
#include "linearAlgebra/DofManager.hpp"
 
namespace geos
{
namespace multiscale
{
 
class MeshObjectManager;
 
class DofManager
{
public:
 
  explicit DofManager( string name );
 
  DofManager( DofManager const & ) = delete;
 
  DofManager( DofManager && ) = default;
 
  DofManager & operator=( DofManager const & ) = delete;
 
  DofManager & operator=( DofManager && ) = default;
 
  void clear();
 
  void setDomain( DomainPartition & domain );
 
  void addField( string const & fieldName,
                 integer const components,
                 MeshObjectManager & manager );
 
  void reorderByRank();
 
  std::size_t numFields() const
  {
    return m_fields.size();
  }
 
  bool fieldExists( string const & name ) const;
 
  string const & key( string const & fieldName ) const;
 
  globalIndex numGlobalDofs( string const & fieldName ) const;
 
  globalIndex numGlobalDofs() const;
 
  localIndex numLocalDofs( string const & fieldName ) const;
 
  localIndex numLocalDofs() const;
 
  localIndex localOffset( string const & fieldName ) const;
 
  globalIndex rankOffset( string const & fieldName ) const;
 
  globalIndex rankOffset() const;
 
  integer numComponents( string const & fieldName ) const;
 
  integer numComponents() const;
 
  globalIndex globalOffset( string const & fieldName ) const;
 
  MeshObjectManager const & manager( string const & fieldName ) const;
 
  DomainPartition & domain() { return *m_domain; }
 
  template< typename MATRIX >
  void makeRestrictor( string const & fieldName,
                       MPI_Comm const & comm,
                       bool transpose,
                       MATRIX & restrictor ) const;
 
private:
 
  struct FieldDescription
  {
    string name;                   
    string key;                    
    MeshObjectManager * manager{}; 
    integer numComponents = 1;     
    localIndex numLocalDof = 0;    
    localIndex localOffset = 0;    
    globalIndex numGlobalDof = 0;  
    globalIndex blockOffset = 0;   
    globalIndex rankOffset = 0;    
    globalIndex globalOffset = 0;  
  };
 
  FieldDescription const & getField( string const & name ) const;
 
  void createIndexArray( FieldDescription const & field );
 
  string m_name;
 
  DomainPartition * m_domain = nullptr;
 
  std::vector< FieldDescription > m_fields;
 
  bool m_reordered = false;
};
 
} // multiscale
} // geos
 
#endif //GEOS_LINEARALGEBRA_MULTISCALE_DOFMANAGER_HPP