int
main(int argc, char * argv[])
{
if (argc < 3)
{
std::cerr << "Usage: " << std::endl;
std::cerr << argv[0] << " <inputImage> <outputImage>";
std::cerr << " [maximumRMSError]";
std::cerr << " [numberOfIterations]";
std::cerr << " [numberOfLayers]" << std::endl;
return EXIT_FAILURE;
}
const char * inputFileName = argv[1];
const char * outputFileName = argv[2];
double maximumRMSError = 0.001;
if (argc > 3)
{
maximumRMSError = std::stod(argv[3]);
}
int numberOfIterations = 50;
if (argc > 4)
{
numberOfIterations = std::stoi(argv[4]);
}
int numberOfLayers = 2;
if (argc > 5)
{
numberOfLayers = std::stoi(argv[5]);
}
using PixelType = float;
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(inputFileName);
FilterType::Pointer filter = FilterType::New();
filter->SetInput(reader->GetOutput());
filter->SetMaximumRMSError(maximumRMSError);
filter->SetNumberOfIterations(numberOfIterations);
filter->SetNumberOfLayers(numberOfLayers);
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;
}