/**
  Pankaj Kukreja
  github.com/proton0001
  Indian Institute of Technology Hyderabad
*/
#include "ImageHelper.h"
#include "dilate.h"
#include <iostream>
#include <cstdlib>

#define BENCHMARK_LIB
#ifdef BENCHMARK_LIB
#include "benchmark/benchmark.h"
#endif

int *inputImage;
extern "C" void dilateKernel(int height, int width, int *inputImage,
                             int *outputImage, int *temporary);

int main(int argc, char *argv[]) {
#ifdef BENCHMARK_LIB
  ::benchmark::Initialize(&argc, argv);
#endif

  const char *dilateOutputFileName = (const char *)"./dilateOutput.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);
  int *temp = (int *)malloc(sizeof(int) * HEIGHT * WIDTH);

  if (outputImage == NULL || temp == NULL) {
    std::cerr << "Insufficient memory\n";
    exit(EXIT_FAILURE);
  }

  dilateKernel(HEIGHT, WIDTH, inputImage, outputImage, temp);

  for (int j = 0; j < WIDTH; j++) {
    outputImage[0 * WIDTH + j] = 0;
    outputImage[(HEIGHT - 1) * WIDTH + j] = 0;
  }

  for (int i = 0; i < HEIGHT; i++) {
    outputImage[i * WIDTH + 0] = 0;
    outputImage[i * WIDTH + (WIDTH - 1)] = 0;
  }

  saveImage(outputImage, dilateOutputFileName, HEIGHT, WIDTH);
  free(temp);
  free(outputImage);
  free(inputImage);
  return (EXIT_SUCCESS);
}

#ifdef BENCHMARK_LIB
void BENCHMARK_DILATE(benchmark::State &state) {
  /* taking height = width */
  int height = state.range(0);
  int width = state.range(0);

  int *outputImage = (int *)malloc(sizeof(int) * height * width);
  int *temp = (int *)malloc(sizeof(int) * height * width);
  if (outputImage == NULL) {
    std::cerr << "Insufficient memory\n";
    exit(EXIT_FAILURE);
  }

  /* This call is to warm up the cache */
  dilateKernel(height, width, inputImage, outputImage, temp);

  while (state.KeepRunning()) {
    dilateKernel(height, width, inputImage, outputImage, temp);
  }

  /* 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 optimize out */
  if (state.range(0) == 20) {
    saveImage(outputImage, (const char *)"testFailed.txt", height, width);
  }
  free(temp);
  free(outputImage);
}

#define MINIMUM_DIM (HEIGHT > WIDTH) ? WIDTH : HEIGHT
#if MINIMUM_DIM > 512
BENCHMARK(BENCHMARK_DILATE)
    ->RangeMultiplier(2)
    ->Range(128, MINIMUM_DIM)
    ->Unit(benchmark::kMicrosecond);
#else
BENCHMARK(BENCHMARK_DILATE)
    ->RangeMultiplier(2)
    ->Range(1, MINIMUM_DIM)
    ->Unit(benchmark::kMicrosecond);
#endif
#endif