// test harness
#include "TestManager.h"

#include "HepTuple/HepHBookNtuple.h"
#include "HepTuple/HepNtuple.h"
#include "HepTuple/Column.h"
#include "HepTuple/Block.h"
ZM_USING_NAMESPACE( zmht )

#include "ZMutility/iostream"

Int4 c;

Int4 func () {
  return c;
}

void readFunc (Int4 i) {
  c = i;
}

class Test {
  int _a;
public:
  Int4 a() {return _a;};
  void seta(Int4 f) { _a = f; };
};

#include <cmath>
#ifndef DBL_EPSILON
#define    DBL_EPSILON     2.2204460492503131E-16
#endif

#define IS_DBL_EQUAL(a,b) ( a == b \
			    || ( ::fabs( (a) - (b) ) < 100*DBL_EPSILON \
			    || ( ::fabs( (a) - (b) ) / (b) ) < 100*DBL_EPSILON ) )

#include "ZMutility/fstream"

USING( namespace std )

std::string Starter = "Starter";
std::string Ender = "Ender";
const int dim1 = 10;
const int dim2 = 20;
// under 1006 behavior with readColumn is weird :(
#if 0
int nr1  = 2000;
int nr2  = 2006;
#else
int nr1 = 50;
int nr2 = 50;
#endif

HepNtuple& create ( HepFileManager * man, const string& name ) {

  HepNtuple& hep = man->ntuple(name);
  hep.setColumnWise();
  hep.column("Col1",3.2,7.0);
  hep.columnArray("ColArray",2,(Float8)8).dimension(dim1,dim2);
  hep.columnDirect("ColAt",0);
  hep.columnDirect("ColAt2",0);
  return hep;

}

HepNtuple& create_start ( HepFileManager * man ) {
  Int4 atInt1 = 55;
  Float8 arr[dim1][dim2];
  Int4 atInt2 = 55;

  Int4 i,j,k;

  for ( i = 0; i < dim1 ; i++) {
    for ( j = 0; j < dim2 ; j++ ) {
      arr[i][j] = 1.0;
    }
  }

  HepNtuple& hep = create ( man, Starter );

  hep.columnAt("ColAt",&atInt1);
  hep.columnAt("ColAt2",&atInt2);

  hep.capture();
  hep.storeCapturedData();

  for (i=0; i<nr1; i++) {
    hep.capture("Col1",3.2+i);
    atInt1 = -3*i;
    for (j=0; j < dim1; j++) {
      for (k=0; k< dim2; k++) {
	arr[j][k] = 8.2/(i+1)*j*k;
      }
    }
    hep.capture("ColArray",&(arr[0][0]));
    hep.capture();
    hep.storeCapturedData();
  }

  return hep;
}

