FEM2D_POISSON_CG
Finite Element 2D,
Poisson Equation,
Conjugate Gradient Solver

FEM2D_POISSON_CG is a FORTRAN90 program which applies the finite element method to solve a form of Poisson's equation over an arbitrary triangulated region, using sparse matrix storage and a conjugate gradient solver.

The storage format chosen is known as DSP or "sparse triplet" format, which essentially simply saves in three vectors A, IA, JA, which record the value, row and column of every nonzero entry.

To solve the linear system, the CG_RC routine is used, which uses reverse communication to carry out a conjugate gradient procedure.

The Triangulated Region:

The computational region is unknown by the program. The user specifies it by preparing a file containing the coordinates of the nodes, and a file containing the indices of nodes that make up triangles that form a triangulation of the region.

Normally, the user does not type in this information by hand, but has a program fill in the nodes, and perhaps another program that constructs the triangulation. However, in the simplest case, the user might construct a very crude triangulation by hand, and have TRIANGULATION_REFINE refine it to something more reasonable.

For the following ridiculously small example:

        4----5
        |\   |\
        | \  | \
        |  \ |  \
        |   \|   \
        1----2----3
      

         0.0 0.0
         1.0 0.0
         2.0 0.0
         0.0 1.0
         1.0 1.0
      

        1 2 4
        5 4 2
        2 3 5
      

The Poisson Equation:

The program is set up to handle the linear Poisson equation with a right hand side function, and nonhomogeneous Dirichlet boundary conditions. The state variable U(X,Y) is then constrained by:

        - Del H(x,y) Del U(x,y) + K(x,y) * U(x,y) = F(x,y)  inside the region;
                                           U(x,y) = G(x,y)  on the boundary.
      

User Interface:

To specify the right hand side function F(x,y), the linear coefficients H(x,y) and K(x,y) and the boundary condition function G(x,y), the user has to modify a file containing three routines,

• void rhs ( int node_num, double node_xy[], double node_rhs[] ) evaluates the right hand side of function F(x,y) at a list of nodes.
• void h_coef ( int node_num, double node_xy[], double node_h[] ) evaluates the coefficient function H(x,y) at a list of nodes.
• void k_coef ( int node_num, double node_xy[], double node_k[] ) evaluates the coefficient function K(x,y) at a list of nodes.
• void dirichlet_condition ( int node_num, double node_xy[], double node_g[] ) evaluates the Dirichlet boundary condition G(X,Y) at a list of nodes.

To run the program, the user compiles the user routines, links them with FEM2D_POISSON_CG, and runs the executable.

The program writes out a file containing an Encapsulated PostScript image of the nodes and elements, with numbers. If there are too many nodes, the plot may be too cluttered to read. For lower values, however, it is a valuable map of what is going on in the geometry.

The program is also able to write out a file containing the solution value at every node. This file may be used to create contour plots of the solution.

Usage:

Assuming the executable program is called "my_problem", then the program is executed by

my_problem prefix

• prefix_nodes.txt contains the X, Y coordinates of nodes;
• prefix_elements.txt contains triples of node indices that form triangles.

Licensing:

The computer code and data files described and made available on this web page are distributed under the GNU LGPL license.

Languages:

FEM2D_POISSON_CG is available in a C++ version and a FORTRAN90 version and a MATLAB version.

Related Data and Programs:

CG_RC, a FORTRAN90 library which implements the conjugate gradient method for solving a positive definite sparse linear system A*x=b, using reverse communication.

FEM2D_POISSON, a FORTRAN90 program which solves Poisson's equation on a triangulated region, using the finite element method and a banded solver.

FEM2D_POISSON_SPARSE, a FORTRAN90 program which solves the steady (time independent) Poisson equation on an arbitrary 2D triangulated region using a version of GMRES for a sparse solver.

FEM2D_POISSON_CG_BAFFLE, a FORTRAN90 library which defines the geometry of a channel with 13 hexagonal baffles, as well as boundary conditions for a given Poisson problem, and is called by fem2d_poisson_cg as part of a solution procedure.

FEM2D_POISSON_CG_ELL, a FORTRAN90 library which defines the geometry of an L-shaped region, as well as boundary conditions for a given Poisson problem, and is called by fem2d_poisson_cg as part of a solution procedure.

