Mask.h

Go to the documentation of this file.

#ifndef __D_MASK__
#define __D_MASK__

#include <opencv/cv.h>
#include <opencv/highgui.h>
#include <vector>

class Mask
{
public:

  Mask(const char *filename);
  
  Mask(const cv::Mat &mask);
  
  virtual ~Mask();
  
  void shrink(int margin); 
  
  inline bool empty() const;
  
  void reset();
  
  void maskKeyPoints(std::vector<cv::KeyPoint> &keys);
  
  void getPointsOutsideMask(const std::vector<cv::KeyPoint> &keys, 
    std::vector<unsigned int> &i_out) const;

  inline bool test(int x, int y) const
  { 
    return x >= 0 && x < m_current_mask.cols &&
      y >= 0 && y < m_current_mask.rows &&
      m_current_mask.at<unsigned char>(y, x) != 0;
  }

protected:
  
  cv::Mat m_original_mask;
  
  cv::Mat m_current_mask;

};

// ---------------------------------------------------------------------------

inline bool Mask::empty() const
{
  return this->m_original_mask.empty();
}

// ---------------------------------------------------------------------------

#endif