OpenCV的Mat与ATL/MFC的CImage相互转换

本文主要是介绍OpenCV的Mat与ATL/MFC的CImage相互转换，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

转自：http://blog.csdn.net/liangjialang/article/details/33320093

头文件
[cpp]  view plain copy  
 class 你的类名  
 {  
 public:   
     /*MatToCImage 
     *简介： 
     *   OpenCV的Mat转ATL/MFC的CImage，仅支持单通道灰度或三通道彩色 
     *参数： 
     *   mat：OpenCV的Mat 
     *   cimage：ATL/MFC的CImage 
     */  
     void MatToCImage(Mat& mat, CImage& cimage);  
   
   
     /*CImageToMat 
     *简介： 
     *   ATL/MFC的CImage转OpenCV的Mat，仅支持单通道灰度或三通道彩色 
     *参数： 
     *   cimage：ATL/MFC的CImage 
     *   mat：OpenCV的Mat 
     */  
     void CImageToMat(CImage& cimage, Mat& mat);  
 };  
实现文件
[cpp]  view plain copy  
 void 你的类名::MatToCImage(Mat& mat, CImage& cimage)  
 {  
     if (0 == mat.total())  
     {  
         return;  
     }  
   
   
     int nChannels = mat.channels();  
     if ((1 != nChannels) && (3 != nChannels))  
     {  
         return;  
     }  
     int nWidth    = mat.cols;  
     int nHeight   = mat.rows;  
   
   
     //重建cimage  
     cimage.Destroy();  
     cimage.Create(nWidth, nHeight, 8 * nChannels);  
   
   
     //拷贝数据  
   
   
     uchar* pucRow;                                  //指向数据区的行指针  
     uchar* pucImage = (uchar*)cimage.GetBits();     //指向数据区的指针  
     int nStep = cimage.GetPitch();                  //每行的字节数,注意这个返回值有正有负  
   
   
     if (1 == nChannels)                             //对于单通道的图像需要初始化调色板  
     {  
         RGBQUAD* rgbquadColorTable;  
         int nMaxColors = 256;  
         rgbquadColorTable = new RGBQUAD[nMaxColors];  
         cimage.GetColorTable(0, nMaxColors, rgbquadColorTable);  
         for (int nColor = 0; nColor < nMaxColors; nColor++)  
         {  
             rgbquadColorTable[nColor].rgbBlue = (uchar)nColor;  
             rgbquadColorTable[nColor].rgbGreen = (uchar)nColor;  
             rgbquadColorTable[nColor].rgbRed = (uchar)nColor;  
         }  
         cimage.SetColorTable(0, nMaxColors, rgbquadColorTable);  
         delete []rgbquadColorTable;  
     }  
   
   
     for (int nRow = 0; nRow < nHeight; nRow++)  
     {  
         pucRow = (mat.ptr<uchar>(nRow));  
         for (int nCol = 0; nCol < nWidth; nCol++)  
         {  
             if (1 == nChannels)  
             {  
                 *(pucImage + nRow * nStep + nCol) = pucRow[nCol];  
             }  
             else if (3 == nChannels)  
             {  
                 for (int nCha = 0 ; nCha < 3; nCha++)  
                 {  
                     *(pucImage + nRow * nStep + nCol * 3 + nCha) = pucRow[nCol * 3 + nCha];  
                 }             
             }  
         }     
     }  
 }  
   
 void 你的类名::CImageToMat(CImage& cimage, Mat& mat)  
 {  
     if (true == cimage.IsNull())  
     {  
         return;  
     }  
   
   
     int nChannels = cimage.GetBPP() / 8;  
     if ((1 != nChannels) && (3 != nChannels))  
     {  
         return;  
     }  
     int nWidth    = cimage.GetWidth();  
     int nHeight   = cimage.GetHeight();  
   
   
     //重建mat  
     if (1 == nChannels)  
     {  
         mat.create(nHeight, nWidth, CV_8UC1);  
     }  
     else if(3 == nChannels)  
     {  
         mat.create(nHeight, nWidth, CV_8UC3);  
     }  
   
   
     //拷贝数据  
   
   
     uchar* pucRow;                                  //指向数据区的行指针  
     uchar* pucImage = (uchar*)cimage.GetBits();     //指向数据区的指针  
     int nStep = cimage.GetPitch();                  //每行的字节数,注意这个返回值有正有负  
   
   
     for (int nRow = 0; nRow < nHeight; nRow++)  
     {  
         pucRow = (mat.ptr<uchar>(nRow));  
         for (int nCol = 0; nCol < nWidth; nCol++)  
         {  
             if (1 == nChannels)  
             {  
                 pucRow[nCol] = *(pucImage + nRow * nStep + nCol);  
             }  
             else if (3 == nChannels)  
             {  
                 for (int nCha = 0 ; nCha < 3; nCha++)  
                 {  
                     pucRow[nCol * 3 + nCha] = *(pucImage + nRow * nStep + nCol * 3 + nCha);  
                 }             
             }  
         }     
     }  
 }  
       