Function main

Modules/Segmentation/cmdapps/ContoursToImage.cpp:165–297

}

int main(int argc, char* argv[])
{
  int returnValue = EXIT_SUCCESS;

  mitkCommandLineParser parser;
  InitializeCommandLineParser(parser);

  auto args = parser.parseArguments(argc, argv);

  if (args.empty())
    return EXIT_FAILURE;

  try
  {
    auto inputFilename = us::any_cast<std::string>(args["input"]);
    auto referenceFilename = us::any_cast<std::string>(args["reference"]);
    auto outputFilename = us::any_cast<std::string>(args["output"]);
    auto format = ParseOutputFormat(args);

    auto referenceImage = mitk::IOUtil::Load<mitk::Image>(referenceFilename);
    auto inputs = FilterValidInputs(mitk::IOUtil::Load(inputFilename));

    MITK_INFO << "Found " << inputs.size() << " input contour set(s)";

    fs::path outputPath(outputFilename);
    CreateParentDirectories(outputPath);

    mitk::MultiLabelSegmentation::Pointer labelSetImage; // Only used for "multilabel" output
    unsigned int nonameCounter = 0; // Helper variable to generate placeholder names for nameless contour sets

    for (auto input : inputs)
    {
      // If the input file contains multiple contour sets but the output format is not set to "multilabel",
      // we create separate output files for each contour set. In this case the specified output filename
      // is used only as a base filename and the names of the individual contour sets are appended accordingly.

      if (inputs.size() > 1 && format != OutputFormat::Multilabel)
      {
        outputPath = outputFilename;
        auto name = GetSafeName(input);

        if (name.empty())
          name = "nameless_" + std::to_string(nonameCounter++);

        outputPath.replace_filename(outputPath.stem().string() + '_' + name + outputPath.extension().string());
      }

      // Do the actual conversion from a contour set to an image with a background pixel value of 0.
      // - For "binary" output, use pixel value 1 and unsigned char as pixel type.
      // - For "label" output, use pixel value 1 and our label pixel type.
      // - For "multilabel" output, use the next available label value instead.

      const auto labelValue = labelSetImage.IsNotNull()
        ? labelSetImage->GetUnusedLabelValue()
        : 1;

      auto filter = mitk::ContourModelSetToImageFilter::New();
      filter->SetMakeOutputLabelPixelType(format != OutputFormat::Binary);