<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0 Transitional//EN">
<html><head><meta name="robots" content="noindex">
<meta http-equiv="Content-Type" content="text/html;charset=iso-8859-1">
<title>ITK: itkFEMSolverCrankNicolson.h Source File</title>
<link href="DoxygenStyle.css" rel="stylesheet" type="text/css">
</head><body bgcolor="#ffffff">
<!-- Section customized for INSIGHT : Tue Jul 17 01:02:45 2001 -->
<center>
<a href="index.html" class="qindex">Main Page</a>
<a href="modules.html" class="qindex">Groups</a>
<a href="namespaces.html" class="qindex">Namespace List</a>
<a href="hierarchy.html" class="qindex">Class Hierarchy</a>
<a href="classes.html" class="qindex">Alphabetical List</a>
<a href="annotated.html" class="qindex">Compound List</a>
<a href="files.html" class="qindex">File
List</a>
<a href="namespacemembers.html" class="qindex">Namespace Members</a>
<a href="functions.html" class="qindex">Compound Members</a>
<a href="globals.html" class="qindex">File Members</a>
<a href="pages.html" class="qindex">Concepts</a></center>
<!-- Generated by Doxygen 1.5.9 -->
<div class="navpath"><a class="el" href="dir_02e23d841c198393a6e961e9a0678fff.html">Code</a> » <a class="el" href="dir_2c33022d19e8077162952ea9ed57656d.html">Numerics</a> » <a class="el" href="dir_316e5ab5c8a9cb56a9dce1c9fc454a23.html">FEM</a>
</div>
<div class="contents">
<h1>itkFEMSolverCrankNicolson.h</h1><a href="itkFEMSolverCrankNicolson_8h.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*=========================================================================</span>
<a name="l00002"></a>00002 <span class="comment"></span>
<a name="l00003"></a>00003 <span class="comment"> Program: Insight Segmentation & Registration Toolkit</span>
<a name="l00004"></a>00004 <span class="comment"> Module: $RCSfile: itkFEMSolverCrankNicolson.h,v $</span>
<a name="l00005"></a>00005 <span class="comment"> Language: C++</span>
<a name="l00006"></a>00006 <span class="comment"> Date: $Date: 2009-01-30 21:53:03 $</span>
<a name="l00007"></a>00007 <span class="comment"> Version: $Revision: 1.19 $</span>
<a name="l00008"></a>00008 <span class="comment"></span>
<a name="l00009"></a>00009 <span class="comment"> Copyright (c) Insight Software Consortium. All rights reserved.</span>
<a name="l00010"></a>00010 <span class="comment"> See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.</span>
<a name="l00011"></a>00011 <span class="comment"></span>
<a name="l00012"></a>00012 <span class="comment"> This software is distributed WITHOUT ANY WARRANTY; without even </span>
<a name="l00013"></a>00013 <span class="comment"> the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR </span>
<a name="l00014"></a>00014 <span class="comment"> PURPOSE. See the above copyright notices for more information.</span>
<a name="l00015"></a>00015 <span class="comment"></span>
<a name="l00016"></a>00016 <span class="comment">=========================================================================*/</span>
<a name="l00017"></a>00017
<a name="l00018"></a>00018 <span class="preprocessor">#ifndef __itkFEMSolverCrankNicolson_h</span>
<a name="l00019"></a>00019 <span class="preprocessor"></span><span class="preprocessor">#define __itkFEMSolverCrankNicolson_h</span>
<a name="l00020"></a>00020 <span class="preprocessor"></span>
<a name="l00021"></a>00021 <span class="preprocessor">#include "<a class="code" href="itkFEMSolver_8h.html">itkFEMSolver.h</a>"</span>
<a name="l00022"></a>00022 <span class="preprocessor">#include "<a class="code" href="itkFEMElementBase_8h.html">itkFEMElementBase.h</a>"</span>
<a name="l00023"></a>00023 <span class="preprocessor">#include "<a class="code" href="itkFEMMaterialBase_8h.html">itkFEMMaterialBase.h</a>"</span>
<a name="l00024"></a>00024 <span class="preprocessor">#include "<a class="code" href="itkFEMLoadBase_8h.html">itkFEMLoadBase.h</a>"</span>
<a name="l00025"></a>00025 <span class="preprocessor">#include "<a class="code" href="itkFEMLinearSystemWrapperVNL_8h.html">itkFEMLinearSystemWrapperVNL.h</a>"</span>
<a name="l00026"></a>00026
<a name="l00027"></a>00027 <span class="preprocessor">#include "vnl/vnl_sparse_matrix.h"</span>
<a name="l00028"></a>00028 <span class="preprocessor">#include "vnl/vnl_matrix.h"</span>
<a name="l00029"></a>00029 <span class="preprocessor">#include "vnl/vnl_vector.h"</span>
<a name="l00030"></a>00030 <span class="preprocessor">#include "vnl/algo/vnl_svd.h"</span>
<a name="l00031"></a>00031 <span class="preprocessor">#include "vnl/algo/vnl_cholesky.h"</span>
<a name="l00032"></a>00032 <span class="preprocessor">#include <vnl/vnl_sparse_matrix_linear_system.h></span>
<a name="l00033"></a>00033 <span class="preprocessor">#include <vnl/algo/vnl_lsqr.h></span>
<a name="l00034"></a>00034 <span class="preprocessor">#include <math.h></span>
<a name="l00035"></a>00035
<a name="l00036"></a>00036
<a name="l00037"></a>00037 <span class="keyword">namespace </span>itk {
<a name="l00038"></a>00038 <span class="keyword">namespace </span>fem {
<a name="l00039"></a>00039
<a name="l00040"></a>00040
<a name="l00063"></a>00063 <span class="keyword">class </span>SolverCrankNicolson : <span class="keyword">public</span> Solver
<a name="l00064"></a>00064 {
<a name="l00065"></a>00065 <span class="keyword">public</span>:
<a name="l00067"></a>00067
<a name="l00071"></a>00071 <span class="keywordtype">void</span> InitializeForSolution();
<a name="l00072"></a>00072
<a name="l00077"></a>00077 <span class="keywordtype">void</span> AssembleKandM();
<a name="l00078"></a>00078
<a name="l00086"></a>00086 <span class="keywordtype">void</span> AssembleFforTimeStep(<span class="keywordtype">int</span> dim=0);
<a name="l00087"></a>00087
<a name="l00091"></a>00091 <span class="keywordtype">void</span> Solve();
<a name="l00092"></a>00092
<a name="l00097"></a>00097 <span class="keywordtype">void</span> AddToDisplacements(Float optimum=1.0);
<a name="l00098"></a>00098 <span class="keywordtype">void</span> AverageLastTwoDisplacements(Float t=0.5);
<a name="l00099"></a>00099 <span class="keywordtype">void</span> ZeroVector(<span class="keywordtype">int</span> which=0);
<a name="l00100"></a>00100 <span class="keywordtype">void</span> PrintDisplacements();
<a name="l00101"></a>00101 <span class="keywordtype">void</span> PrintForce();
<a name="l00103"></a>00103
<a name="l00105"></a>00105 <span class="keyword">inline</span> <span class="keywordtype">void</span> SetAlpha(Float a = 0.5) { m_alpha=a; }
<a name="l00106"></a>00106
<a name="l00108"></a>00108 <span class="keyword">inline</span> <span class="keywordtype">void</span> SetDeltatT(Float T) { m_deltaT=T; }
<a name="l00109"></a>00109
<a name="l00111"></a>00111 <span class="keyword">inline</span> <span class="keywordtype">void</span> SetRho(Float rho) { m_rho=rho; }
<a name="l00112"></a>00112
<a name="l00116"></a>00116 <span class="keywordtype">void</span> RecomputeForceVector(<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> index);
<a name="l00117"></a>00117
<a name="l00118"></a>00118 <span class="comment">/* Finds a triplet that brackets the energy minimum. From Numerical Recipes.*/</span>
<a name="l00119"></a>00119 <span class="keywordtype">void</span> FindBracketingTriplet(Float* a,Float* b,Float* c);
<a name="l00120"></a>00120
<a name="l00124"></a>00124 Float GoldenSection(Float tol=0.01,<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> MaxIters=25);
<a name="l00125"></a>00125 <span class="comment">/* Brents method from Numerical Recipes. */</span>
<a name="l00126"></a>00126 Float BrentsMethod(Float tol=0.01,<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> MaxIters=25);
<a name="l00127"></a>00127 Float EvaluateResidual(Float t=1.0);
<a name="l00128"></a>00128 Float GetDeformationEnergy(Float t=1.0);
<a name="l00129"></a>00129 <span class="keyword">inline</span> Float GSSign(Float a,Float b) { <span class="keywordflow">return</span> (b > 0.0 ? vcl_fabs(a) : -1.*vcl_fabs(a)); }
<a name="l00130"></a>00130 <span class="keyword">inline</span> Float GSMax(Float a,Float b) { <span class="keywordflow">return</span> (a > b ? a : b); }
<a name="l00132"></a>00132
<a name="l00133"></a>00133 <span class="keywordtype">void</span> SetEnergyToMin(Float xmin);
<a name="l00134"></a>00134 <span class="keyword">inline</span> LinearSystemWrapper* GetLS(){ <span class="keywordflow">return</span> m_ls;}
<a name="l00135"></a>00135
<a name="l00136"></a>00136 Float GetCurrentMaxSolution() { <span class="keywordflow">return</span> m_CurrentMaxSolution; }
<a name="l00137"></a>00137
<a name="l00140"></a>00140 <span class="keywordtype">void</span> PrintMinMaxOfSolution();
<a name="l00141"></a>00141
<a name="l00146"></a>00146 SolverCrankNicolson()
<a name="l00147"></a>00147 {
<a name="l00148"></a>00148 m_deltaT=0.5;
<a name="l00149"></a>00149 m_rho=1.;
<a name="l00150"></a>00150 m_alpha=0.5;
<a name="l00151"></a>00151 <span class="comment">// BUG FIXME NOT SURE IF SOLVER IS USING VECTOR INDEX 1 FOR BCs</span>
<a name="l00152"></a>00152 ForceTIndex=0; <span class="comment">// vector</span>
<a name="l00153"></a>00153 ForceTMinus1Index=2; <span class="comment">// vector</span>
<a name="l00154"></a>00154 SolutionVectorTMinus1Index=3; <span class="comment">// vector</span>
<a name="l00155"></a>00155 DiffMatrixBySolutionTMinus1Index=4; <span class="comment">// vector</span>
<a name="l00156"></a>00156 ForceTotalIndex=5; <span class="comment">// vector</span>
<a name="l00157"></a>00157 SolutionTIndex=0; <span class="comment">// solution</span>
<a name="l00158"></a>00158 TotalSolutionIndex=1; <span class="comment">// solution</span>
<a name="l00159"></a>00159 SolutionTMinus1Index=2; <span class="comment">// solution</span>
<a name="l00160"></a>00160 SumMatrixIndex=0; <span class="comment">// matrix</span>
<a name="l00161"></a>00161 DifferenceMatrixIndex=1; <span class="comment">// matrix</span>
<a name="l00162"></a>00162 m_CurrentMaxSolution=1.0;
<a name="l00163"></a>00163 }
<a name="l00164"></a>00164
<a name="l00165"></a>00165
<a name="l00166"></a>00166 ~SolverCrankNicolson() { }
<a name="l00167"></a>00167
<a name="l00168"></a>00168 Float m_deltaT;
<a name="l00169"></a>00169 Float m_rho;
<a name="l00170"></a>00170 Float m_alpha;
<a name="l00171"></a>00171 Float m_CurrentMaxSolution;
<a name="l00172"></a>00172
<a name="l00173"></a>00173 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> ForceTIndex;
<a name="l00174"></a>00174 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> ForceTotalIndex;
<a name="l00175"></a>00175 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> ForceTMinus1Index;
<a name="l00176"></a>00176 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SolutionTIndex;
<a name="l00177"></a>00177 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SolutionTMinus1Index;
<a name="l00178"></a>00178 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SolutionVectorTMinus1Index;
<a name="l00179"></a>00179 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> TotalSolutionIndex;
<a name="l00180"></a>00180 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> DifferenceMatrixIndex;
<a name="l00181"></a>00181 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> SumMatrixIndex;
<a name="l00182"></a>00182 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> DiffMatrixBySolutionTMinus1Index;
<a name="l00183"></a>00183
<a name="l00184"></a>00184 };
<a name="l00185"></a>00185
<a name="l00186"></a>00186 }} <span class="comment">// end namespace itk::fem</span>
<a name="l00187"></a>00187
<a name="l00188"></a>00188 <span class="preprocessor">#endif // #ifndef __itkFEMSolverCrankNicolson_h</span>
<a name="l00189"></a>00189 <span class="preprocessor"></span>
</pre></div></div>
<hr><address><small>
Generated at Thu May 7 22:42:07 2009 for ITK by <a href="http://www.stack.nl/~dimitri/doxygen/index.html"> <img
src="http://www.stack.nl/~dimitri/doxygen/doxygen.png" alt="doxygen"
align="middle" border=0 width=110 height=53>
</a> 1.5.9 written by <a href="mailto:dimitri@stack.nl">Dimitri van Heesch</a>,
© 1997-2000</small></address>
</body>
</html>
distrib
>
Mandriva
>
2010.0
>
i586
>
media
>
contrib-release
>
by-pkgid
>
d5e74628f0e673bb8680aebce32b2c04
>
files
>
15779
