// =========================================================================
// @author Leonardo Florez Valencia
// @email florez-l@javeriana.edu.co
// =========================================================================
#ifndef __CTBronchi__Functions__h__
#define __CTBronchi__Functions__h__

#include <chrono>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>

namespace CTBronchi
{
  // -----------------------------------------------------------------------
  double MeasureTime( itk::ProcessObject* f )
  {
    std::chrono::time_point< std::chrono::high_resolution_clock > s, e;
    std::chrono::duration< double > t;
    s = std::chrono::high_resolution_clock::now( );
    f->Update( );
    e = std::chrono::high_resolution_clock::now( );
    t = e - s;
    return( t.count( ) );
  }

  // -----------------------------------------------------------------------
  template< class _TImagePtr >
  void ReadImage( _TImagePtr& image, const std::string& fname )
  {
    typedef typename _TImagePtr::ObjectType _TImage;
    typedef itk::ImageFileReader< _TImage > _TReader;
    typename _TReader::Pointer reader = _TReader::New( );
    reader->SetFileName( fname );
    double t = MeasureTime( reader );
    std::cout << "Read " << fname << " in " << t << " s" << std::endl;
    image = reader->GetOutput( );
    image->DisconnectPipeline( );
  }

  // -----------------------------------------------------------------------
  template< class _TImagePtr >
  void WriteImage( const _TImagePtr& image, const std::string& fname )
  {
    typedef typename _TImagePtr::ObjectType _TImage;
    typedef itk::ImageFileWriter< _TImage > _TWriter;
    typename _TWriter::Pointer writer = _TWriter::New( );
    writer->SetFileName( fname );
    writer->SetInput( image );
    double t = MeasureTime( writer );
    std::cout << "Wrote " << fname << " in " << t << " s" << std::endl;
  }

} // ecapseman

#endif // __CTBronchi__Functions__h__

// eof - $RCSfile$