ImageMaskCV.h

Go to the documentation of this file.

/*******************************************************************************
 *  ImageMaskCV.h
 *
 *  (C) 2007 AG Aktives Sehen <agas@uni-koblenz.de>
 *           Universitaet Koblenz-Landau
 *
 *  Additional information:
 *  $Id: $
 *******************************************************************************/

#ifndef ImageMaskCV_H
#define ImageMaskCV_H

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>

#include "Workers/Math/Box2D.h"


class ImageMaskCV
{
  public:

    enum MaskValues {
      MASKED = 0,
      VISIBLE = 255
    };

    ImageMaskCV( );

    ImageMaskCV( const ImageMaskCV& other );
    ImageMaskCV& operator=( const ImageMaskCV& other );

    ImageMaskCV( unsigned width, unsigned height, unsigned char* data=0 );

    ImageMaskCV( unsigned width, unsigned height, unsigned char* data, char maskedMin, char maskedMax );

    ImageMaskCV( cv::Mat& image, unsigned char minVal, unsigned char maxVal=255 );

    ImageMaskCV(cv::Mat& foregroundY, cv::Mat& foregroundUv,
              cv::Mat& backgroundY, cv::Mat& backgroundUv,
              int threshold );

    ~ImageMaskCV();

    void apply( cv::Mat& image, unsigned char fillValue=0 );
    void apply( cv::Mat& image, unsigned char fillR, unsigned char fillG, unsigned char fillB );

    void grayOut( cv::Mat& colorImage, cv::Mat& graimageY );
    void grayOut( cv::Mat& colorImage );

    bool findValue( int x, int y, unsigned char value, float radius );

    void erode( float radius=1.0 );

    void dilate( float radius=1.0 );

    void fill( unsigned char value );

    void expand( const ImageMaskCV& other );

    void findBorders( );

    Point2D getGravCenter( );

    Box2D<int> getBoundingBox();

    ImageMaskCV* subMask( Box2D<int> area );

    unsigned char* getData() { return m_Data; }

    inline unsigned getWidth() { return m_Width; }
    inline unsigned getHeight() { return m_Height; }

  private:

    enum maskOperationT{
      dilateOperation,
      erodeOperation
    };

    void maskOperation( maskOperationT operation, float radius );

    void createCircularKernel( float radius, int*& maskOffset, int& halfMaskSize, unsigned& maskLength );

    unsigned char* m_Data;

    unsigned m_Width;
    unsigned m_Height;

};

#endif