CLStaple.cpp

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#include "mitkCommandLineParser.h"
#include "mitkIOUtil.h"
#include <mitkImageCast.h>

#include <itkSTAPLEImageFilter.h>

int main(int argc, char* argv[])
{
  typedef itk::Image<unsigned char, 3> MaskImageType;
  typedef itk::Image<float, 3> ImageType;
  typedef itk::STAPLEImageFilter<MaskImageType, ImageType> FilterType;

  FilterType::Pointer filter = FilterType::New();
  filter->SetForegroundValue(2);
  for (int i = 2; i < argc; ++i)
  {
    MITK_INFO << argv[i];
    MaskImageType::Pointer itkImg = MaskImageType::New();
    mitk::Image::Pointer img = mitk::IOUtil::LoadImage(argv[i]);
    mitk::CastToItkImage(img,itkImg);
    filter->SetInput(i-2, itkImg);
  }
  filter->Update();
  auto out = filter->GetOutput();
  mitk::Image::Pointer outImg = mitk::Image::New();
  mitk::CastToMitkImage(out, outImg);
  mitk::IOUtil::SaveImage(outImg, argv[1]);

  return EXIT_SUCCESS;
}