FaceAoSLexicographicEnumerator.h

Go to the documentation of this file.

// This file is part of the ExaHyPE2 project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#pragma once
 
#include <functional>
#include <string>
 
#include "peano4/utils/Globals.h"
#include "tarch/accelerator/accelerator.h"
#include "tarch/compiler/CompilerSpecificSettings.h"
#include "tarch/la/Vector.h"
#include "tarch/multicore/multicore.h"
 
namespace exahype2::enumerator {
  struct FaceAoSLexicographicEnumerator {
    constexpr GPUCallableInlineMethod FaceAoSLexicographicEnumerator(int faceNumber, int numberOfDoFsPerAxisInCell, int haloSize, int unknowns, int numberOfAuxiliaryVariables):
      _faceNumber(faceNumber),
      _numberOfDoFsPerAxisInCell(numberOfDoFsPerAxisInCell),
      _haloSize(haloSize),
      _unknowns(unknowns),
      _numberOfAuxiliaryVariables(numberOfAuxiliaryVariables) {}
 
    GPUCallableInlineMethod int operator()(const tarch::la::Vector<Dimensions, int>& dofIndex, int unknown) const {
      int base   = 1;
      int result = 0;
      for (int d = 0; d < Dimensions; d++) {
        result += dofIndex(d) * base;
        if (d == _faceNumber % Dimensions) {
#if defined(GPUOffloadingOff)
          assertion2(dofIndex(d) >= 0, toString(), dofIndex);
          assertion2(dofIndex(d) < 2 * _haloSize, toString(), dofIndex);
#endif
          base *= (2 * _haloSize);
        } else {
#if defined(GPUOffloadingOff)
          assertion2(dofIndex(d) >= 0, toString(), dofIndex);
          assertion2(dofIndex(d) < _numberOfDoFsPerAxisInCell, toString(), dofIndex);
#endif
          base *= _numberOfDoFsPerAxisInCell;
        }
      }
      return result * (_unknowns + _numberOfAuxiliaryVariables) + unknown;
    }
 
#if defined(GPUOffloadingOff)
    std::string toString() const;
#endif
 
    GPUCallableInlineMethod int size() const {
      if constexpr (Dimensions == 2) {
        return (_numberOfDoFsPerAxisInCell) * (2 * _haloSize) * (_unknowns + _numberOfAuxiliaryVariables);
      } else {
        return (_numberOfDoFsPerAxisInCell * _numberOfDoFsPerAxisInCell) * (2 * _haloSize) * (_unknowns + _numberOfAuxiliaryVariables);
      }
    }
 
    const int _faceNumber;
    const int _numberOfDoFsPerAxisInCell;
    const int _haloSize;
    const int _unknowns;
    const int _numberOfAuxiliaryVariables;
  };
} // namespace exahype2::enumerator