bool check_start ( HepNtuple& hep ) {

  Float8 arr[dim1][dim2];
  bool status = true;
  Int4 i,j,k;
  Float8 float1;
  Int4 int1,int2;

  for ( i = 0; i < dim1 ; i++) {
    for ( j = 0; j < dim2 ; j++ ) {
      arr[i][j] = 1.0;
    }
  }

  Float8 col1;
  hep.readColumn(1,"Col1",&col1);

  if ( ! IS_DBL_EQUAL( col1, 3.2) ) {
    cerr << "first col1 is wrong " << col1 << endl;
    status = false;
  }
  hep.readColumnArray(1,"ColArray",&(arr[0][0]));
  for ( i = 0; i < dim1*dim2 ; i++) {
    if ( ! IS_DBL_EQUAL( ((double*)arr)[i] , 8.0 ) ) {
      cerr << "first ColArray of " << i << " is wrong " << ((double*)arr)[i]
	<< endl;
      status = false;
   }
  }

  // HBook does not like it if we only read some of the columns :(
  hep.readColumn(1,"ColAt",&int1);
  hep.readColumn(1,"ColAt2",&int2);


  hep.destinationAt("Col1",&float1);
  hep.destinationAt("ColAt",&int1);
  hep.destinationAt("ColAt2",&int2);
  hep.destinationAt("ColArray",&(arr[0][0]));
    i = 0;
    hep.readRow(i+2); // the first row is "1"

    if ( ! IS_DBL_EQUAL( float1 ,3.2+i ) ) {
      cerr << "Col1, row " << i+2 << " wrong " << float1 << " vs " << (3.2+i)
	   << endl;
      status = false;
    }
    if ( int1 != -3*i ) {
      cerr << "ColAt, row " << i+2 << " wrong " << int1 << " vs " << -3*i
	   << endl;
      status = false;
    }
    if ( int2 != 55 ) {
      cerr << "ColAt2, row " << i+2 << " wrong " << int2 << " vs " << 55
	   << endl;
      status = false;
    }
    for (j=0; j < dim1; j++) {
      for (k=0; k< dim2; k++) {
	if ( ! IS_DBL_EQUAL( arr[j][k], 8.2/(i+1)*j*k) ) {
	  cerr << "ColArray, row " << i+2 << " wrong " << j << "," << k << ": "
	       << arr[j][k] << " vs " << (8.2/(i+1)*j*k)
	       << ";" << endl;
	  status = false;
	}
      }
    }
  // now readColumn back and compare !!
  for (i=1; i<nr1; i++) {

    hep.readRow(i+2); // the first row is "1"

    if ( ! IS_DBL_EQUAL( float1 ,3.2+i ) ) {
      cerr << "Col1, row " << i+2 << " wrong " << float1 << " vs " << (3.2+i)
	   << endl;
      status = false;
    }
    if ( int1 != -3*i ) {
      cerr << "ColAt, row " << i+2 << " wrong " << int1 << " vs " << -3*i
	   << endl;
      status = false;
    }
    if ( int2 != 55 ) {
      cerr << "ColAt2, row " << i+2 << " wrong " << int2 << " vs " << 55
	   << endl;
      status = false;
    }
    for (j=0; j < dim1; j++) {
      for (k=0; k< dim2; k++) {
	if ( ! IS_DBL_EQUAL( arr[j][k], 8.2/(i+1)*j*k) ) {
	  cerr << "ColArray, row " << i+2 << " wrong " << j << "," << k << ": "
	       << arr[j][k] << " vs " << (8.2/(i+1)*j*k)
	       << ";" << endl;
	  status = false;
	}
      }
    }
  }
  return status;
}

HepNtuple& create_end ( HepFileManager * man ) {
  Int4 atInt1 = 55;
  Float8 arr[dim1][dim2];
  Int4 atInt2 = 55;

  Int4 i,j,k;

  for ( i = 0; i < dim1 ; i++) {
    for ( j = 0; j < dim2 ; j++ ) {
      arr[i][j] = 1.0;
    }
  }

  HepNtuple& hep = create ( man, Ender );

  hep.columnAt("ColAt",&atInt1);
  hep.columnAt("ColAt2",&atInt2);

  hep.capture();
  hep.storeCapturedData();

  for (i=0; i<nr2; i++) {
    hep.capture("Col1",3.2+i);
    atInt1 = -3*i;
    for (j=0; j < dim1; j++) {
      for (k=0; k< dim2; k++) {
	arr[j][k] = 8.2/(i+1)*j*k;
      }
    }
    hep.capture("ColArray",&(arr[0][0]));
    hep.capture();
    hep.storeCapturedData();
  }

  return hep;
}

