int
main(int argc, char * argv[])
{
if (argc != 5)
{
std::cerr << "Usage: " << std::endl;
std::cerr << argv[0];
std::cerr << " <InputFileName> <OutputFileName> <label> <radius>";
std::cerr << std::endl;
return EXIT_FAILURE;
}
using PixelType = unsigned char;
const char * inputFileName = argv[1];
const char * outputFileName = argv[2];
const auto label = static_cast<PixelType>(std::stoi(argv[3]));
const unsigned int radiusValue = std::stoi(argv[4]);
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(inputFileName);
LabelImageToLabelMapFilterType::Pointer labelMapConverter = LabelImageToLabelMapFilterType::New();
labelMapConverter->SetInput(reader->GetOutput());
SelectorType::Pointer selector = SelectorType::New();
selector->SetInput(labelMapConverter->GetOutput());
selector->SetLabel(label);
StructuringElementType::RadiusType radius;
radius.Fill(radiusValue);
StructuringElementType structuringElement = StructuringElementType::Ball(radius);
using MorphologicalFilterType =
MorphologicalFilterType::Pointer closingFilter = MorphologicalFilterType::New();
ObjectByObjectLabelMapFilterType::Pointer objectByObjectLabelMapFilter = ObjectByObjectLabelMapFilterType::New();
objectByObjectLabelMapFilter->SetInput(selector->GetOutput(0));
objectByObjectLabelMapFilter->SetBinaryInternalOutput(true);
objectByObjectLabelMapFilter->SetFilter(closingFilter);
MergeLabelFilterType::Pointer merger = MergeLabelFilterType::New();
merger->SetInput(0, objectByObjectLabelMapFilter->GetOutput(0));
merger->SetInput(1, selector->GetOutput(1));
UniqueLabelMapFilterType::Pointer unique = UniqueLabelMapFilterType::New();
unique->SetInput(merger->GetOutput());
LabelMapToLabelImageFilterType::Pointer labelImageConverter = LabelMapToLabelImageFilterType::New();
labelImageConverter->SetInput(unique->GetOutput());
WriterType::Pointer writer = WriterType::New();
writer->SetFileName(outputFileName);
writer->SetInput(labelImageConverter->GetOutput());
try
{
writer->Update();
}
catch (itk::ExceptionObject & error)
{
std::cerr << "Error: " << error << std::endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}