1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkFEMLoadGrav_h
#define itkFEMLoadGrav_h
#include "itkFEMLoadElementBase.h"
#include "ITKFEMExport.h"
#include "vnl/vnl_vector.h"
namespace itk
{
namespace fem
{
/**
* \class LoadGrav
* \brief Abstract gravity load class.
*
* This load is integrated over a whole element. The load vector is returned in a
* Fg member function defined in a derived class. The Fg function accepts a vector
* specifying a point in global coordinate system and returns a load vector
* defined at the point. Derived LoadClasses must define this function.
* \ingroup ITKFEM
*/
class ITKFEM_EXPORT LoadGrav : public LoadElement
{
public:
/** Standard class type aliases. */
using Self = LoadGrav;
using Superclass = LoadElement;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(LoadGrav);
virtual vnl_vector<Float> GetGravitationalForceAtPoint(vnl_vector<Float>) = 0;
protected:
void
PrintSelf(std::ostream & os, Indent indent) const override;
};
/**
* \class LoadGravConst
* \brief Constant gravity load class.
*
* This is a special case of LoadGrav. The load vector is the same on
* every point in space.
* \ingroup ITKFEM
*/
class ITKFEM_EXPORT LoadGravConst : public LoadGrav
{
public:
/** Standard class type aliases. */
using Self = LoadGravConst;
using Superclass = LoadGrav;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** Method for creation through the object factory. */
itkSimpleNewMacro(Self);
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(LoadGravConst);
/** CreateAnother method will clone the existing instance of this type,
* including its internal member variables. */
itk::LightObject::Pointer
CreateAnother() const override;
vnl_vector<Float> GetGravitationalForceAtPoint(vnl_vector<Float>) override;
/**
* Set the gravity force that exists at every point
*/
void
SetForce(const vnl_vector<itk::fem::Element::Float> force);
/**
* Get the gravity force that exists at every point
*/
vnl_vector<itk::fem::Element::Float> &
GetForce();
const vnl_vector<itk::fem::Element::Float> &
GetForce() const;
/** Apply the load to the specified element */
void
ApplyLoad(Element::ConstPointer element, Element::VectorType & Fe) override;
protected:
void
PrintSelf(std::ostream & os, Indent indent) const override;
vnl_vector<Float> m_GravityForce{};
};
} // end namespace fem
} // end namespace itk
#endif // itkFEMLoadGrav_h
|