CorrelationParams.hpp

/* $Id$ */

#ifndef CORRELATION_PARAMS_HPP
#define CORRELATION_PARAMS_HPP

#include "stereo/CalibrageMatrices.hpp"
#include "stereo/ImagesSizes.hpp"

namespace jafar {
  namespace stereo {

    class CorrelationParams
    {
    public:
      enum STATE {NOT_INITIALIZED, INITIALIZED};
      
      enum CORRELATION_TYPE{ ZNCC, 
           ZNCC_SIGMA,
           CENSUS, 
           CENSUS_SIGMA};

      enum INTERPOLATION_TYPE{ NIL, 
             PARABOLIQUE_ONLY,
             BILINERAIRE_ONLY,
             BOTH};


      // Indique le type de transformation census qu'on applique
      enum CENSUS_TYPE {CENSUS_NIL, 
      CENSUS_8BITS_SQUARE,
      CENSUS_8BITS_DIAMOND,
      CENSUS_MEAN_8BITS_SQUARE,
      CENSUS_MEAN_8BITS_DIAMOND,
      CENSUS_MEAN_MEAN_8BITS_SQUARE,
      CENSUS_MEAN_MEAN_8BITS_DIAMOND};
      
      CorrelationParams();
      ~CorrelationParams();
      
      CorrelationParams& copy(CorrelationParams& srcCorrelationParams);

      void info();

      void  initialize(CorrelationParams::CORRELATION_TYPE type_correlation,
           CorrelationParams::INTERPOLATION_TYPE type_interpolation,
           int win_nb_lig,
           int win_nb_col,
           int disp_min,
           int disp_max,
           double var_th,
           double score_th,
           double diff_2_max_th,
           double diff_neighbor_max_th,
           int inv_th);
      
      
      // data members :
      CorrelationParams::STATE state;
      CorrelationParams::CORRELATION_TYPE type_correlation;
      CorrelationParams::INTERPOLATION_TYPE type_interpolation;

      // Taille de la fenêtre de corrélation
      int win_nb_lig;
      int win_nb_col;

      // Écart de disparité
      int disp_min;
      int disp_max;
      
      
      // 'th' stands for 'threshold'. Les différents seuils pour valider un
      // apariement. Dans l'ordre : seuil sur l'écart type de la fenêtre considérée,
      // sur la valeur du score max, sur l'écart entre les 2 plus grand pics, sur la
      // différence max / voisins, et sur l'écart (en pixels) corrélation directe /
      // corrélation inverse. 
      double var_th;
      double score_th;
      double diff_2_max_th;
      double diff_neighbor_max_th;
      int    inv_th;  
    };
    
  }// namespace stereo
}// namespace jafar


#endif // CORRELATION_PARAMS_HPP