ThermalDiffusionDispersionFluxComputeKernel.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_PHYSICSSOLVERS_FLUIDFLOW_COMPOSITIONAL_THERMALDIFFUSIONDISPERSIONFLUXCOMPUTEKERNEL_HPP
#define GEOS_PHYSICSSOLVERS_FLUIDFLOW_COMPOSITIONAL_THERMALDIFFUSIONDISPERSIONFLUXCOMPUTEKERNEL_HPP
 
#include "physicsSolvers/fluidFlow/kernels/compositional/DiffusionDispersionFluxComputeKernel.hpp"
 
namespace geos
{
 
namespace thermalCompositionalMultiphaseFVMKernels
{
 
/******************************** DiffusionDispersionFluxComputeKernel ********************************/
 
template< integer NUM_COMP, integer NUM_DOF, typename STENCILWRAPPER >
class DiffusionDispersionFluxComputeKernel :
  public isothermalCompositionalMultiphaseFVMKernels::DiffusionDispersionFluxComputeKernel< NUM_COMP, NUM_DOF, STENCILWRAPPER >
{
public:
 
  template< typename VIEWTYPE >
  using ElementViewConst = ElementRegionManager::ElementViewConst< VIEWTYPE >;
 
  using AbstractBase = isothermalCompositionalMultiphaseFVMKernels::FluxComputeKernelBase;
  using DofNumberAccessor = AbstractBase::DofNumberAccessor;
  using CompFlowAccessors = AbstractBase::CompFlowAccessors;
  using MultiFluidAccessors = AbstractBase::MultiFluidAccessors;
  using AbstractBase::m_dt;
  using AbstractBase::m_dPhaseCompFrac;
  using AbstractBase::m_dPhaseVolFrac;
 
  using Base = typename isothermalCompositionalMultiphaseFVMKernels::DiffusionDispersionFluxComputeKernel< NUM_COMP, NUM_DOF, STENCILWRAPPER >;
  using DiffusionAccessors = typename Base::DiffusionAccessors;
  using DispersionAccessors = typename Base::DispersionAccessors;
  using PorosityAccessors = typename Base::PorosityAccessors;
  using Base::numFluxSupportPoints;
  using Base::numEqn;
  using Base::numComp;
  using Base::numDof;
  using Base::m_referencePorosity;
  using Base::m_phaseVolFrac;
  using Base::m_phaseDens;
  using Base::m_dPhaseDens;
  using Base::m_phaseDiffusivityMultiplier;
 
  DiffusionDispersionFluxComputeKernel( integer const numPhases,
                                        globalIndex const rankOffset,
                                        STENCILWRAPPER const & stencilWrapper,
                                        DofNumberAccessor const & dofNumberAccessor,
                                        CompFlowAccessors const & compFlowAccessors,
                                        MultiFluidAccessors const & multiFluidAccessors,
                                        DiffusionAccessors const & diffusionAccessors,
                                        DispersionAccessors const & dispersionAccessors,
                                        PorosityAccessors const & porosityAccessors,
                                        real64 const dt,
                                        CRSMatrixView< real64, globalIndex const > const & localMatrix,
                                        arrayView1d< real64 > const & localRhs,
                                        BitFlags< isothermalCompositionalMultiphaseFVMKernels::KernelFlags > kernelFlags )
    : Base( numPhases,
            rankOffset,
            stencilWrapper,
            dofNumberAccessor,
            compFlowAccessors,
            multiFluidAccessors,
            diffusionAccessors,
            dispersionAccessors,
            porosityAccessors,
            dt,
            localMatrix,
            localRhs,
            kernelFlags )
  {}
 
  struct StackVariables : public Base::StackVariables
  {
public:
 
    GEOS_HOST_DEVICE
    StackVariables( localIndex const size, localIndex numElems )
      : Base::StackVariables( size, numElems )
    {}
 
    using Base::StackVariables::transmissibility;
    using Base::StackVariables::localFlux;
    using Base::StackVariables::localFluxJacobian;
 
  };
 
  GEOS_HOST_DEVICE
  inline
  void computeDiffusionFlux( localIndex const iconn,
                             StackVariables & stack ) const
  {
    using Deriv = constitutive::multifluid::DerivativeOffset;
 
    // ***********************************************
    // First, we call the base computeFlux to compute the diffusionFlux and its derivatives (including derivatives wrt temperature),
    //
    // We use the lambda below (called **inside** the phase loop of the base computeFlux) to access these variables
    Base::computeDiffusionFlux( iconn, stack, [&] ( integer const ip,
                                                    integer const ic,
                                                    localIndex const (&k)[2],
                                                    localIndex const (&seri)[2],
                                                    localIndex const (&sesri)[2],
                                                    localIndex const (&sei)[2],
                                                    localIndex const connectionIndex,
                                                    localIndex const k_up,
                                                    localIndex const er_up,
                                                    localIndex const esr_up,
                                                    localIndex const ei_up,
                                                    real64 const compFracGrad,
                                                    real64 const upwindCoefficient )
    {
      // We are in the loop over phases and components, ip provides the current phase index.
 
      real64 dCompFracGrad_dT[numFluxSupportPoints]{};
      real64 dDiffusionFlux_dT[numFluxSupportPoints]{};
 
      for( integer i = 0; i < numFluxSupportPoints; i++ )
      {
        localIndex const er  = seri[i];
        localIndex const esr = sesri[i];
        localIndex const ei  = sei[i];
 
        dCompFracGrad_dT[i] += stack.transmissibility[connectionIndex][i] * m_dPhaseCompFrac[er][esr][ei][0][ip][ic][Deriv::dT];
      }
 
      // add contributions of the derivatives of component fractions wrt pressure/component fractions
      for( integer ke = 0; ke < numFluxSupportPoints; ++ke )
      {
        dDiffusionFlux_dT[ke] += upwindCoefficient * dCompFracGrad_dT[ke];
      }
 
      // add contributions of the derivatives of upwind coefficient wrt temperature
      real64 const dUpwindCoefficient_dT =
        m_referencePorosity[er_up][esr_up][ei_up] *
        m_phaseDiffusivityMultiplier[er_up][esr_up][ei_up][0][ip] *
        ( m_dPhaseDens[er_up][esr_up][ei_up][0][ip][Deriv::dT] * m_phaseVolFrac[er_up][esr_up][ei_up][ip]
          + m_phaseDens[er_up][esr_up][ei_up][0][ip] * m_dPhaseVolFrac[er_up][esr_up][ei_up][ip][Deriv::dT] );
      dDiffusionFlux_dT[k_up] += dUpwindCoefficient_dT * compFracGrad;
 
      // finally, increment local flux and local Jacobian
      integer const eqIndex0 = k[0] * numEqn + ic;
      integer const eqIndex1 = k[1] * numEqn + ic;
 
      for( integer ke = 0; ke < numFluxSupportPoints; ++ke )
      {
        localIndex const localDofIndexTemp = k[ke] * numDof + numComp + 1;
        stack.localFluxJacobian[eqIndex0][localDofIndexTemp] += m_dt * dDiffusionFlux_dT[ke];
        stack.localFluxJacobian[eqIndex1][localDofIndexTemp] -= m_dt * dDiffusionFlux_dT[ke];
      }
    } );
  }
 
  GEOS_HOST_DEVICE
  inline
  void computeDispersionFlux( localIndex const iconn,
                              StackVariables & stack ) const
  {
    using Deriv = constitutive::multifluid::DerivativeOffset;
 
    // ***********************************************
    // First, we call the base computeFlux to compute the dispersionFlux and its derivatives (including derivatives wrt temperature),
    //
    // We use the lambda below (called **inside** the phase loop of the base computeFlux) to access these variables
    Base::computeDispersionFlux( iconn, stack, [&] ( integer const ip,
                                                     integer const ic,
                                                     localIndex const (&k)[2],
                                                     localIndex const (&seri)[2],
                                                     localIndex const (&sesri)[2],
                                                     localIndex const (&sei)[2],
                                                     localIndex const connectionIndex,
                                                     localIndex const k_up,
                                                     localIndex const er_up,
                                                     localIndex const esr_up,
                                                     localIndex const ei_up,
                                                     real64 const compFracGrad )
    {
      // We are in the loop over phases and components, ip provides the current phase index.
 
      real64 dCompFracGrad_dT[numFluxSupportPoints]{};
      real64 dDispersionFlux_dT[numFluxSupportPoints]{};
 
      for( integer i = 0; i < numFluxSupportPoints; i++ )
      {
        localIndex const er  = seri[i];
        localIndex const esr = sesri[i];
        localIndex const ei  = sei[i];
 
        dCompFracGrad_dT[i] += stack.transmissibility[connectionIndex][i] * m_dPhaseCompFrac[er][esr][ei][0][ip][ic][Deriv::dT];
      }
 
      // add contributions of the derivatives of component fractions wrt pressure/component fractions
      for( integer ke = 0; ke < numFluxSupportPoints; ++ke )
      {
        dDispersionFlux_dT[ke] += m_phaseDens[er_up][esr_up][ei_up][0][ip] * dCompFracGrad_dT[ke];
      }
 
      // add contributions of the derivatives of upwind coefficient wrt temperature
      dDispersionFlux_dT[k_up] += m_dPhaseDens[er_up][esr_up][ei_up][0][ip][Deriv::dT] * compFracGrad;
 
      // finally, increment local flux and local Jacobian
      integer const eqIndex0 = k[0] * numEqn + ic;
      integer const eqIndex1 = k[1] * numEqn + ic;
 
      for( integer ke = 0; ke < numFluxSupportPoints; ++ke )
      {
        localIndex const localDofIndexTemp = k[ke] * numDof + numComp + 1;
        stack.localFluxJacobian[eqIndex0][localDofIndexTemp] += m_dt * dDispersionFlux_dT[ke];
        stack.localFluxJacobian[eqIndex1][localDofIndexTemp] -= m_dt * dDispersionFlux_dT[ke];
      }
    } );
  }
};
 
class DiffusionDispersionFluxComputeKernelFactory
{
public:
 
