#include <iostream>
#include <math.h>

#include <ff/farm.hpp>
#include <ff/node.hpp>

using namespace std;
using namespace ff; 


// should be global to be accessible from workers 
#define MAX 1024
double x[MAX];
double y[MAX];
  
#define ITERNO 100000
int iterno = ITERNO; 

class Worker: public ff_node {
  int svc_init() {
    cout << "Worker initialized" << endl; 
    return 0; 
  }

  void * svc(void * in) {
    int i = * ((int *) in);
    // cout << "Computing " << i << endl;
    for(int j=0; j<iterno; j++) 
       x[i] = sin(x[i]);
    y[i] += x[i];
    return in; 
  }
};


#define NW 2 

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

  cout << "init " << argc  << endl; 
  int nw = (argc==1 ? NW : atoi(argv[1]));
  iterno = (argc<=2 ? ITERNO : atoi(argv[2]));

  cout << "using " << nw << " workers iterating " << iterno << " times "<< endl; 

  // init input (fake)
  for(int i=0; i<MAX; i++) {
    x[i] = (double)(i*10);
    y[i] = (double)(-i*13);
  }
  cout << "Setting up farm" << endl; 
  ff_farm<> farm(true,nw);    // true for offloading, NW for the number of workers ... 

  std::vector<ff_node *> w;   // prepare workers
  for(int i=0;i<nw;++i) 
    w.push_back(new Worker);
  farm.add_workers(w);        // add them to the farm
  farm.run_then_freeze();     // run farm asynchronously
  
  cout << "Sending tasks ..." << endl; 
  int tasks[MAX]; 
  for(int i=0; i<MAX; i++) {
     tasks[i]=i;
     farm.offload((void *) &tasks[i]);
  }
  farm.offload((void *) FF_EOS);

  cout << "Waiting termination" << endl;
  farm.wait();

  cout << "Farm terminated after computing for " << farm.ffTime() << endl; 

  ffTime(STOP_TIME);
  cout << "Spent overall " << ffTime(GET_TIME) << endl;

}