opm-simulators/opm/models/discretization/vcfv/p1fegradientcalculator.hh
Bård Skaflestad 326449fe0b Fix Build After "Unused.hpp" Removal
Needed for configurations that don't have Dune-localfunctions or
which *do* have ALUGrid.
2022-08-04 15:12:33 +02:00

358 lines
15 KiB
C++

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*
This file is part of the Open Porous Media project (OPM).
OPM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.
OPM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with OPM. If not, see <http://www.gnu.org/licenses/>.
Consult the COPYING file in the top-level source directory of this
module for the precise wording of the license and the list of
copyright holders.
*/
/*!
* \file
*
* \copydoc Opm::P1FeGradientCalculator
*/
#ifndef EWOMS_P1FE_GRADIENT_CALCULATOR_HH
#define EWOMS_P1FE_GRADIENT_CALCULATOR_HH
#include "vcfvproperties.hh"
#include <opm/models/discretization/common/fvbasegradientcalculator.hh>
#include <dune/common/fvector.hh>
#include <dune/common/version.hh>
#include <dune/geometry/type.hh>
#if HAVE_DUNE_LOCALFUNCTIONS
#include <dune/localfunctions/lagrange/pqkfactory.hh>
#endif // HAVE_DUNE_LOCALFUNCTIONS
#include <dune/common/fvector.hh>
#include <vector>
namespace Opm {
/*!
* \ingroup FiniteElementDiscretizations
*
* \brief This class calculates gradients of arbitrary quantities at flux integration
* points using first order finite elemens ansatz functions.
*
* This approach can also be used for the vertex-centered finite volume (VCFV)
* discretization.
*/
template<class TypeTag>
class P1FeGradientCalculator : public FvBaseGradientCalculator<TypeTag>
{
using ParentType = FvBaseGradientCalculator<TypeTag>;
using Scalar = GetPropType<TypeTag, Properties::Scalar>;
using Evaluation = GetPropType<TypeTag, Properties::Evaluation>;
using GridView = GetPropType<TypeTag, Properties::GridView>;
using ElementContext = GetPropType<TypeTag, Properties::ElementContext>;
enum { dim = GridView::dimension };
// set the maximum number of degrees of freedom and the maximum
// number of flux approximation points per elements. For this, we
// assume cubes as the type of element with the most vertices...
enum { maxDof = (2 << dim) };
enum { maxFap = maxDof };
using CoordScalar = typename GridView::ctype;
using DimVector = Dune::FieldVector<Scalar, dim>;
#if HAVE_DUNE_LOCALFUNCTIONS
using LocalFiniteElementCache = Dune::PQkLocalFiniteElementCache<CoordScalar, Scalar, dim, 1>;
using LocalFiniteElement = typename LocalFiniteElementCache::FiniteElementType;
using LocalBasisTraits = typename LocalFiniteElement::Traits::LocalBasisType::Traits;
using ShapeJacobian = typename LocalBasisTraits::JacobianType;
#endif // HAVE_DUNE_LOCALFUNCTIONS
public:
/*!
* \brief Precomputes the common values to calculate gradients and
* values of quantities at any flux approximation point.
*
* \param elemCtx The current execution context
*/
template <bool prepareValues = true, bool prepareGradients = true>
void prepare([[maybe_unused]] const ElementContext& elemCtx,
[[maybe_unused]] unsigned timeIdx)
{
if (getPropValue<TypeTag, Properties::UseP1FiniteElementGradients>()) {
#if !HAVE_DUNE_LOCALFUNCTIONS
// The dune-localfunctions module is required for P1 finite element gradients
throw std::logic_error("The dune-localfunctions module is required in oder to use"
" finite element gradients");
#else
const auto& stencil = elemCtx.stencil(timeIdx);
const LocalFiniteElement& localFE = feCache_.get(elemCtx.element().type());
localFiniteElement_ = &localFE;
// loop over all face centeres
for (unsigned faceIdx = 0; faceIdx < stencil.numInteriorFaces(); ++faceIdx) {
const auto& localFacePos = stencil.interiorFace(faceIdx).localPos();
// Evaluate the P1 shape functions and their gradients at all
// flux approximation points.
if (prepareValues)
localFE.localBasis().evaluateFunction(localFacePos, p1Value_[faceIdx]);
if (prepareGradients) {
// first, get the shape function's gradient in local coordinates
std::vector<ShapeJacobian> localGradient;
localFE.localBasis().evaluateJacobian(localFacePos, localGradient);
// convert to a gradient in global space by
// multiplying with the inverse transposed jacobian of
// the position
const auto& geom = elemCtx.element().geometry();
const auto& jacInvT =
geom.jacobianInverseTransposed(localFacePos);
size_t numVertices = elemCtx.numDof(timeIdx);
for (unsigned vertIdx = 0; vertIdx < numVertices; vertIdx++) {
jacInvT.mv(/*xVector=*/localGradient[vertIdx][0],
/*destVector=*/p1Gradient_[faceIdx][vertIdx]);
}
}
}
#endif
}
else
ParentType::template prepare<prepareValues, prepareGradients>(elemCtx, timeIdx);
}
/*!
* \brief Calculates the value of an arbitrary quantity at any
* interior flux approximation point.
*
* \param elemCtx The current execution context
* \param fapIdx The local index of the flux approximation point
* in the current element's stencil.
* \param quantityCallback A callable object returning the value
* of the quantity at an index of a degree of
* freedom
*/
template <class QuantityCallback>
auto calculateScalarValue([[maybe_unused]] const ElementContext& elemCtx,
[[maybe_unused]] unsigned fapIdx,
[[maybe_unused]] const QuantityCallback& quantityCallback) const
-> typename std::remove_reference<typename QuantityCallback::ResultType>::type
{
if (getPropValue<TypeTag, Properties::UseP1FiniteElementGradients>()) {
#if !HAVE_DUNE_LOCALFUNCTIONS
// The dune-localfunctions module is required for P1 finite element gradients
throw std::logic_error("The dune-localfunctions module is required in oder to use"
" finite element gradients");
#else
using QuantityConstType = typename std::remove_reference<typename QuantityCallback::ResultType>::type;
using QuantityType = typename std::remove_const<QuantityConstType>::type;
using Toolbox = MathToolbox<QuantityType>;
// If the user does not want to use two-point gradients, we
// use P1 finite element gradients..
QuantityType value(0.0);
for (unsigned vertIdx = 0; vertIdx < elemCtx.numDof(/*timeIdx=*/0); ++vertIdx) {
if (std::is_same<QuantityType, Scalar>::value ||
elemCtx.focusDofIndex() == vertIdx)
value += quantityCallback(vertIdx)*p1Value_[fapIdx][vertIdx];
else
value += Toolbox::value(quantityCallback(vertIdx))*p1Value_[fapIdx][vertIdx];
}
return value;
#endif
}
else
return ParentType::calculateScalarValue(elemCtx, fapIdx, quantityCallback);
}
/*!
* \brief Calculates the value of an arbitrary quantity at any
* interior flux approximation point.
*
* \param elemCtx The current execution context
* \param fapIdx The local index of the flux approximation point
* in the current element's stencil.
* \param quantityCallback A callable object returning the value
* of the quantity at an index of a degree of
* freedom
*/
template <class QuantityCallback>
auto calculateVectorValue([[maybe_unused]] const ElementContext& elemCtx,
[[maybe_unused]] unsigned fapIdx,
[[maybe_unused]] const QuantityCallback& quantityCallback) const
-> typename std::remove_reference<typename QuantityCallback::ResultType>::type
{
if (getPropValue<TypeTag, Properties::UseP1FiniteElementGradients>()) {
#if !HAVE_DUNE_LOCALFUNCTIONS
// The dune-localfunctions module is required for P1 finite element gradients
throw std::logic_error("The dune-localfunctions module is required in oder to use"
" finite element gradients");
#else
using QuantityConstType = typename std::remove_reference<typename QuantityCallback::ResultType>::type;
using QuantityType = typename std::remove_const<QuantityConstType>::type;
using RawFieldType = decltype(std::declval<QuantityType>()[0]);
using FieldType = typename std::remove_const<typename std::remove_reference<RawFieldType>::type>::type;
using Toolbox = MathToolbox<FieldType>;
// If the user does not want to use two-point gradients, we
// use P1 finite element gradients..
QuantityType value(0.0);
for (unsigned vertIdx = 0; vertIdx < elemCtx.numDof(/*timeIdx=*/0); ++vertIdx) {
if (std::is_same<QuantityType, Scalar>::value ||
elemCtx.focusDofIndex() == vertIdx)
{
const auto& tmp = quantityCallback(vertIdx);
for (unsigned k = 0; k < tmp.size(); ++k)
value[k] += tmp[k]*p1Value_[fapIdx][vertIdx];
}
else {
const auto& tmp = quantityCallback(vertIdx);
for (unsigned k = 0; k < tmp.size(); ++k)
value[k] += Toolbox::value(tmp[k])*p1Value_[fapIdx][vertIdx];
}
}
return value;
#endif
}
else
return ParentType::calculateVectorValue(elemCtx, fapIdx, quantityCallback);
}
/*!
* \brief Calculates the gradient of an arbitrary quantity at any
* flux approximation point.
*
* \param elemCtx The current execution context
* \param fapIdx The local index of the flux approximation point
* in the current element's stencil.
* \param quantityCallback A callable object returning the value
* of the quantity at an index of a degree of
* freedom
*/
template <class QuantityCallback, class EvalDimVector>
void calculateGradient([[maybe_unused]] EvalDimVector& quantityGrad,
[[maybe_unused]] const ElementContext& elemCtx,
[[maybe_unused]] unsigned fapIdx,
[[maybe_unused]] const QuantityCallback& quantityCallback) const
{
if (getPropValue<TypeTag, Properties::UseP1FiniteElementGradients>()) {
#if !HAVE_DUNE_LOCALFUNCTIONS
// The dune-localfunctions module is required for P1 finite element gradients
throw std::logic_error("The dune-localfunctions module is required in oder to use"
" finite element gradients");
#else
using QuantityConstType = typename std::remove_reference<typename QuantityCallback::ResultType>::type;
using QuantityType = typename std::remove_const<QuantityConstType>::type;
// If the user does not want two-point gradients, we use P1 finite element
// gradients...
quantityGrad = 0.0;
for (unsigned vertIdx = 0; vertIdx < elemCtx.numDof(/*timeIdx=*/0); ++vertIdx) {
if (std::is_same<QuantityType, Scalar>::value ||
elemCtx.focusDofIndex() == vertIdx)
{
const auto& dofVal = quantityCallback(vertIdx);
const auto& tmp = p1Gradient_[fapIdx][vertIdx];
for (int dimIdx = 0; dimIdx < dim; ++ dimIdx)
quantityGrad[dimIdx] += dofVal*tmp[dimIdx];
}
else {
const auto& dofVal = quantityCallback(vertIdx);
const auto& tmp = p1Gradient_[fapIdx][vertIdx];
for (int dimIdx = 0; dimIdx < dim; ++ dimIdx)
quantityGrad[dimIdx] += scalarValue(dofVal)*tmp[dimIdx];
}
}
#endif
}
else
ParentType::calculateGradient(quantityGrad, elemCtx, fapIdx, quantityCallback);
}
/*!
* \brief Calculates the value of an arbitrary quantity at any
* flux approximation point on the grid boundary.
*
* Boundary values are always calculated using the two-point
* approximation.
*
* \param elemCtx The current execution context
* \param fapIdx The local index of the flux approximation point
* in the current element's stencil.
* \param quantityCallback A callable object returning the value
* of the quantity at an index of a degree of
* freedom
*/
template <class QuantityCallback>
auto calculateBoundaryValue(const ElementContext& elemCtx,
unsigned fapIdx,
const QuantityCallback& quantityCallback)
-> decltype(ParentType::calculateBoundaryValue(elemCtx, fapIdx, quantityCallback))
{ return ParentType::calculateBoundaryValue(elemCtx, fapIdx, quantityCallback); }
/*!
* \brief Calculates the gradient of an arbitrary quantity at any
* flux approximation point on the boundary.
*
* Boundary gradients are always calculated using the two-point
* approximation.
*
* \param elemCtx The current execution context
* \param fapIdx The local index of the flux approximation point
* in the current element's stencil.
* \param quantityCallback A callable object returning the value
* of the quantity at an index of a degree of
* freedom
*/
template <class QuantityCallback, class EvalDimVector>
void calculateBoundaryGradient(EvalDimVector& quantityGrad,
const ElementContext& elemCtx,
unsigned fapIdx,
const QuantityCallback& quantityCallback) const
{ ParentType::calculateBoundaryGradient(quantityGrad, elemCtx, fapIdx, quantityCallback); }
#if HAVE_DUNE_LOCALFUNCTIONS
static LocalFiniteElementCache& localFiniteElementCache()
{ return feCache_; }
#endif
private:
#if HAVE_DUNE_LOCALFUNCTIONS
static LocalFiniteElementCache feCache_;
const LocalFiniteElement* localFiniteElement_;
std::vector<Dune::FieldVector<Scalar, 1>> p1Value_[maxFap];
DimVector p1Gradient_[maxFap][maxDof];
#endif // HAVE_DUNE_LOCALFUNCTIONS
};
#if HAVE_DUNE_LOCALFUNCTIONS
template<class TypeTag>
typename P1FeGradientCalculator<TypeTag>::LocalFiniteElementCache
P1FeGradientCalculator<TypeTag>::feCache_;
#endif
} // namespace Opm
#endif