VectorComponentFunctor.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "MooseFunctor.h"

namespace Moose
{
template <typename T>
class VectorComponentFunctor : public FunctorBase<T>
{
public:
  using typename FunctorBase<T>::ValueType;
  using typename FunctorBase<T>::GradientType;
  using typename FunctorBase<T>::DotType;
  using VectorArg = typename libMesh::TensorTools::IncrementRank<T>::type;
  using VectorFunctor = FunctorBase<VectorArg>;

  VectorComponentFunctor(const VectorFunctor & vector, const unsigned int component)
    : FunctorBase<T>(vector.functorName() + "_" + std::to_string(component)),
      _vector(vector),
      _component(component)
  {
  }

  bool isExtrapolatedBoundaryFace(const FaceInfo & fi,
                                  const Elem * elem,
                                  const Moose::StateArg & state) const override;
  bool hasBlocks(SubdomainID sub_id) const override { return _vector.hasBlocks(sub_id); }

  bool supportsFaceArg() const override final { return _vector.supportsFaceArg(); }
  bool supportsElemSideQpArg() const override final { return _vector.supportsElemSideQpArg(); }

private:
  const VectorFunctor & _vector;

  const unsigned int _component;

  ValueType evaluate(const ElemArg & elem, const StateArg & state) const override final
  {
    return _vector(elem, state)(_component);
  }

  ValueType evaluate(const FaceArg & face, const StateArg & state) const override final
  {
    return _vector(face, state)(_component);
  }

  ValueType evaluate(const ElemQpArg & elem_qp, const StateArg & state) const override final
  {
    return _vector(elem_qp, state)(_component);
  }

  ValueType evaluate(const ElemSideQpArg & elem_side_qp,
                     const StateArg & state) const override final
  {
    return _vector(elem_side_qp, state)(_component);
  }

  ValueType evaluate(const ElemPointArg & elem_point, const StateArg & state) const override final
  {
    return _vector(elem_point, state)(_component);
  }

  ValueType evaluate(const NodeArg & node, const StateArg & state) const override final
  {
    return _vector(node, state)(_component);
  }

  using FunctorBase<T>::evaluateGradient;
  GradientType evaluateGradient(const ElemArg & elem_arg,
                                const StateArg & state) const override final
  {
    return _vector.gradient(elem_arg, state).row(_component);
  }

  GradientType evaluateGradient(const FaceArg & face, const StateArg & state) const override final
  {
    return _vector.gradient(face, state).row(_component);
  }

  using FunctorBase<T>::evaluateGradDot;
  GradientType evaluateGradDot(const ElemArg & elem_arg,
                               const StateArg & state) const override final
  {
    return _vector.gradDot(elem_arg, state).row(_component);
  }

  GradientType evaluateGradDot(const FaceArg & face, const StateArg & state) const override final
  {
    return _vector.gradDot(face, state).row(_component);
  }
};

template <typename T>
bool
VectorComponentFunctor<T>::isExtrapolatedBoundaryFace(const FaceInfo & fi,
                                                      const Elem * elem,
                                                      const Moose::StateArg & state) const
{
  return _vector.isExtrapolatedBoundaryFace(fi, elem, state);
}
}