StructuredGridUtilities.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_MESH_UTILITIES_STRUCTUREDGRIDUTILITIES_HPP
#define GEOS_MESH_UTILITIES_STRUCTUREDGRIDUTILITIES_HPP
 
#include "common/logger/Logger.hpp"
#include "common/GeosxMacros.hpp"
#include <vector>
 
namespace structuredGrid
{
 
template< int dim >
int dimpower( int n );
 
 
template<> inline int dimpower< 1 >( int n ) { return n; }
template<> inline int dimpower< 2 >( int n ) { return n*n; }
template<> inline int dimpower< 3 >( int n ) { return n*n*n; }
 
 
template< int dim >
void map_index( const int index,
                const int nnx,
                std::vector< int > & indices );
 
 
template<>
inline
void map_index< 1 >( const int index,
                     const int nnx,
                     std::vector< int > & indices )
{
  GEOS_ASSERT_GT( nnx, index );
  GEOS_DEBUG_VAR( nnx );
  indices[0] = index;
}
 
template<>
inline
void map_index< 2 >( const int index,
                     const int nnx,
                     std::vector< int > & indices )
{
  GEOS_ASSERT_GT( nnx*nnx, index );
  indices[0] = index % nnx;
  indices[1] = index / nnx;
}
 
template<>
inline
void map_index< 3 >( const int index,
                     const int nnx,
                     std::vector< int > & indices )
{
  GEOS_ASSERT_GT( nnx*nnx*nnx, index );
  indices[0] = index % nnx;
  indices[1] = (index / nnx) % nnx;
  indices[2] = index / (nnx*nnx);
}
 
 
} /* namespace structuredGrid */
 
#endif /* GEOS_MESH_UTILITIES_STRUCTUREDGRIDUTILITIES_HPP */