← Back to branch summary

~paparazzi-uav/paparazzi/v5.0-manual

« back to all changes in this revision

Viewing changes to sw/ext/opencv_bebop/opencv/samples/cpp/morphology2.cpp

  • Committer: Paparazzi buildbot
  • Date: 2016-05-18 15:00:29 UTC
  • Revision ID: felix.ruess+docbot@gmail.com-20160518150029-e8lgzi5kvb4p7un9
Manual import commit 4b8bbb730080dac23cf816b98908dacfabe2a8ec from v5.0 branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
sw/ext/opencv_bebop/opencv/samples/cpp/morphology2.cpp
 
1
#include "opencv2/imgproc/imgproc.hpp"
 
2
#include "opencv2/imgcodecs.hpp"
 
3
#include "opencv2/highgui/highgui.hpp"
 
4
#include <stdlib.h>
 
5
#include <stdio.h>
 
6
#include <string>
 
7
 
 
8
using namespace cv;
 
9
 
 
10
static void help()
 
11
{
 
12
 
 
13
printf("\nShow off image morphology: erosion, dialation, open and close\n"
 
14
    "Call:\n   morphology2 [image]\n"
 
15
    "This program also shows use of rect, ellipse and cross kernels\n\n");
 
16
printf( "Hot keys: \n"
 
17
    "\tESC - quit the program\n"
 
18
    "\tr - use rectangle structuring element\n"
 
19
    "\te - use elliptic structuring element\n"
 
20
    "\tc - use cross-shaped structuring element\n"
 
21
    "\tSPACE - loop through all the options\n" );
 
22
}
 
23
 
 
24
Mat src, dst;
 
25
 
 
26
int element_shape = MORPH_RECT;
 
27
 
 
28
//the address of variable which receives trackbar position update
 
29
int max_iters = 10;
 
30
int open_close_pos = 0;
 
31
int erode_dilate_pos = 0;
 
32
 
 
33
// callback function for open/close trackbar
 
34
static void OpenClose(int, void*)
 
35
{
 
36
    int n = open_close_pos - max_iters;
 
37
    int an = n > 0 ? n : -n;
 
38
    Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
 
39
    if( n < 0 )
 
40
        morphologyEx(src, dst, MORPH_OPEN, element);
 
41
    else
 
42
        morphologyEx(src, dst, MORPH_CLOSE, element);
 
43
    imshow("Open/Close",dst);
 
44
}
 
45
 
 
46
// callback function for erode/dilate trackbar
 
47
static void ErodeDilate(int, void*)
 
48
{
 
49
    int n = erode_dilate_pos - max_iters;
 
50
    int an = n > 0 ? n : -n;
 
51
    Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
 
52
    if( n < 0 )
 
53
        erode(src, dst, element);
 
54
    else
 
55
        dilate(src, dst, element);
 
56
    imshow("Erode/Dilate",dst);
 
57
}
 
58
 
 
59
 
 
60
int main( int argc, char** argv )
 
61
{
 
62
    cv::CommandLineParser parser(argc, argv, "{help h||}{ @image | ../data/baboon.jpg | }");
 
63
    if (parser.has("help"))
 
64
    {
 
65
        help();
 
66
        return 0;
 
67
    }
 
68
    std::string filename = parser.get<std::string>("@image");
 
69
    if( (src = imread(filename,1)).empty() )
 
70
    {
 
71
        help();
 
72
        return -1;
 
73
    }
 
74
 
 
75
    //create windows for output images
 
76
    namedWindow("Open/Close",1);
 
77
    namedWindow("Erode/Dilate",1);
 
78
 
 
79
    open_close_pos = erode_dilate_pos = max_iters;
 
80
    createTrackbar("iterations", "Open/Close",&open_close_pos,max_iters*2+1,OpenClose);
 
81
    createTrackbar("iterations", "Erode/Dilate",&erode_dilate_pos,max_iters*2+1,ErodeDilate);
 
82
 
 
83
    for(;;)
 
84
    {
 
85
        int c;
 
86
 
 
87
        OpenClose(open_close_pos, 0);
 
88
        ErodeDilate(erode_dilate_pos, 0);
 
89
        c = waitKey(0);
 
90
 
 
91
        if( (char)c == 27 )
 
92
            break;
 
93
        if( (char)c == 'e' )
 
94
            element_shape = MORPH_ELLIPSE;
 
95
        else if( (char)c == 'r' )
 
96
            element_shape = MORPH_RECT;
 
97
        else if( (char)c == 'c' )
 
98
            element_shape = MORPH_CROSS;
 
99
        else if( (char)c == ' ' )
 
100
            element_shape = (element_shape + 1) % 3;
 
101
    }
 
102
 
 
103
    return 0;
 
104
}