itkFiberTractsReader.h

Go to the documentation of this file.

#ifndef __itkFiberTractsReader_h
#define __itkFiberTractsReader_h

#include <itkLightProcessObject.h>

#include "itkFiberTracts.h"
#include "utlCoreMacro.h"

namespace itk
{

class ITK_EXPORT FiberTractsReader : public LightProcessObject
{
public:
  typedef FiberTractsReader       Self;
  typedef LightProcessObject   Superclass;
  typedef SmartPointer<Self>   Pointer;
  
  itkNewMacro(Self);

  itkTypeMacro(FiberTractsReader, LightProcessObject);

  typedef FiberTracts<double>                     FiberTractsType;
  typedef typename FiberTractsType::Pointer       FiberTractsPointer;
  typedef typename FiberTractsType::FiberType     FiberType;
  typedef typename FiberTractsType::FiberPointer  FiberPointer;
  typedef typename FiberType::STDVectorType       STDVectorType;
  typedef typename FiberType::VertexType          VertexType;

  itkSetGetMacro(FileName, std::string);
  // itkSetGetMacro(FiberTracts, FiberTractsPointer);

  virtual void Update();

  void ReadTractsTRK();

  void ReadTractsTCK();
  
  void ReadTractsVTK();

  FiberTractsPointer GetOutput() const
    {
    return m_FiberTracts;
    }
    
protected:
  FiberTractsReader(){}
  ~FiberTractsReader(){}

  void PrintSelf(std::ostream& os, Indent indent) const ITK_OVERRIDE
    {
    Superclass::PrintSelf(os, indent);
    PrintVar(true, os<< indent, m_FileName);
    m_FiberTracts->Print(os, indent);
    }
  
  std::string m_FileName;

  FiberTractsPointer m_FiberTracts = FiberTractsType::New();


private:
  FiberTractsReader(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

};

inline bool 
ReadFibers (const std::string& filename, SmartPointer<FiberTracts<double> >& fibers, const std::string& printInfo="Reading fibers:") 
{
  typename itk::FiberTractsReader::Pointer reader = itk::FiberTractsReader::New();
  reader->SetFileName(filename);
  try 
    {
    if (utl::IsLogNormal())
      std::cout << printInfo << " " << filename << std::endl;
    reader->Update(); 
    } 
  catch (itk::ExceptionObject & err) 
    { 
    std::cout << "ExceptionObject caught !" << std::endl; 
    std::cout << err << std::endl; 
    return false;
    }
  fibers = reader->GetOutput();
  return true;
}

}

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


#endif