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 __itkGaussianMixtureModelComponent_h 00019 #define __itkGaussianMixtureModelComponent_h 00020 00021 #include "itkMixtureModelComponentBase.h" 00022 #include "itkGaussianMembershipFunction.h" 00023 #include "itkWeightedMeanSampleFilter.h" 00024 #include "itkWeightedCovarianceSampleFilter.h" 00025 00026 namespace itk 00027 { 00028 namespace Statistics 00029 { 00049 template< class TSample > 00050 class ITK_EXPORT GaussianMixtureModelComponent: 00051 public MixtureModelComponentBase< TSample > 00052 { 00053 public: 00055 typedef GaussianMixtureModelComponent Self; 00056 typedef MixtureModelComponentBase< TSample > Superclass; 00057 typedef SmartPointer< Self > Pointer; 00058 typedef SmartPointer< const Self > ConstPointer; 00059 00061 itkTypeMacro(GaussianMixtureModelComponent, MixtureModelComponentBase); 00062 itkNewMacro(Self); 00064 00066 typedef typename Superclass::MeasurementVectorType MeasurementVectorType; 00067 typedef typename Superclass::MeasurementVectorSizeType MeasurementVectorSizeType; 00068 typedef typename Superclass::MembershipFunctionType MembershipFunctionType; 00069 typedef typename Superclass::WeightArrayType WeightArrayType; 00070 typedef typename Superclass::ParametersType ParametersType; 00071 00073 typedef GaussianMembershipFunction< MeasurementVectorType > 00074 NativeMembershipFunctionType; 00075 00078 typedef WeightedMeanSampleFilter< TSample > MeanEstimatorType; 00079 typedef WeightedCovarianceSampleFilter< TSample > CovarianceEstimatorType; 00080 00082 typedef typename MeanEstimatorType::OutputType MeanVectorType; 00083 00085 typedef typename CovarianceEstimatorType::OutputType CovarianceMatrixType; 00086 00088 void SetSample(const TSample *sample); 00089 00091 void SetParameters(const ParametersType & parameters); 00092 00093 protected: 00094 GaussianMixtureModelComponent(); 00095 virtual ~GaussianMixtureModelComponent() {} 00096 void PrintSelf(std::ostream & os, Indent indent) const; 00097 00100 double CalculateParametersChange(); 00101 00103 void GenerateData(); 00104 00105 private: 00106 typename NativeMembershipFunctionType::Pointer m_GaussianMembershipFunction; 00107 00108 typename MeanEstimatorType::MeasurementVectorType m_Mean; 00109 00110 typename CovarianceEstimatorType::MatrixType m_Covariance; 00111 00112 typename MeanEstimatorType::Pointer m_MeanEstimator; 00113 00114 typename CovarianceEstimatorType::Pointer m_CovarianceEstimator; 00115 }; // end of class 00116 } // end of namespace Statistics 00117 } // end of namespace itk 00118 00119 #ifndef ITK_MANUAL_INSTANTIATION 00120 #include "itkGaussianMixtureModelComponent.hxx" 00121 #endif 00122 00123 #endif 00124