itkFiberTractsWriter.h

Go to the documentation of this file.

#ifndef __itkFiberTractsWriter_h
#define __itkFiberTractsWriter_h

#include <itkLightProcessObject.h>

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

namespace itk
{

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

  itkTypeMacro(FiberTractsWriter, 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 WriteTractsTRK();

  void WriteTractsTCK();
  
  // void WriteTractsVTK();
    
protected:
  FiberTractsWriter(){}
  ~FiberTractsWriter(){}

  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:
  FiberTractsWriter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

};

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

}

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


#endif