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

  Program:   Tensor ToolKit - TTK
  Module:    $URL: svn+ssh://ntoussaint@scm.gforge.inria.fr/svn/ttk/trunk/Common/itkAffineTensorTransform.h $
  Language:  C++
  Date:      $Date: 2010-06-07 12:39:13 +0100 (Mon, 07 Jun 2010) $
  Version:   $Revision: 68 $

  Copyright (c) INRIA 2010. All rights reserved.
  See LICENSE.txt for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkGradientFileWriter_h
#define __itkGradientFileWriter_h

#include "itkLightProcessObject.h"
#include "itkVector.h"
#include <string>
#include <vector>
#include <itksys/SystemTools.hxx>
//#include <itksys/ios/sstream>
#include <sstream>
#include <fstream>

#include <ITKTensorExport.h>

namespace itk
{

/** \class GradientFileWriter
 * 
 * \brief 
 */
class ITKTENSOR_EXPORT GradientFileWriter : public LightProcessObject
{
public:

  /** SmartPointer typedef support */
  typedef GradientFileWriter Self;
  typedef SmartPointer<Self>  Pointer;

  typedef double ScalarType;
  typedef Vector<ScalarType, 3> VectorType;
  typedef std::vector<VectorType> VectorListType;
    
  /** Method for creation through the object factory */
  itkNewMacro(Self);

  /** Run-time type information (and related methods). */
  typedef Object Superclass;
  itkTypeMacro(GradientFileWriter, LightProcessObject);

  /** Set the filename  */
  itkSetStringMacro(FileName);

  /** Get the filename */
  itkGetStringMacro(FileName);

  /** Read out the gradients the transform */
  void Update();

  /** Get the gradient list */
  void SetGradientList(VectorListType list)
  { m_GradientList = list; }

protected:
  GradientFileWriter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  void WriteInsightStyleFile (std::ofstream& out);
  void WriteBvecFile (std::ofstream& out);
  void WriteSimpleTextFile (std::ofstream& out);
  
  GradientFileWriter();
  virtual ~GradientFileWriter();

private:

  std::string m_FileName;

  VectorListType m_GradientList;
  
};

} // namespace itk

#endif // __itkGradientFileWriter_h