Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkVoronoiSegmentationImageFilterBase_h
00018 #define __itkVoronoiSegmentationImageFilterBase_h
00019
00020 #include "itkImageToImageFilter.h"
00021 #include "itkVoronoiDiagram2D.h"
00022 #include "itkVoronoiDiagram2DGenerator.h"
00023 #include "itkImage.h"
00024
00025 namespace itk
00026 {
00027
00058 template <class TInputImage, class TOutputImage,class TBinaryPriorImage=Image<unsigned char,2> >
00059 class ITK_EXPORT VoronoiSegmentationImageFilterBase:
00060 public ImageToImageFilter<TInputImage,TOutputImage>
00061 {
00062 public:
00064 typedef VoronoiSegmentationImageFilterBase Self;
00065 typedef ImageToImageFilter<TInputImage,TOutputImage> Superclass;
00066 typedef SmartPointer <Self> Pointer;
00067 typedef SmartPointer<const Self> ConstPointer;
00068
00070 itkNewMacro(Self);
00071
00073 itkTypeMacro(VoronoiSegmentationImageFilterBase,ImageToImageFilter);
00074
00076 itkStaticConstMacro(ImageDimension, unsigned int,
00077 TInputImage::ImageDimension);
00078
00080 typedef TInputImage InputImageType;
00081 typedef typename TInputImage::Pointer InputImagePointer;
00082 typedef typename TInputImage::ConstPointer InputImageConstPointer;
00083 typedef typename TInputImage::IndexType IndexType;
00084 typedef typename TInputImage::SizeType SizeType;
00085 typedef typename TInputImage::RegionType RegionType;
00086 typedef typename TInputImage::PixelType PixelType;
00087
00088 typedef TOutputImage OutputImageType;
00089 typedef typename TOutputImage::PixelType OutputPixelType;
00090
00091 typedef VoronoiDiagram2D<double> VoronoiDiagram;
00092 typedef VoronoiDiagram2DGenerator<double> VoronoiDiagramGenerator;
00093 typedef typename VoronoiDiagram::PointType PointType;
00094 typedef typename VoronoiDiagram::CellType CellType;
00095 typedef typename VoronoiDiagram::CellAutoPointer CellAutoPointer;
00096 typedef typename VoronoiDiagram::Pointer VoronoiPointer;
00097 typedef typename CellType::PointIdIterator PointIdIterator;
00098 typedef typename VoronoiDiagram::SeedsType SeedsType;
00099 typedef typename VoronoiDiagram::SeedsIterator SeedsIterator;
00100 typedef typename VoronoiDiagram::NeighborIdIterator NeighborIdIterator;
00101 typedef typename VoronoiDiagram::VoronoiEdgeIterator EdgeIterator;
00102 typedef typename VoronoiDiagram::VoronoiEdge EdgeInfo;
00103 typedef std::vector<PointType> PointTypeVector;
00104 typedef std::deque<PointType> PointTypeDeque;
00105 typedef TBinaryPriorImage BinaryObjectImage;
00106 typedef typename BinaryObjectImage::Pointer BinaryObjectImagePointer;
00107 typedef std::vector<IndexType> IndexList;
00108
00110 typedef Image<unsigned char,2> VDImage;
00111 typedef typename VDImage::Pointer VDImagePointer;
00112
00114 itkSetMacro(NumberOfSeeds, int);
00115 itkGetConstMacro(NumberOfSeeds, int);
00117
00119 itkSetMacro(MinRegion, int);
00120 itkGetConstMacro(MinRegion, int);
00122
00125 itkSetMacro(Steps, int);
00126 itkGetConstMacro(Steps, int);
00128
00130 itkGetConstMacro(LastStepSeeds, int);
00131
00133 itkGetConstMacro(NumberOfSeedsToAdded, int);
00134
00136 itkSetMacro(UseBackgroundInAPrior, bool);
00137 itkGetConstMacro(UseBackgroundInAPrior, bool);
00139
00141 itkSetMacro(OutputBoundary, bool);
00142 itkGetConstMacro(OutputBoundary, bool);
00144
00147 itkSetMacro(InteractiveSegmentation, bool);
00148 itkGetConstMacro(InteractiveSegmentation, bool);
00149 itkBooleanMacro(InteractiveSegmentation);
00151
00153 itkSetMacro(MeanDeviation, double);
00154 itkGetConstMacro(MeanDeviation, double);
00156
00158 itkSetMacro(Size,SizeType);
00159 itkGetConstMacro(Size,SizeType);
00161
00164 virtual void TakeAPrior(const BinaryObjectImage*){}
00165
00167 void RunSegment(void);
00168
00170 void RunSegmentOneStep(void);
00171
00173 virtual void MakeSegmentBoundary(void);
00174 virtual void MakeSegmentObject(void);
00176
00178 VoronoiPointer GetVoronoiDiagram(void)
00179 { return m_WorkingVD; }
00180
00181 #if !defined(CABLE_CONFIGURATION) // generates invalid iterator instantiation with msvc
00182
00185 void SetSeeds(int num, SeedsIterator begin)
00186 {
00187 m_NumberOfSeeds = num;
00188 m_WorkingVD->SetSeeds(num,begin);
00189 }
00190 #endif
00191
00192
00196 void SetSeeds(SeedsType & seeds)
00197 {
00198 m_NumberOfSeeds = seeds.size();
00199 typename SeedsType::iterator it = seeds.begin();
00200 m_WorkingVD->SetSeeds(m_NumberOfSeeds, it);
00201 }
00203
00205 PointType GetSeed(int SeedID)
00206 { return m_WorkingVD->GetSeed(SeedID); }
00207
00209 void DrawDiagram(VDImagePointer result,unsigned char incolor,
00210 unsigned char outcolor,unsigned char boundcolor);
00211
00212 void BeforeNextStep(void);
00213
00216 virtual void GenerateInputRequestedRegion();
00217
00220 virtual void EnlargeOutputRequestedRegion(DataObject *output);
00221
00222 protected:
00223 VoronoiSegmentationImageFilterBase();
00224 ~VoronoiSegmentationImageFilterBase();
00225 virtual void PrintSelf(std::ostream& os, Indent indent) const;
00226
00227 void GenerateData(void);
00228
00229 SizeType m_Size;
00230 int m_NumberOfSeeds;
00231 int m_MinRegion;
00232 int m_Steps;
00233 int m_LastStepSeeds;
00234 int m_NumberOfSeedsToAdded;
00235 int m_NumberOfBoundary;
00236
00237 std::vector<int> m_NumberOfPixels;
00238 std::vector<unsigned char> m_Label;
00239
00240 double m_MeanDeviation;
00241 bool m_UseBackgroundInAPrior;
00242 bool m_OutputBoundary;
00243 bool m_InteractiveSegmentation;
00244
00245 typename VoronoiDiagram::Pointer m_WorkingVD;
00246 typename VoronoiDiagramGenerator::Pointer m_VDGenerator;
00247
00248 std::vector<PointType> m_SeedsToAdded;
00249
00250
00251
00252 virtual void ClassifyDiagram(void);
00253
00254
00255 virtual void GenerateAddingSeeds(void);
00256
00257
00258 void GetPixelIndexFromPolygon(PointTypeDeque VertList, IndexList *PixelPool);
00259 virtual bool TestHomogeneity(IndexList&)
00260 {return 1;}
00261
00262 void FillPolygon(PointTypeDeque vertlist, OutputPixelType color=1);
00263
00264
00265 void drawLine(PointType p1,PointType p2);
00266
00267
00268 void drawVDline(VDImagePointer result,PointType p1,PointType p2, unsigned char color);
00269
00270 private:
00271 VoronoiSegmentationImageFilterBase(const Self&);
00272 void operator=(const Self&);
00273
00274 };
00275
00276 }
00277
00278
00279 #ifndef ITK_MANUAL_INSTANTIATION
00280 #include "itkVoronoiSegmentationImageFilterBase.txx"
00281 #endif
00282
00283 #endif
00284