// (c) EB Nov 2009

#include "Config.h"

#include "MemoryTasks.h"
#include "ArithmeticTasks.h"

#if CONFIG_USE_QT
#include <QtGui/QApplication>
#include <QtGui/QMainWindow>
#include <QtGui/QTextEdit>
#include <QtGui/QClipboard>
#endif

#include <string.h>
#include <iostream>

struct TestResult
{
  std::string title;
  std::vector<int> mbps; // MB/s by logSize
  void update(int ls,int x) { if (x > mbps[ls]) mbps[ls] = x; } // keep max
};

void print(FILE * f,const TestResult & r,int minLogSize,int maxLogSize)
{
  fprintf(f,"%-25s",r.title.c_str());
  for (int i=minLogSize;i<=maxLogSize;i++)
    {
      int x = r.mbps[i];
      fprintf(f,"%8d",x);
    }
  fprintf(f,"\n");
}

void print(Logger * log,const TestResult & r,int minLogSize,int maxLogSize)
{
  char aux[200];
  std::string s;
  _snprintf(aux,200,"%-25s",r.title.c_str()); s.append(aux);
  for (int i=minLogSize;i<=maxLogSize;i++)
    {
      int x = r.mbps[i];
      _snprintf(aux,200,"%8d",x); s.append(aux);
    }
  log->append(s);
}

void help(void)
{
  std::cout<<"Usage: tocl {-cpu | -gpu} [-copy|-add|-zero|-mul]\n";
}

