getLines.h

Go to the documentation of this file.

// Copyright (C) 2011, Albert-Ludwig-Universität Freiburg
// Author: Ina Baumgarten <ina.baumgarten@gmail.com>

#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <string>
#include <utility>
using namespace cv;
using namespace std;

namespace portrait_painter {
class GetLines {
public:
        GetLines();

        vector<pair<CvPoint2D32f, vector<CvPoint2D32f> > > getLines(string& image);

        vector<pair<CvPoint2D32f, vector<CvPoint2D32f> > > scaleTo(size_t size,
                        bool preserveRatio);

        void drawLines();

        void drawPR2(float percentualHeightOfImage);

private:
        struct Lines {
                CvPoint2D32f start; 
                CvPoint2D32f end; 
                vector<CvPoint2D32f> relative; 
                unsigned int id; 
        };

        void readInFile(string& image, vector<string>* output);

        void minimize();

        void eraseDoubleLines(size_t maxDistance);

        void eraseNearestNeighbours(size_t maxDistance, size_t distance,
                        vector<CvPoint2D32f>& absolutePoints);

        vector<Lines> connectLines(size_t maxDistance,
                        vector<pair<CvPoint2D32f, vector<CvPoint2D32f> > >& linesToConnect);

        void minimizeLines(size_t maxDistance, size_t threshold);

        void setPotraceParams(size_t turdSize, bool invert);

        void setErasingParams(size_t maxDistanceBetweenPoints,
                        size_t minDistanceWithinVector, float splitFactor,
                        size_t minLineSize);

        void setConnectingParams(size_t maxDistanceBetweenLines);

        void setMinimizeLinesParams(size_t maxDistance, size_t threshold);

        vector<pair<CvPoint2D32f, vector<CvPoint2D32f> > > lines;

        int _size;

        Mat src;

        string image_name;

        struct getLinesParams {
                size_t turdSize; 
                bool invert; 
                size_t maxDistanceBetweenPoints; 
                size_t minDistanceWithinVector; 
                float splitFactor; 
                size_t minLineSize; 
                size_t maxDistanceBetweenLines; 
                size_t maxDistance; 
                size_t threshold; 
        };

        getLinesParams params;
};
}