One dimensional FEM in Concepts. This code features first order Ansatz functions on a (possibly) uniformly refined mesh.
The equation solved is
on (-1,1) with homogeneous Dirichlet boundary conditions. The linear system is solved using CG.
The exact solution is u = -3 cosh(x)/cosh(1) + x^2 + 2.
The relative energy error of the FE solution is computed at the end of the program
System include files for input and output operations.
#include <iostream>
#include "basics.hh" #include "geometry.hh" #include "space.hh" #include "toolbox.hh" #include "function.hh" #include "operator.hh" #include "integration.hh" #include "linearFEM.hh"
concepts:: before Real everytime. using concepts::Real;
The mesh is taken from the meshes tutorial: Line. Here, the relevant code is included:
#include "meshes.cc"
The linear form is used to compute the element contributions to the load vector. The computations are done in the application operator of this class.
class RHS : public concepts::LinearForm<Real> { private:
quad_ of type concepts::Quadrature. The template parameter 4 indicates the type of quadrature rule. It contains the quadrature points and weights for given number of quadrature points (one in our case).
The constructor takes as argument the number of quadrature points and initialises the quadrature object quad_.
const concepts::Quadrature<4> quad_; // gauss jacobi type
elm and is responsible for computing the contributions of this element to the global load vector. The local load vector is stored in em. virtual void operator()(const concepts::Element<Real>& elm, concepts::ElementMatrix<Real>& em) {
em. Check if the general element elm really is an element of type linearFEM::Line which can be handled by our routine. This is achieved by trying to cast elm to linearFEM::Line. If the cast succeeds (ie. elm is really of type linearFEM::Line), then e contains a pointer to this element. If the cast fails, e is equal to 0 and the assertion fails too. If the assertions fails, an error message is displayed (indicating the name of the file, the line number in the file, the name of the routine where the error happened and the check which failed) and the execution of the program stops. const linearFEM::Line* e = dynamic_cast<const linearFEM::Line*>(&elm); conceptsAssert(e, concepts::Assertion());
cell which returns a reference to a concepts::Edge1d and this class is able to compute the size of the cell (on which the element is based). The size of the cell is needed to compute the Jacobian in the numerical integration. Next, we get pointers to the first elements of the arrays of quadrature weights and points. These data were computed in the constructor (which initialised the member quad_). const Real* w = quad_.weights(); // quadrature weights const Real* q = quad_.abscissas(); // quadrature points
There are two nested loops here: the outer loop enumerates the shape funtions of this element and the inner loop enumerates the quadrature points.
for (uint i = 0; i < n; ++i) { // loop over shape functions
m is a reference to the entry of the local load vector which corresponds to the current shape function. Now, all data is known to add to m:
w[k] is the quadrature weight associated to the current quadrature point q[k]. m += w[k]
h is the determinant of the Jacobian of the element map which maps the reference coordinates (0,1) to the physical coordinates of this element. 1/2 is the Jacobian of the map from (-1,1) to (0,1). The integration takes place on (-1,1) (as described in concepts::Quadrature). * h / 2.0
e->shapefct(i, q[k]) is the value of the shape function i at the point q[k]. As you can see in linearFEM::Line, it takes values in (-1,1). * e->shapefct(i, q[k])
pow(e->cell().chi( (1.0+q[k])/2.0 ), 2.0) computes the value of the right hand side funtion f = x2. (1.0+q[k])/2.0 maps a point in (-1,1) to (0,1) and the function pow from the standard math library computes ab. Set up the mesh. The values -1, 0 and 1 give the coordinates of the left, middle and right points in the mesh.
int main(int argc, char** argv) { try { Line msh(-1, 0, 1); std::cout << "Mesh: " << msh << std::endl;
The class Boundary holds a type (Dirichlet or Neumann) and (if necessary) a function or value. These objects can be added to BoundaryConditions which manages them.
concepts::BoundaryConditions bc; bc.add(concepts::Attribute(2), concepts::Boundary(concepts::Boundary::DIRICHLET)); std::cout << "Boundary conditions: " << bc << std::endl;
linearFEM::Linear1d spc(msh, 1, &bc); std::cout << "Space: " << spc << std::endl;
RHS which is a linear form. This linear form is then given to the vector. The vector fills its entries by evaluating the linear form on every element of the space and assembling the local load vectors according to the assembling information contained in the elements. RHS rhs_lf; concepts::Vector<Real> rhs(spc, rhs_lf);
linearFEM::Laplace1d stiff_bf; concepts::SparseMatrix<Real> stiff(spc, stiff_bf); linearFEM::Identity1d mass_bf; concepts::SparseMatrix<Real> mass(spc, mass_bf);
L is formed from the mass and stiffness matrix. The linear system is going to be solved iteratively. This linear combination is used to initialize a CG solver together with the maximal residual and the maximal number of iterations. concepts::LiCo<Real> L(stiff, mass, 1.0, 1.0); concepts::CG<Real> solver(L, 1e-6, 2000);
concepts::Vector<Real> sol(spc);
The CG solver throws a exception of type concepts::NoConvergence which is caught here. Although the solver might not have converged, the solution might give some hints to the user. Therefore, the execution of the program is not stopped here and only a warning message is displayed.
catch(concepts::NoConvergence& e) { std::cout << "solver did not converge!" << std::endl << e << std::endl; }
std::cout << "solver = " << solver << std::endl; std::cout << "solution = " << sol << std::endl;
FE energy error.
const Real exEnergie = 206.0/15.0 + 36.0*exp(2.0)/(exp(2.0)+1.0)/(exp(2.0)+1.0)*sinh(2.0) - 72.0*exp(1.0)/(exp(2.0)+1.0)*sinh(1.0); const Real feEnergie(sol*rhs); std::cout << "energy error = " << fabs(exEnergie-feEnergie)/exEnergie << std::endl; }
Here, the exceptions are caught and some output is generated. Normally, this should not happen. We return a value different from 0 in case an exception is caught.
catch(concepts::ExceptionBase& e) { std::cout << e << std::endl; return 1; }
return 0;
}
Output of the program:
Mesh: Line(Edge1d(idx = (0, 0), cntr = Edge(Key(0), (Vertex(Key(0)), Vertex(Key(1))), Attribute(0)), vtx = [-1, 0], map = MapEdge1d(-1, 0)), Edge1d(idx = (0, 0), cntr = Edge(Key(1), (Vertex(Key(1)), Vertex(Key(2))), Attribute(0)), vtx = [0, 1], map = MapEdge1d(0, 1))) Boundary conditions: BoundaryConditions(2: Boundary(DIRICHLET, (0)), 0: Boundary(FREE, (0))) Space: Linear1d(dim = 3, nelm = 4) solver = CG(solves LiCo(1 * SparseMatrix(3x3, HashedSparseMatrix: 7 (77.7778%) entries bound.) + 1 * SparseMatrix(3x3, HashedSparseMatrix: 7 (77.7778%) entries bound.)), eps = 9.5313e-34, it = 2, relres = 0) solution = Vector(3, [0.0644745, 0.0642467, 0.0644745]) energy error = 0.100759
#include <iostream> #include "basics.hh" #include "geometry.hh" #include "space.hh" #include "toolbox.hh" #include "function.hh" #include "operator.hh" #include "integration.hh" #include "linearFEM.hh" using concepts::Real; #include "meshes.cc" // ************************************************************* Linear Form ** class RHS : public concepts::LinearForm<Real> { private: const concepts::Quadrature<4> quad_; // gauss jacobi type public: RHS() : quad_(1) {} virtual void operator()(const concepts::Element<Real>& elm, concepts::ElementMatrix<Real>& em) { const uint n = elm.T().m(); em.resize(n, 1); em.zeros(); const linearFEM::Line* e = dynamic_cast<const linearFEM::Line*>(&elm); conceptsAssert(e, concepts::Assertion()); const Real h = e->cell().size(); const Real* w = quad_.weights(); // quadrature weights const Real* q = quad_.abscissas(); // quadrature points for (uint i = 0; i < n; ++i) { // loop over shape functions Real& m = em(i,0); for (uint k = 0; k < quad_.n(); ++k) { // loop over quadrature points m += w[k] * h / 2.0 * e->shapefct(i, q[k]) * pow(e->cell().chi( (1.0+q[k])/2.0 ), 2.0); } // for k } // for i } }; // ************************************************************ Main Program ** int main(int argc, char** argv) { try { Line msh(-1, 0, 1); std::cout << "Mesh: " << msh << std::endl; concepts::BoundaryConditions bc; bc.add(concepts::Attribute(2), concepts::Boundary(concepts::Boundary::DIRICHLET)); std::cout << "Boundary conditions: " << bc << std::endl; linearFEM::Linear1d spc(msh, 1, &bc); std::cout << "Space: " << spc << std::endl; RHS rhs_lf; concepts::Vector<Real> rhs(spc, rhs_lf); linearFEM::Laplace1d stiff_bf; concepts::SparseMatrix<Real> stiff(spc, stiff_bf); linearFEM::Identity1d mass_bf; concepts::SparseMatrix<Real> mass(spc, mass_bf); concepts::LiCo<Real> L(stiff, mass, 1.0, 1.0); concepts::CG<Real> solver(L, 1e-6, 2000); concepts::Vector<Real> sol(spc); try { solver(rhs, sol); } catch(concepts::NoConvergence& e) { std::cout << "solver did not converge!" << std::endl << e << std::endl; } std::cout << "solver = " << solver << std::endl; std::cout << "solution = " << sol << std::endl; const Real exEnergie = 206.0/15.0 + 36.0*exp(2.0)/(exp(2.0)+1.0)/(exp(2.0)+1.0)*sinh(2.0) - 72.0*exp(1.0)/(exp(2.0)+1.0)*sinh(1.0); const Real feEnergie(sol*rhs); std::cout << "energy error = " << fabs(exEnergie-feEnergie)/exEnergie << std::endl; } catch(concepts::ExceptionBase& e) { std::cout << e << std::endl; return 1; } return 0; }