00001 /*========================================================================= 00002 00003 Program: Insight Segmentation & Registration Toolkit 00004 Module: $RCSfile: itkGaussianMixtureModelComponent.h,v $ 00005 Language: C++ 00006 Date: $Date: 2005-07-26 15:54:55 $ 00007 Version: $Revision: 1.10 $ 00008 00009 Copyright (c) Insight Software Consortium. All rights reserved. 00010 See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details. 00011 00012 This software is distributed WITHOUT ANY WARRANTY; without even 00013 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 00014 PURPOSE. See the above copyright notices for more information. 00015 00016 =========================================================================*/ 00017 00018 #ifndef __itkGaussianMixtureModelComponent_h 00019 #define __itkGaussianMixtureModelComponent_h 00020 00021 #include "itkMixtureModelComponentBase.h" 00022 #include "itkGaussianDensityFunction.h" 00023 #include "itkWeightedMeanCalculator.h" 00024 #include "itkWeightedCovarianceCalculator.h" 00025 00026 namespace itk{ 00027 namespace Statistics{ 00028 00047 template< class TSample > 00048 class GaussianMixtureModelComponent : 00049 public MixtureModelComponentBase< TSample > 00050 { 00051 public: 00053 typedef GaussianMixtureModelComponent Self; 00054 typedef MixtureModelComponentBase< TSample > Superclass ; 00055 typedef SmartPointer<Self> Pointer; 00056 typedef SmartPointer<const Self> ConstPointer; 00057 00059 itkTypeMacro(GaussianMixtureModelComponent, MixtureModelComponentBase); 00060 itkNewMacro(Self) ; 00062 00063 00065 typedef typename Superclass::MeasurementVectorType MeasurementVectorType ; 00066 typedef typename Superclass::MeasurementVectorSizeType MeasurementVectorSizeType ; 00067 typedef typename Superclass::MembershipFunctionType MembershipFunctionType ; 00068 typedef typename Superclass::WeightArrayType WeightArrayType ; 00069 typedef typename Superclass::ParametersType ParametersType ; 00070 00072 typedef GaussianDensityFunction< MeasurementVectorType > 00073 NativeMembershipFunctionType ; 00074 00077 typedef WeightedMeanCalculator< TSample > MeanEstimatorType ; 00078 typedef WeightedCovarianceCalculator< TSample > CovarianceEstimatorType ; 00079 00081 typedef typename MeanEstimatorType::OutputType MeanType ; 00082 00084 typedef typename CovarianceEstimatorType::OutputType CovarianceType ; 00085 00087 void SetSample(const TSample* sample) ; 00088 00090 void SetParameters(const ParametersType ¶meters) ; 00091 00092 protected: 00093 GaussianMixtureModelComponent() ; 00094 virtual ~GaussianMixtureModelComponent() {} 00095 void PrintSelf(std::ostream& os, Indent indent) const; 00096 00099 double CalculateParametersChange() ; 00100 00102 void GenerateData() ; 00103 00104 private: 00105 typename NativeMembershipFunctionType::Pointer m_GaussianDensityFunction ; 00106 MeanType m_Mean ; 00107 CovarianceType m_Covariance ; 00108 typename MeanEstimatorType::Pointer m_MeanEstimator ; 00109 typename CovarianceEstimatorType::Pointer m_CovarianceEstimator ; 00110 } ; // end of class 00111 00112 } // end of namespace Statistics 00113 } // end of namespace itk 00114 00115 #ifndef ITK_MANUAL_INSTANTIATION 00116 #include "itkGaussianMixtureModelComponent.txx" 00117 #endif 00118 00119 #endif 00120 00121