KokkosLinearFVDiffusion.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* 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 "KokkosLinearFVFluxKernel.h"
#include "KokkosParsedFunction.h"

class KokkosLinearFVDiffusion : public Moose::Kokkos::LinearFVFluxKernel
{
public:
  static InputParameters validParams();

  KokkosLinearFVDiffusion(const InputParameters & parameters);

  virtual bool needsBoundaryNormalGradientData() const override { return true; }
  virtual bool hasInternalRightHandSideContribution() const override { return false; }

  template <typename Derived>
  KOKKOS_FUNCTION Real computeInternalMatrixContribution(const FVDatum & datum) const;
  template <typename Derived>
  KOKKOS_FUNCTION Real computeInternalNeighborMatrixContribution(const FVDatum & datum) const;
  template <typename Derived>
  KOKKOS_FUNCTION Real computeBoundaryMatrixContribution(
      const FVDatum & datum, const int bc_index, const MOOSE_KOKKOS_INDEX_TYPE bc_face_index) const;
  template <typename Derived>
  KOKKOS_FUNCTION Real computeBoundaryRightHandSideContribution(
      const FVDatum & datum, const int bc_index, const MOOSE_KOKKOS_INDEX_TYPE bc_face_index) const;

private:
  KOKKOS_FUNCTION Real faceConductance(const FVDatum & datum) const;

  KOKKOS_FUNCTION Real faceDiffusionCoefficient(const FVDatum & datum) const;

  Moose::Kokkos::ReferenceWrapper<const KokkosParsedFunction> _diffusion_coeff;
};

template <typename Derived>
KOKKOS_FUNCTION Real
KokkosLinearFVDiffusion::computeInternalMatrixContribution(const FVDatum & datum) const
{
  return faceConductance(datum);
}

template <typename Derived>
KOKKOS_FUNCTION Real
KokkosLinearFVDiffusion::computeInternalNeighborMatrixContribution(const FVDatum & datum) const
{
  return -faceConductance(datum);
}

template <typename Derived>
KOKKOS_FUNCTION Real
KokkosLinearFVDiffusion::computeBoundaryMatrixContribution(
    const FVDatum & datum, const int bc_index, const MOOSE_KOKKOS_INDEX_TYPE bc_face_index) const
{
  return -faceDiffusionCoefficient(datum) * datum.faceArea() *
         boundaryNormalGradientCoefficient(bc_index, bc_face_index);
}

template <typename Derived>
KOKKOS_FUNCTION Real
KokkosLinearFVDiffusion::computeBoundaryRightHandSideContribution(
    const FVDatum & datum, const int bc_index, const MOOSE_KOKKOS_INDEX_TYPE bc_face_index) const
{
  return faceDiffusionCoefficient(datum) * datum.faceArea() *
         boundaryNormalGradientSource(bc_index, bc_face_index);
}

KOKKOS_FUNCTION inline Real
KokkosLinearFVDiffusion::faceConductance(const FVDatum & datum) const
{
  return faceDiffusionCoefficient(datum) * datum.faceArea() / datum.faceDCNMag();
}

KOKKOS_FUNCTION inline Real
KokkosLinearFVDiffusion::faceDiffusionCoefficient(const FVDatum & datum) const
{
  return _diffusion_coeff->value(_t, datum.faceCentroid());
}