MFEM  v4.3.0
Finite element discretization library
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Pages
bilininteg_diffusion_mf.cpp
Go to the documentation of this file.
1 // Copyright (c) 2010-2021, Lawrence Livermore National Security, LLC. Produced
2 // at the Lawrence Livermore National Laboratory. All Rights reserved. See files
3 // LICENSE and NOTICE for details. LLNL-CODE-806117.
4 //
5 // This file is part of the MFEM library. For more information and source code
6 // availability visit https://mfem.org.
7 //
8 // MFEM is free software; you can redistribute it and/or modify it under the
9 // terms of the BSD-3 license. We welcome feedback and contributions, see file
10 // CONTRIBUTING.md for details.
11 
12 #include "../general/forall.hpp"
13 #include "bilininteg.hpp"
14 #include "gridfunc.hpp"
15 #include "ceed/diffusion.hpp"
16 
17 using namespace std;
18 
19 namespace mfem
20 {
21 
22 void DiffusionIntegrator::AssembleMF(const FiniteElementSpace &fes)
23 {
24  // Assuming the same element type
25  fespace = &fes;
26  Mesh *mesh = fes.GetMesh();
27  if (mesh->GetNE() == 0) { return; }
28  const FiniteElement &el = *fes.GetFE(0);
29  const IntegrationRule *ir = IntRule ? IntRule : &GetRule(el, el);
30  if (DeviceCanUseCeed())
31  {
32  delete ceedOp;
33  MFEM_VERIFY(!VQ && !MQ,
34  "Only scalar coefficient supported for DiffusionIntegrator"
35  " with libCEED");
36  ceedOp = new ceed::MFDiffusionIntegrator(fes, *ir, Q);
37  return;
38  }
39  MFEM_ABORT("Error: DiffusionIntegrator::AssembleMF only implemented with"
40  " libCEED");
41 }
42 
43 void DiffusionIntegrator::AssembleDiagonalMF(Vector &diag)
44 {
45  if (DeviceCanUseCeed())
46  {
47  ceedOp->GetDiagonal(diag);
48  }
49  else
50  {
51  MFEM_ABORT("Error: DiffusionIntegrator::AssembleDiagonalMF only"
52  " implemented with libCEED");
53  }
54 }
55 
56 void DiffusionIntegrator::AddMultMF(const Vector &x, Vector &y) const
57 {
58  if (DeviceCanUseCeed())
59  {
60  ceedOp->AddMult(x, y);
61  }
62  else
63  {
64  MFEM_ABORT("Error: DiffusionIntegrator::AddMultMF only implemented with"
65  " libCEED");
66  }
67 }
68 
69 }
Abstract class for all finite elements.
Definition: fe.hpp:243
Class for an integration rule - an Array of IntegrationPoint.
Definition: intrules.hpp:90
int GetNE() const
Returns number of elements.
Definition: mesh.hpp:846
Mesh * GetMesh() const
Returns the mesh.
Definition: fespace.hpp:410
Class FiniteElementSpace - responsible for providing FEM view of the mesh, mainly managing the set of...
Definition: fespace.hpp:87
virtual const FiniteElement * GetFE(int i) const
Returns pointer to the FiniteElement in the FiniteElementCollection associated with i'th element in t...
Definition: fespace.cpp:2388
Vector data type.
Definition: vector.hpp:60