irdm.hpp

/* $Id$ */
#ifndef _SAS_IRDM_HPP_
#define _SAS_IRDM_HPP_

#include <map>
#include <list>
#include <string>
#include "camera/cameraBarreto.hpp"
#include "image/Image.hpp"
namespace jafar {
  namespace sas {
        
    struct sir {
      CvHistogram * hist;
      float x,y,w,h,area;
      std::string c,f;
      
      sir() : hist(0) {};
      ~sir() {
        if (hist) cvReleaseHist(&hist);
      }
    }; /* struct sir */
    
    
    struct sTrack {
      int m_ID; 
      bool m_tracked; 
      int m_nFrames; 
      int m_firstFrame; 
      std::list<sir *> m_track; 
      
      ~sTrack();
    }; /* struct sTrack */
    
    class IRDM {
      public:
      IRDM () : m_next_ID(0), m_cur_Frame(0), m_camModelInit(false) {};
      virtual ~IRDM ();

      inline sTrack * createSTrack(sir * s) {
        sTrack * stk = new sTrack();
        stk->m_ID = m_next_ID++;
        stk->m_firstFrame = m_cur_Frame;
        stk->m_nFrames = 1;
        stk->m_tracked = true;
        stk->m_track.push_back(s);
        m_stracks.push_back(stk);
        return stk;
      };
      
      //void pruneTracks(int min_size=5);

      inline void attachImg(std::string const & label, jafar::image::Image const & img) {
        m_imgs[label] = img;
      };
      inline void setImg(std::string const & label, jafar::image::Image const & img) {
        m_imgs[label] = img.clone();
      };

      inline jafar::image::Image & getImg(std::string const & label) {
        return m_imgs[label];
      };

      inline bool isCamModelInit() const {
        return m_camModelInit;
      };

      inline jafar::camera::CameraParabolicBarreto const & getCamModel() const {
        return m_camModel;
      };

      inline std::list<sTrack *> const & getSTrackList () const {return m_stracks;};

      private:
      int m_next_ID;
      int m_cur_Frame;
      bool m_camModelInit;
      std::list<sir *> m_sirs;
      std::list<sTrack *> m_stracks;
      std::map<std::string, jafar::image::Image> m_imgs;
      jafar::camera::CameraParabolicBarreto m_camModel;
      
      //void m_sirs_clear();
      
      friend class IRD;
      friend class CV_IRT_BlobDetector;
      friend class CV_IRT;
      friend class BK_IRT;
      friend class Engine;
    }; /* class IRDM */
    
  } /* ns sas */
} /* ns jafar */


#endif /* _SAS_IRDM_HPP_ */