TextDetection.h

/*
    Copyright 2012 Andrew Perrault and Saurav Kumar.

    This file is part of DetectText.

    DetectText is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    DetectText is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with DetectText.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef TEXTDETECTION_H
#define TEXTDETECTION_H

#include <opencv/cv.h>

using namespace std;

struct Point2d {
    int x;
    int y;
    float SWT;
};

struct Point2dFloat {
    float x;
    float y;
};

struct Ray {
        Point2d p;
        Point2d q;
        std::vector<Point2d> points;
};

struct Point3dFloat {
    float x;
    float y;
    float z;
};

struct Component 
{
	std::vector<Point2d> componentPoints;
	int componentPointsNum;
	Point2dFloat center; //中心坐标
	float medianSWT; //笔划宽度的中值
	Point2dFloat demension; //长与宽
	Point3dFloat color;
	std::pair<Point2d,Point2d> compBB;

	Component() {};
	Component(vector<Point2d> points, 
		int nComPoints,
		Point2dFloat center, 
		float medianSWT,
		Point2dFloat demension,
		std::pair<Point2d,Point2d> compBB,
		Point3dFloat color = Point3dFloat()):
		componentPoints(points), componentPointsNum(nComPoints), center(center), medianSWT(medianSWT), 
	demension(demension), compBB(compBB), color(color) {}
};

struct Chain {
    int p;
    int q;
    float dist;
    bool merged;
    Point2dFloat direction;
    std::vector<int> components;
};

bool Point2dSort (Point2d const & lhs,
                  Point2d const & rhs);

IplImage * textDetection (IplImage *    float_input,
                          bool dark_on_light);

void strokeWidthTransform (IplImage * edgeImage,
                           IplImage * gradientX,
                           IplImage * gradientY,
                           bool dark_on_light,
                           IplImage * SWTImage,
                           std::vector<Ray> & rays);

void SWTMedianFilter (IplImage * SWTImage,
                     std::vector<Ray> & rays);

void findLegallyConnectedComponents (IplImage * SWTImage,
                                std::vector<Ray> & rays,
								std::vector<std::vector<Point2d> > &components);

void findLegallyCC(cv::Mat SWTImage,
								std::vector<std::vector<Point2d> > &components);

std::vector< std::vector<Point2d> >
findLegallyConnectedComponentsRAY (IplImage * SWTImage,
                                std::vector<Ray> & rays);

void componentStats(IplImage * SWTImage,
                                        const std::vector<Point2d> & component,
                                        float & mean, float & variance, float & median,
                                        int & minx, int & miny, int & maxx, int & maxy);

vector<Component> filterComponents(IplImage *SWTImage, std::vector<std::vector<Point2d> > & validComponents);
                      
std::vector<Chain> makeChains(vector<Component> &components);

void computeAverageComponentColor(IplImage *colorImage, 
								  vector<Component> &components);

void clusterChineseWord( IplImage * colorImage, vector<Component> &components);
#endif // TEXTDETECTION_H