contour.h

Go to the documentation of this file.

#ifndef CONTOUR_H
#define CONTOUR_H

#include <vector>
#include <valarray>

#include <opencv2/core/core.hpp>

namespace tuw {


class Contour {
public:

    Contour();
    ~Contour();
    static const int ANGLE_8U;
    static const int ANGLE_32F;
    static const int ANGLE_64F;

    static const int MODE_SIMPLE;
    static const int MODE_CONTOUR;
    static const int MODE_GRAIDENT;
    static const int MODE_COMPLEX;



    void Init ( unsigned int iImgWidth, unsigned int iImgHeight, bool bAllowToModifyTheSources = false, unsigned char iEdgeToProcess = 0xFF, unsigned char iEdgeInProcess = 0xFF-1, unsigned char iEdgeProcessed = 0xFF-2, unsigned char iEdgeStrengthRemoved = 0 );
    void Perform ( unsigned char* pImgEdgeStrength, int defEdgeLinkMode = MODE_CONTOUR, void* pImgEdgeDirection = NULL, int defImgEdgeDirectionType = ANGLE_8U );

    void Draw( unsigned char* pImgRGB );

    unsigned int GetNrOfEdges() {
        return mNrOfEdges;
    };

    int GetEdgeListSplittedXY (std::vector<cv::Point_<int> > &rEdges, std::vector<unsigned char> **ppAngle8Bit = NULL);
    
    int getContours( std::vector<std::vector<cv::Point> > &contours, unsigned min_length = 0);

    std::vector<cv::Range> getSegmentIndexes();
    
    void GetAbnormitiesInEdgesImage ( IplImage *ptImgEdge, std::vector<CvPoint> *pAbnormities, uchar iEdgeStrength = 0 );

private:
    unsigned char *mpImgEdge;
    void *mpImgEdgeDirection;
    int mdImgEdgeDirectionType;
    int mdefEdgeLinkMode;
    int mImgWidth;
    int mImgHeight;
    bool mbAllowToModifyTheSources;
    unsigned char mEdgeToProcess;
    unsigned char mEdgeInProcess;
    unsigned char mEdgeProcessed;
    static const int pppDirectionWeightsField[4][3][3];
    static const int ppGradientWeightsField[4][9];
    static const int pppCommingFromEdgeWeightsField[9][3][3];
    static const int ppContourWeightsField[9][9];
    static const float fPi;
    static const double dPi;
    //CvPoint *mpEdges;
    std::vector<CvPoint> mEdges;
    std::vector<unsigned char> mAngle8Bit;
    unsigned int mNrOfEdges;
    std::vector<cv::Range> mSegments;

    void AllocateMemory();
    void RelaseMemory();
    void Linking_Simple();
    int GetImgDirectionIndex ( CvPoint tPoint );
    void Trace_Simple ( CvPoint tPoint, int *pEnd );
    void Linking_Complex();
    void Trace_Complex ( CvPoint tPoint, int *pEnd, unsigned int iCommingFromEdge );
    void Linking_Contour();
    void Trace_Contour ( CvPoint tPoint, int *pEnd, unsigned int iCommingFromEdge );
    void Linking_Gradient();
    void Trace_Gradient ( CvPoint tPoint, int *pEnd );

    static const CvPoint GetNeighborPoint ( CvPoint pPtrCenter, int iNeighborIndex );
    static const void SortArrayIndexes ( int *pArray, int *pIndexes, const int iSize );

//Inlines
    inline unsigned char* getImgEdge ( CvPoint tPoint ) {
        return mpImgEdge + ( tPoint.y * mImgWidth + tPoint.x );
    };
    inline void SumArrayMatrix ( int *pMatrixA, int *pMatrixB, int *pSum, const int iSize ) {
        for ( int i = 0; i < iSize; i++ ) pSum[i] = pMatrixA[i] + pMatrixB[i];
    };
    inline bool isInImage( const CvPoint &p){
        return ( ( p.x > 0 ) && ( p.x < mImgWidth-1 ) && ( p.y > 0 ) && ( p.y < mImgHeight-1 ) );
    }
    inline void markNeighborEdgesAsProcessed(unsigned char *pPix){
        pPix[ mImgWidth-1] = mEdgeProcessed;
        pPix[ mImgWidth  ] = mEdgeProcessed;
        pPix[ mImgWidth+1] = mEdgeProcessed;
        pPix[-1] = mEdgeProcessed;
        pPix[ 1] = mEdgeProcessed;
        pPix[-mImgWidth-1] = mEdgeProcessed;
        pPix[-mImgWidth  ] = mEdgeProcessed;
        pPix[-mImgWidth+1] = mEdgeProcessed;
    }
};

}

#endif //CONTOUR_H
// kate: indent-mode cstyle; space-indent on; indent-width 4;