dataBase.hpp

/* $Id$ */

#ifndef SIFT_DATA_BASE_HPP
#define SIFT_DATA_BASE_HPP

#include <iostream>
#include <list>

#include "boost/numeric/ublas/vector.hpp"

namespace jafar {
  namespace sift {

    typedef boost::numeric::ublas::bounded_vector<int, 128> DescriptorType;

    int descriptorDistSquared(DescriptorType const& d1, DescriptorType const& d2);

    struct DBKey {

      typedef unsigned int IdType;
      static const unsigned int NoId = 0;

      IdType id;
      DescriptorType descriptor;

      DBKey(IdType id_, DescriptorType descriptor_) : 
  id(id_), descriptor(descriptor_) {}

    };

    struct SiftPoint {

      SiftPoint() : id(DBKey::NoId) {}
      SiftPoint(SiftPoint const& pt) :
  id(pt.id), u(pt.u), v(pt.v), scale(pt.scale), orientation(pt.orientation), descriptor(pt.descriptor) {}

      DBKey::IdType id;

      float u, v;
      float scale, orientation;
      DescriptorType descriptor;

    };

    std::ostream& operator <<(std::ostream& s, const SiftPoint& pt);

    class DataBase {

    private:

      DBKey::IdType idFactory;
      DBKey::IdType getId() {return ++idFactory;}

      typedef std::list<DBKey> DBKeysType;

      DBKeysType p_dbKeys;

    public:

      DataBase() : idFactory(DBKey::NoId) {};
      ~DataBase() {};

      bool findPoint(SiftPoint& pt, bool addToDb = false);
      
      template<class Container>
      void processPoints(Container& siftPoints, bool addToDb = false) {
  for (typename Container::iterator it = siftPoints.begin() ; it != siftPoints.end() ; ++it) {
    findPoint(*it, addToDb);
  }
      }

    };

  }
}

#endif // SIFT_DATA_BASE_HPP