/*=========================================================================
  Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv

  Authors belong to: 
  - University of LYON              http://www.universite-lyon.fr/
  - Léon Bérard cancer center       http://www.centreleonberard.fr
  - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr

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

  It is distributed under dual licence

  - BSD        See included LICENSE.txt file
  - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  ===========================================================================**/

#ifndef CLITKRelativePositionDataBaseBuilderFILTER_H
#define CLITKRelativePositionDataBaseBuilderFILTER_H

// clitk
#include "clitkFilterWithAnatomicalFeatureDatabaseManagement.h"
#include "clitkFilterBase.h"
#include "clitkRelativePositionAnalyzerFilter.h"
#include "clitkRelativePositionDataBase.h"

// itk
#include <itkImageToImageFilter.h>

namespace clitk {
  
  //--------------------------------------------------------------------
  /*
    Analyze the relative position of a Target (mask image) according
    to some Object, in a given Support. Indicate the main important
    position of this Target according the Object. 
  */
  //--------------------------------------------------------------------
  
  template <class ImageType>
  class RelativePositionDataBaseBuilderFilter:
    public virtual FilterBase,
    public clitk::FilterWithAnatomicalFeatureDatabaseManagement,
    public itk::ImageToImageFilter<ImageType, ImageType>
  {

  public:
    /** Standard class typedefs. */
    typedef itk::ImageToImageFilter<ImageType, ImageType>      Superclass;
    typedef RelativePositionDataBaseBuilderFilter<ImageType>          Self;
    typedef itk::SmartPointer<Self>                            Pointer;
    typedef itk::SmartPointer<const Self>                      ConstPointer;
       
    /** Method for creation through the object factory. */
    itkNewMacro(Self);
    
    /** Run-time type information (and related methods). */
    itkTypeMacro(RelativePositionDataBaseBuilderFilter, ImageToImageFilter);

    /** Some convenient typedefs. */
    typedef typename ImageType::ConstPointer ImageConstPointer;
    typedef typename ImageType::Pointer      ImagePointer;
    typedef typename ImageType::RegionType   RegionType; 
    typedef typename ImageType::PixelType    PixelType;
    typedef typename ImageType::SpacingType  SpacingType;
    typedef typename ImageType::SizeType     SizeType;
    typedef typename ImageType::IndexType    IndexType;
    typedef typename ImageType::PointType    PointType;
    
    /** ImageDimension constants */
    itkStaticConstMacro(ImageDimension, unsigned int, ImageType::ImageDimension);
    FILTERBASE_INIT;
    typedef itk::Image<float, ImageDimension> FloatImageType;
   
    // Inputs
    itkGetConstMacro(SupportName, std::string);
    itkSetMacro(SupportName, std::string);

    itkGetConstMacro(TargetName, std::string);
    itkSetMacro(TargetName, std::string);

    void AddObjectName(std::string s) { m_ObjectNames.push_back(s); }
    std::string & GetObjectName(int i) { return m_ObjectNames[i]; }
    int GetNumberOfObjects() { return m_ObjectNames.size(); }

    // Options   
    itkGetConstMacro(BackgroundValue, PixelType);
    itkSetMacro(BackgroundValue, PixelType);

    itkGetConstMacro(ForegroundValue, PixelType);
    itkSetMacro(ForegroundValue, PixelType);
    
    itkGetConstMacro(NumberOfBins, int);
    itkSetMacro(NumberOfBins, int);

    itkGetConstMacro(NumberOfAngles, int);
    itkSetMacro(NumberOfAngles, int);

    itkGetConstMacro(AreaLossTolerance, double);
    itkSetMacro(AreaLossTolerance, double);

    // For debug
    void PrintOptions();

    // I dont want to verify inputs information
    virtual void VerifyInputInformation() { }
    
   protected:
    RelativePositionDataBaseBuilderFilter();
    virtual ~RelativePositionDataBaseBuilderFilter() {}
    
    PixelType m_BackgroundValue;
    PixelType m_ForegroundValue;

    ImagePointer m_Support;
    ImagePointer m_Object;
    ImagePointer m_Target;

    std::string m_SupportName;
    std::string m_TargetName;
    std::vector<std::string> m_ObjectNames;

    int m_NumberOfAngles;
    std::vector<double> m_ListOfAngles;
    std::vector<clitk::RelativePositionDirectionType> m_ListOfDirections;

    int m_NumberOfBins;
    double m_AreaLossTolerance;
    
    virtual void GenerateOutputInformation();
    virtual void GenerateInputRequestedRegion();
    virtual void GenerateData();

    typename FloatImageType::Pointer
    ComputeFuzzyMap(ImageType * object, ImageType * target, double angle);
    
    void
    ComputeOptimalThresholds(FloatImageType * map, ImageType * target, int bins, double tolerance, 
                             double & threshold, double & reverseThreshold);
  private:
    RelativePositionDataBaseBuilderFilter(const Self&); //purposely not implemented
    void operator=(const Self&); //purposely not implemented
    
  }; // end class
  //--------------------------------------------------------------------

} // end namespace clitk
//--------------------------------------------------------------------

#include "clitkRelativePositionDataBaseBuilderFilter.txx"

#endif