ITK  4.4.0
Insight Segmentation and Registration Toolkit
itkFEMElement2DStress.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 __itkFEMElement2DStress_h
20 #define __itkFEMElement2DStress_h
21 
22 #include "itkFEMElementBase.h"
24 
25 namespace itk
26 {
27 namespace fem
28 {
44 template <class TBaseClass = Element>
45 class Element2DStress : public TBaseClass
46 {
47 public:
48 
51  typedef TBaseClass Superclass;
54 
56  itkTypeMacro(Element2DStress, TBaseClass);
57 
58  // Repeat the required typedefs and enums from parent class
59  typedef typename Superclass::Float Float;
62 
67 
68  // ////////////////////////////////////////////////////////////////////////
69  /*
70  * Methods related to the physics of the problem.
71  */
72 
76  virtual void GetStrainDisplacementMatrix(MatrixType & B, const MatrixType & shapeDgl) const;
77 
81  virtual void GetMaterialMatrix(MatrixType & D) const;
82 
86  void GetMassMatrix(MatrixType & Me) const;
87 
91  virtual unsigned int GetNumberOfDegreesOfFreedomPerNode(void) const
92  {
93  return 2;
94  }
95 
99  virtual Material::ConstPointer GetMaterial(void) const
100  {
101  return dynamic_cast<const Material *>(m_mat);
102  }
103 
105  {
106  this->SetMaterialInternal(mat_.GetPointer());
107  }
108  virtual void SetMaterial(Material::Pointer mat_)
109  {
110  this->SetMaterialInternal(mat_.GetPointer());
111  }
112 
113 protected:
114 
115  virtual void PrintSelf(std::ostream& os, Indent indent) const;
116 
121  virtual void SetMaterialInternal(const Material *mat_)
122  {
123  m_mat =
124  dynamic_cast<const MaterialLinearElasticity *>( mat_ );
125  }
127 
128 }; // class Element2DStress
129 
130 }
131 } // end namespace itk::fem
132 
133 #ifndef ITK_MANUAL_INSTANTIATION
134 #include "itkFEMElement2DStress.hxx"
135 #endif
136 
137 #endif // #ifndef __itkFEMElement2DStress_h
138