Galerkin FEM for elliptic PDEs
Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.

FEM4EllipticPDE.h 10KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290
  1. // ===========================================================================
  2. //
  3. // Filename: FEM4EllipticPDE.h
  4. //
  5. // Created: 08/16/12 18:19:41
  6. // Compiler: Tested with g++, icpc, and MSVC 2010
  7. //
  8. // Author: Trevor Irons (ti)
  9. //
  10. // Organisation: Colorado School of Mines (CSM)
  11. // United States Geological Survey (USGS)
  12. //
  13. // Email: tirons@mines.edu, tirons@usgs.gov
  14. //
  15. // This program is free software: you can redistribute it and/or modify
  16. // it under the terms of the GNU General Public License as published by
  17. // the Free Software Foundation, either version 3 of the License, or
  18. // (at your option) any later version.
  19. //
  20. // This program is distributed in the hope that it will be useful,
  21. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  22. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  23. // GNU General Public License for more details.
  24. //
  25. // You should have received a copy of the GNU General Public License
  26. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  27. //
  28. // ===========================================================================
  29. /**
  30. @file
  31. @author Trevor Irons
  32. @date 08/16/12
  33. @version $Rev$
  34. **/
  35. #ifndef FEM4ELLIPTICPDE_INC
  36. #define FEM4ELLIPTICPDE_INC
  37. #include "LemmaObject.h"
  38. #include "vtkImplicitFunction.h"
  39. #include "vtkDataSet.h"
  40. #include "vtkXMLDataSetWriter.h"
  41. #include "vtkSmartPointer.h"
  42. #include "vtkIdList.h"
  43. #include "vtkCell.h"
  44. #include "vtkDoubleArray.h"
  45. #include "vtkPointData.h"
  46. #include "vtkDataSetSurfaceFilter.h"
  47. #include "vtkCellArray.h"
  48. #include "vtkCellData.h"
  49. #include <Eigen/IterativeLinearSolvers>
  50. #include <Eigen/SparseLU>
  51. #include "DCSurvey.h"
  52. namespace Lemma {
  53. /** @defgroup FEM4EllipticPDE
  54. @brief Finite element solver for elliptic PDE's
  55. @details Uses Galerkin method.
  56. */
  57. /** \addtogroup FEM4EllipticPDE
  58. @{
  59. */
  60. // ===================================================================
  61. // Class: FEM4EllipticPDE
  62. /**
  63. @class FEM4EllipticPDE
  64. @brief Galerkin FEM solver for Elliptic PDE problems
  65. @details Solves problems of the type
  66. \f[ - \nabla \cdot ( \sigma(\mathbf{r}) \nabla u(\mathbf{r}) ) = g(\mathbf{r})
  67. \f]
  68. */
  69. class FEM4EllipticPDE : public LemmaObject {
  70. friend std::ostream &operator<<(std::ostream &stream,
  71. const FEM4EllipticPDE &ob);
  72. public:
  73. // ==================== LIFECYCLE =======================
  74. /** Returns a pointer to a new object of type FEM4EllipticPDE.
  75. * It allocates all necessary memory.
  76. */
  77. static FEM4EllipticPDE* New();
  78. /** Deletes this object. Delete also disconnects any
  79. * attachments to this object.
  80. */
  81. void Delete();
  82. // ==================== OPERATORS =======================
  83. // ==================== OPERATIONS =======================
  84. /** Sets the vtkImplictFunction that is used as sigma in
  85. \f$ - \nabla \cdot ( \sigma(\mathbf{r}) \nabla u(\mathbf{r}) ) = g(\mathbf{r})
  86. \f$. If this is not set, sigma evaluates as 1.
  87. */
  88. void SetSigmaFunction(vtkImplicitFunction* sigma);
  89. /** Sets the step to be used in applying boundary conditions.
  90. */
  91. void SetBoundaryStep(const Real& h);
  92. /** Sets the vtkImplictFunction that is used
  93. */
  94. void SetGFunction(vtkImplicitFunction* G);
  95. /** Uses a function pointer instead of a vtkImplicitFunction. For functions that can
  96. * be written in closed form, this can be much faster and more accurate then interpolating
  97. * off a grid.
  98. */
  99. void SetGFunction( Real (*pFcn3)(const Real&, const Real&, const Real&) );
  100. /** Sets the vtkDataSet that defines the calculation grid.
  101. */
  102. void SetGrid(vtkDataSet* Grid);
  103. /** Sets up a DC problem with a survey
  104. * @param[in] ij is the injection index
  105. */
  106. void SetupDC(DCSurvey* Survey, const int& ij);
  107. /** Sets up the potential problem from the VTK file
  108. */
  109. void SetupPotential();
  110. /** Performs solution */
  111. void Solve(const std::string& fname);
  112. /** Performs solution */
  113. void SolveOLD2(const std::string& fname);
  114. // ==================== ACCESS =======================
  115. // ==================== INQUIRY =======================
  116. protected:
  117. // ==================== LIFECYCLE =======================
  118. /// Default protected constructor.
  119. FEM4EllipticPDE (const std::string& cname);
  120. /** Default protected constructor. */
  121. ~FEM4EllipticPDE ();
  122. /**
  123. * @copybrief LemmaObject::Release()
  124. * @copydetails LemmaObject::Release()
  125. */
  126. void Release();
  127. // ==================== OPERATIONS =======================
  128. /** Used internally to construct stiffness matrix, A
  129. */
  130. void ConstructAMatrix();
  131. /* Used internally to construct the load vector, g
  132. */
  133. //void ConstructLoadVector();
  134. // This function is copyright (C) 2012 Joseph Capriotti
  135. /**
  136. Returns a vtkIdList of points in member vtkGrid connected to point ido.
  137. @param[in] ido is the point of interest
  138. @param[in] grid is a pointer to a vtkDataSet
  139. */
  140. vtkSmartPointer<vtkIdList> GetConnectedPoints(const int& id0);
  141. /** Uses Simpson's rule to numerically integrate
  142. * (Shamelessly adapted from http://en.wikipedia.org/wiki/Simpson's_rule
  143. */
  144. Real CompositeSimpsons(vtkImplicitFunction* f, Real r0[3], Real r1[3], int numInt);
  145. /** Uses Simpson's rule to numerically integrate
  146. * (Shamelessly adapted from http://en.wikipedia.org/wiki/Simpson's_rule
  147. * this method is for a static f
  148. */
  149. Real CompositeSimpsons(const Real& f, Real r0[3], Real r1[3], int numInt);
  150. /**
  151. * Uses Simpson's rule to numerically integrate two functions in 1 dimension over the points
  152. * r0 and r1
  153. */
  154. Real CompositeSimpsons2(vtkImplicitFunction* f, Real r0[3], Real r1[3], int numInt);
  155. /**
  156. * Uses Simpson's rule to numerically integrate two functions in 1 dimension over the points
  157. * r0 and r1, uses the Hat function and the function pointer fPtr
  158. */
  159. Real CompositeSimpsons2( Real (*fPtr)(const Real&, const Real&, const Real&), Real r0[3], Real r1[3], int numInt);
  160. /**
  161. * Uses Simpson's rule to numerically integrate two functions in 1 dimension over the points
  162. * r0 and r1, uses the Hat function. Assumes a constand function value f
  163. */
  164. Real CompositeSimpsons2( const Real& f, Real r0[3], Real r1[3], int numInt);
  165. /**
  166. * Uses Simpson's rule to numerically integrate two functions in 1 dimension over the points
  167. * r0 and r1, uses the Hat function. Assumes a linear function from f0 to f1.
  168. */
  169. Real CompositeSimpsons2( const Real& f0, const Real& f1, Real r0[3], Real r1[3], int numInt);
  170. /**
  171. * Uses Simpson's rule to numerically integrate three functions in 1 dimension over the points
  172. * r0 and r1, uses two Hat functions. Assumes a linear function from f0 to f1.
  173. */
  174. Real CompositeSimpsons3( const Real& f0, const Real& f1, Real r0[3], Real r1[3], int numInt);
  175. /** Hat function for use in Galerkin FEM method, actually a 1/2 Hat.
  176. */
  177. //Real hat(Real r[3], Real ri[3], Real rm1[3] );
  178. Real Hat(const Vector3r &r, const Vector3r& ri, const Vector3r& rm1 );
  179. /** Computes distance between r0 and r1
  180. */
  181. Real dist(Real r0[3], Real r1[3]);
  182. /** Computes distance between r0 and r1
  183. */
  184. Real dist(const Vector3r& r0, const Vector3r& r1);
  185. /** Sets up the potential problem from the VTK file
  186. */
  187. void SetupLineSourcePotential();
  188. /** Sets up the potential problem from the VTK file
  189. */
  190. void SetupSurfaceSourcePotential();
  191. /** Sets up the potential problem from the VTK file
  192. */
  193. void SetupPointSourcePotential();
  194. /** Sets up the potential problem from the VTK file
  195. */
  196. void SetupVolumeSourcePotential();
  197. // ==================== DATA MEMBERS =========================
  198. /** The effective step at the boundary condition. Defaults to 1
  199. */
  200. Real BndryH;
  201. /** The sigma value at the boundary condition. Defaults to 1
  202. */
  203. Real BndrySigma;
  204. /** Implicit function defining the \f$\sigma\f$ term in
  205. \f$ - \nabla \cdot ( \sigma(\mathbf{r}) \nabla u(\mathbf{r}) ) = g(\mathbf{r})
  206. \f$.
  207. If set to 1, the Poisson equation is solved. Any type of vtkImplicitFunction may
  208. be used, including vtkImplicitDataSet.
  209. */
  210. vtkSmartPointer<vtkImplicitFunction> vtkSigma;
  211. /** Implicit function defining the \f$g\f$ term in
  212. \f$ - \nabla \cdot ( \sigma(\mathbf{r}) \nabla u(\mathbf{r}) ) = g(\mathbf{r})
  213. \f$.
  214. Typically this is used to define the source term.
  215. */
  216. vtkSmartPointer<vtkImplicitFunction> vtkG;
  217. /** Any type of vtkDataSet may be used as a calculation Grid */
  218. vtkSmartPointer<vtkDataSet> vtkGrid;
  219. /** Function pointer to function describing g */
  220. Real (*gFcn3)(const Real&, const Real&, const Real&);
  221. Eigen::SparseMatrix<Real> A;
  222. VectorXr g;
  223. private:
  224. }; // ----- end of class FEM4EllipticPDE -----
  225. /*! @} */
  226. } // ----- end of Lemma name -----
  227. #endif // ----- #ifndef FEM4ELLIPTICPDE_INC -----