79 lines
2.1 KiB
C++
79 lines
2.1 KiB
C++
// Copyright (c) 2010-2023, 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 "../general/forall.hpp"
|
|
#include "bilininteg.hpp"
|
|
#include "gridfunc.hpp"
|
|
#include "ceed/integrators/diffusion/diffusion.hpp"
|
|
|
|
using namespace std;
|
|
|
|
namespace mfem
|
|
{
|
|
|
|
void DiffusionIntegrator::AssembleMF(const FiniteElementSpace &fes)
|
|
{
|
|
// Assuming the same element type
|
|
fespace = &fes;
|
|
Mesh *mesh = fes.GetMesh();
|
|
if (mesh->GetNE() == 0) { return; }
|
|
const FiniteElement &el = *fes.GetFE(0);
|
|
const IntegrationRule *ir = IntRule ? IntRule : &GetRule(el, el);
|
|
if (DeviceCanUseCeed())
|
|
{
|
|
delete ceedOp;
|
|
MFEM_VERIFY(!VQ && !MQ,
|
|
"Only scalar coefficient supported for DiffusionIntegrator"
|
|
" with libCEED");
|
|
const bool mixed = mesh->GetNumGeometries(mesh->Dimension()) > 1 ||
|
|
fes.IsVariableOrder();
|
|
if (mixed)
|
|
{
|
|
ceedOp = new ceed::MixedMFDiffusionIntegrator(*this, fes, Q);
|
|
}
|
|
else
|
|
{
|
|
ceedOp = new ceed::MFDiffusionIntegrator(fes, *ir, Q);
|
|
}
|
|
return;
|
|
}
|
|
MFEM_ABORT("Error: DiffusionIntegrator::AssembleMF only implemented with"
|
|
" libCEED");
|
|
}
|
|
|
|
void DiffusionIntegrator::AssembleDiagonalMF(Vector &diag)
|
|
{
|
|
if (DeviceCanUseCeed())
|
|
{
|
|
ceedOp->GetDiagonal(diag);
|
|
}
|
|
else
|
|
{
|
|
MFEM_ABORT("Error: DiffusionIntegrator::AssembleDiagonalMF only"
|
|
" implemented with libCEED");
|
|
}
|
|
}
|
|
|
|
void DiffusionIntegrator::AddMultMF(const Vector &x, Vector &y) const
|
|
{
|
|
if (DeviceCanUseCeed())
|
|
{
|
|
ceedOp->AddMult(x, y);
|
|
}
|
|
else
|
|
{
|
|
MFEM_ABORT("Error: DiffusionIntegrator::AddMultMF only implemented with"
|
|
" libCEED");
|
|
}
|
|
}
|
|
|
|
}
|