#pragma once
/*=========================================================================

 medInria

 Copyright (c) INRIA 2013 - 2020. 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.

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

#include <itkImageIOBase.h>
#include <itkImageRegion.h>
#include <itkMultiThreaderBase.h>

#include <medImageIOExport.h>

namespace itk
{
  class MEDIMAGEIO_EXPORT MultiThreadedImageIOBase: public ImageIOBase
  {
    
  public:
    typedef MultiThreadedImageIOBase Self;
    typedef ImageIOBase              Superclass;
    typedef SmartPointer<Self>       Pointer;
    typedef SmartPointer<const Self> ConstPointer;

    itkTypeMacro (MultiThreadedImageIOBase, ImageIOBase)

    typedef std::vector<std::string> FileNameVectorType;
    typedef ImageRegion<1>           RegionType;
    
    itkGetObjectMacro (MultiThreaderBase, MultiThreaderBase)

    /**
       Set/Get the number of threads for this reader.
     */
    itkSetMacro (NumberOfThreads, int)
    itkGetMacro (NumberOfThreads, int)

    virtual void AddFileName (const char* filename);

    virtual void SetFileNames (FileNameVectorType filenames);
    
    virtual FileNameVectorType GetFileNames() const
    { return m_FileNames; }
    
    virtual void Read(void* buffer);
    
  protected:
    MultiThreadedImageIOBase();
    ~MultiThreadedImageIOBase(){}

    virtual void ThreadedRead (void* buffer, RegionType region, int threadId){}

    static ITK_THREAD_RETURN_TYPE ThreaderCallback( void *arg );
    
    struct ThreadStruct
    {
      Pointer Reader;
      void*   Buffer;
    };

    virtual unsigned int SplitRequestedRegion (unsigned int id, unsigned int total, RegionType& region);
    
  private:
    MultiThreadedImageIOBase (const Self&);
    void operator=(const Self&);

    FileNameVectorType m_FileNames;
    
    MultiThreaderBase::Pointer m_MultiThreaderBase;
    int                    m_NumberOfThreads;
    
  };
  
  itkEventMacroDeclaration(SliceReadEvent, itk::AnyEvent);
} // end of namespace