83 lines
2.9 KiB
C++
83 lines
2.9 KiB
C++
// Copyright (c) 2010-2020, Lawrence Livermore National Security, LLC. Produced
|
|
// at the Lawrence Livermore National Laboratory. All Rights reserved. See files
|
|
// LICENSE and NOTICE for details. LLNL-CODE-806117.
|
|
//
|
|
// This file is part of the MFEM library. For more information and source code
|
|
// availability visit https://mfem.org.
|
|
//
|
|
// MFEM is free software; you can redistribute it and/or modify it under the
|
|
// terms of the BSD-3 license. We welcome feedback and contributions, see file
|
|
// CONTRIBUTING.md for details.
|
|
|
|
#include "convection.hpp"
|
|
|
|
#include "../../config/config.hpp"
|
|
#ifdef MFEM_USE_CEED
|
|
#include "convection.h"
|
|
#endif
|
|
|
|
namespace mfem
|
|
{
|
|
|
|
CeedPAConvectionIntegrator::CeedPAConvectionIntegrator(
|
|
const FiniteElementSpace &fes,
|
|
const mfem::IntegrationRule &irm,
|
|
VectorCoefficient *Q,
|
|
const double alpha)
|
|
: CeedPAIntegrator()
|
|
{
|
|
#ifdef MFEM_USE_CEED
|
|
Mesh &mesh = *fes.GetMesh();
|
|
// Perform checks for some assumptions made in the Q-functions.
|
|
MFEM_VERIFY(mesh.Dimension() == mesh.SpaceDimension(), "case not supported");
|
|
MFEM_VERIFY(fes.GetVDim() == 1 || fes.GetVDim() == mesh.Dimension(),
|
|
"case not supported");
|
|
int dim = mesh.Dimension();
|
|
ConvectionContext ctx;
|
|
InitCeedVecCoeff(Q, mesh, irm, coeff, ctx);
|
|
ctx.alpha = alpha;
|
|
bool const_coeff = coeff->IsConstant();
|
|
std::string build_func = const_coeff ? ":f_build_conv_const" : ":f_build_conv_quad";
|
|
CeedQFunctionUser build_qf = const_coeff ? f_build_conv_const : f_build_conv_quad;
|
|
CeedPAOperator convOp = {fes, irm,
|
|
dim * (dim + 1) / 2, "/convection.h",
|
|
build_func, build_qf,
|
|
":f_apply_conv", f_apply_conv,
|
|
EvalMode::Grad,
|
|
EvalMode::Interp
|
|
};
|
|
Assemble(convOp, ctx);
|
|
#else
|
|
mfem_error("MFEM must be built with MFEM_USE_CEED=YES to use libCEED.");
|
|
#endif
|
|
}
|
|
|
|
CeedMFConvectionIntegrator::CeedMFConvectionIntegrator(
|
|
const FiniteElementSpace &fes,
|
|
const mfem::IntegrationRule &irm,
|
|
VectorCoefficient *Q,
|
|
const double alpha)
|
|
: CeedMFIntegrator()
|
|
{
|
|
#ifdef MFEM_USE_CEED
|
|
Mesh &mesh = *fes.GetMesh();
|
|
ConvectionContext ctx;
|
|
InitCeedVecCoeff(Q, mesh, irm, coeff, ctx);
|
|
ctx.alpha = alpha;
|
|
bool const_coeff = coeff->IsConstant();
|
|
std::string apply_func = const_coeff ? ":f_apply_conv_mf_const" : ":f_apply_conv_mf_quad";
|
|
CeedQFunctionUser apply_qf = const_coeff ? f_apply_conv_mf_const : f_apply_conv_mf_quad;
|
|
CeedMFOperator convOp = {fes, irm,
|
|
"/convection.h",
|
|
apply_func, apply_qf,
|
|
EvalMode::Grad,
|
|
EvalMode::Interp
|
|
};
|
|
Assemble(convOp, ctx);
|
|
#else
|
|
mfem_error("MFEM must be built with MFEM_USE_CEED=YES to use libCEED.");
|
|
#endif
|
|
}
|
|
|
|
} // namespace mfem
|