00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
#ifndef _itkExtensionVelocitiesImageFilter_h
00018
#define _itkExtensionVelocitiesImageFilter_h
00019
00020
#include "itkLevelSetVelocityNeighborhoodExtractor.h"
00021
#include "itkFastMarchingExtensionImageFilter.h"
00022
#include "itkReinitializeLevelSetImageFilter.h"
00023
00024
namespace itk
00025 {
00047
template <
00048
class TLevelSet,
00049
class TAuxValue =
float,
00050
unsigned int VAuxDimension = 1
00051 >
00052 class ITK_EXPORT ExtensionVelocitiesImageFilter :
00053
public ReinitializeLevelSetImageFilter<TLevelSet>
00054 {
00055
public:
00056
00058 typedef ExtensionVelocitiesImageFilter
Self;
00059 typedef ReinitializeLevelSetImageFilter<TLevelSet> Superclass;
00060 typedef SmartPointer<Self> Pointer;
00061 typedef SmartPointer<const Self> ConstPointer;
00062
00064
itkNewMacro(
Self);
00065
00067
itkTypeMacro(ExtensionVelocitiesImageFilter,
ReinitializeLevelSetImageFilter);
00068
00070 typedef LevelSetTypeDefault<TLevelSet> LevelSetType;
00071 typedef typename LevelSetType::LevelSetPointer
LevelSetPointer;
00072 typedef typename LevelSetType::LevelSetConstPointer
LevelSetConstPointer;
00073 typedef typename LevelSetType::PixelType
PixelType;
00074 typedef typename LevelSetType::NodeType
NodeType;
00075 typedef typename LevelSetType::NodeContainer
NodeContainer;
00076 typedef typename LevelSetType::NodeContainerPointer
NodeContainerPointer;
00077
00079
itkStaticConstMacro(SetDimension,
unsigned int,LevelSetType::SetDimension);
00080
00082
typedef AuxVarTypeDefault<TAuxValue,VAuxDimension,
00083 itkGetStaticConstMacro(SetDimension)>
AuxVarType;
00084 typedef typename AuxVarType::AuxValueType
AuxValueType;
00085 typedef typename AuxVarType::AuxValueVectorType
AuxValueVectorType;
00086 typedef typename AuxVarType::AuxValueContainer
AuxValueContainer;
00087 typedef typename AuxVarType::AuxImageType
AuxImageType;
00088 typedef typename AuxVarType::AuxImagePointer
AuxImagePointer;
00089 typedef typename AuxVarType::AuxImageConstPointer
AuxImageConstPointer;
00090
00092
itkStaticConstMacro(AuxDimension,
unsigned int,VAuxDimension);
00093
00095
void SetInputVelocityImage(
const AuxImageType * ptr,
unsigned int idx = 0);
00096
const AuxImageType * GetInputVelocityImage(
unsigned int idx = 0);
00097
00099
AuxImageType * GetOutputVelocityImage(
unsigned int idx = 0 );
00100
00101
protected:
00102 ExtensionVelocitiesImageFilter();
00103 ~ExtensionVelocitiesImageFilter(){};
00104
00105
virtual void GenerateDataFull();
00106
virtual void GenerateDataNarrowBand();
00107
virtual void AllocateOutput();
00108
00109
virtual void EnlargeOutputRequestedRegion(
DataObject * );
00110
00111
private:
00112 ExtensionVelocitiesImageFilter(
const Self&);
00113
void operator=(
const Self&);
00114
00118
typedef Image<float, itkGetStaticConstMacro(SetDimension) > SpeedImageType;
00119
typedef LevelSetVelocityNeighborhoodExtractor<TLevelSet,TAuxValue,VAuxDimension>
00120
LocatorType;
00121
typedef FastMarchingExtensionImageFilter<TLevelSet,TAuxValue,VAuxDimension,SpeedImageType>
00122
FastMarchingImageFilterType;
00123
00124
typename LocatorType::Pointer m_Locator;
00125
typename FastMarchingImageFilterType::Pointer m_Marcher;
00126
00127 };
00128
00129 }
00130
00131
#ifndef ITK_MANUAL_INSTANTIATION
00132
#include "itkExtensionVelocitiesImageFilter.txx"
00133
#endif
00134
00135
#endif