天天看點
傳回

OPenCV操作ImageView.cpp

  1. // ImageView.cpp : implementation of the CImageView class
  2. //
  3. #include "stdafx.h"
  4. #include "Image.h"
  5. #include "ImageDoc.h"
  6. #include "ImageView.h"
  7. /////////begin/////////////////////////////////
  8. #include "cv.h"      // include core library interface
  9. #include "highgui.h" // include GUI library interface
  10. #pragma comment(lib, "cv" )
  11. #pragma comment(lib, "highgui" )
  12. #pragma comment(lib, "cxcore" )
  13. CString imgfilepath;
  14. IplImage* img=0; // Declare IPL/OpenCV image pointer
  15. ////////end/////////////////////////////////
  16. #ifdef _DEBUG
  17. #define new DEBUG_NEW
  18. #undef THIS_FILE
  19. static char THIS_FILE[] = __FILE__;
  20. #endif
  21. /////////////////////////////////////////////////////////////////////////////
  22. // CImageView
  23. IMPLEMENT_DYNCREATE(CImageView, CView)
  24. BEGIN_MESSAGE_MAP(CImageView, CView)
  25.         //{{AFX_MSG_MAP(CImageView)
  26.         ON_COMMAND(ID_IMAGE_OPENFILE, OnImageOpenFile)
  27.         ON_COMMAND(ID_IMAGE_SOBEL, OnImageSobel)
  28.         ON_COMMAND(ID_IMAGE_LAPLACE, OnImageLaplace)
  29.         ON_COMMAND(ID_IMAGE_CANNY, OnImageCanny)
  30.         ON_COMMAND(ID_IMAGE_HISTGRAM, OnImageHistgram)
  31.         ON_COMMAND(ID_IMAGE_RESIZE, OnImageResize)
  32.         ON_COMMAND(ID_IMAGE_ROTATION, OnImageRotation)
  33.         ON_COMMAND(ID_IMAGE_EROSION, OnImageErosion)
  34.         ON_COMMAND(ID_IMAGE_OPEN, OnImageOpen)
  35.         ON_COMMAND(ID_IMAGE_HOUGH_LINE, OnImageHoughLine)
  36.         ON_COMMAND(ID_IMAGE_HOUGH_CIRCLE, OnImageHoughCircle)
  37.         //}}AFX_MSG_MAP
  38.         // Standard printing commands
  39.         ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
  40.         ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
  41.         ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
  42. END_MESSAGE_MAP()
  44. // CImageView construction/destruction
  45. CImageView::CImageView()
  46. {
  47.         // TODO: add construction code here
  48. }
  49. CImageView::~CImageView()
  52. BOOL CImageView::PreCreateWindow(CREATESTRUCT& cs)
  54.         // TODO: Modify the Window class or styles here by modifying
  55.         //  the CREATESTRUCT cs
  56.         return CView::PreCreateWindow(cs);
  59. // CImageView drawing
  60. void CImageView::OnDraw(CDC* pDC)
  62.         CImageDoc* pDoc = GetDocument();
  63.         ASSERT_VALID(pDoc);
  64.         // TODO: add draw code for native data here
  67. // CImageView printing
  68. BOOL CImageView::OnPreparePrinting(CPrintInfo* pInfo)
  70.         // default preparation
  71.         return DoPreparePrinting(pInfo);
  73. void CImageView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  75.         // TODO: add extra initialization before printing
  77. void CImageView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  79.         // TODO: add cleanup after printing
  82. // CImageView diagnostics
  84. void CImageView::AssertValid() const
  86.         CView::AssertValid();
  88. void CImageView::Dump(CDumpContext& dc) const
  90.         CView::Dump(dc);
  92. CImageDoc* CImageView::GetDocument() // non-debug version is inline
  94.         ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CImageDoc)));
  95.         return (CImageDoc*)m_pDocument;
  97. #endif //_DEBUG
  99. // CImageView message handlers
  100. void CImageView::OnImageOpenFile()
  102.         // TODO: Add your command handler code here
  103.                 CFileDialog dlg(TRUE, _T("*.bmp"), "",
  104.                OFN_FILEMUSTEXIST|OFN_PATHMUSTEXIST|OFN_HIDEREADONLY,
  105.                "image files (*.bmp; *.jpg) |*.bmp;*.jpg|AVI files (*.avi) |*.avi|All Files (*.*)|*.*||",NULL);
  106.         char title[]= {"Open Image"};
  107.         dlg.m_ofn.lpstrTitle= title;
  108.         if (dlg.DoModal() == IDOK)
  109.         {
  110.                 imgfilepath= dlg.GetPathName();
  111.                 if (img!=0)
  112.                         cvReleaseImage( &img );
  113.                 img = cvvLoadImage( imgfilepath ); // load image
  114.         cvvNamedWindow( "Original Image", 1 );  // create a window
  115.         cvvShowImage( "Original Image", img );  // display the image on window
  116.         }
  118. void CImageView::OnImageSobel()
  121.     //定義的變量
  122.         IplImage* pImage= NULL; // 聲明IplImage 變量
  123.         IplImage* pImgSobelgray= NULL;// 聲明IplImage 變量，用于灰階圖像Sobel變換
  124.         IplImage* pImg8u= NULL;// 聲明IplImage 變量，用于圖像格式轉換
  125.         IplImage* pImg8uSmooth= NULL;// 聲明IplImage 變量，用于存儲平滑後的圖像
  126.         IplImage* pImgColor= NULL;// 聲明IplImage 變量，用于Sobel變換
  127.         IplImage* pImgSobelcolor= NULL;// 聲明IplImage 變量，用于彩色圖像Sobel變換
  128.     IplImage* pImgPlanes[3] = { 0, 0, 0 };
  129.         //将已讀入系統的圖像複制一份
  130.         pImage=cvCloneImage( img );
  131.         //建立和原始圖像一樣圖像記憶體區，圖像元素的位深度設為IPL_DEPTH_8U
  132.         //即無符号8位整型
  133.         pImg8u = cvCreateImage(cvGetSize(pImage),IPL_DEPTH_8U, 1);
  134.         pImg8uSmooth = cvCreateImage(cvGetSize(pImage),IPL_DEPTH_8U, 1);
  135.         //對灰階圖像進行Sobel變換
  136.         //将彩色圖像轉換為灰階圖像
  137.         cvCvtColor(pImage, pImg8u, CV_BGR2GRAY);
  138.         //對圖像進行高斯濾波
  139.         cvSmooth( pImg8u, pImg8uSmooth,CV_GAUSSIAN,3,0,0);
  140.         //建立一新圖像記憶體區，圖像元素的位深度設為IPL_DEPTH_16S有符号16位整型
  141.         //因為cvSobel函數要求目标圖像必須是16-bit圖像
  142.         pImgSobelgray = cvCreateImage(cvGetSize(pImage),IPL_DEPTH_16S, 1);
  143.         //計算一階 x方向的圖像差分，可根據需要設定參數
  144.         cvSobel(pImg8uSmooth, pImgSobelgray,0,1,3);
  145.         //将圖像格式再轉換回來，用于顯示
  146.     cvConvertScaleAbs(pImgSobelgray,pImg8u,1,0 ) ;
  147.         //建立視窗，顯示圖像
  148.         cvvNamedWindow( "Sobel gray Image", 1 );  
  149.     cvvShowImage( "Sobel gray Image", pImg8u  );
  150.         //對彩色圖像進行Sobel變換
  151.         //建立3個圖像記憶體區，分别存儲圖像3個通道，圖像元素的位深度設為IPL_DEPTH_8U
  152.         int i;
  153.     for( i = 0; i < 3; i++ )
  154.         pImgPlanes[i] = cvCreateImage( cvSize(pImage ->width, pImage ->height), 8, 1 );
  155.     //建立一新圖像記憶體區，圖像元素的位深度設為IPL_DEPTH_16S有符号16位整型
  156.         pImgSobelcolor = cvCreateImage( cvSize(pImage ->width, pImage ->height), IPL_DEPTH_16S, 1 );
  157.      //要求輸出圖像是 16 位有符号的
  158.     pImgColor = cvCreateImage( cvSize(pImage ->width, pImage ->height), 8, 3 );
  159.         //将彩色圖像分成3 個單通道圖像
  160.     cvCvtPixToPlane(pImage, pImgPlanes[0], pImgPlanes[1], pImgPlanes[2], 0 );
  161.         for( i = 0; i < 3; i++ )
  162.     {
  163.         //分别對每通道圖像進行Sobel變換
  164.                 cvSobel( pImgPlanes[i], pImgSobelcolor,0,1,3 );
  165.                 //轉化為8位的圖像
  166.         cvConvertScaleAbs(pImgSobelcolor, pImgPlanes[i], 1, 0 );  
  167.     }
  168.         //将各通道圖像進行合并
  169.         cvCvtPlaneToPix( pImgPlanes[0], pImgPlanes[1], pImgPlanes[2], 0, pImgColor);
  171.         cvvNamedWindow( "Sobel color Image", 1 );  
  172.     cvvShowImage( "Sobel color Image", pImgColor);  
  173.         //等待按鍵
  174.         cvWaitKey(0);
  175.         //鎖毀視窗
  176.         cvDestroyWindow( " Sobel gray Image " );       
  177.         cvDestroyWindow( " Sobel color Image " );      
  178.         //将程式開始定義的變量釋放
  179.         cvReleaseImage( & pImage);     
  180.         cvReleaseImage( & pImgSobelgray);
  181.         cvReleaseImage( & pImgSobelcolor);
  182.         cvReleaseImage( & pImg8u);     
  183.         cvReleaseImage( & pImg8uSmooth);
  185. void CImageView::OnImageLaplace()
  188.         //定義的變量
  189.         IplImage* pImage= NULL;// 聲明IplImage 變量
  190.         IplImage* pImgLaplace= NULL;// 聲明IplImage 變量，用于Laplace變換
  198.         pImg8uSmooth= cvCreateImage(cvGetSize(pImage),IPL_DEPTH_8U, 1);
  200.         cvCvtColor(pImage,pImg8u, CV_BGR2GRAY);
  201.         //對灰階圖像進行高斯濾波
  203.         //此函數和cvSobel函數一樣要求目标圖像必須是16-bit圖像
  204.         //如果給8-bit目标圖像會報錯
  205.         pImgLaplace = cvCreateImage(cvGetSize(img),IPL_DEPTH_16S, 1);
  206.         cvLaplace(pImg8uSmooth, pImgLaplace,3);
  207.     //将圖像格式再轉換回來，用于顯示
  208.         cvConvertScaleAbs(pImgLaplace,pImg8u,1,0 ) ;
  210.         cvvNamedWindow( "laplace gray Image", 1 );  
  211.         cvvShowImage( "laplace gray Image", pImg8u  );  
  214.         //銷毀視窗
  215.         cvDestroyWindow( " laplace gray Image " );     
  218.         cvReleaseImage( & pImgLaplace);
  220.         //cvReleaseImage( & pImg8uSmooth);
  222. void CImageView::OnImageCanny()
  226.         IplImage* pImgCanny= NULL;// 聲明IplImage 變量，用于灰階圖像Canny變換
  234.         pImgCanny = cvCreateImage(cvGetSize(pImage),IPL_DEPTH_8U, 1);
  236.         //對灰階圖像進行Canny變換
  241.         //cvCanny的源圖像必須是灰階圖像
  242.         cvCanny(pImg8uSmooth,pImgCanny, 100, 200, 3 );
  244.         cvConvertScaleAbs(pImgCanny,pImg8u,1,0 ) ;
  246.         cvvNamedWindow( "Canny Image", 1 );  
  247.         cvvShowImage( "Canny Image", pImg8u);  
  251.         cvDestroyWindow( " Canny Image " );    
  254.         cvReleaseImage( & pImgCanny);
  258. void CImageView::OnImageHistgram()
  261.         int hist_size = 255; //直方圖的針數
  262.         float range_0[]={0,256}; //第0維數值的變化範圍
  263.         float* ranges[] = { range_0 };//第1維數值的變化範圍
  264.         int i, bin_w; //下标号
  265.         float max_value = 0, min_value = 0; //直方圖數值的max和min
  266.         int min_idx = 0, max_idx = 0; //對應上述max和min數值時的下标号
  267.         double mean = 0, variance = 0; //用于存放平均值(mean)和變化量(variance)的變量
  269.         IplImage* pImgGray= NULL;// 聲明IplImage 變量
  270.         //将已讀入系統的圖像一份
  273.         pImgGray = cvCreateImage(cvSize(pImage->width,pImage->height),
  274.                         IPL_DEPTH_8U, 1);
  275.         //轉換成灰階圖像
  276.         cvCvtColor(pImage, pImgGray, CV_BGR2GRAY);
  277.         //建立一個矩形區域
  278.         CvRect rect = cvRect(0, 0, 500, 600 );
  279.         //把矩形作用于圖像以建立一個感興趣區(region of interest, ROI)
  280.         cvSetImageROI(pImgGray, rect);    // 設定 ROI
  281.         //建立一個圖像用來存放直方圖
  282.         IplImage* histImage = cvCreateImage(cvSize(320,200), 8, 1);
  283.         //建立一個指定尺寸的直方圖，用于存放從圖像獲得的資訊
  284.         CvHistogram* hist =
  285.                 cvCreateHist(1, &hist_size, CV_HIST_ARRAY, ranges, 1);
  286.         //計算直方圖并作用到hist變量
  287.         cvCalcHist( &pImgGray, hist, 0, NULL );
  288.         //得到直方圖中的最大最小值及其标号
  289.         cvGetMinMaxHistValue( hist, &min_value, &max_value, &min_idx, &max_idx);
  290.         //縮放這些針值以容入圖像内
  291.         cvScale( hist->bins, hist->bins, ((double)histImage->height)/max_value, 0 );
  292.         //設所有直方圖的數值為255
  293.         cvSet( histImage, cvScalarAll(255), 0 );
  294.         //建立一個比例因子以沿寬度縮放
  295.         bin_w = cvRound((double)histImage->width/hist_size);
  296.         for( i = 0; i < hist_size; i++ ) {
  297.                 //把直方圖畫到圖像中
  298.                 cvRectangle( histImage, cvPoint(i*bin_w, histImage->height),
  299.                         cvPoint((i+1)*bin_w,
  300.                         histImage->height - cvRound(cvGetReal1D(hist->bins,i))),
  301.                         cvScalarAll(0), -1, 8, 0 );
  302.                 //得到目前數值
  303.                 float* bins = cvGetHistValue_1D(hist,i);
  304.                 //增加均值
  305.                 mean += bins[0];
  306.         //      std::cout<< bins[0] << " " << bins[1] << std::endl;
  308.         //均值計算完畢
  309.         mean /= hist_size;
  310.         //有了均值後就可以計算變化量了(variance)
  313.                 variance += pow((bins[0] - mean),2);
  315.         //變化量計算完畢
  316.         variance /= hist_size;
  317. //      std::cout << "Histogram Mean: " << mean << std::endl;
  318. //      std::cout << "Variance: " << variance << std::endl;
  319. //      std::cout << "Standard Deviation: " << sqrt(variance) << std::endl;
  321.         cvNamedWindow("Original", 0);
  322.         cvShowImage("Original", pImage );
  323.         cvNamedWindow("Gray", 0);
  324.         cvShowImage("Gray", pImgGray );
  325.         cvNamedWindow("Histogram", 0);
  326.         cvShowImage("Histogram", histImage );
  327.         CvFont *pfont=new CvFont;
  328.         cvInitFont(pfont,CV_FONT_HERSHEY_SIMPLEX  ,0.8f,0.8f,0,2);
  329.         CString Result="Histogram Mean: ";
  330.         CString str; str.Format("%f", mean);
  331.         Result+=str+"\n";
  332.         cvPutText(histImage,Result,cvPoint(80,80),pfont,CV_RGB(255,0,0));
  333.         delete pfont;
  337.         cvReleaseImage(&histImage);
  338.         cvReleaseImage(&pImgGray);
  340.         cvDestroyWindow("Original");
  341.         cvDestroyWindow("Gray");
  342.         cvDestroyWindow("Histogram");
  344. void CImageView::OnImageResize()
  348.         IplImage* pImgNew= NULL;// 聲明IplImage 變量
  351.         CFileDialog dlg(TRUE, _T("*.bmp"), "",
  352.             OFN_FILEMUSTEXIST|OFN_PATHMUSTEXIST|OFN_HIDEREADONLY,
  354.         char title[]= {"Open Image again"};
  359.                 if (pImgNew!=0)
  360.                         cvReleaseImage( &pImgNew );
  361.                 //載入第二幅圖像
  362.                 pImgNew = cvvLoadImage( imgfilepath );
  363.                 //建立視窗，顯示第二幅圖像
  364.                 cvvNamedWindow( "Second Image", 1 );  
  365.                         cvvShowImage( "Second Image", pImgNew );  
  367.         //縮放圖像，使其尺寸與第二幅圖像相等
  368.         cvResize( pImage, pImgNew, CV_INTER_NN );
  369.         //建立視窗，顯示縮放後的幅圖像
  370.         cvvNamedWindow( "Resized Image", 1 );  
  371.         cvvShowImage( "Resized Image", pImgNew );  
  375.         cvReleaseImage(&pImage);
  376.         cvReleaseImage(&pImgNew);
  378.         cvDestroyWindow("Second Image");
  379.         cvDestroyWindow("Resized Image");
  381. void CImageView::OnImageRotation()
  384.         int angle = 30; //旋轉30度
  385.         int opt = 0;     // 1： 加縮放
  386.                                 // 0:  僅僅旋轉
  387.         double factor; //縮放因子
  389.         IplImage* pImgRotation= NULL;// 聲明IplImage 變量
  392.         pImgRotation=cvCloneImage( img );
  393.         //建立M矩陣
  394.         float m[6];
  395.                         // Matrix m looks like:
  396.                         //
  397.                         // [ m0  m1  m2 ] ===>  [ A11  A12   b1 ]
  398.                         // [ m3  m4  m5 ]       [ A21  A22   b2 ]
  400.         CvMat M = cvMat( 2, 3, CV_32F, m );
  401.         int w = img->width;
  402.         int h = img->height;
  403.         if(opt) // 旋轉加縮放
  404.                                 factor = (cos(angle*CV_PI/180.) + 1.0)*2;
  405.         else //  僅僅旋轉
  406.                                 factor = 1;
  407.         m[0] = (float)(factor*cos(-angle*CV_PI/180.));
  408.         m[1] = (float)(factor*sin(-angle*CV_PI/180.));
  409.         m[3] = -m[1];
  410.         m[4] = m[0];
  411.         // 将旋轉中心移至圖像中間
  412.         m[2] = w*0.5f;
  413.         m[5] = h*0.5f;
  414.         //  dst(x,y) = A * src(x,y) + b
  415.         cvZero(pImgRotation);
  416.         cvGetQuadrangleSubPix( pImage, pImgRotation, &M);
  417.         //建立視窗，顯示圖像  
  418.         cvNamedWindow( "Rotation Image", 1 );
  419.         cvShowImage( "Rotation Image", pImgRotation );
  424.         cvReleaseImage(&pImgRotation);
  426.         cvDestroyWindow("Rotation Image");
  428. void CImageView::OnImageErosion()
  431.         IplImage* pImgErode= NULL;// 聲明IplImage 變量
  432.         IplImage* pImgDilate= NULL;// 聲明IplImage 變量
  434.         pImgErode=cvCloneImage( img );
  435.         pImgDilate=cvCloneImage( img );
  436.         //對輸入圖像使用預設的3×3 長方形結構元素進行腐蝕
  437.         cvErode( pImgErode, pImgErode, 0, 2 );
  438.         //對輸入圖像使用預設的3×3 長方形結構元素進行膨脹
  439.         cvDilate( pImgDilate, pImgDilate, 0, 2 );
  441.         cvNamedWindow("Erosion", 0);
  442.         cvShowImage("Erosion", pImgErode );
  443.         cvNamedWindow("Dilation", 0);
  444.         cvShowImage("Dilation", pImgDilate );
  448.         cvReleaseImage(&pImgErode);
  449.         cvReleaseImage(&pImgDilate);
  451.         cvDestroyWindow("Erosion");
  452.         cvDestroyWindow("Dilation");
  454. void CImageView::OnImageOpen()
  457.         IplImage* pImgOpen= NULL;// 聲明IplImage 變量
  458.         IplImage* pImgClose= NULL;// 聲明IplImage 變量
  459.         IplConvKernel* pKernel =NULL;// 聲明IplConvKernel 變量
  461.         pImgOpen=cvCloneImage( img );
  462.         pImgClose=cvCloneImage( img );
  463.         //建立立結構元素
  464.         pKernel = cvCreateStructuringElementEx(15,15,8,8,CV_SHAPE_ELLIPSE,NULL);
  465.         //進行開運算
  466.         cvErode(pImgOpen,pImgOpen,pKernel,1);
  467.     cvDilate(pImgOpen,pImgOpen,pKernel,1);  
  468.         //進行閉運算
  469.     cvDilate(pImgClose,pImgClose,pKernel,1);  
  470.         cvErode(pImgClose,pImgClose,pKernel,1);
  472.         cvNamedWindow("Open", 0);
  473.         cvShowImage("Open", pImgOpen );
  474.         cvNamedWindow("Close", 0);
  475.         cvShowImage("Close", pImgClose );
  479.         cvReleaseImage(&pImgOpen);
  480.         cvReleaseImage(&pImgClose);
  482.         cvDestroyWindow("Open");
  483.         cvDestroyWindow("Close");
  485. void CImageView::OnImageHoughLine()
  491.         CvMemStorage* storage = NULL;// 聲明storage 變量，用于存儲檢測到的線段
  492.     CvSeq* lines = NULL;   //聲明lines變量，用于存儲直線的輪廓
  493.         //将已讀入系統的圖像複制兩份，一份作為背景圖像
  501.         //建立記憶體空間
  502.         storage = cvCreateMemStorage(0);
  503.         //利用Canny變換找出圖像邊緣
  504.     cvCanny( pImg8u, pImgCanny, 50, 500, 3 );
  505.         //Hough線變換
  506.     lines = cvHoughLines2( pImgCanny, storage, CV_HOUGH_PROBABILISTIC, 1, CV_PI/180, 80, 30, 10 );
  507.         //在原圖上畫紅直線
  509.         for( i = 0; i < lines->total; i++ )
  511.         CvPoint* line = (CvPoint*)cvGetSeqElem(lines,i);
  512.         cvLine( pImage, line[0], line[1], CV_RGB(255,0,0), 3, 8 );
  515.     cvNamedWindow( "Hough Line Transform", 1 );
  516.     cvShowImage( "Hough Line Transform", pImage);
  520.         cvDestroyWindow( " Hough Line Transform " );   
  526. void CImageView::OnImageHoughCircle()
  532.     CvSeq* circles = NULL;
  539.     if(img->nChannels != 1)
  540.          cvCvtColor( pImage, pImg8u, CV_BGR2GRAY );
  541.     else
  542.          cvCopy(pImage, pImg8u);
  543.     //平滑化
  544.     cvSmooth( pImg8u, pImg8u, CV_GAUSSIAN, 7, 7 );
  547.         //Hough圓變換
  548.     circles = cvHoughCircles( pImg8u, storage, CV_HOUGH_GRADIENT, 2, pImg8u->height/4, 250, 55 );
  549.         // 畫出識别出的圓
  551.         for( i = 0; i < circles->total; i++ )
  553.                   float* p = (float*)cvGetSeqElem( circles, i );
  554.                   cvCircle(pImage, cvPoint(cvRound(p[0]),cvRound(p[1])), cvRound(p[2]), CV_RGB(255,0,0), 3, 8, 0 );
  557.     cvNamedWindow( "Hough Circle Transform", 1 );
  558.     cvShowImage( "Hough Circle Transform", pImage);
  562.         cvDestroyWindow( " Hough Circle Transform " ); 
灰階圖像 顯示圖像 直方圖 縮放 彩色圖像
上一篇: C/C++ 字元編碼的轉換(ut8、gb2312)
下一篇: Word列印預覽隻能顯示一半？

繼續閱讀