int
main(int argc, char * argv[])
{
if ((argc != 6) && (argc != 9))
{
std::cerr << "Usage: " << std::endl;
std::cerr << argv[0];
std::cerr << " <InputFileName> <LabelMapFileName> <OutputFileName> <Label> <Background>";
std::cerr << " [<negated (bool)> <crop (bool)> <border size>]";
std::cerr << std::endl;
return EXIT_FAILURE;
}
using PixelType = unsigned char;
const char * inputFileName = argv[1];
const char * labelMapFileName = argv[2];
const char * outputFileName = argv[3];
const auto label = static_cast<PixelType>(std::stoi(argv[4]));
const auto background = static_cast<PixelType>(std::stoi(argv[5]));
bool negated = false;
bool crop = false;
if (argc == 9)
{
negated = (std::stoi(argv[6]) == 1);
crop = (std::stoi(argv[7]) == 1);
borderSize = static_cast<ImageType::SizeValueType>(std::stoi(argv[8]));
}
ReaderType::Pointer reader1 = ReaderType::New();
reader1->SetFileName(inputFileName);
ReaderType::Pointer reader2 = ReaderType::New();
reader2->SetFileName(labelMapFileName);
LabelImage2LabelMapType::Pointer convert = LabelImage2LabelMapType::New();
convert->SetInput(reader2->GetOutput());
FilterType::Pointer filter = FilterType::New();
filter->SetInput(convert->GetOutput());
filter->SetFeatureImage(reader1->GetOutput());
filter->SetLabel(label);
filter->SetBackgroundValue(background);
filter->SetNegated(negated);
filter->SetCrop(crop);
filter->SetCropBorder(border);
WriterType::Pointer writer = WriterType::New();
writer->SetFileName(outputFileName);
writer->SetInput(filter->GetOutput());
try
{
writer->Update();
}
catch (itk::ExceptionObject & error)
{
std::cerr << "Error: " << error << std::endl;
return EXIT_FAILURE;
}
return EXIT_SUCCESS;
}