engine.hpp

/* $Id$ */

#ifndef LOCPANO_ENGINE_HPP
#define LOCPANO_ENGINE_HPP

#include <ostream>
#include <map>

#include "boost/tuple/tuple.hpp"

#include "image/Image.hpp"

#include "camera/cameraBarreto.hpp"

#include "locpano/LocPano_struct.h"

namespace jafar {
  namespace locpano {

    class Engine {

    private:

      double imageScale;

      // data structure imported from calife
      LocPanoParams* params;
      LocPanoInternal* internal;
      LocPanoData* dataBase;

      int internalIndex;

      typedef std::map<int, int> ImageIndexMap;

      ImageIndexMap imageIndexInternalToUser;
      ImageIndexMap imageIndexUserToInternal;

    public:

      Engine();
      Engine(jafar::camera::CameraParabolicBarreto const& pCam, jafar::image::Image const& initImage, int maxNbImagesInDataBase);

      ~Engine();

      void setImageScale(double imageScale_) {imageScale = imageScale_;}

      int getNbImages() const {return dataBase->nbImages;}

      void init(jafar::camera::CameraParabolicBarreto const& pCam, jafar::image::Image const& initImage, int nbImagesInDataBase);

      void addImage(jafar::image::Image const& image, bool updateProfile = true);

      void addImage(jafar::image::Image const& image, int imageIndex, bool updateProfile = true);

      double getImageDistance(jafar::image::Image const& image, 
            int index,
            bool useProfile = true, int distanceType = 3);
      
      boost::tuple<int, double> getClosestImage(jafar::image::Image const& image,
            bool useProfile = true, int distanceType = 3);

      boost::tuple<int, double> getClosestImage(jafar::image::Image const& image,
            int index1, int index2,
            bool useProfile = true, int distanceType = 3);
    private:

      boost::tuple<int, double> getClosestImageWrapper(jafar::image::Image const& image,
                   int internIndex1, int internIndex2,
                   bool useProfile = true, int distanceType = 3);

    }; // class Engine

    std::ostream& operator <<(std::ostream& s, const boost::tuple<int, double>& t);

  } // namespace locpano
} // namespace jafar

#endif // LOCPANO_ENGINE_HPP