int main(int argc,char ** argv)
{
#if CONFIG_USE_QT
  // Qt windows
  QApplication app(argc,argv);
  QMainWindow * w = new QMainWindow();
  QTextEdit * te = new QTextEdit();
  te->setReadOnly(true);
  w->setCentralWidget(te);
  w->setMinimumSize(QSize(800,600));
  w->show();
  app.processEvents();
  Logger * log = new QtLogger(te);
#else
  // Command line
  Logger * log = new Logger();
#endif

  int do_cpu = 0;
  int do_gpu = 0;
  int do_all = 0;
  int do_copy = 0;
  int do_add = 0;
  int do_zero = 0;
  int do_mul = 0;

  if (argc < 2) {
    help();
    return -1;
  }

  if (!strcmp(argv[1], "-cpu"))
    do_cpu = 1;

  if (!strcmp(argv[1], "-gpu"))
    do_gpu = 1;

  if (do_cpu == do_gpu) {
    help();
    return -1;
  }

  if (argc == 2)
    do_all = 1;
  else {
    int i = 2;
    while (argv[i]) {
      if (!strcmp(argv[i], "-copy"))
	do_copy = 1;
      if (!strcmp(argv[i], "-add"))
	  do_add = 1;
      if (!strcmp(argv[i], "-zero"))
	do_zero = 1;
      if (!strcmp(argv[i], "-mul"))
	do_mul = 1;
      i++;
    }
  }

  if (do_gpu) {
    
    // Display device info
    cl::Context * c = cl::Context::create();
    if (c != 0)
      {
	std::string s;
	if (c->getAllDeviceInfo(0,s)) log->append(s.c_str());
	delete c;
	c = 0;
      }
  }

  int minLogSize = 10;
  int maxLogSize = 29;

  std::vector<TestResult> allTests;

  // Run all CPU tests

  if (do_cpu) {
  
    for (int task=0;task<100;task++)
      {
	// Select tests here
	// if (task != 30) continue;
	// continue;

	CPUTask * g = 0;
	// Default values
	std::string title;
	int minNThreads = 1;
	int maxNThreads = 32;

	switch (task)
	  {
	  case 0:
	    if (do_all || do_copy) {
	      title.assign("CPU copyN");
	      g = new CopyCPUTask();
	    }
	    break;
	  case 1:
	    if (do_all || do_zero) {
	      title.assign("CPU zeroN");
	      g = new ZeroCPUTask();
	    }
	    break;

	  case 10:
	    if (do_all || do_add) {
	      title.assign("CPU addN v1 16-bit");
	      g = new AddNCPUTask(ADDN_V1,16);
	    }
	    break;
	  case 11:
	    if (do_all || do_add) {
	      title.assign("CPU addN v1 32-bit");
	      g = new AddNCPUTask(ADDN_V1,32);
	    }
	    break;
	  case 12:
	    if (do_all || do_add) {
	      title.assign("CPU addN v1 64-bit");
	      g = new AddNCPUTask(ADDN_V1,64);
	    }
	    break;

	  case 20:
	    if (do_all || do_add) {
	      title.assign("CPU addN v2 16-bit");
	      g = new AddNCPUTask(ADDN_V2,16);
	    }
	    break;
	  case 21:
	    if (do_all || do_add) {
	      title.assign("CPU addN v2 32-bit");
	      g = new AddNCPUTask(ADDN_V2,32);
	    }
	    break;
	  case 22:
	    if (do_all || do_add) {
	      title.assign("CPU addN v2 64-bit");
	      g = new AddNCPUTask(ADDN_V2,64);
	    }
	    break;

#ifndef Linux
	  case 30:
	    if (do_all || do_add) {
	      title.assign("CPU addN v3 32-bit");
	      g = new AddNCPUTask(ADDN_V3,32);
	    }
	    break;
#endif
	  }
	if (g == 0) continue; // nothing to do

	TestResult tr;
	tr.title.assign(title);
	tr.mbps.resize(1+maxLogSize,0);
	log->appendTitle(title.c_str());
	for (int nt=minNThreads;nt<=maxNThreads;nt<<=1)
	  for (int ls=minLogSize;ls<=maxLogSize;ls++)
	    {
	      size_t sz = 1<<ls;
	      double mbps = g->run(nt,sz);
	      char aux[200];
	      std::string usz;
	      getUserSize(ls,usz);
	      _snprintf(aux,200,"NTHREAD=%3d  SZ=%7s   %6.0f MB/s",nt,usz.c_str(),mbps);
	      log->append(aux);
	      tr.update(ls,mbps);
	    }

	print(stdout,tr,minLogSize,maxLogSize);
#if CONFIG_USE_QT
	print(log,tr,minLogSize,maxLogSize);
#endif
	allTests.push_back(tr);
	delete g;
      }

  }

  if (do_gpu) {

    // Run all GPU tests
    for (int task=0;task<100;task++)
      {
	// Select tests here
	//continue;
	//if (task != 30) continue;

	GPUTask * g = 0;
	// Default values
	std::string title;
	int minWorkgroupSize = 16;
	int maxWorkgroupSize = 512;

	switch (task)
	  {
	  case 0:
	    if (do_all || do_copy) {
	      title.assign("Host to device copy");
	      g = new CopyGPUTask(CopyGPUTask::HOST_TO_DEVICE_COPY,log);
	      minWorkgroupSize = maxWorkgroupSize;
	    }
	    break;
	  case 1:
	    if (do_all || do_copy) {
	      title.assign("Device to host copy");
	      g = new CopyGPUTask(CopyGPUTask::DEVICE_TO_HOST_COPY,log);
	      minWorkgroupSize = maxWorkgroupSize;
	    }
	    break;
	  case 2:
	    if (do_all || do_copy) {
	      title.assign("copyN");
	      g = new CopyGPUTask(CopyGPUTask::DEVICE_TO_DEVICE_COPY,log);
	      minWorkgroupSize = maxWorkgroupSize;
	    }
	    break;
	  case 3:
	    if (do_all || do_zero) {
	      title.assign("zeroN 32-bit");
	      g = new ZeroGPUTask(32,log);
	    }
	    break;
	  case 4:
	    if (do_all || do_zero) {
	      title.assign("zeroN 64-bit");
	      g = new ZeroGPUTask(64,log);
	    }
	    break;
	  case 10:
	    if (do_all || do_add) {
	      title.assign("AddN v1 16-bit");
	      g = new AddNGPUTask(ADDN_V1,16,log);
	    }
	    break;
	  case 11:
	    if (do_all || do_add) {
	      title.assign("AddN v1 32-bit");
	      g = new AddNGPUTask(ADDN_V1,32,log);
	    }
	    break;
	  case 12:
	    if (do_all || do_add) {
	      title.assign("AddN v1 64-bit");
	      g = new AddNGPUTask(ADDN_V1,64,log);
	    }
	    break;

	  case 20:
	    if (do_all || do_add) {
	      title.assign("AddN v2 16-bit");
	      g = new AddNGPUTask(ADDN_V2,16,log);
	    }
	    break;
	  case 21:
	    if (do_all || do_add) {
	      title.assign("AddN v2 32-bit");
	      g = new AddNGPUTask(ADDN_V2,32,log);
	    }
	    break;
	  case 22:
	    if (do_all || do_add) {
	      title.assign("AddN v2 64-bit");
	      g = new AddNGPUTask(ADDN_V2,64,log);
	    }
	    break;

	  case 40:
	    if (do_all || do_mul) {
	      title.assign("Mul1 v1 32-bit");
	      g = new Mul1GPUTask(MUL1_V1,1,log);
	    }
	    break;
	  }
	if (g == 0) continue; // nothing to do

	TestResult tr;
	tr.title.assign(title);
	tr.mbps.resize(1+maxLogSize,0);
	log->appendTitle(title.c_str());
	for (int wg=minWorkgroupSize;wg<=maxWorkgroupSize;wg<<=1)
	  {
	    char aux[1000];
	    _snprintf(aux,1000,"%s - workgroup_size=%d",title.c_str(),wg);
	    log->append(aux);
	    for (int ls=minLogSize;ls<=maxLogSize;ls++)
	      {
		size_t sz = 1<<ls;
		double mbps = g->run(wg,sz);
		std::string usz;
		getUserSize(ls,usz);
		_snprintf(aux,1000,"WG=%3d  SZ=%7s   %6.0f MB/s",wg,usz.c_str(),mbps);
		log->append(aux);
		tr.update(ls,mbps);
	      }
	  }
      
	print(stdout,tr,minLogSize,maxLogSize);
#if CONFIG_USE_QT
	print(log,tr,minLogSize,maxLogSize);
#endif
	allTests.push_back(tr);
	delete g;
      }

  }

  // Dump all results
  int nt = (int)allTests.size();
  log->appendTitle("RESULTS");
  printf("\n----\n");
  for (int i=0;i<nt;i++)
    {
      TestResult & tr = allTests[i];
      print(stdout,tr,minLogSize,maxLogSize);
#if CONFIG_USE_QT
      print(log,tr,minLogSize,maxLogSize);
#endif
    }

#if CONFIG_USE_QT
  // Enter event loop to terminate
  log->appendTitle("DONE");
  app.exec();
#endif
}