mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-26 06:38:16 +08:00
Apply consistent code formatting across the repo. Add clang-format and pre-commit hooks.
This commit is contained in:
Rob Armstrong
@@ -36,285 +36,290 @@
|
||||
#include <windows.h>
|
||||
#endif
|
||||
|
||||
#include "FreeImage.h"
|
||||
#include "Exceptions.h"
|
||||
// Common Helpers
|
||||
|
||||
#include <string.h>
|
||||
#include "Exceptions.h"
|
||||
#include "FreeImage.h"
|
||||
|
||||
// Other Includes
|
||||
#include <cuda_runtime.h>
|
||||
#include <fstream>
|
||||
#include <iostream>
|
||||
#include <npp.h> // CUDA NPP Definitions
|
||||
#include <string.h>
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
#include <npp.h> // CUDA NPP Definitions
|
||||
// Other helpers
|
||||
#include <helper_cuda.h> // helper for CUDA Error handling and initialization
|
||||
#include <helper_string.h> // helper for string parsing
|
||||
|
||||
#include <helper_cuda.h> // helper for CUDA Error handling and initialization
|
||||
#include <helper_string.h> // helper for string parsing
|
||||
inline int cudaDeviceInit(int argc, const char **argv)
|
||||
{
|
||||
int deviceCount;
|
||||
checkCudaErrors(cudaGetDeviceCount(&deviceCount));
|
||||
|
||||
inline int cudaDeviceInit(int argc, const char **argv) {
|
||||
int deviceCount;
|
||||
checkCudaErrors(cudaGetDeviceCount(&deviceCount));
|
||||
if (deviceCount == 0) {
|
||||
std::cerr << "CUDA error: no devices supporting CUDA." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (deviceCount == 0) {
|
||||
std::cerr << "CUDA error: no devices supporting CUDA." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
int dev = findCudaDevice(argc, argv);
|
||||
|
||||
int dev = findCudaDevice(argc, argv);
|
||||
cudaDeviceProp deviceProp;
|
||||
cudaGetDeviceProperties(&deviceProp, dev);
|
||||
std::cerr << "cudaSetDevice GPU" << dev << " = " << deviceProp.name << std::endl;
|
||||
|
||||
cudaDeviceProp deviceProp;
|
||||
cudaGetDeviceProperties(&deviceProp, dev);
|
||||
std::cerr << "cudaSetDevice GPU" << dev << " = " << deviceProp.name
|
||||
<< std::endl;
|
||||
checkCudaErrors(cudaSetDevice(dev));
|
||||
|
||||
checkCudaErrors(cudaSetDevice(dev));
|
||||
|
||||
return dev;
|
||||
return dev;
|
||||
}
|
||||
|
||||
// Error handler for FreeImage library.
|
||||
// In case this handler is invoked, it throws an NPP exception.
|
||||
extern "C" void FreeImageErrorHandler(FREE_IMAGE_FORMAT oFif,
|
||||
const char *zMessage) {
|
||||
throw npp::Exception(zMessage);
|
||||
}
|
||||
extern "C" void FreeImageErrorHandler(FREE_IMAGE_FORMAT oFif, const char *zMessage) { throw npp::Exception(zMessage); }
|
||||
|
||||
std::ostream &operator<<(std::ostream &rOutputStream, const FIBITMAP &rBitmap) {
|
||||
unsigned int nImageWidth =
|
||||
FreeImage_GetWidth(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nImageHeight =
|
||||
FreeImage_GetHeight(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nPitch = FreeImage_GetPitch(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nBPP = FreeImage_GetBPP(const_cast<FIBITMAP *>(&rBitmap));
|
||||
std::ostream &operator<<(std::ostream &rOutputStream, const FIBITMAP &rBitmap)
|
||||
{
|
||||
unsigned int nImageWidth = FreeImage_GetWidth(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nImageHeight = FreeImage_GetHeight(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nPitch = FreeImage_GetPitch(const_cast<FIBITMAP *>(&rBitmap));
|
||||
unsigned int nBPP = FreeImage_GetBPP(const_cast<FIBITMAP *>(&rBitmap));
|
||||
|
||||
FREE_IMAGE_COLOR_TYPE eType =
|
||||
FreeImage_GetColorType(const_cast<FIBITMAP *>(&rBitmap));
|
||||
FREE_IMAGE_COLOR_TYPE eType = FreeImage_GetColorType(const_cast<FIBITMAP *>(&rBitmap));
|
||||
|
||||
rOutputStream << "Size (" << nImageWidth << ", " << nImageHeight << ")\n";
|
||||
rOutputStream << "Pitch " << nPitch << "\n";
|
||||
rOutputStream << "Type ";
|
||||
rOutputStream << "Size (" << nImageWidth << ", " << nImageHeight << ")\n";
|
||||
rOutputStream << "Pitch " << nPitch << "\n";
|
||||
rOutputStream << "Type ";
|
||||
|
||||
switch (eType) {
|
||||
switch (eType) {
|
||||
case FIC_MINISWHITE:
|
||||
rOutputStream << "FIC_MINISWHITE\n";
|
||||
break;
|
||||
rOutputStream << "FIC_MINISWHITE\n";
|
||||
break;
|
||||
|
||||
case FIC_MINISBLACK:
|
||||
rOutputStream << "FIC_MINISBLACK\n";
|
||||
break;
|
||||
rOutputStream << "FIC_MINISBLACK\n";
|
||||
break;
|
||||
|
||||
case FIC_RGB:
|
||||
rOutputStream << "FIC_RGB\n";
|
||||
break;
|
||||
rOutputStream << "FIC_RGB\n";
|
||||
break;
|
||||
|
||||
case FIC_PALETTE:
|
||||
rOutputStream << "FIC_PALETTE\n";
|
||||
break;
|
||||
rOutputStream << "FIC_PALETTE\n";
|
||||
break;
|
||||
|
||||
case FIC_RGBALPHA:
|
||||
rOutputStream << "FIC_RGBALPHA\n";
|
||||
break;
|
||||
rOutputStream << "FIC_RGBALPHA\n";
|
||||
break;
|
||||
|
||||
case FIC_CMYK:
|
||||
rOutputStream << "FIC_CMYK\n";
|
||||
break;
|
||||
rOutputStream << "FIC_CMYK\n";
|
||||
break;
|
||||
|
||||
default:
|
||||
rOutputStream << "Unknown pixel format.\n";
|
||||
}
|
||||
rOutputStream << "Unknown pixel format.\n";
|
||||
}
|
||||
|
||||
rOutputStream << "BPP " << nBPP << std::endl;
|
||||
rOutputStream << "BPP " << nBPP << std::endl;
|
||||
|
||||
return rOutputStream;
|
||||
return rOutputStream;
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
printf("%s Starting...\n\n", argv[0]);
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
printf("%s Starting...\n\n", argv[0]);
|
||||
|
||||
try {
|
||||
std::string sFilename;
|
||||
char *filePath;
|
||||
try {
|
||||
std::string sFilename;
|
||||
char *filePath;
|
||||
|
||||
// set your own FreeImage error handler
|
||||
FreeImage_SetOutputMessage(FreeImageErrorHandler);
|
||||
// set your own FreeImage error handler
|
||||
FreeImage_SetOutputMessage(FreeImageErrorHandler);
|
||||
|
||||
cudaDeviceInit(argc, (const char **)argv);
|
||||
cudaDeviceInit(argc, (const char **)argv);
|
||||
|
||||
NppStreamContext nppStreamCtx;
|
||||
nppStreamCtx.hStream = 0; // The NULL stream by default, set this to whatever your stream ID is if not the NULL stream.
|
||||
NppStreamContext nppStreamCtx;
|
||||
nppStreamCtx.hStream =
|
||||
0; // The NULL stream by default, set this to whatever your stream ID is if not the NULL stream.
|
||||
|
||||
cudaError_t cudaError = cudaGetDevice(&nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess)
|
||||
{
|
||||
printf("CUDA error: no devices supporting CUDA.\n");
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
cudaError_t cudaError = cudaGetDevice(&nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess) {
|
||||
printf("CUDA error: no devices supporting CUDA.\n");
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
}
|
||||
|
||||
const NppLibraryVersion *libVer = nppGetLibVersion();
|
||||
|
||||
printf("NPP Library Version %d.%d.%d\n", libVer->major, libVer->minor, libVer->build);
|
||||
|
||||
int driverVersion, runtimeVersion;
|
||||
cudaDriverGetVersion(&driverVersion);
|
||||
cudaRuntimeGetVersion(&runtimeVersion);
|
||||
|
||||
printf("CUDA Driver Version: %d.%d\n", driverVersion / 1000, (driverVersion % 100) / 10);
|
||||
printf("CUDA Runtime Version: %d.%d\n\n", runtimeVersion / 1000, (runtimeVersion % 100) / 10);
|
||||
|
||||
cudaError = cudaDeviceGetAttribute(&nppStreamCtx.nCudaDevAttrComputeCapabilityMajor,
|
||||
cudaDevAttrComputeCapabilityMajor,
|
||||
nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess)
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
|
||||
cudaError = cudaDeviceGetAttribute(&nppStreamCtx.nCudaDevAttrComputeCapabilityMinor,
|
||||
cudaDevAttrComputeCapabilityMinor,
|
||||
nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess)
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
|
||||
cudaError = cudaStreamGetFlags(nppStreamCtx.hStream, &nppStreamCtx.nStreamFlags);
|
||||
|
||||
cudaDeviceProp oDeviceProperties;
|
||||
|
||||
cudaError = cudaGetDeviceProperties(&oDeviceProperties, nppStreamCtx.nCudaDeviceId);
|
||||
|
||||
nppStreamCtx.nMultiProcessorCount = oDeviceProperties.multiProcessorCount;
|
||||
nppStreamCtx.nMaxThreadsPerMultiProcessor = oDeviceProperties.maxThreadsPerMultiProcessor;
|
||||
nppStreamCtx.nMaxThreadsPerBlock = oDeviceProperties.maxThreadsPerBlock;
|
||||
nppStreamCtx.nSharedMemPerBlock = oDeviceProperties.sharedMemPerBlock;
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "input")) {
|
||||
getCmdLineArgumentString(argc, (const char **)argv, "input", &filePath);
|
||||
}
|
||||
else {
|
||||
filePath = sdkFindFilePath("teapot512.pgm", argv[0]);
|
||||
}
|
||||
|
||||
if (filePath) {
|
||||
sFilename = filePath;
|
||||
}
|
||||
else {
|
||||
sFilename = "teapot512.pgm";
|
||||
}
|
||||
|
||||
// if we specify the filename at the command line, then we only test
|
||||
// sFilename otherwise we will check both sFilename[0,1]
|
||||
int file_errors = 0;
|
||||
std::ifstream infile(sFilename.data(), std::ifstream::in);
|
||||
|
||||
if (infile.good()) {
|
||||
std::cout << "freeImageInteropNPP opened: <" << sFilename.data() << "> successfully!" << std::endl;
|
||||
file_errors = 0;
|
||||
infile.close();
|
||||
}
|
||||
else {
|
||||
std::cout << "freeImageInteropNPP unable to open: <" << sFilename.data() << ">" << std::endl;
|
||||
file_errors++;
|
||||
infile.close();
|
||||
}
|
||||
|
||||
if (file_errors > 0) {
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
std::string sResultFilename = sFilename;
|
||||
|
||||
std::string::size_type dot = sResultFilename.rfind('.');
|
||||
|
||||
if (dot != std::string::npos) {
|
||||
sResultFilename = sResultFilename.substr(0, dot);
|
||||
}
|
||||
|
||||
sResultFilename += "_boxFilterFII.pgm";
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "output")) {
|
||||
char *outputFilePath;
|
||||
getCmdLineArgumentString(argc, (const char **)argv, "output", &outputFilePath);
|
||||
sResultFilename = outputFilePath;
|
||||
}
|
||||
|
||||
FREE_IMAGE_FORMAT eFormat = FreeImage_GetFileType(sFilename.c_str());
|
||||
|
||||
// no signature? try to guess the file format from the file extension
|
||||
if (eFormat == FIF_UNKNOWN) {
|
||||
eFormat = FreeImage_GetFIFFromFilename(sFilename.c_str());
|
||||
}
|
||||
|
||||
NPP_ASSERT(eFormat != FIF_UNKNOWN);
|
||||
// check that the plugin has reading capabilities ...
|
||||
FIBITMAP *pBitmap;
|
||||
|
||||
if (FreeImage_FIFSupportsReading(eFormat)) {
|
||||
pBitmap = FreeImage_Load(eFormat, sFilename.c_str());
|
||||
}
|
||||
|
||||
NPP_ASSERT(pBitmap != 0);
|
||||
// Dump the bitmap information to the console
|
||||
std::cout << (*pBitmap) << std::endl;
|
||||
// make sure this is an 8-bit single channel image
|
||||
NPP_ASSERT(FreeImage_GetColorType(pBitmap) == FIC_MINISBLACK);
|
||||
NPP_ASSERT(FreeImage_GetBPP(pBitmap) == 8);
|
||||
|
||||
unsigned int nImageWidth = FreeImage_GetWidth(pBitmap);
|
||||
unsigned int nImageHeight = FreeImage_GetHeight(pBitmap);
|
||||
unsigned int nSrcPitch = FreeImage_GetPitch(pBitmap);
|
||||
unsigned char *pSrcData = FreeImage_GetBits(pBitmap);
|
||||
|
||||
int nSrcPitchCUDA;
|
||||
Npp8u *pSrcImageCUDA = nppiMalloc_8u_C1(nImageWidth, nImageHeight, &nSrcPitchCUDA);
|
||||
NPP_ASSERT_NOT_NULL(pSrcImageCUDA);
|
||||
// copy image loaded via FreeImage to into CUDA device memory, i.e.
|
||||
// transfer the image-data up to the GPU's video-memory
|
||||
NPP_CHECK_CUDA(cudaMemcpy2D(
|
||||
pSrcImageCUDA, nSrcPitchCUDA, pSrcData, nSrcPitch, nImageWidth, nImageHeight, cudaMemcpyHostToDevice));
|
||||
|
||||
// define size of the box filter
|
||||
const NppiSize oMaskSize = {7, 7};
|
||||
const NppiPoint oMaskAchnor = {0, 0};
|
||||
// compute maximal result image size
|
||||
const NppiSize oSizeROI = {(int)nImageWidth - (oMaskSize.width - 1),
|
||||
(int)nImageHeight - (oMaskSize.height - 1)};
|
||||
// allocate result image memory
|
||||
int nDstPitchCUDA;
|
||||
Npp8u *pDstImageCUDA = nppiMalloc_8u_C1(oSizeROI.width, oSizeROI.height, &nDstPitchCUDA);
|
||||
NPP_ASSERT_NOT_NULL(pDstImageCUDA);
|
||||
NPP_CHECK_NPP(nppiFilterBox_8u_C1R_Ctx(pSrcImageCUDA,
|
||||
nSrcPitchCUDA,
|
||||
pDstImageCUDA,
|
||||
nDstPitchCUDA,
|
||||
oSizeROI,
|
||||
oMaskSize,
|
||||
oMaskAchnor,
|
||||
nppStreamCtx));
|
||||
// create the result image storage using FreeImage so we can easily
|
||||
// save
|
||||
FIBITMAP *pResultBitmap = FreeImage_Allocate(oSizeROI.width, oSizeROI.height, 8 /* bits per pixel */);
|
||||
NPP_ASSERT_NOT_NULL(pResultBitmap);
|
||||
unsigned int nResultPitch = FreeImage_GetPitch(pResultBitmap);
|
||||
unsigned char *pResultData = FreeImage_GetBits(pResultBitmap);
|
||||
|
||||
NPP_CHECK_CUDA(cudaMemcpy2D(pResultData,
|
||||
nResultPitch,
|
||||
pDstImageCUDA,
|
||||
nDstPitchCUDA,
|
||||
oSizeROI.width,
|
||||
oSizeROI.height,
|
||||
cudaMemcpyDeviceToHost));
|
||||
// now save the result image
|
||||
bool bSuccess;
|
||||
bSuccess = FreeImage_Save(FIF_PGM, pResultBitmap, sResultFilename.c_str(), 0) == TRUE;
|
||||
NPP_ASSERT_MSG(bSuccess, "Failed to save result image.");
|
||||
|
||||
// free nppiImage
|
||||
nppiFree(pSrcImageCUDA);
|
||||
nppiFree(pDstImageCUDA);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
const NppLibraryVersion *libVer = nppGetLibVersion();
|
||||
|
||||
printf("NPP Library Version %d.%d.%d\n", libVer->major, libVer->minor, libVer->build);
|
||||
|
||||
int driverVersion, runtimeVersion;
|
||||
cudaDriverGetVersion(&driverVersion);
|
||||
cudaRuntimeGetVersion(&runtimeVersion);
|
||||
|
||||
printf("CUDA Driver Version: %d.%d\n", driverVersion/1000, (driverVersion%100)/10);
|
||||
printf("CUDA Runtime Version: %d.%d\n\n", runtimeVersion/1000, (runtimeVersion%100)/10);
|
||||
|
||||
cudaError = cudaDeviceGetAttribute(&nppStreamCtx.nCudaDevAttrComputeCapabilityMajor,
|
||||
cudaDevAttrComputeCapabilityMajor,
|
||||
nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess)
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
|
||||
cudaError = cudaDeviceGetAttribute(&nppStreamCtx.nCudaDevAttrComputeCapabilityMinor,
|
||||
cudaDevAttrComputeCapabilityMinor,
|
||||
nppStreamCtx.nCudaDeviceId);
|
||||
if (cudaError != cudaSuccess)
|
||||
return NPP_NOT_SUFFICIENT_COMPUTE_CAPABILITY;
|
||||
|
||||
cudaError = cudaStreamGetFlags(nppStreamCtx.hStream, &nppStreamCtx.nStreamFlags);
|
||||
|
||||
cudaDeviceProp oDeviceProperties;
|
||||
|
||||
cudaError = cudaGetDeviceProperties(&oDeviceProperties, nppStreamCtx.nCudaDeviceId);
|
||||
|
||||
nppStreamCtx.nMultiProcessorCount = oDeviceProperties.multiProcessorCount;
|
||||
nppStreamCtx.nMaxThreadsPerMultiProcessor = oDeviceProperties.maxThreadsPerMultiProcessor;
|
||||
nppStreamCtx.nMaxThreadsPerBlock = oDeviceProperties.maxThreadsPerBlock;
|
||||
nppStreamCtx.nSharedMemPerBlock = oDeviceProperties.sharedMemPerBlock;
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "input")) {
|
||||
getCmdLineArgumentString(argc, (const char **)argv, "input", &filePath);
|
||||
} else {
|
||||
filePath = sdkFindFilePath("teapot512.pgm", argv[0]);
|
||||
catch (npp::Exception &rException) {
|
||||
std::cerr << "Program error! The following exception occurred: \n";
|
||||
std::cerr << rException << std::endl;
|
||||
std::cerr << "Aborting." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (filePath) {
|
||||
sFilename = filePath;
|
||||
} else {
|
||||
sFilename = "teapot512.pgm";
|
||||
catch (...) {
|
||||
std::cerr << "Program error! An unknow type of exception occurred. \n";
|
||||
std::cerr << "Aborting." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
// if we specify the filename at the command line, then we only test
|
||||
// sFilename otherwise we will check both sFilename[0,1]
|
||||
int file_errors = 0;
|
||||
std::ifstream infile(sFilename.data(), std::ifstream::in);
|
||||
|
||||
if (infile.good()) {
|
||||
std::cout << "freeImageInteropNPP opened: <" << sFilename.data()
|
||||
<< "> successfully!" << std::endl;
|
||||
file_errors = 0;
|
||||
infile.close();
|
||||
} else {
|
||||
std::cout << "freeImageInteropNPP unable to open: <" << sFilename.data()
|
||||
<< ">" << std::endl;
|
||||
file_errors++;
|
||||
infile.close();
|
||||
}
|
||||
|
||||
if (file_errors > 0) {
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
std::string sResultFilename = sFilename;
|
||||
|
||||
std::string::size_type dot = sResultFilename.rfind('.');
|
||||
|
||||
if (dot != std::string::npos) {
|
||||
sResultFilename = sResultFilename.substr(0, dot);
|
||||
}
|
||||
|
||||
sResultFilename += "_boxFilterFII.pgm";
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "output")) {
|
||||
char *outputFilePath;
|
||||
getCmdLineArgumentString(argc, (const char **)argv, "output",
|
||||
&outputFilePath);
|
||||
sResultFilename = outputFilePath;
|
||||
}
|
||||
|
||||
FREE_IMAGE_FORMAT eFormat = FreeImage_GetFileType(sFilename.c_str());
|
||||
|
||||
// no signature? try to guess the file format from the file extension
|
||||
if (eFormat == FIF_UNKNOWN) {
|
||||
eFormat = FreeImage_GetFIFFromFilename(sFilename.c_str());
|
||||
}
|
||||
|
||||
NPP_ASSERT(eFormat != FIF_UNKNOWN);
|
||||
// check that the plugin has reading capabilities ...
|
||||
FIBITMAP *pBitmap;
|
||||
|
||||
if (FreeImage_FIFSupportsReading(eFormat)) {
|
||||
pBitmap = FreeImage_Load(eFormat, sFilename.c_str());
|
||||
}
|
||||
|
||||
NPP_ASSERT(pBitmap != 0);
|
||||
// Dump the bitmap information to the console
|
||||
std::cout << (*pBitmap) << std::endl;
|
||||
// make sure this is an 8-bit single channel image
|
||||
NPP_ASSERT(FreeImage_GetColorType(pBitmap) == FIC_MINISBLACK);
|
||||
NPP_ASSERT(FreeImage_GetBPP(pBitmap) == 8);
|
||||
|
||||
unsigned int nImageWidth = FreeImage_GetWidth(pBitmap);
|
||||
unsigned int nImageHeight = FreeImage_GetHeight(pBitmap);
|
||||
unsigned int nSrcPitch = FreeImage_GetPitch(pBitmap);
|
||||
unsigned char *pSrcData = FreeImage_GetBits(pBitmap);
|
||||
|
||||
int nSrcPitchCUDA;
|
||||
Npp8u *pSrcImageCUDA =
|
||||
nppiMalloc_8u_C1(nImageWidth, nImageHeight, &nSrcPitchCUDA);
|
||||
NPP_ASSERT_NOT_NULL(pSrcImageCUDA);
|
||||
// copy image loaded via FreeImage to into CUDA device memory, i.e.
|
||||
// transfer the image-data up to the GPU's video-memory
|
||||
NPP_CHECK_CUDA(cudaMemcpy2D(pSrcImageCUDA, nSrcPitchCUDA, pSrcData,
|
||||
nSrcPitch, nImageWidth, nImageHeight,
|
||||
cudaMemcpyHostToDevice));
|
||||
|
||||
// define size of the box filter
|
||||
const NppiSize oMaskSize = {7, 7};
|
||||
const NppiPoint oMaskAchnor = {0, 0};
|
||||
// compute maximal result image size
|
||||
const NppiSize oSizeROI = {(int)nImageWidth - (oMaskSize.width - 1),
|
||||
(int)nImageHeight - (oMaskSize.height - 1)};
|
||||
// allocate result image memory
|
||||
int nDstPitchCUDA;
|
||||
Npp8u *pDstImageCUDA =
|
||||
nppiMalloc_8u_C1(oSizeROI.width, oSizeROI.height, &nDstPitchCUDA);
|
||||
NPP_ASSERT_NOT_NULL(pDstImageCUDA);
|
||||
NPP_CHECK_NPP(nppiFilterBox_8u_C1R_Ctx(pSrcImageCUDA, nSrcPitchCUDA,
|
||||
pDstImageCUDA, nDstPitchCUDA, oSizeROI,
|
||||
oMaskSize, oMaskAchnor, nppStreamCtx));
|
||||
// create the result image storage using FreeImage so we can easily
|
||||
// save
|
||||
FIBITMAP *pResultBitmap = FreeImage_Allocate(
|
||||
oSizeROI.width, oSizeROI.height, 8 /* bits per pixel */);
|
||||
NPP_ASSERT_NOT_NULL(pResultBitmap);
|
||||
unsigned int nResultPitch = FreeImage_GetPitch(pResultBitmap);
|
||||
unsigned char *pResultData = FreeImage_GetBits(pResultBitmap);
|
||||
|
||||
NPP_CHECK_CUDA(cudaMemcpy2D(pResultData, nResultPitch, pDstImageCUDA,
|
||||
nDstPitchCUDA, oSizeROI.width, oSizeROI.height,
|
||||
cudaMemcpyDeviceToHost));
|
||||
// now save the result image
|
||||
bool bSuccess;
|
||||
bSuccess = FreeImage_Save(FIF_PGM, pResultBitmap, sResultFilename.c_str(),
|
||||
0) == TRUE;
|
||||
NPP_ASSERT_MSG(bSuccess, "Failed to save result image.");
|
||||
|
||||
// free nppiImage
|
||||
nppiFree(pSrcImageCUDA);
|
||||
nppiFree(pDstImageCUDA);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
} catch (npp::Exception &rException) {
|
||||
std::cerr << "Program error! The following exception occurred: \n";
|
||||
std::cerr << rException << std::endl;
|
||||
std::cerr << "Aborting." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
} catch (...) {
|
||||
std::cerr << "Program error! An unknow type of exception occurred. \n";
|
||||
std::cerr << "Aborting." << std::endl;
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user