#ifdef WIN32
#include <windows.h>
#include <winbase.h>
#endif

#include <stdio.h>
#include <stdlib.h>

#include "thing.h"

  /* various parameters: */
#define NSET 1       /* Number of datasets per node */
#define NDATA 1000   /* size of interior, in words */
#define NBDRY 100    /* size of each boundary, in words */
#define MAXNODES 32

static int Myid, Nodes;

static volatile int called= 0;

void callme(thing_t *status, void *p) {
  called= status->nbytes_xferred;
  if ( !called ) called= -1;
}

void serror( char *string, char *name ){
  fprintf(stderr,"%s: '%s'\n", string, name );
  exit(1);
}

void ierror( char *string, int i ) {
	fprintf(stderr,"%s: %d\n", string, i );
	exit(1);
}

void shuffle( int *ia, int n ) {
  int i,j,k,m;
  srand( 123 );
  if ( n==1 ) return;
  for (i=0; i<10; ++i ) 
	  for (j=1; j<=n/2; ++j ) {
		  /* find random other node to interchange with */
		  do k= (rand()%n)+1; while ( k == j );
		  m=ia[j]; ia[j]=ia[k]; ia[k]= m;
	  }
}

/*----------------------------------------------------------------
 * task manager
 */
void tm() {
  int i, ierr, icm;
  int task,  ndone, icycle;
  int nworkers;
  int worklist[MAXNODES], workcycle[MAXNODES];

    fprintf(stderr,"==== starting TM nodes=%d\n", Nodes);

    nworkers= Nodes-1;

	for (i=1; i<Nodes; ++i ) {
		worklist[i]= i;
		workcycle[i]= 0;
	}

	/* The test will be to set up a series of work cycles, where cm's
	* are instructed to read/check/write a context.  */

	for ( icycle=1; icycle<=10; ++icycle ) {
		/* scramble worklist */
		shuffle(worklist, nworkers);
		for ( i=1; i<=nworkers; ++i ) {
			task= worklist[i];
			if ( ierr=assignWork_thing( i, &task, sizeof(task)) ) serror("assignWork","!");
		}
		ndone= 0;
		do {
			icm= waitForWorkDone_thing( &task, sizeof(task), -1 );
			if ( icm == -1 ) serror("waitForWorkDone","!");
			/* if task is nonzero, it did a RIO */
			if ( task ) fprintf(stderr,"Cm%d ",icm);
			else fprintf(stderr,"cm%d ",icm);
		} while( ++ndone < nworkers );
		fprintf(stderr,"  cycle complete\n");
	}

	/* shut them down */
  task= 666;
  for ( i=1; i<=nworkers; ++i ) assignWork_thing( i, &task, sizeof(task) );
}

/*----------------------------------------------------------------
* Check data for consistency - should start at home node number,
* increment.  Returns -1 if no error, else the index at which a problem
* was found.
*/

int verify( int *data, int node ) {
	int i,k=node;
	for(i=0; i<NDATA; ++i ) if ( data[i] != k++ ) return i;
	return -1;
}

/*----------------------------------------------------------------
 * computer
 */
