ITK
4.1.0
Insight Segmentation and Registration Toolkit
|
00001 /*========================================================================= 00002 * 00003 * Copyright Insight Software Consortium 00004 * 00005 * Licensed under the Apache License, Version 2.0 (the "License"); 00006 * you may not use this file except in compliance with the License. 00007 * You may obtain a copy of the License at 00008 * 00009 * http://www.apache.org/licenses/LICENSE-2.0.txt 00010 * 00011 * Unless required by applicable law or agreed to in writing, software 00012 * distributed under the License is distributed on an "AS IS" BASIS, 00013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00014 * See the License for the specific language governing permissions and 00015 * limitations under the License. 00016 * 00017 *=========================================================================*/ 00018 #ifndef __itkScalarImageKmeansImageFilter_h 00019 #define __itkScalarImageKmeansImageFilter_h 00020 00021 00022 #include "itkKdTree.h" 00023 #include "itkKdTreeBasedKmeansEstimator.h" 00024 #include "itkWeightedCentroidKdTreeGenerator.h" 00025 00026 #include "itkSampleClassifierFilter.h" 00027 #include "itkImageToListSampleAdaptor.h" 00028 #include "itkMinimumDecisionRule.h" 00029 00030 #include "itkRegionOfInterestImageFilter.h" 00031 00032 #include <vector> 00033 00034 namespace itk 00035 { 00062 template< class TInputImage, 00063 class TOutputImage = Image< unsigned char, ::itk::GetImageDimension< TInputImage >::ImageDimension > > 00064 class ITK_EXPORT ScalarImageKmeansImageFilter: 00065 public ImageToImageFilter< TInputImage, TOutputImage > 00066 { 00067 public: 00069 itkStaticConstMacro(ImageDimension, unsigned int, 00070 TInputImage::ImageDimension); 00071 00073 typedef TInputImage InputImageType; 00074 typedef TOutputImage OutputImageType; 00075 00077 typedef ScalarImageKmeansImageFilter Self; 00078 typedef ImageToImageFilter< InputImageType, OutputImageType > Superclass; 00079 typedef SmartPointer< Self > Pointer; 00080 typedef SmartPointer< const Self > ConstPointer; 00081 00083 itkNewMacro(Self); 00084 00086 itkTypeMacro(ScalarImageKmeansImageFilter, ImageToImageFilter); 00087 00089 typedef typename InputImageType::PixelType InputPixelType; 00090 typedef typename OutputImageType::PixelType OutputPixelType; 00091 00093 typedef typename NumericTraits< InputPixelType >::RealType RealPixelType; 00094 00096 typedef itk::Statistics::ImageToListSampleAdaptor< InputImageType > AdaptorType; 00097 00099 typedef typename AdaptorType::MeasurementVectorType MeasurementVectorType; 00100 00101 typedef itk::Statistics::DistanceToCentroidMembershipFunction< MeasurementVectorType > MembershipFunctionType; 00102 typedef itk::Statistics::SampleClassifierFilter< AdaptorType > ClassifierType; 00103 typedef itk::Statistics::MinimumDecisionRule DecisionRuleType; 00104 00105 typedef typename ClassifierType::ClassLabelVectorType ClassLabelVectorType; 00106 00107 typedef typename ClassifierType::MembershipFunctionVectorType MembershipFunctionVectorType; 00108 typedef typename MembershipFunctionType::CentroidType MembershipFunctionOriginType; 00109 00110 typedef typename MembershipFunctionType::Pointer MembershipFunctionPointer; 00111 00113 typedef itk::Statistics::WeightedCentroidKdTreeGenerator< AdaptorType > TreeGeneratorType; 00114 typedef typename TreeGeneratorType::KdTreeType TreeType; 00115 typedef itk::Statistics::KdTreeBasedKmeansEstimator< TreeType > EstimatorType; 00116 00117 typedef typename EstimatorType::ParametersType ParametersType; 00118 00119 typedef typename InputImageType::RegionType ImageRegionType; 00120 00121 typedef RegionOfInterestImageFilter< 00122 InputImageType, 00123 InputImageType > RegionOfInterestFilterType; 00124 00126 void AddClassWithInitialMean(RealPixelType mean); 00127 00129 itkGetConstReferenceMacro(FinalMeans, ParametersType); 00130 00136 itkSetMacro(UseNonContiguousLabels, bool); 00137 itkGetConstReferenceMacro(UseNonContiguousLabels, bool); 00138 itkBooleanMacro(UseNonContiguousLabels); 00140 00142 void SetImageRegion(const ImageRegionType & region); 00143 00145 itkGetConstReferenceMacro(ImageRegion, ImageRegionType); 00146 00147 #ifdef ITK_USE_CONCEPT_CHECKING 00148 00149 itkConceptMacro( InputHasNumericTraitsCheck, 00150 ( Concept::HasNumericTraits< InputPixelType > ) ); 00151 00153 #endif 00154 protected: 00155 ScalarImageKmeansImageFilter(); 00156 virtual ~ScalarImageKmeansImageFilter() {} 00157 void PrintSelf(std::ostream & os, Indent indent) const; 00159 00165 void GenerateData(); 00166 00167 private: 00168 ScalarImageKmeansImageFilter(const Self &); //purposely not implemented 00169 void operator=(const Self &); //purposely not implemented 00170 00171 typedef std::vector< RealPixelType > MeansContainer; 00172 00173 MeansContainer m_InitialMeans; 00174 00175 ParametersType m_FinalMeans; 00176 00177 bool m_UseNonContiguousLabels; 00178 00179 ImageRegionType m_ImageRegion; 00180 00181 bool m_ImageRegionDefined; 00182 }; 00183 } // end namespace itk 00184 00185 #ifndef ITK_MANUAL_INSTANTIATION 00186 #include "itkScalarImageKmeansImageFilter.hxx" 00187 #endif 00188 00189 #endif 00190