ImageMask.h

Go to the documentation of this file.

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

#ifndef ImageMask_H
#define ImageMask_H

#include "GrayLevelImage8.h"
#include "ColorImageRGB8.h"
#include "ColorImageUV8.h"

#include "Workers/Math/Box2D.h"

class ImageMask
{
  public:

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

    ImageMask( );

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

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

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

    ImageMask( puma2::GrayLevelImage8& image, unsigned char minVal, unsigned char maxVal=255 );
    ImageMask( puma2::ColorImageUV8& image, unsigned char minValU, unsigned char minValV );

    ImageMask(puma2::GrayLevelImage8& foregroundY, puma2::ColorImageUV8& foregroundUv,
              puma2::GrayLevelImage8& backgroundY, puma2::ColorImageUV8& backgroundUv,
              int threshold );

    ~ImageMask();

    void apply( puma2::GrayLevelImage8& image, unsigned char fillValue=0 );
    void apply( puma2::ColorImageUV8& image, unsigned char fillU=0, unsigned char fillV=0 );
    void apply( puma2::ColorImageRGB8& image, unsigned char fillR, unsigned char fillG, unsigned char fillB );

    void grayOut( puma2::ColorImageRGB8& colorImage, puma2::GrayLevelImage8& graimageY );
    void grayOut( puma2::ColorImageRGB8& 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 ImageMask& other );

    void findBorders( );

    Point2D getGravCenter( );

    Box2D<int> getBoundingBox();

    ImageMask* 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