/**
  Pankaj Kukreja
  github.com/proton0001
  Indian Institute of Technology Hyderabad
*/

#include "ImageHelper.h"
#include "bilateralFiltering.h"
#include <iostream> // std::cerr
#include <cstdlib>

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

int *inputImage;
extern "C" void bilateralFilterKernel(int height, int width, int *inputImage,
                                      int *outputImage, int diameter);

int main(int argc, char *argv[]) {

#ifdef BENCHMARK_LIB
  ::benchmark::Initialize(&argc, argv);
#endif

  const char *bilateralFilterOutputFileName =
      (const char *)"./bilateralFilterOutput.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);
  }
  for (int i = 0; i < HEIGHT; i++) {
    for (int j = 0; j < WIDTH; j++) {
      outputImage[i * WIDTH + j] = 0;
    }
  }

  bilateralFilterKernel(HEIGHT, WIDTH, inputImage, outputImage, DIAMETER);

  saveImage(outputImage, bilateralFilterOutputFileName, HEIGHT, WIDTH);

  free(outputImage);
  free(inputImage);
  return (EXIT_SUCCESS);
}

#ifdef BENCHMARK_LIB
void BENCHMARK_BILATERAL_FILTER(benchmark::State &state) {

  int height = state.range(0);
  int width = state.range(0);
  int diameter = state.range(1);

  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 the cache*/
  bilateralFilterKernel(height, width, inputImage, outputImage, diameter);
  while (state.KeepRunning()) {
    bilateralFilterKernel(height, width, inputImage, outputImage, diameter);
  }

  /* Since we are not passing state.range as 20 this if case will always be
   * false. This check is to prevent above kernel calls from getting
   * optimized out
   */
  if (state.range(0) == 20) {
    saveImage(outputImage, (const char *)"testFailed.txt", height, width);
  }

  free(outputImage);
}

#if (HEIGHT < WIDTH)
#define MINIMUM_DIM HEIGHT
#else
#define MINIMUM_DIM WIDTH
#endif
static void CustomArguments(benchmark::internal::Benchmark *b) {
  int limit = MINIMUM_DIM;
  int start = 16;
  if (start > limit) {
    start = 1;
  }
  for (int i = start; i < limit; i <<= 1) {
    b->ArgPair(i, 2);
    b->ArgPair(i, 4);
  }
}
BENCHMARK(BENCHMARK_BILATERAL_FILTER)
    ->Apply(CustomArguments)
    ->Unit(benchmark::kMicrosecond);
#endif