forked from oreillymedia/Learning-OpenCV-3_examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
example_09-02.cpp
104 lines (96 loc) · 2.58 KB
/
example_09-02.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Example 9-2. Toy program for using a mouse to draw boxes on the screen
#include <opencv2/opencv.hpp>
// Define our callback which we will install for
// mouse events
//
void my_mouse_callback(
int event, int x, int y, int flags, void* param
);
cv::Rect box;
bool drawing_box = false;
// A little subroutine to draw a box onto an image
//
void draw_box( cv::Mat& img, cv::Rect box ) {
cv::rectangle(
img,
box.tl(),
box.br(),
cv::Scalar(0x00,0x00,0xff) /* red */
);
}
void help(char** argv) {
std::cout << "Example 9-2. Toy program for using a mouse to draw boxes on the screen"
<< "\nCall:\n" << argv[0] <<
"\n\nshows how to use a mouse to draw regions in an image. Esc to quit\n" << std::endl;
}
int main( int argc, char** argv ) {
help(argv);
box = cv::Rect(-1,-1,0,0);
cv::Mat image(200, 200, CV_8UC3), temp;
image.copyTo(temp);
box = cv::Rect(-1,-1,0,0);
image = cv::Scalar::all(0);
cv::namedWindow( "Box Example" );
// Here is the crucial moment where we actually install
// the callback. Note that we set the value of 'params' to
// be the image we are working with so that the callback
// will have the image to edit.
//
cv::setMouseCallback(
"Box Example",
my_mouse_callback,
(void*)&image
);
// The main program loop. Here we copy the working image
// to the temp image, and if the user is drawing, then
// put the currently contemplated box onto that temp image.
// Display the temp image, and wait 15ms for a keystroke,
// then repeat.
//
for(;;) {
image.copyTo(temp);
if( drawing_box ) draw_box( temp, box );
cv::imshow( "Box Example", temp );
if( cv::waitKey( 15 ) == 27 ) break;
}
return 0;
}
// This is our mouse callback. If the user
// presses the left button, we start a box.
// When the user releases that button, then we
// add the box to the current image. When the
// mouse is dragged (with the button down) we
// resize the box.
//
void my_mouse_callback(
int event, int x, int y, int flags, void* param)
{
cv::Mat& image = *(cv::Mat*) param;
switch( event ) {
case cv::EVENT_MOUSEMOVE: {
if( drawing_box ) {
box.width = x-box.x;
box.height = y-box.y;
}
}
break;
case cv::EVENT_LBUTTONDOWN: {
drawing_box = true;
box = cv::Rect( x, y, 0, 0 );
}
break;
case cv::EVENT_LBUTTONUP: {
drawing_box = false;
if( box.width < 0 ) {
box.x += box.width;
box.width *= -1;
}
if( box.height < 0 ) {
box.y += box.height;
box.height *= -1;
}
draw_box( image, box );
}
break;
}
}