#include <iostream>

using namespace std;

extern void test_cuda(float* input, int width, int height, int iterations);

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

  // Parse Command Line and extract arguments
  if (argc < 2)
  {
    cout << endl << "Usage: test ITERATIONS" << endl;
    exit( 0 );
  }

  // load parameters
  int max_iter =      atoi(argv[1]);
  int width = 512;
  int height = 512;

  // allocate arrays for input and output
  float* input = new float[width*height];

  // Copy data to float image
  for (int y=0; y<height; y++) {
    for (int x=0; x<width; x++) {
      if ((x+y)%2 == 0)
        input[y*width+x] = 1.0f;
      else
        input[y*width+x] = 0.0f;
    }
  }

  test_cuda(input, width, height, max_iter);

  delete [] input;
}