| /** |
| Pankaj Kukreja |
| github.com/proton0001 |
| Indian Institute of Technology Hyderabad |
| */ |
| |
| #include "ImageHelper.h" |
| #include "diffusion.h" |
| #include <cstdlib> |
| #include <iostream> // std::cerr |
| |
| #define BENCHMARK_LIB |
| #ifdef BENCHMARK_LIB |
| #include "benchmark/benchmark.h" |
| #endif |
| |
| int *inputImage; |
| extern "C" void anisotropicDiffusionKernel(int height, int width, int *source, |
| int *destination, int num_iteration); |
| int main(int argc, char *argv[]) { |
| |
| #ifdef BENCHMARK_LIB |
| ::benchmark::Initialize(&argc, argv); |
| #endif |
| |
| const char *anisotropicDiffusionOutputFileName = |
| (const char *)"./anisotropicDiffusionOutput.txt"; |
| |
| inputImage = (int *)malloc(sizeof(int) * (HEIGHT) * (WIDTH)); |
| |
| if (inputImage == NULL) { |
| std::cerr << "Insufficient memory\n"; |
| exit(EXIT_FAILURE); |
| } |
| |
| initializeRandomImage(inputImage, HEIGHT, WIDTH); |
| |
| #ifdef BENCHMARK_LIB |
| ::benchmark::RunSpecifiedBenchmarks(); |
| #endif |
| |
| int *outputImage = (int *)malloc(sizeof(int) * (HEIGHT) * (WIDTH)); |
| if (outputImage == NULL) { |
| std::cerr << "Insufficient memory\n"; |
| exit(EXIT_FAILURE); |
| } |
| |
| anisotropicDiffusionKernel(HEIGHT, WIDTH, inputImage, outputImage, ITERATION); |
| |
| for (int j = 0; j < WIDTH; j++) { |
| outputImage[0 + j] = 0; |
| outputImage[1 * WIDTH + j] = 0; |
| outputImage[(HEIGHT - 1) * WIDTH + j] = 0; |
| outputImage[(HEIGHT - 2) * WIDTH + j] = 0; |
| } |
| |
| for (int i = 0; i < HEIGHT; i++) { |
| outputImage[i * WIDTH + 0] = 0; |
| outputImage[i * WIDTH + 1] = 0; |
| outputImage[i * WIDTH + (WIDTH - 1)] = 0; |
| outputImage[i * WIDTH + (WIDTH - 2)] = 0; |
| } |
| |
| saveImage(outputImage, anisotropicDiffusionOutputFileName, HEIGHT, WIDTH); |
| |
| free(outputImage); |
| free(inputImage); |
| |
| return EXIT_SUCCESS; |
| } |
| |
| #ifdef BENCHMARK_LIB |
| void BENCHMARK_ANISTROPIC_DIFFUSION(benchmark::State &state) { |
| |
| int height = state.range(0); |
| int width = state.range(0); |
| |
| int *outputImage = (int *)malloc(sizeof(int) * (height) * (width)); |
| if (outputImage == NULL) { |
| std::cerr << "Insufficient memory\n"; |
| exit(EXIT_FAILURE); |
| } |
| /* first call made to warm up cache */ |
| anisotropicDiffusionKernel(height, width, inputImage, outputImage, ITERATION); |
| |
| while (state.KeepRunning()) { |
| anisotropicDiffusionKernel(height, width, inputImage, outputImage, |
| ITERATION); |
| } |
| /* Since we are not passing state.range as 20 this if case will always be |
| * false. |
| * This call is to make compiler think that outputImage may be used |
| * later so that above kernel calls will not get optimize out */ |
| if (state.range(0) == 20) { |
| saveImage(outputImage, (const char *)"testFailed.txt", height, width); |
| } |
| |
| free(outputImage); |
| } |
| |
| BENCHMARK(BENCHMARK_ANISTROPIC_DIFFUSION) |
| ->Arg(32) |
| ->Arg(64) |
| ->Arg(128) |
| ->Arg(256) |
| ->Unit(benchmark::kMicrosecond); |
| #endif |