FEM2D_POISSON_CG_LAKE, a FORTRAN90 library which defines the geometry of a lake-shaped region, as well as boundary conditions for a given Poisson problem, and is called by fem2d_poisson_cg as part of a solution procedure.

Reference:

1. Hans Rudolf Schwarz,
   Finite Element Methods,
   Academic Press, 1988,
   ISBN: 0126330107,
   LC: TA347.F5.S3313.
2. Gilbert Strang, George Fix,
   An Analysis of the Finite Element Method,
   Cambridge, 1973,
   ISBN: 096140888X,
   LC: TA335.S77.
3. Olgierd Zienkiewicz,
   The Finite Element Method,
   Sixth Edition,
   Butterworth-Heinemann, 2005,
   ISBN: 0750663200,
   LC: TA640.2.Z54

Source Code:

• fem2d_poisson_cg.f90, the source code;

List of Routines:

• MAIN is the main program of FEM2D_POISSON_CG.
• ASSEMBLE_POISSON_DSP assembles the system for the Poisson equation.
• BASIS_ONE_T3 evaluates a linear basis function.
• CG_RC is a reverse communication conjugate gradient routine.
• CH_CAP capitalizes a single character.
• CH_EQI is a case insensitive comparison of two characters for equality.
• CH_TO_DIGIT returns the integer value of a base 10 digit.
• DIAG_INDEX determines where the diagonal matrix entries are stored.
• DIRICHLET_APPLY_DSP accounts for Dirichlet boundary conditions.
• DSP_IJ_TO_K seeks the compressed index of the (I,J) entry of A.
• DSP_PRINT_SOME prints some of a DSP matrix.
• FILE_COLUMN_COUNT counts the number of columns in the first line of a file.
• FILE_NAME_SPECIFICATION determines the names of the input files.
• FILE_ROW_COUNT counts the number of row records in a file.
• GET_UNIT returns a free FORTRAN unit number.
• I4_HUGE returns a "huge" I4.
• I4_MODP returns the nonnegative remainder of integer division.
• I4_WRAP forces an integer to lie between given limits by wrapping.
• I4COL_COMPARE compares columns I and J of an I4COL.
• I4COL_SORT_A ascending sorts an I4COL.
• I4COL_SWAP swaps columns I and J of an I4COL.
• I4MAT_TRANSPOSE_PRINT_SOME prints some of the transpose of an I4mat.
• I4VEC2_COMPARE compares pairs of integers stored in two vectors.
• I4VEC2_SORT_A ascending sorts a vector of pairs of integers.
• I4MAT_DATA_READ reads data from an I4MAT file.
• I4MAT_HEADER_READ reads the header from an I4MAT.
• QUAD_RULE sets the quadrature rule for assembly.
• R8MAT_DATA_READ reads data from an R8MAT file.
• R8MAT_HEADER_READ reads the header from an R8MAT file.
• R8MAT_TRANSPOSE_PRINT_SOME prints some of an R8MAT, transposed.
• R8MAT_WRITE writes an R8MAT file.
• R8VEC_PRINT_SOME prints "some" of an R8VEC.
• R8VEC_UNIFORM_01 returns a unit pseudorandom R8VEC.
• REFERENCE_TO_PHYSICAL_T3 maps reference points to physical points.
• S_TO_I4 reads an I4 from a string.
• S_TO_I4VEC reads an I4VEC from a string.
• S_TO_R8 reads an R8 from a string.
• S_TO_R8VEC reads an R8VEC from a string.
• S_WORD_COUNT counts the number of "words" in a string.
• SOLUTION_EVALUATE evaluates the solution at a point in an element.
• SOLVE_CG solves a linear system using the conjugate gradient method.
• SORT_HEAP_EXTERNAL externally sorts a list of items into ascending order.
• TIMESTAMP prints the current YMDHMS date as a time stamp.
• TRIANGLE_AREA_2D computes the area of a triangle in 2D.
• TRIANGULATION_ORDER3_ADJ_COUNT counts adjacencies in a triangulation.
• TRIANGULATION_ORDER3_ADJ_SET2 sets adjacencies in a triangulation.
• TRIANGULATION_ORDER3_BOUNDARY_NODE indicates which nodes are on the boundary.
• TRIANGULATION_ORDER3_NEIGHBOR_TRIANGLES determines triangle neighbors.

You can go up one level to the FORTRAN90 source codes.