bool check_end ( HepNtuple& hep, Int4 start_row ) {
  Float8 float1;
  Int4 int1,int2;
  Float8 arr[dim1][dim2];
  bool status = true;
  Int4 i,j,k;

  for ( i = 0; i < dim1 ; i++) {
    for ( j = 0; j < dim2 ; j++ ) {
      arr[i][j] = 1.0;
    }
  }

#if 0
  Float8 col1;
  hep.readColumn(start_row+1,"Col1",&col1);
  if ( ! IS_DBL_EQUAL( col1, 3.2) ) {
    cerr << "first col1 is wrong " << col1 << endl;
    status = false;
  }
  hep.readColumnArray(start_row+1,"ColArray",&(arr[0][0]));
  for ( i = 0; i < dim1*dim2 ; i++) {
    if ( ! IS_DBL_EQUAL( ((double*)arr)[i] , 8.0 ) ) {
      cerr << "first ColArray of " << i << " is wrong " << ((double*)arr)[i]
	<< endl;
      status = false;
   }
  }
  // HBook does not like it if we only read some of the columns :(
  hep.readColumn(1,"ColAt",&int1);
  hep.readColumn(1,"ColAt2",&int2);
#endif
  hep.destinationAt("Col1",&float1);
  hep.destinationAt("ColAt",&int1);
  hep.destinationAt("ColAt2",&int2);
  hep.destinationAt("ColArray",&(arr[0][0]));
  // now readColumn back and compare !!
  for (i=0; i<nr2; i++) {

    hep.readRow(start_row+i+2); // the first row is "1"

    if ( ! IS_DBL_EQUAL( float1 ,3.2+i ) ) {
      cerr << "Col1, row " << i+2 << " wrong " << float1 << " vs " << (3.2+i)
	   << endl;
      status = false;
    }
    if ( int1 != -3*i ) {
      cerr << "ColAt, row " << i+2 << " wrong " << int1 << " vs " << -3*i
	   << endl;
      status = false;
    }
    if ( int2 != 55 ) {
      cerr << "ColAt2, row " << i+2 << " wrong " << int2 << " vs " << 55
	   << endl;
      status = false;
    }
    for (j=0; j < dim1; j++) {
      for (k=0; k< dim2; k++) {
	if ( ! IS_DBL_EQUAL( arr[j][k], 8.2/(i+1)*j*k) ) {
	  cerr << "ColArray, row " << i+2 << " wrong " << j << "," << k << ": "
	       << arr[j][k] << " vs " << (8.2/(i+1)*j*k)
	       << ";" << endl;
	  status = false;
	}
      }
    }
  }
  return status;
}



void test (HepFileManager* man1, HepFileManager* man2 ) {
  std::cerr << "Test from " << man1->fileName() << " to "
	    << man2->fileName() << endl;

  HepNtuple& start = create_start( man1 );
  check_start( start );

  HepNtuple& end = create_end( man2 );
  check_end( end, 0 );

  Int4 start_nrow = start.nrows();

  start.append( end );

  check_start( start );
  check_end  ( start, start_nrow );

}

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

  HepFileManager* files[3];
  string managerType[3];
  Int4 nFiles = 0;

#ifdef USE_HBOOK
  files[nFiles] =  OPEN_HBOOK_FILE("append");
  managerType[nFiles] = "hbook";
  if ( files[nFiles] != NULL ) nFiles++;
#endif
#ifdef USE_HISTOSCOPE
  files[nFiles] =  OPEN_HISTO_FILE("append");
  managerType[nFiles] = "histoscope";
  if ( files[nFiles] != NULL ) nFiles++;
#endif
  // Root does not handle ntuples yet #ifdef USE_ROOT
#ifdef USE_ROOT
  files[nFiles] =  OPEN_ROOT_FILE("append");
  managerType[nFiles] = "root";
  if ( files[nFiles] != NULL ) nFiles++;
#endif

  Int4 i,j;

  for ( i = 0; i < nFiles; i++ ) {
    for ( j = 0; j < nFiles; j++ ) {
      if ( i!=j ) {
	string new_dir_i = string("from_")+managerType[j];
	string new_dir_j = string("to_")+managerType[i];
	files[i]->cd();
	files[i]->mkdir( new_dir_i );
	files[i]->cd( new_dir_i );
	files[j]->cd();
	files[j]->mkdir( new_dir_j );
	files[j]->cd( new_dir_j );
      } else {
	files[i]->cd();
	files[i]->mkdir( "internal" );
	files[i]->cd   ( "internal" );
      }
      test ( files[i], files[j] );
    }
  }
  for ( i=0; i<nFiles; ++i ) {
    if( files[i] != NULL ) delete files[i];
  }
  return 0;
}