cxAlgorithmHelpers.cpp

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/
#include "cxAlgorithmHelpers.h"

#include "cxImage.h"
#include "cxTypeConversions.h"
#include "itkImageFileReader.h"
#include "vtkMetaImageWriter.h"
#include "cxSettings.h"
#include <QDir>
#include "cxUtilHelpers.h"
#include "cxDataLocations.h"
#include "cxLogger.h"
#include <itkGrayscaleFillholeImageFilter.h>

namespace cx
{

//---------------------------------------------------------------------------------------------------------------------
itkImageType::ConstPointer AlgorithmHelper::getITKfromVTKImage(vtkImageDataPtr image)
{
  //HACK
  return AlgorithmHelper::getITKfromVTKImageViaFile(image);
}
//---------------------------------------------------------------------------------------------------------------------

itkImageType::ConstPointer AlgorithmHelper::getITKfromSSCImage(ImagePtr input)
{
        if (!input)
                return getITKfromVTKImage(vtkImageDataPtr());
        else
                return getITKfromVTKImage(input->getBaseVtkImageData());
}
//---------------------------------------------------------------------------------------------------------------------

itkImageType::ConstPointer AlgorithmHelper::getITKfromVTKImageViaFile(vtkImageDataPtr input)
{
  if(!input)
  {
    std::cout << "getITKfromSSCImage(): NO image!!!" << std::endl;
    return itkImageType::ConstPointer();
  }

  double minVal = input->GetScalarRange()[0];
  double maxVal = input->GetScalarRange()[1];

  if(maxVal > SHRT_MAX || minVal < SHRT_MIN)
        reportWarning("Image values out of range. max: " + qstring_cast(maxVal)
                        + " min: " + qstring_cast(minVal) + " See bug #363 if this needs to be fixed");

  QString tempFolder = DataLocations::getCachePath() + "/vtk2itk/";
  QDir().mkpath(tempFolder);

  // write to disk
  vtkMetaImageWriterPtr writer = vtkMetaImageWriterPtr::New();

  QString filename = tempFolder + "/"+qstring_cast(writer.GetPointer())+".mhd";


  writer->SetInputData(input);
  writer->SetFileName(cstring_cast(filename));
  writer->SetCompression(false);
  writer->Write();

  // read from disk
  typedef itk::ImageFileReader<itkImageType> ReaderType;
  ReaderType::Pointer reader = ReaderType::New();
#if ITK_VERSION_MAJOR==3
  reader->SetFileName(cstring_cast(filename));
#else
  reader->SetFileName(string_cast(filename));
#endif

//  reader->SetFileName(string_cast(filename));
  reader->Update();
  itkImageType::ConstPointer retval = reader->GetOutput();

  QFile(filename).remove(); // cleanup
  QFile(changeExtension(filename, "raw")).remove(); // cleanup

  return retval;
}
//---------------------------------------------------------------------------------------------------------------------

vtkImageDataPtr AlgorithmHelper::getVTKFromITK(itkImageType::ConstPointer input)
{
        //Convert ITK to VTK
        itkToVtkFilterType::Pointer itkToVtkFilter = itkToVtkFilterType::New();
        itkToVtkFilter->SetInput(input);
        itkToVtkFilter->Update();

        vtkImageDataPtr rawResult = vtkImageDataPtr::New();
        rawResult->DeepCopy(itkToVtkFilter->GetOutput());
        // TODO: possible memory problem here - check debug mem system of itk/vtk

        return rawResult;
}

vtkImageDataPtr AlgorithmHelper::execute_itk_GrayscaleFillholeImageFilter(vtkImageDataPtr input)
{
  itkImageType::ConstPointer itkImage = AlgorithmHelper::getITKfromVTKImage(input);

  typedef itk::GrayscaleFillholeImageFilter<itkImageType, itkImageType> FilterType;
  FilterType::Pointer filter = FilterType::New();

  filter->SetInput(itkImage);
  filter->Update();

  return AlgorithmHelper::getVTKFromITK(filter->GetOutput());
}


}