  template< typename POLICY, typename STENCILWRAPPER >
  static void
  createAndLaunch( integer const numComps,
                   integer const numPhases,
                   globalIndex const rankOffset,
                   string const & dofKey,
                   BitFlags< isothermalCompositionalMultiphaseFVMKernels::KernelFlags > kernelFlags,
                   string const & solverName,
                   ElementRegionManager const & elemManager,
                   STENCILWRAPPER const & stencilWrapper,
                   real64 const dt,
                   CRSMatrixView< real64, globalIndex const > const & localMatrix,
                   arrayView1d< real64 > const & localRhs )
  {
    isothermalCompositionalMultiphaseBaseKernels::
      internal::kernelLaunchSelectorCompSwitch( numComps, [&]( auto NC )
    {
      integer constexpr NUM_COMP = NC();
      integer constexpr NUM_DOF = NC() + 2;
 
      ElementRegionManager::ElementViewAccessor< arrayView1d< globalIndex const > > dofNumberAccessor =
        elemManager.constructArrayViewAccessor< globalIndex, 1 >( dofKey );
      dofNumberAccessor.setName( solverName + "/accessors/" + dofKey );
 
      using kernelType = DiffusionDispersionFluxComputeKernel< NUM_COMP, NUM_DOF, STENCILWRAPPER >;
      typename kernelType::CompFlowAccessors compFlowAccessors( elemManager, solverName );
      typename kernelType::MultiFluidAccessors multiFluidAccessors( elemManager, solverName );
      typename kernelType::DiffusionAccessors diffusionAccessors( elemManager, solverName );
      typename kernelType::DispersionAccessors dispersionAccessors( elemManager, solverName );
      typename kernelType::PorosityAccessors porosityAccessors( elemManager, solverName );
 
      kernelType kernel( numPhases, rankOffset, stencilWrapper,
                         dofNumberAccessor, compFlowAccessors, multiFluidAccessors,
                         diffusionAccessors, dispersionAccessors, porosityAccessors,
                         dt, localMatrix, localRhs, kernelFlags );
      kernelType::template launch< POLICY >( stencilWrapper.size(),
                                             kernelFlags.isSet( isothermalCompositionalMultiphaseFVMKernels::KernelFlags::Diffusion ),
                                             kernelFlags.isSet( isothermalCompositionalMultiphaseFVMKernels::KernelFlags::Dispersion ),
                                             kernel );
    } );
  }
};
 
} // namespace thermalCompositionalMultiphaseFVMKernels
 
} // namespace geos
 
 
#endif //GEOS_PHYSICSSOLVERS_FLUIDFLOW_COMPOSITIONAL_THERMALDIFFUSIONDISPERSIONFLUXCOMPUTEKERNEL_HPP