#include <iostream>
class canvas{
public:
bool setupQ;
int minDims,maxDims;
double scale;
int rows, cols;
void init(int minD, int maxD){
minDims = minD; maxDims = maxD;
scale = 1.0;
rows = 0;
cols = 0;
setupQ = false;
}
if(setupQ){
if(corner.
x < max.x){corner.
x = (int)(max.x + 1.0);};
if(corner.
y < max.y){corner.
y = (int)(max.y + 1.0);};
if(origin.
x > min.x){origin.
x = (int) min.x;};
if(origin.
y > min.y){origin.
y = (int) min.y;};
} else {
corner =
cv::Point((
int)(max.x + 1.0), (
int)(max.y + 1.0));
}
int c = (int)(scale*((corner.
x + 1.0) - origin.
x));
if(c<minDims){
scale = scale * (double)minDims/(double)c;
} else {
if(c>maxDims){
scale = scale * (double)maxDims/(double)c;
}
}
int r = (int)(scale*((corner.
y + 1.0) - origin.
y));
if(r<minDims){
scale = scale * (double)minDims/(double)r;
} else {
if(r>maxDims){
scale = scale * (double)maxDims/(double)r;
}
}
cols = (int)(scale*((corner.
x + 1.0) - origin.
x));
rows = (int)(scale*((corner.
y + 1.0) - origin.
y));
setupQ = true;
}
void stretch(vector<Point2f> pts)
{
for(size_t i=1; i < pts.size(); i++){
if(max.x < pnt.
x){max.x = pnt.
x;};
if(max.y < pnt.
y){max.y = pnt.
y;};
if(min.x > pnt.
x){min.x = pnt.
x;};
if(min.y > pnt.
y){min.y = pnt.
y;};
};
stretch(min, max);
}
{
for( int i = 0; i < 4; i++ ){
if(max.x < pnt.
x){max.x = pnt.
x;};
if(max.y < pnt.
y){max.y = pnt.
y;};
if(min.x > pnt.
x){min.x = pnt.
x;};
if(min.y > pnt.
y){min.y = pnt.
y;};
}
stretch(min, max);
}
{
stretch(box);
}
ellipse(img, box, color, lineThickness, LINE_AA);
for( int j = 0; j < 4; j++ ){
line(img, vtx[j], vtx[(j+1)%4], color, lineThickness, LINE_AA);
}
}
void drawPoints(vector<Point2f> pts,
cv::Scalar color)
{
stretch(pts);
}
for(size_t i=0; i < pts.size(); i++){
};
}
void drawLabels( std::vector<std::string> text, std::vector<cv::Scalar> colors)
{
}
int vPos = 0;
for (size_t i=0; i < text.size(); i++) {
std::string txt = text[i];
Size textsize = getTextSize(txt, FONT_HERSHEY_COMPLEX, 1, 1, 0);
vPos += (int)(1.3 * textsize.
height);
cv::putText(img, txt, org, FONT_HERSHEY_COMPLEX, 1, color, 1, LINE_8);
}
}
};
static void help(char** argv)
{
cout << "\nThis program is demonstration for ellipse fitting. The program finds\n"
"contours and approximate it by ellipses. Three methods are used to find the \n"
"elliptical fits: fitEllipse, fitEllipseAMS and fitEllipseDirect.\n"
"Call:\n"
<< argv[0] << " [image_name -- Default ellipses.jpg]\n" << endl;
}
int sliderPos = 70;
bool fitEllipseQ, fitEllipseAMSQ, fitEllipseDirectQ;
void processImage(int, void*);
int main(
int argc,
char** argv )
{
fitEllipseQ = true;
fitEllipseAMSQ = true;
fitEllipseDirectQ = true;
if (parser.has("help"))
{
help(argv);
return 0;
}
string filename = parser.get<string>("@image");
image =
imread(samples::findFile(filename), 0);
if( image.empty() )
{
cout << "Couldn't open image " << filename << "\n";
return 0;
}
createTrackbar(
"threshold",
"result", &sliderPos, 255, processImage );
processImage(0, 0);
return 0;
}
}
void processImage(int , void*)
{
vector<vector<Point> > contours;
Mat bimage = image >= sliderPos;
findContours(bimage, contours, RETR_LIST, CHAIN_APPROX_NONE);
canvas paper;
std::vector<std::string> text;
std::vector<cv::Scalar> color;
if (fitEllipseQ) {
text.push_back("OpenCV");
color.push_back(fitEllipseColor);
}
if (fitEllipseAMSQ) {
text.push_back("AMS");
color.push_back(fitEllipseAMSColor);
}
if (fitEllipseDirectQ) {
text.push_back("Direct");
color.push_back(fitEllipseDirectColor);
}
paper.drawLabels(text, color);
int margin = 2;
vector< vector<Point2f> > points;
for(size_t i = 0; i < contours.size(); i++)
{
size_t count = contours[i].size();
if( count < 6 )
continue;
vector<Point2f>pts;
for (
int j = 0; j < pointsf.
rows; j++) {
if ((pnt.
x > margin && pnt.
y > margin && pnt.
x < bimage.
cols-margin && pnt.
y < bimage.
rows-margin)) {
if(j%20==0){
pts.push_back(pnt);
}
}
}
points.push_back(pts);
}
for(size_t i = 0; i < points.size(); i++)
{
vector<Point2f> pts = points[i];
if (pts.size()<5) {
continue;
}
if (fitEllipseQ) {
if (isGoodBox(box)) {
paper.drawEllipseWithBox(box, fitEllipseColor, 3);
}
}
if (fitEllipseAMSQ) {
if (isGoodBox(boxAMS)) {
paper.drawEllipseWithBox(boxAMS, fitEllipseAMSColor, 2);
}
}
if (fitEllipseDirectQ) {
if (isGoodBox(boxDirect)){
paper.drawEllipseWithBox(boxDirect, fitEllipseDirectColor, 1);
}
}
paper.drawPoints(pts, fitEllipseTrueColor);
}
}
Designed for command line parsing.
Definition utility.hpp:820
n-dimensional dense array class
Definition mat.hpp:812
static CV_NODISCARD_STD MatExpr zeros(int rows, int cols, int type)
Returns a zero array of the specified size and type.
_Tp & at(int i0=0)
Returns a reference to the specified array element.
int cols
Definition mat.hpp:2138
bool empty() const
Returns true if the array has no elements.
int rows
the number of rows and columns or (-1, -1) when the matrix has more than 2 dimensions
Definition mat.hpp:2138
void convertTo(OutputArray m, int rtype, double alpha=1, double beta=0) const
Converts an array to another data type with optional scaling.
_Tp y
y coordinate of the point
Definition types.hpp:202
_Tp x
x coordinate of the point
Definition types.hpp:201
The class represents rotated (i.e. not up-right) rectangles on a plane.
Definition types.hpp:531
Point2f center
returns the rectangle mass center
Definition types.hpp:563
Size2f size
returns width and height of the rectangle
Definition types.hpp:565
void points(Point2f pts[]) const
Template class for specifying the size of an image or rectangle.
Definition types.hpp:335
_Tp height
the height
Definition types.hpp:363
_Tp width
the width
Definition types.hpp:362
Template class for short numerical vectors, a partial case of Matx.
Definition matx.hpp:369
Point2i Point
Definition types.hpp:209
Point_< float > Point2f
Definition types.hpp:207
#define CV_32F
Definition interface.h:78
unsigned char uchar
Definition interface.h:51
#define CV_8UC3
Definition interface.h:90
#define MIN(a, b)
Definition cvdef.h:513
#define MAX(a, b)
Definition cvdef.h:517
void imshow(const String &winname, InputArray mat)
Displays an image in the specified window.
int waitKey(int delay=0)
Waits for a pressed key.
void namedWindow(const String &winname, int flags=WINDOW_AUTOSIZE)
Creates a window.
int createTrackbar(const String &trackbarname, const String &winname, int *value, int count, TrackbarCallback onChange=0, void *userdata=0)
Creates a trackbar and attaches it to the specified window.
CV_EXPORTS_W Mat imread(const String &filename, int flags=IMREAD_COLOR)
Loads an image from a file.
void putText(InputOutputArray img, const String &text, Point org, int fontFace, double fontScale, Scalar color, int thickness=1, int lineType=LINE_8, bool bottomLeftOrigin=false)
Draws a text string.
RotatedRect fitEllipseDirect(InputArray points)
Fits an ellipse around a set of 2D points.
RotatedRect fitEllipseAMS(InputArray points)
Fits an ellipse around a set of 2D points.
void findContours(InputArray image, OutputArrayOfArrays contours, OutputArray hierarchy, int mode, int method, Point offset=Point())
Finds contours in a binary image.
RotatedRect fitEllipse(InputArray points)
Fits an ellipse around a set of 2D points.
int main(int argc, char *argv[])
Definition highgui_qt.cpp:3
"black box" representation of the file storage associated with a file on disk.
Definition core.hpp:102