StencilAccessors.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_PHYSICSSOLVERS_FLUIDFLOW_STENCILACCESSORS_HPP_
#define GEOS_PHYSICSSOLVERS_FLUIDFLOW_STENCILACCESSORS_HPP_
 
#include "mesh/ElementRegionManager.hpp"
#include "codingUtilities/traits.hpp"
#include "codingUtilities/Utilities.hpp"
 
#include <tuple>
 
namespace geos
{
 
template< typename ... TRAITS >
class StencilAccessors
{
public:
 
  template< typename TRAIT >
  auto get( TRAIT ) const
  {
    constexpr std::size_t idx = traits::type_list_index< TRAIT, std::tuple< TRAITS ... > >;
    static_assert( idx != std::tuple_size< std::tuple< TRAITS... > >::value, "input trait/stencil does not match the available traits/stencils." );
    return std::get< idx >( m_accessors ).toNestedViewConst();
  }
 
  template< typename TRAIT >
  auto get() const
  {
    return get( TRAIT{} );
  }
 
  StencilAccessors( ElementRegionManager const & elemManager,
                    string const & solverName )
  {
    forEachArgInTuple( std::tuple< TRAITS ... >{}, [&]( auto t, auto idx )
    {
      GEOS_UNUSED_VAR( t );
      using TRAIT = TYPEOFREF( t );
 
      auto & acc = std::get< idx() >( m_accessors );
      acc = elemManager.constructFieldAccessor< TRAIT >();
      acc.setName( solverName + "/accessors/" + TRAIT::key() );
    } );
  }
 
protected:
 
  std::tuple< ElementRegionManager::ElementViewAccessor< traits::ViewTypeConst< typename TRAITS::type > > ... > m_accessors;
 
  StencilAccessors() = default;
};
 
template< typename MATERIAL_TYPE, typename ... TRAITS >
class StencilMaterialAccessors : public StencilAccessors< TRAITS ... >
{
public:
 
  using StencilAccessors< TRAITS ... >::m_accessors;
 
  StencilMaterialAccessors( ElementRegionManager const & elemManager,
                            string const & solverName ):
    StencilAccessors< TRAITS ... >()
  {
    forEachArgInTuple( std::tuple< TRAITS ... >{}, [&]( auto t, auto idx )
    {
      GEOS_UNUSED_VAR( t );
      using TRAIT = TYPEOFREF( t );
 
      auto & acc = std::get< idx() >( m_accessors );
      bool const allowMissingViews = false;
      acc = elemManager.constructMaterialFieldAccessor< MATERIAL_TYPE, TRAIT >( allowMissingViews );
      acc.setName( solverName + "/accessors/" + TRAIT::key() );
    } );
  }
};
 
 
}
 
#endif //GEOS_PHYSICSSOLVERS_FLUIDFLOW_STENCILACCESSORS_HPP_