cuda npp库旋转图片

  1 #include <stdio.h>
  2 #include <stdlib.h>
  3 #include <string.h>
  4 #include <assert.h>
  5 #include <math.h>
  6 #include <windows.h>
  7 #include <FreeImage.h>
  8 #include <cuda_runtime.h>
  9 #include <npp.h>
 10 
 11 
 12 #pragma comment(lib, "FreeImage.lib")
 13 #pragma comment(lib, "cudart.lib")
 14 #pragma comment(lib, "nppi.lib")
 15 
 16 
 17 FIBITMAP* LoadImg( const char* szFile )
 18 {
 19     FREE_IMAGE_FORMAT nFif;
 20 
 21     if ( szFile == NULL || *szFile == 0 )
 22     {
 23         return NULL;
 24     }
 25 
 26     if ( ( nFif = FreeImage_GetFileType( szFile, 0 ) ) == FIF_UNKNOWN )
 27     {
 28         if ( ( nFif = FreeImage_GetFIFFromFilename( szFile ) ) == FIF_UNKNOWN )
 29         {
 30             return NULL;
 31         }
 32     }
 33 
 34     if ( !FreeImage_FIFSupportsReading( nFif ) )
 35     {
 36         return NULL;
 37     }
 38 
 39     return FreeImage_Load( nFif, szFile );
 40 }
 41 
 42 
 43 int main( int argc, char* argv[] )
 44 {
 45     cudaError_t cuRet;
 46     NppStatus nppRet;
 47     BOOL fiRet;
 48     FIBITMAP* pSrcBmp;
 49     FIBITMAP* pDstBmp;
 50     unsigned char* pSrcData;
 51     unsigned char* pDstData;
 52     Npp8u* pSrcDataCUDA;
 53     Npp8u* pDstDataCUDA;
 54     NppiSize oSrcSize;
 55     NppiSize oDstSize;
 56     NppiRect oSrcROI;
 57     NppiRect oDstROI;
 58     int nImgBpp;
 59     int nSrcPitch;
 60     int nDstPitch;
 61     int nSrcPitchCUDA;
 62     int nDstPitchCUDA;
 63     double aBoundingBox[2][2];
 64     double nAngle;
 65 
 66     if ( argc < 3 )
 67     {
 68         return -1;
 69     }
 70 
 71     /* 载入文件 */
 72     pSrcBmp = LoadImg( argv[1] );
 73     assert( pSrcBmp != NULL );
 74 
 75     nImgBpp = ( FreeImage_GetBPP( pSrcBmp ) >> 3 );
 76     pSrcData = FreeImage_GetBits( pSrcBmp );
 77 
 78     oSrcSize.width = ( int )FreeImage_GetWidth( pSrcBmp );
 79     oSrcSize.height = ( int )FreeImage_GetHeight( pSrcBmp );
 80     nSrcPitch = ( int )FreeImage_GetPitch( pSrcBmp );
 81 
 82     oSrcROI.x = oSrcROI.y = 0;
 83     oSrcROI.width = oSrcSize.width;
 84     oSrcROI.height = oSrcSize.height;
 85 
 86     nAngle = atof( argv[2] );
 87 
 88     /* 设置显卡,构建上下文 */
 89     cuRet = cudaSetDevice( 0 );
 90     assert( cuRet == cudaSuccess );
 91 
 92     /* 分配显存 */
 93     switch ( nImgBpp )
 94     {
 95         case 1:
 96             pSrcDataCUDA = nppiMalloc_8u_C1( oSrcSize.width, oSrcSize.height, &nSrcPitchCUDA );
 97             break;
 98         case 3:
 99             pSrcDataCUDA = nppiMalloc_8u_C3( oSrcSize.width, oSrcSize.height, &nSrcPitchCUDA );
100             break;
101         case 4:
102             pSrcDataCUDA = nppiMalloc_8u_C4( oSrcSize.width, oSrcSize.height, &nSrcPitchCUDA );
103             break;
104         default:
105             assert( 0 );
106             break;
107     }
108     assert( pSrcDataCUDA != NULL );
109 
110     /* 将原图传入显存 */
111     cudaMemcpy2D( pSrcDataCUDA, nSrcPitchCUDA, pSrcData, nSrcPitch, oSrcSize.width * nImgBpp, oSrcSize.height, cudaMemcpyHostToDevice );
112 
113     /* 计算旋转后长宽 */
114     nppiGetRotateBound( oSrcROI, aBoundingBox, nAngle, 0, 0 );
115     oDstSize.width = ( int )ceil( fabs( aBoundingBox[1][0] - aBoundingBox[0][0] ) );
116     oDstSize.height = ( int )ceil( fabs( aBoundingBox[1][1] - aBoundingBox[0][1] ) );
117 
118     /* 建目标图 */
119     pDstBmp = FreeImage_Allocate( oDstSize.width, oDstSize.height, nImgBpp << 3 );
120     assert( pDstBmp != NULL );
121 
122     pDstData = FreeImage_GetBits( pDstBmp );
123 
124     nDstPitch = ( int )FreeImage_GetPitch( pDstBmp );
125     oDstROI.x = oDstROI.y = 0;
126     oDstROI.width = oDstSize.width;
127     oDstROI.height = oDstSize.height;
128 
129     /* 分配显存 */
130     switch ( nImgBpp )
131     {
132         case 1:
133             pDstDataCUDA = nppiMalloc_8u_C1( oDstSize.width, oDstSize.height, &nDstPitchCUDA );
134             break;
135         case 3:
136             pDstDataCUDA = nppiMalloc_8u_C3( oDstSize.width, oDstSize.height, &nDstPitchCUDA );
137             break;
138         case 4:
139             pDstDataCUDA = nppiMalloc_8u_C4( oDstSize.width, oDstSize.height, &nDstPitchCUDA );
140             break;
141     }
142     assert( pDstDataCUDA != NULL );
143     cudaMemset2D( pDstDataCUDA, nDstPitchCUDA, 0, oDstSize.width * nImgBpp, oDstSize.height );
144 
145     /* 处理 */
146     switch ( nImgBpp )
147     {
148         case 1:
149             nppRet = nppiRotate_8u_C1R( pSrcDataCUDA, oSrcSize, nSrcPitchCUDA, oSrcROI,
150                                         pDstDataCUDA, nDstPitchCUDA, oDstROI,
151                                         nAngle, - aBoundingBox[0][0], - aBoundingBox[0][1], NPPI_INTER_CUBIC );
152             break;
153         case 3:
154             nppRet = nppiRotate_8u_C3R( pSrcDataCUDA, oSrcSize, nSrcPitchCUDA, oSrcROI,
155                                         pDstDataCUDA, nDstPitchCUDA, oDstROI,
156                                         nAngle, - aBoundingBox[0][0], - aBoundingBox[0][1], NPPI_INTER_CUBIC );
157             break;
158         case 4:
159             nppRet = nppiRotate_8u_C4R( pSrcDataCUDA, oSrcSize, nSrcPitchCUDA, oSrcROI,
160                                         pDstDataCUDA, nDstPitchCUDA, oDstROI,
161                                         nAngle, - aBoundingBox[0][0], - aBoundingBox[0][1], NPPI_INTER_CUBIC );
162             break;
163     }
164     assert( nppRet == NPP_NO_ERROR );
165 
166     cudaMemcpy2D( pDstData, nDstPitch, pDstDataCUDA, nDstPitchCUDA, oDstSize.width * nImgBpp, oDstSize.height, cudaMemcpyDeviceToHost );
167 
168     fiRet = FreeImage_Save( FIF_BMP, pDstBmp, "ret.bmp" );
169     assert( fiRet );
170 
171     nppiFree( pSrcDataCUDA );
172     nppiFree( pDstDataCUDA );
173 
174     cudaDeviceReset();
175 
176     FreeImage_Unload( pSrcBmp );
177     FreeImage_Unload( pDstBmp );
178 
179     return 0;
180 }

