mitkToFImageWriter.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 <mitkToFImageWriter.h>
#include <mitkCommon.h>
 
// itk includes
#include "itksys/SystemTools.hxx"
 
namespace mitk
{
  ToFImageWriter::ToFImageWriter():m_Extension(".nrrd"),
    m_DistanceImageFileName(), m_AmplitudeImageFileName(), m_IntensityImageFileName(), m_RGBImageFileName(),
    m_NumOfFrames(0), m_DistanceImageSelected(true), m_AmplitudeImageSelected(false),
    m_IntensityImageSelected(false), m_RGBImageSelected(false), m_ToFCaptureWidth(200),m_ToFCaptureHeight(200),
    m_RGBCaptureWidth(200),m_RGBCaptureHeight(200),
    m_ToFPixelNumber(0), m_ToFImageSizeInBytes(0),
    m_RGBPixelNumber(0), m_RGBImageSizeInBytes(0),
    m_ToFImageType(ToFImageWriter::ToFImageType3D)
  {
  }
 
  ToFImageWriter::~ToFImageWriter()
  {
  }
 
  void ToFImageWriter::CheckForFileExtension(std::string& fileName)
  {
    std::string baseFilename = itksys::SystemTools::GetFilenameWithoutLastExtension( fileName );
    std::string extension = itksys::SystemTools::GetFilenameLastExtension( fileName );
 
    if( extension.length() != 0 && extension != this->m_Extension)
    {
      MITK_ERROR << "Wrong file extension for " << baseFilename <<
                    " The default extension is " << this->m_Extension.c_str() <<
                    ", currently the requested file extension is " <<
                    extension.c_str() <<"!";
      this->m_Extension = extension;
    }
 
    size_t found = fileName.find( this->m_Extension ); // !!! HAS to be at the very end of the filename (not somewhere in the middle)
    if( found == std::string::npos)
    {
      fileName.append(this->m_Extension);
    }
  }
 
  ToFImageWriter::ToFImageType ToFImageWriter::GetToFImageType()
  {
    return this->m_ToFImageType;
  }
 
  void ToFImageWriter::SetToFImageType(ToFImageWriter::ToFImageType toFImageType)
  {
    this->m_ToFImageType = toFImageType;
  }
 
} // end namespace mitk