void cm(int iarg) {

  /* First, each cm constructs its local test data. 
   * Each node creates an 'interior' dataset named dNN.M (NN node number,
   * M dataset 1..9) and two boundaries named blNN.M and brNN.M
   *
   * The datasets are filled in with the owner node number + offset,
   */

  int set, i;
  char name[256];
  int data[NDATA], bdry[NBDRY];
  int task;

  thing_t status1, status2, status3;
  int ierr;

  fprintf(stderr,"=== starting CM %d arg=%d\n", Myid, iarg );

  if ( iarg <=0 ) 
  for (set=1; set<=NSET; ++set ) {
	sprintf(name,"d%02d.%1d", Myid, set );
	if ( create_thing( name, DISK_THING|REPLACE_THING) ) serror("can't create ",name);

	sprintf(name,"bd%c%02d.%1d", 'l',Myid, set );
	if ( create_thing(name, DISK_THING|REPLACE_THING) ) serror("can't create",name);

	sprintf(name,"bd%c%02d.%1d", 'r',Myid, set );
	if ( create_thing(name, DISK_THING|REPLACE_THING) ) serror("can't create",name);  
  }

  synchAllCreated_thing();

  for (i=0; i<NDATA; ++i ) data[i]= Myid+i;
  for (i=0; i<NBDRY; ++i ) bdry[i]= Myid+i;

  /* Write each set - the only difference is that word #set is negated. */
	if ( iarg <=1 )
  for (set=1; set<=NSET; ++set ) {

	sprintf(name,"d%02d.%1d", Myid, set );
	if ( ierr=write_thing( name,0, data, NDATA*sizeof(int), &status1 ) == -1 )
	  ierror("write_thing returned",ierr);

	sprintf(name,"bd%c%02d.%1d", 'l',Myid, set );
	if ( ierr=write_thing( name,0, bdry, NBDRY*sizeof(int), &status2 ) == -1 )
	  ierror("write_thing returned",ierr);

	sprintf(name,"bd%c%02d.%1d", 'r',Myid, set );
	if ( ierr=write_thing( name,0, bdry, NBDRY*sizeof(int), &status3 ) == -1 )
	  ierror("write_thing returned",ierr);

	if ( ierr= wait_thing( &status1, TILDONE_THING ) == -1 )
	  ierror("wait_thing returned ",ierr );
	if ( ierr= wait_thing( &status2, TILDONE_THING ) == -1 )
	  ierror("wait_thing returned ",ierr );
	if ( ierr= wait_thing( &status3, TILDONE_THING ) == -1 )
	  ierror("wait_thing returned ",ierr );

  }

  /* Now work is an integer; do a read of the specified node
   * data, do something with it and write it back
   */

  do {

	ierr= getWork_thing( -1, &task, sizeof(task) );
	if ( ierr ) ierror("getWork_thing", ierr );
	
	if ( task == 666 ) break;

	/* read the specified set and print its first 10 words */
	sprintf(name,"d%02d.%1d", task, 1 );

	if ( ierr=read_thing( name,0,data,NDATA*sizeof(int), NULL ) == -1 )
		ierror("read_thing",ierr);

    if ( ierr=	verify( data, task ) >=0 ) {
		i=ierr-5; if (i<0) i=0;
		fprintf(stderr,"ERROR found by cm%d at %d: ",Myid,ierr);
		while (i<ierr+5) { fprintf(stderr,"%d=%d ", i,data[i] ); ++i; }
		fprintf(stderr,"\n");
	}

	if ( ierr=write_thing( name,0,data,NDATA*sizeof(int), NULL ) == -1 )
		ierror("write_thing",ierr);

	/* else
	fprintf(stderr,"CM %d) %d %d %d %d %d %d %d %d %d\n", Myid,
		data[0],data[1],data[2],data[3],data[4],data[5],data[6],data[7],data[8] );
		*/

	/* return flag for RIO use */
	if ( task != Myid ) task= 1;
	else task= 0;
	workDone_thing( 0, &task, sizeof(task) );

  } while(1);

	fprintf(stderr,"***CM %d given work assignment %d\n", Myid, task );
	return;
}

/*================================================================
* simple numeric argument is phase to start CMs in.  0 create, 1 init, 2 process
* optional 2nd argument is 'thing home'
 */

void main( int argc, char **argv ) {
	int i;
    char home[80];

	/*fprintf(stderr,"sizeof thing_t is %d \n", sizeof(thing_t) ); */
	
	if ( argc == 3 ) { sprintf(home,"%s\\",argv[2]); strcat(home,"%1d"); }
	else strcpy( home, "D:\\disk\\%1d" );

	Myid =start_thing( home, 0, &argc, &argv );
	if ( Myid == -1 ) { serror("Could not start things @",home); exit(1); }

	Nodes= getNodeCount_thing();

	/* if given an argument, pass to cm */
	if ( argc >= 2 ) i= atoi(argv[1]);
	else i=0;

	if ( Myid == 0 ) tm();
	else cm(i);

	stop_thing();
	exit(0);
}