下面是所有通道的，扩展了一点。以后自己用

  1 #include <stdio.h>
  2 #include <stdlib.h>
  3 #include <string.h>
  4 #include <assert.h>
  5 #include <math.h>
  6 #include <windows.h>
  7 
  8 #include <FreeImage.h>
  9 
 10 #include <cuda_runtime.h>
 11 #include <npp.h>
 12 
 13 
 14 int main( int argc, char* argv[] )
 15 {
 16     cudaError_t cuRet;
 17     NppStatus nppRet;
 18     BOOL fiRet;
 19     FIBITMAP* pSrcBmp;
 20     FIBITMAP* pDstBmp;
 21     unsigned char* pSrcData;
 22     unsigned char* pDstData;
 23     Npp8u* pSrcDataCUDA;
 24     Npp8u* pDstDataCUDA;
 25     NppiSize oSrcSize;
 26     NppiSize oDstSize;
 27     NppiRect oSrcROI;
 28     NppiRect oDstROI;
 29     int nSrcPitch;
 30     int nDstPitch;
 31     int nSrcPitchCUDA;
 32     int nDstPitchCUDA;
 33     double aBoundingBox[2][2];
 34     double nAngle;
 35 
 36 
 37     /* 设置显卡,构建上下文 */
 38     cuRet = cudaSetDevice( 0 );
 39     assert( cuRet == cudaSuccess );
 40 
 41     /* 打开文件 */
 42     pSrcBmp = FreeImage_Load( FIF_BMP, "1.bmp" );
 43     assert( pSrcBmp != NULL );
 44 
 45     pSrcData = FreeImage_GetBits( pSrcBmp );
 46     assert( pSrcData != NULL );
 47 
 48     oSrcSize.width = ( int )FreeImage_GetWidth( pSrcBmp );
 49     oSrcSize.height = ( int )FreeImage_GetHeight( pSrcBmp );
 50     nSrcPitch = ( int )FreeImage_GetPitch( pSrcBmp );
 51 
 52     oSrcROI.x = oSrcROI.y = 0;
 53     oSrcROI.width = oSrcSize.width;
 54     oSrcROI.height = oSrcSize.height;
 55 
 56     nAngle = 45;
 57 
 58 
 59     /* 分配显存 */
 60     pSrcDataCUDA = nppiMalloc_8u_C1( oSrcSize.width, oSrcSize.height, &nSrcPitchCUDA );
 61     assert( pSrcDataCUDA != NULL );
 62 
 63 
 64     /* 计算旋转后长宽 */
 65     nppiGetRotateBound( oSrcROI, aBoundingBox, nAngle, 0, 0 );
 66     oDstSize.width = ( int )ceil( fabs( aBoundingBox[1][0] - aBoundingBox[0][0] ) );
 67     oDstSize.height = ( int )ceil( fabs( aBoundingBox[1][1] - aBoundingBox[0][1] ) );
 68 
 69     /* 建目标图 */
 70     pDstBmp = FreeImage_Allocate( oDstSize.width, oDstSize.height, 8 );
 71     assert( pDstBmp != NULL );
 72 
 73     pDstData = FreeImage_GetBits( pDstBmp );
 74 
 75     nDstPitch = ( int )FreeImage_GetPitch( pDstBmp );
 76     oDstROI.x = oDstROI.y = 0;
 77     oDstROI.width = oDstSize.width;
 78     oDstROI.height = oDstSize.height;
 79 
 80     /* 分配显存 */
 81     pDstDataCUDA = nppiMalloc_8u_C1( oDstSize.width, oDstSize.height, &nDstPitchCUDA );
 82     assert( pDstDataCUDA != NULL );
 83 
 84     cudaMemcpy2D( pSrcDataCUDA, nSrcPitchCUDA, pSrcData, nSrcPitch, oSrcSize.width, oSrcSize.height, cudaMemcpyHostToDevice );
 85     cudaMemset2D( pDstDataCUDA, nDstPitchCUDA, 0, oDstSize.width, oDstSize.height );
 86 
 87     /* 处理 */
 88     nppRet = nppiRotate_8u_C1R( pSrcDataCUDA, oSrcSize, nSrcPitchCUDA, oSrcROI,
 89                                 pDstDataCUDA, nDstPitchCUDA, oDstROI,
 90                                 nAngle, - aBoundingBox[0][0], - aBoundingBox[0][1], NPPI_INTER_CUBIC );
 91     assert( nppRet == NPP_NO_ERROR );
 92 
 93     cudaMemcpy2D( pDstData, nDstPitch, pDstDataCUDA, nDstPitchCUDA, oDstSize.width, oDstSize.height, cudaMemcpyDeviceToHost );
 94 
 95     fiRet = FreeImage_Save( FIF_BMP, pDstBmp, "2.bmp" );
 96     assert( fiRet );
 97 
 98     nppiFree( pSrcDataCUDA );
 99     nppiFree( pDstDataCUDA );
100 
101     cudaDeviceReset();
102 
103     FreeImage_Unload( pSrcBmp );
104     FreeImage_Unload( pDstBmp );
105 
106     return 0;
107 }