itkTransformIOBase.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef __itkTransformIOBase_h
#define __itkTransformIOBase_h

#include "itkLightProcessObject.h"
#include "itkTransformBase.h"
#include <list>
#include <iostream>
#include <fstream>
#include <string>

namespace itk
{

template<typename TScalar>
class TransformIOBaseTemplate:public LightProcessObject
{
public:
  typedef TransformIOBaseTemplate   Self;
  typedef LightProcessObject        Superclass;
  typedef SmartPointer< Self >      Pointer;

  itkTypeMacro(TransformIOBaseTemplate, Superclass);

  typedef TScalar                           ScalarType;
  typedef TransformBaseTemplate<ScalarType> TransformType;

  typedef typename TransformType::Pointer             TransformPointer;
  typedef std::list< TransformPointer >               TransformListType;
  typedef typename TransformType::ConstPointer        ConstTransformPointer;
  typedef std::list< ConstTransformPointer >          ConstTransformListType;

  itkSetStringMacro(FileName);
  itkGetStringMacro(FileName);

  virtual void Read() = 0;

  virtual void Write() = 0;

  virtual bool CanReadFile(const char *) = 0;

  virtual bool CanWriteFile(const char *)  = 0;

  TransformListType & GetTransformList() { return m_ReadTransformList; }
  TransformListType & GetReadTransformList() { return m_ReadTransformList; }
  ConstTransformListType & GetWriteTransformList() { return m_WriteTransformList; }

  void SetTransformList(ConstTransformListType & transformList);

  itkSetMacro(AppendMode, bool);
  itkGetConstMacro(AppendMode, bool);
  itkBooleanMacro(AppendMode);

  static inline void CorrectTransformPrecisionType( std::string & itkNotUsed(inputTransformName) )
    {
    itkGenericExceptionMacro(<< "Unknown ScalarType" << typeid(ScalarType).name());
    }

protected:
  TransformIOBaseTemplate();
  virtual ~TransformIOBaseTemplate();
  virtual void PrintSelf(std::ostream & os, Indent indent) const ITK_OVERRIDE;

  void OpenStream(std::ofstream & outputStream, bool binary);

  void CreateTransform(TransformPointer & ptr, const std::string & ClassName);

  std::string            m_FileName;
  TransformListType      m_ReadTransformList;
  ConstTransformListType m_WriteTransformList;
  bool                   m_AppendMode;

  /* The following struct returns the string name of computation type */
  /* default implementation */
  static inline const std::string GetTypeNameString()
    {
    itkGenericExceptionMacro(<< "Unknown ScalarType" << typeid(ScalarType).name());
    }
};


template <>
inline void
TransformIOBaseTemplate<float>
::CorrectTransformPrecisionType( std::string & inputTransformName )
{
  // output precision type is not found in input transform.
 if(inputTransformName.find("float") == std::string::npos)
   {
   const std::string::size_type begin = inputTransformName.find("double");
   inputTransformName.replace(begin, 6, "float");
   }
}

template <>
inline void
TransformIOBaseTemplate<double>
::CorrectTransformPrecisionType( std::string & inputTransformName )
{
  // output precision type is not found in input transform.
 if(inputTransformName.find("double") == std::string::npos)
   {
   const std::string::size_type begin = inputTransformName.find("float");
   inputTransformName.replace(begin, 5, "double");
   }
}

template <>
inline const std::string
TransformIOBaseTemplate<float>
::GetTypeNameString()
{
  return std::string("float");
}

template <>
inline const std::string
TransformIOBaseTemplate<double>
::GetTypeNameString()
{
  return std::string("double");
}

typedef itk::TransformIOBaseTemplate<double> TransformIOBase;

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkTransformIOBase.hxx"
#endif

#endif // __itkTransformIOBase.h