frame.hpp

/* $Id$ */

#ifndef SAMS_FRAME_HPP
#define SAMS_FRAME_HPP

#include <string>
#include <vector>
#include "image/Image.hpp"
#include "sams/detector.hpp"
#include "sams/descriptor.hpp"
#include "sams/feature.hpp"

namespace jafar {
  namespace sams {

    /* Frame class. contains all data according one image
    */
    class Frame {
    public:

      // constructor/desctructor
      Frame() : mImg(0) {};
      Frame(int width_, int height_) : mImg(new jafar::image::Image(width_,height_,IPL_DEPTH_8U,JfrImage_CS_GRAY)) {};
      Frame(const jafar::image::Image& img_) : mImg(0) {
        attachImage(img_);
      };
      Frame(const std::string filename_) : mImg(0) {
        attachImage(filename_);
      };

      ~Frame() {
        clean();
        if (mImg)
          delete mImg;
      };

      // accessors
      operator jafar::image::Image const & () const {
        return *mImg;
      };

      operator jafar::image::Image & () {
        return *mImg;
      };

      operator jafar::image::Image * const () const {
        return mImg;
      };

      operator jafar::image::Image * () {
        return mImg;
      };

      inline int size() const {
        return mFeatures.size();
      };

      inline Feature * const ptr(int i) const {
        return mFeatures[i];
      };

      inline Feature * ptr(int i) {
        return mFeatures[i];
      };

      inline Feature const & operator[] (int i) const {
        return *(mFeatures[i]);
      };

      inline Feature& operator[] (int i) {
        return *(mFeatures[i]);
      };

      // methods
      void attachImage(const jafar::image::Image& img_) {
        JFR_PRECOND((img_.depth() == IPL_DEPTH_8U) && (img_.channels() == 1),
          "Frame: wrong source image type. must be U8 single channel");
        if (mImg)
          delete mImg;
        mImg = new jafar::image::Image(img_);       
      };

      void attachImage(const std::string filename_) {
        if (mImg)
          delete mImg;
        mImg = jafar::image::Image::loadImage(filename_.c_str(),0);
      };

      void detachImage(bool erase_=false) {
        clean();
        if (erase_)
          delete mImg;
        mImg = 0;
      };

      void generateFeatures(const jafar::sams::Detector& det_); 

      void generateDescriptors(const jafar::sams::DescriptorFactory& desc_);
      
      inline void clean() {
        for(int i=0; i< size(); i++)
          delete mFeatures[i];
        mFeatures.clear();
      };
    private:
      jafar::image::Image * mImg;
      std::vector<jafar::sams::Feature *> mFeatures;      

    }; // class Frame
  } // namespace sams
} // namespace jafar

#endif // SAMS_FRAME_HPP