TargetDetector.h

Go to the documentation of this file.

#ifndef TARGETDETECTOR_H
#define TARGETDETECTOR_H

#include <opencv2/imgproc/imgproc.hpp>

using namespace cv;

struct LineDetails {
    int length, 
        xPos; 
};

#ifdef __cplusplus
extern "C" {
#endif

enum rotation_t {
    ANTICLOCKWISE = -1, 
    NONE = 0, 
    CLOCKWISE = 1 
};

typedef struct LineResult {
    bool isProcessed, 
         isGood; 
    float xPos; 
    float width; 
    rotation_t rotation; 
} LineResult;

#ifdef __cplusplus
}
#endif

class TargetDetector {
public:
    TargetDetector(Mat &_image);
    virtual ~TargetDetector();
    void prepareImage();
    void findContours();
    bool filterContours(); // Check and eliminate invalid contours - if this returns false, this frame doesn't contain the target
    LineResult getContours();
    int thresholdVal;
    
private:
    Mat &image;
    vector<vector<Point> > contours, validContours;
    vector<Vec4i> hierarchy;
    LineResult result;
    bool shouldFilterContours;

};

#endif  /* TARGETDETECTOR_H */