OpenCV学习之形态学操作

//形态学操作
#include "cv.h"
#include "highgui.h"
IplImage* src = 0;
IplImage* dst = 0;
IplConvKernel* element = 0;
int element_shape = CV_SHAPE_RECT;
int max_iters = 10;
int open_close_pos = 0;
int erode_dilate_pos = 0;
void OpenClose(int pos) {
    int n = open_close_pos - max_iters;
    int an = n > 0 ? n : -n;
    element = cvCreateStructuringElementEx(an * 2 + 1, an * 2 + 1, an, an, element_shape, 0);
    if (n < 0) {
        cvErode(src, dst, element, 1);
        cvDilate(dst, dst, element, 1);
    }
    else {
        cvDilate(src, dst, element, 1);
        cvErode(dst, dst, element, 1);
    }
    cvReleaseStructuringElement(&element);
    cvShowImage("Open/Close", dst);
}
void ErodeDilate(int pos) {
    int n = erode_dilate_pos - max_iters;
    int an = n > 0 ? n : -n;
    element = cvCreateStructuringElementEx(an * 2 + 1, an * 2 + 1, an, an, element_shape, 0);
    if (n > 0) {
        cvErode(src, dst, element, 1);
    }
    else {
        cvDilate(src, dst, element, 1);
    }
    cvReleaseStructuringElement(&element);
    cvShowImage("Erode/Dilate", dst);
}
int main(int argc, char** argv) {
    src = cvLoadImage("Lena.tif", 1);
    printf("Hot Key:\n"
        "\tEsc - quit\n"
        "\tr - use rectangle structuring element\n"
        "\te - use elloptic structuring element\n"
        "\tc - use cross - shaped structuring element\n"
        "\tENTER - loop through all the options\n");
    dst = cvCloneImage(src);
    cvNamedWindow("Open/Close", 1);
    cvNamedWindow("Erode/Dilate");
    open_close_pos = erode_dilate_pos = max_iters;
    cvCreateTrackbar("iterations", "Open/Close", &open_close_pos, max_iters * 2 + 1, OpenClose);
    cvCreateTrackbar("iterations", "Erode/Dilate", &erode_dilate_pos, max_iters * 2 + 1, ErodeDilate);
    while (1) {
        int c;
        OpenClose(open_close_pos);
        ErodeDilate(erode_dilate_pos);
        c = cvWaitKey(0);
        if (c == 27)
            break;
        else if (c == 'r')
            element_shape = CV_SHAPE_RECT;
        else if (c == 'e')
            element_shape = CV_SHAPE_ELLIPSE;
        else if (c == 'c')
            element_shape = CV_SHAPE_CROSS;
        else if (c == '\r')
            element_shape = (element_shape + 1) % 3;
    }
    cvReleaseImage(&src);
    cvReleaseImage(&dst);
    cvDestroyWindow("Open/Close");
    cvDestroyWindow("Erode/Dilate");
    return 0;
}