00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkSampleClassifierWithMask_h
00018 #define __itkSampleClassifierWithMask_h
00019
00020 #include <vector>
00021
00022 #include "itkObject.h"
00023 #include "itkExceptionObject.h"
00024 #include "itkSubsample.h"
00025 #include "itkMembershipSample.h"
00026 #include "itkSampleClassifier.h"
00027
00028 namespace itk{
00029 namespace Statistics{
00030
00053 template< class TSample, class TMaskSample >
00054 class ITK_EXPORT SampleClassifierWithMask :
00055 public SampleClassifier< TSample >
00056 {
00057 public:
00059 typedef SampleClassifierWithMask Self;
00060 typedef SampleClassifier< TSample > Superclass;
00061 typedef SmartPointer< Self > Pointer;
00062 typedef SmartPointer<const Self> ConstPointer;
00063
00065 itkTypeMacro(SampleClassifierWithMask, SampleClassifier);
00066 itkNewMacro(Self) ;
00067
00069 typedef typename Superclass::OutputType OutputType ;
00070 typedef typename Superclass::ClassLabelType ClassLabelType ;
00071 typedef typename Superclass::ClassLabelVectorType ClassLabelVectorType ;
00072
00074 typedef typename TSample::MeasurementType MeasurementType ;
00075 typedef typename TSample::MeasurementVectorType MeasurementVectorType ;
00076
00077 itkStaticConstMacro(MeasurementVectorSize, unsigned int,
00078 TSample::MeasurementVectorSize);
00079
00081 typedef typename Superclass::MembershipFunctionPointerVector
00082 MembershipFunctionPointerVector ;
00083
00084 void SetMask( TMaskSample* mask ) ;
00085
00086 TMaskSample* GetMask()
00087 { return m_Mask.GetPointer() ; }
00088
00089 void SetSelectedClassLabels( ClassLabelVectorType& labels)
00090 { m_SelectedClassLabels = labels ; }
00091
00092 void SetOtherClassLabel( ClassLabelType label)
00093 { m_OtherClassLabel = label ; }
00094
00095 protected:
00096 SampleClassifierWithMask() ;
00097 virtual ~SampleClassifierWithMask() {}
00098 void PrintSelf(std::ostream& os, Indent indent) const;
00099
00101 void GenerateData() ;
00102
00103 private:
00105 typename TMaskSample::Pointer m_Mask ;
00106 ClassLabelVectorType m_SelectedClassLabels ;
00107 ClassLabelType m_OtherClassLabel ;
00108 } ;
00109
00110
00111 }
00112 }
00113
00114
00115 #ifndef ITK_MANUAL_INSTANTIATION
00116 #include "itkSampleClassifierWithMask.txx"
00117 #endif
00118
00119 #endif
00120
00121
00122
00123
00124
00125
00126