mitkVtkImageReader.cpp

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#include "mitkVtkImageReader.h"

#include <vtkDataReader.h>
#include <vtkStructuredPoints.h>
#include <vtkStructuredPointsReader.h>

mitk::VtkImageReader::VtkImageReader() : m_FileName(""), m_FilePrefix(""), m_FilePattern("")
{
}

mitk::VtkImageReader::~VtkImageReader()
{
}

void mitk::VtkImageReader::GenerateData()
{
  if (m_FileName != "")
  {
    MITK_INFO << "Loading " << m_FileName << " as vtk" << std::endl;

    vtkDataReader *chooser = vtkDataReader::New();
    chooser->SetFileName(m_FileName.c_str());

    if (chooser->IsFileStructuredPoints())
    {
      MITK_INFO << "StructuredPoints" << std::endl;
      vtkStructuredPointsReader *reader = vtkStructuredPointsReader::New();
      reader->SetFileName(m_FileName.c_str());
      reader->Update();

      if (reader->GetOutput() != nullptr)
      {
        mitk::Image::Pointer output = this->GetOutput();
        output->Initialize(reader->GetOutput());
        output->SetVolume(reader->GetOutput()->GetScalarPointer());
      }
      reader->Delete();
    }
    else
    {
      MITK_ERROR << " ... sorry, this .vtk format is not supported yet." << std::endl;
    }
    chooser->Delete();
  }
}

bool mitk::VtkImageReader::CanReadFile(const std::string filename,
                                       const std::string /*filePrefix*/,
                                       const std::string /*filePattern*/)
{
  // First check the extension
  if (filename == "")
    return false;

  bool extensionFound = false;
  std::string::size_type PVTKPos = filename.rfind(".pvtk");
  if ((PVTKPos != std::string::npos) && (PVTKPos == filename.length() - 5))
  {
    extensionFound = true;
  }

  PVTKPos = filename.rfind(".PVTK");
  if ((PVTKPos != std::string::npos) && (PVTKPos == filename.length() - 5))
  {
    extensionFound = true;
  }

  if (extensionFound)
  {
    vtkDataReader *chooser = vtkDataReader::New();
    chooser->SetFileName(filename.c_str());
    if (!chooser->IsFileStructuredPoints())
      return false;
  }
  else
    return false;

  return true;
}