Mask.cpp

Go to the documentation of this file.

#include <opencv/cv.h>
#include <opencv/highgui.h>
#include "Mask.h"

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

Mask::Mask(const char *filename)
{
  m_original_mask = cv::imread(filename, 0);
  m_current_mask = m_original_mask.clone();
}

// ---------------------------------------------------------------------------
  
Mask::Mask(const cv::Mat &mask)
{
  m_original_mask = mask.clone();
  m_current_mask = m_original_mask.clone();
}

// ---------------------------------------------------------------------------
  
Mask::~Mask()
{
}

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

void Mask::shrink(int margin)
{
  if(margin > 0)
  {
    cv::Mat erosion = cv::Mat::ones(margin*2+1, margin*2+1, CV_8U);
    cv::Mat aux;
    cv::erode(m_current_mask, aux, erosion);
    m_current_mask = aux;
  }
}

// ---------------------------------------------------------------------------
  
void Mask::reset()
{
  m_current_mask = m_original_mask.clone();
}

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

void Mask::maskKeyPoints(std::vector<cv::KeyPoint> &keys)
{
  int deleted = 0;
  const int N = keys.size();
  cv::KeyPoint aux;
  
  int i = 0;
  while(i < N - deleted)
  {
    if(!m_current_mask.at<unsigned char>(keys[i].pt.y, keys[i].pt.x))
    {
      // move to the last position
      aux = keys[ N - 1 - deleted ];
      keys[ N - 1 - deleted ] = keys[i];
      keys[i] = aux;
      ++deleted;
    }
    else
    {
      ++i;
    }
  }

  if(deleted) keys.resize(N - deleted);
}

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

void Mask::getPointsOutsideMask(const std::vector<cv::KeyPoint> &keys, 
    std::vector<unsigned int> &i_out) const
{
  i_out.clear();
  
  for(unsigned int i = 0; i < keys.size(); ++i)
  {
    if(!test(keys[i].pt.x, keys[i].pt.y)) i_out.push_back(i);
  }
  
}

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