itk::Image<unsigned char, 3>::Pointer pdp::EMClassification::classify(itk::Image<float, 3>::Pointer img, itk::Image<unsigned char, 3>::Pointer mask)
{
typedef itk::Vector< double, 1 > MeasurementVectorType;
typedef itk::Image<unsigned char, 3> MaskType;
typedef itk::Image<float, 3> ImageType;
typedef itk::ImageRegionConstIterator< ImageType > ImageIteratorType;
ImageIteratorType imgIt(img, img->GetLargestPossibleRegion());
typedef itk::ImageRegionConstIterator< MaskType> MaskIteratorType;
MaskIteratorType maskIt( mask, mask->GetLargestPossibleRegion());
MaskType::Pointer correctImage = MaskType::New();
correctImage->CopyInformation(mask);
MaskType::RegionType outputRegion = mask->GetLargestPossibleRegion();
correctImage->SetRegions( outputRegion );
correctImage->Allocate();
typedef itk::ImageRegionIterator<MaskType> IteratorType;
IteratorType outputIt( correctImage, outputRegion);
MeasurementVectorType mv;
for ( imgIt.GoToBegin(), maskIt.GoToBegin(), outputIt.GoToBegin(); !maskIt.IsAtEnd(); ++maskIt, ++imgIt, ++outputIt)
{
if (maskIt.Get() == 255)
{
mv[0] = imgIt.Get();
if (genLabel(mv) == 3)
{
outputIt.Set(255);
}
}
}
return correctImage;
}
