VertexCFD_Closure_ElectricPotentialDiffusionFlux_impl.hpp

#ifndef VERTEXCFD_CLOSURE_ELECTRICPOTENTIALDIFFUSIONFLUX_IMPL_HPP
#define VERTEXCFD_CLOSURE_ELECTRICPOTENTIALDIFFUSIONFLUX_IMPL_HPP
 
#include <Panzer_HierarchicParallelism.hpp>
 
#include <string>
 
namespace VertexCFD
{
namespace ClosureModel
{
//---------------------------------------------------------------------------//
template<class EvalType, class Traits>
ElectricPotentialDiffusionFlux<EvalType, Traits>::ElectricPotentialDiffusionFlux(
    const panzer::IntegrationRule& ir,
    const std::string& flux_prefix,
    const std::string& gradient_prefix)
    : _electric_potential_flux(
          flux_prefix + "ELECTRIC_POTENTIAL_FLUX_electric_potential_equation",
          ir.dl_vector)
    , _sigma("electrical_conductivity", ir.dl_scalar)
    , _grad_electric_potential(gradient_prefix + "GRAD_electric_potential",
                               ir.dl_vector)
    , _num_grad_dim(ir.spatial_dimension)
{
    // Evaluated fields
    this->addEvaluatedField(_electric_potential_flux);
 
    // Dependent fields
    this->addDependentField(_sigma);
    this->addDependentField(_grad_electric_potential);
 
    this->setName("Electric Potential Diffusion Flux "
                  + std::to_string(_num_grad_dim) + "D");
}
 
//---------------------------------------------------------------------------//
template<class EvalType, class Traits>
void ElectricPotentialDiffusionFlux<EvalType, Traits>::evaluateFields(
    typename Traits::EvalData workset)
{
    auto policy = panzer::HP::inst().teamPolicy<scalar_type, PHX::Device>(
        workset.num_cells);
    Kokkos::parallel_for(this->getName(), policy, *this);
}
 
//---------------------------------------------------------------------------//
template<class EvalType, class Traits>
KOKKOS_INLINE_FUNCTION void
ElectricPotentialDiffusionFlux<EvalType, Traits>::operator()(
    const Kokkos::TeamPolicy<PHX::exec_space>::member_type& team) const
{
    const int cell = team.league_rank();
    const int num_point = _electric_potential_flux.extent(1);
 
    Kokkos::parallel_for(
        Kokkos::TeamThreadRange(team, 0, num_point), [&](const int point) {
            for (int dim = 0; dim < _num_grad_dim; ++dim)
            {
                _electric_potential_flux(cell, point, dim)
                    = -_sigma(cell, point)
                      * _grad_electric_potential(cell, point, dim);
            }
        });
}
 
//---------------------------------------------------------------------------//
 
} // end namespace ClosureModel
} // end namespace VertexCFD
 
#endif // end VERTEXCFD_CLOSURE_ELECTRICPOTENTIALDIFFUSIONFLUX_IMPL_HPP