ITK  4.9.0
Insight Segmentation and Registration Toolkit
itkFEMRobustSolver.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright Insight Software Consortium
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 
19 #ifndef itkFEMRobustSolver_h
20 #define itkFEMRobustSolver_h
21 
22 #include "itkFEMSolver.h"
23 #include "itkFEMElementBase.h"
24 #include "itkFEMMaterialBase.h"
25 #include "itkFEMLoadBase.h"
29 
30 #include <cmath>
31 
32 namespace itk
33 {
34 namespace fem
35 {
82 
83 template <unsigned int VDimension = 3>
84 class RobustSolver : public Solver<VDimension>
85 {
86 public:
88  typedef RobustSolver Self;
92 
94  itkNewMacro(Self);
95 
97  itkTypeMacro(RobustSolver, Solver);
98 
100  typedef typename Superclass::VectorType VectorType;
101  typedef typename Superclass::Float Float;
109  typedef typename InterpolationGridType::DirectionType InterpolationGridDirectionType;
110 
111  itkStaticConstMacro(FEMDimension, unsigned int, VDimension);
112 
114 
116  typedef typename Element::MatrixType MatrixType;
117  typedef typename FEMObjectType::LoadContainerType LoadContainerType;
118  typedef typename FEMObjectType::NodeContainerType NodeContainerType;
119  typedef typename FEMObjectType::LoadContainerIterator LoadContainerIterator;
120 
124  itkSetMacro(ApproximationSteps, unsigned int);
125  itkGetMacro(ApproximationSteps, unsigned int);
127 
131  itkSetMacro(OutlierRejectionSteps, unsigned int);
132  itkGetMacro(OutlierRejectionSteps, unsigned int);
134 
140  itkSetMacro(ToleranceToLargestDisplacement, double);
141  itkGetMacro(ToleranceToLargestDisplacement, double);
143 
150  itkSetMacro(TradeOffImageMeshEnergy, double);
151  itkGetMacro(TradeOffImageMeshEnergy, double);
153 
158  itkSetMacro(FractionErrorRejected, double);
159  itkGetMacro(FractionErrorRejected, double);
161 
167  itkSetMacro(UseInterpolationGrid, bool);
168  itkGetMacro(UseInterpolationGrid, bool);
170 
171 protected:
172 
177  RobustSolver();
178  ~RobustSolver();
180 
183  virtual void GenerateData() ITK_OVERRIDE;
184 
188  virtual void RunSolver(void) ITK_OVERRIDE;
189 
193  void Initialization();
194 
202 
207  void InitializeLandmarks();
208 
214 
219  virtual void AssembleElementMatrixWithID(const Element::Pointer & e, unsigned int matrixIndex);
220 
226 
231 
235  void AssembleF();
236 
242 
248 
252  void SolveSystem();
253 
260 
265  void ComputeLandmarkTensor();
266 
270  float GetLandmarkTensorPonderation() const;
271 
277  void NthElementWRTDisplacementError(unsigned int numberOfRejectedBlocksPerStep);
278 
285  void UnselectLandmarks(unsigned int numberOfRejectedBlocksPerStep);
286 
292 
297  void DeleteFromLandmarkBeginning(unsigned int numberOfRejectedLandmarksPerStep);
298 
303 
311  void RescaleLandmarkStiffnessMatrix(double oldPointTensorPonderation);
312 
318 
324 
325 private:
326 
327  RobustSolver(const Self &) ITK_DELETE_FUNCTION;
328  void operator=(const Self &) ITK_DELETE_FUNCTION;
329 
335 
339  unsigned int m_ApproximationSteps;
340 
341 
345  typedef unsigned int FEMIndexType;
346 
347  FEMIndexType m_ForceIndex;
348  FEMIndexType m_LandmarkForceIndex;
349  FEMIndexType m_ExternalForceIndex;
350  FEMIndexType m_SolutionIndex;
354 
356 
360 
368 
370 };
371 
379  public std::binary_function<const Load::Pointer &, const Load::Pointer &, bool>
380 {
381 public:
382  bool operator()(const Load::Pointer & L1 , const Load::Pointer & L2)
383  {
384  LoadNoisyLandmark * l1 = dynamic_cast<LoadNoisyLandmark*>(L1.GetPointer());
385  LoadNoisyLandmark * l2 = dynamic_cast<LoadNoisyLandmark*>(L2.GetPointer());
387 
388  return l1->GetErrorNorm() > l2->GetErrorNorm();
389  }
390 };
391 
392 }
393 
394 } // end namespace itk::fem
395 
396 #ifndef ITK_MANUAL_INSTANTIATION
397 #include "itkFEMRobustSolver.hxx"
398 #endif
399 
400 #endif
unsigned int m_OutlierRejectionSteps
void DeleteFromLandmarkBeginning(unsigned int numberOfRejectedLandmarksPerStep)
Light weight base class for most itk classes.
InterpolationGridType::PointType InterpolationGridPointType
Definition: itkFEMSolver.h:109
virtual void AssembleElementMatrixWithID(const Element::Pointer &e, unsigned int matrixIndex)
void RescaleLandmarkStiffnessMatrix(double oldPointTensorPonderation)
Superclass::FEMObjectType FEMObjectType
InterpolationGridType::SpacingType InterpolationGridSpacingType
Definition: itkFEMSolver.h:110
Element::VectorType VectorType
Definition: itkFEMSolver.h:96
void IncrementalSolverWithOutlierRejection()
void AssembleGlobalMatrixFromLandmarksAndMeshMatrices()
Superclass::InterpolationGridSpacingType InterpolationGridSpacingType
itk::Image< Element::ConstPointer, VDimension > InterpolationGridType
Definition: itkFEMSolver.h:105
Superclass::InterpolationGridIndexType InterpolationGridIndexType
ObjectType * GetPointer() const
static const double e
The base of the natural logarithm or Euler&#39;s number
Definition: itkMath.h:47
FEMObjectType::LoadContainerIterator LoadContainerIterator
FEMObjectType::LoadContainerType LoadContainerType
FEMIndexType m_MeshStiffnessMatrixIndex
vnl_matrix< Float > MatrixType
itk::fem::FEMObject< VDimension > FEMObjectType
Definition: itkFEMSolver.h:89
float GetLandmarkTensorPonderation() const
InterpolationGridType::RegionType InterpolationGridRegionType
Definition: itkFEMSolver.h:108
void InitializeInterpolationGrid()
SmartPointer< Self > Pointer
InterpolationGridType::IndexType InterpolationGridIndexType
Definition: itkFEMSolver.h:111
Superclass::VectorType VectorType
LinearSystemWrapperItpack m_Itpack
void AssembleLandmarkStiffnessMatrix()
InterpolationGridType::Pointer InterpolationGridPointerType
Definition: itkFEMSolver.h:106
Superclass::InterpolationGridPointerType InterpolationGridPointerType
FEM solver used to generate a solution for a FE formulation.
Definition: itkFEMSolver.h:69
void AddExternalForcesToSetMeshZeroEnergy()
FEMIndexType m_StiffnessMatrixIndex
Abstract base element class.
InterpolationGridType::SizeType InterpolationGridSizeType
Definition: itkFEMSolver.h:107
bool operator()(const Load::Pointer &L1, const Load::Pointer &L2)
void AssembleMeshStiffnessMatrix()
void RemoveUnselectedLandmarkContributionInPointStiffnessMatrix()
Element::MatrixType MatrixType
void NthElementWRTDisplacementError(unsigned int numberOfRejectedBlocksPerStep)
General abstract load base class.
void IncrementalSolverWithoutOutlierRejection()
InterpolationGridType::DirectionType InterpolationGridDirectionType
LinearSystemWrapper class that uses Itpack numeric library functions to define and solve a sparse lin...
void ComputeLandmarkSimulatedDisplacementAndWeightedError()
void UnselectLandmarks(unsigned int numberOfRejectedBlocksPerStep)
FEMIndexType m_LandmarkStiffnessMatrixIndex
Element::Float Float
Definition: itkFEMSolver.h:95
Superclass::InterpolationGridSizeType InterpolationGridSizeType
virtual void GenerateData() override
Superclass::InterpolationGridType InterpolationGridType
FEMObjectType::NodeContainerType NodeContainerType
static const unsigned int FEMDimension
Superclass::InterpolationGridRegionType InterpolationGridRegionType
Superclass::InterpolationGridPointType InterpolationGridPointType
SmartPointer< const Self > ConstPointer
Solver< VDimension > Superclass
virtual void RunSolver(void) override