main.cpp

来自「OFELI is an object oriented library of C」· C++ 代码 · 共 88 行

/*==============================================================================

                                    O  F  E  L  I

                            Object  Finite  Element  Library

  ==============================================================================

   Copyright (C) 1998 - 2004 Rachid Touzani

   This program is free software; you can redistribute it and/or modify it under
   the terms of the GNU General Public License as published by the Free 
   Software Foundation; Version 2 of the License.

   This program is distributed in the hope that it will be useful, but WITHOUT
   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
   FOR A PARTICULAR PURPOSE. See the GNU General Public License for more 
   details.

   You should have received a copy of the GNU General Public License 
   along with this program; if not, write to the :

   Free Software Foundation
   Inc., 59 Temple Place - Suite 330
   Boston, MA  02111-1307, USA

  ==============================================================================

               An example of a Finite Element Code using OFELI

            Solution of a 2-D Poisson problem using P1 Finite elements  

  ==============================================================================*/

#include "OFELI.h"
#include "Therm.h"
using namespace OFELI;

int main(int argc, char *argv[])
{
   Element *el;
   banner("ex2");

// Read and output mesh data
   if (argc <= 1) {
      cout << " Usage : ex2 <mesh_file>" << endl;
      exit(1);
   }
   Mesh ms(argv[1]);
   ms.Verbose(10);
   cout << ms;

// Declare problem data (matrix, rhs, boundary conditions, body forces)
   SkSMatrix<double> a(ms);
   Vect<double> b(ms.NbDOF()), bc(ms.BC());

// Loop over elements
// ------------------

   for (ms.TopElement(); (el=ms.GetElement());) {

//    Declare an instance of class DC2DT3 (contains equation)
      DC2DT3 eq(el);

//    Diffusion contribution to matrix
      eq.Diffusion();

//    Assemble element matrix and RHS
      a.Assembly(el,eq.A());
      b.Assembly(el,eq.b());
   }

// Impose boundary conditions on matrix and RHS
   a.Prescribe(ms,b,bc);

// Solve the linear system of equations
   a.Factor();
   a.Solve(b);

// Output solution
   cout << "\nSolution :\n" << b;

#ifdef WITH_PAUSE
   system("PAUSE");
#endif
   return 0;
}