//
// This file test the functionalities of the Block class.
// Interectly it also test the functionalities of the ColumnAttribs class.
// 

#include <HepTuple/HepHBookNtuple.h>
#include <HepTuple/HepNtuple.h>
#include <HepTuple/Column.h>
#include <HepTuple/Block.h>
#include <iostream.h>
#include <stdio.h>

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; };
};


int main (int argc, Int1* argv[]) {
  Int4 atInt1 = 55;
  Int4 atInt2 = 55;
  Float8 atDouble = 55.0;
  Test object;
  // Float8 arr[10][10];
  Float8 arr[100][10];
  
  int dim1 = 10;
  int dim2 = 10;
  
  for (int i = 0; i < dim1 ; i++) {
    ((double*)arr)[i] = 1.0;
  }
  
  HepFileManager* manager = new HepHBookFileManager("compCWN.out");
  cout << " will create ntuple \n";
  HepNtuple& hep = manager->ntuple("NTUPLE",1);
  hep.setColumnWise();
  
  
  hep.newColumn("Col1",3.2,7.0);
  hep.newColumn("Col2",8.2);
  
  hep.columnVia("colF",func,(Int4)0);
  hep.columnVia("colM",newUseMethod(&object, object.a));

  hep.newColumn("Col3",8.2,99.0);
  hep.newColumn("Col4",8.2,99.0);

  hep.columnAt("colAt",&atInt1);
  hep.columnAt("colAt2",&atInt2);

  hep.columnArray("colArray",2,(Float8)8).dimension(dim1,dim2);
  
  

  hep.capture();
  hep.storeCapturedData();
  hep.clearData();
  
  cerr << hep.blockFormat(0) << endl;
  cerr << "loop: ";
  for (int i=0; i<50; i++) {
    cerr << i << ":";
    
    hep.capture("Col1",3.2+i);
    hep.capture("Col2",8.2+2*i);
    
    atInt1 = -3*i;
    c = i; 
    object.seta(5*i);
    // cout << "before loop " << dim1 << "," << dim2 << "\n";
    for (int j=0; j < dim1+1; j++) {
      for (int k=0; k< dim2+1; k++) {
	arr[j][k] = 8.2/(i+1)*j*k;
	// cerr << j << "," << k << ":" << arr[j][k] << ";";
      }
    }
    hep.capture("colArray",&(arr[0][0]));
    hep.capture();
    hep.storeCapturedData();
    hep.clearData();
  }
  cout << endl;

  // set the readColumn back destinations.

  Float8 col1;
  hep.readColumn(1,"Col1",&col1);
  if ( col1 != 3.2 ) {
    cerr << "first col1 is wrong " << col1 << endl;
  }
  hep.readColumn(1,"Col2",&col1);
  if ( col1 != 8.2 ) {
    cerr << "first col2 is wrong " << col1 <<  endl;
  }
  hep.readColumn(1,"Col3",&col1);
  if ( col1 != 8.2 ) {
    cerr << "first col3 is wrong " << col1 <<  endl;
  }
  hep.readColumn(1,"Col4",&col1);
  if ( col1 != 8.2 ) {
    cerr << "first col4 is wrong " << col1 <<  endl;
  }
  Int4 creadColumn;
  hep.readColumn(1,"colF",&creadColumn);
  if ( creadColumn != 0 ) {
    cerr << "first colF is wrong " << creadColumn <<endl;
  }
  hep.readColumn(1,"colM",&creadColumn);
  if (  creadColumn != 0 ) {
    cerr << "first colM is wrong " << creadColumn <<endl;
  }
  hep.readColumn(1,"colAt",&creadColumn);
  if ( creadColumn != 55 ) {
    cerr << "first colAt is wrong " << creadColumn <<endl;
  }
  hep.readColumn(1,"colAt2",&creadColumn);
  if ( creadColumn != 55 ) {
    cerr << "first colAt2 is wrong " << creadColumn <<endl;
  }
  hep.readColumnArray(1,"colArray",&(arr[0][0]));
  for (int i = 0; i < dim1*dim2 ; i++) {
    if ( ((double*)arr)[i] != 8.0 ) {
      cerr << "first colArray of " << i << " is wrong " << ((double*)arr)[i] 
	<< endl;
    }
  }

  Float8 float1,float2,float3,float4;
  Int4 int1,int2;
  
  hep.destinationAt("Col1",&float1);
  hep.destinationAt("Col2",&float2);
  hep.destinationVia("colF",readFunc);
  hep.destinationVia("colM",newSetMethod(&object, object.seta));
  hep.destinationAt("Col3",&float3);
  hep.destinationAt("Col4",&float4);
  hep.destinationAt("colAt",&int1);
  hep.destinationAt("colAt2",&int2);
  hep.destinationAt("colArray",&(arr[0][0]));
  

  // now readColumn back and compare !!
  for (int i=0; i<50; i++) {
    cerr << i << ":";
    
    hep.readRow(i+2); // the first row is "1" 

    if ( float1 != 3.2+i ) {
      cerr << "Col1 wrong " << float1 << endl;
    }
    if ( float2 != 8.2+2*i ) {
      cerr << "Col2 wrong " << float1 << endl;
    }
    if ( float3 != 99.0 ) {
      cerr << "Col3 wrong " << float1 << endl;
    }
    if ( float4 != 99.0 ) {
      cerr << "Col4 wrong " << float1 << endl;
    }
    if ( int1 != -3*i ) {
      cerr << "colAt wrong " << float1 << endl;
    }
    if ( int2 != 55 ) {
      cerr << "colAt2 wrong " << float1 << endl;
    }
    if ( c!= i ) {
      cerr << "colF wrong " << c << endl;
    }
    if ( object.a() != 5*i ) {
      cerr << "colM wrong " << object.a() << endl;
    }
    for (int j=0; j < dim1; j++) {
      for (int k=0; k< dim2; k++) {
	if ( arr[j][k] != 8.2/(i+1)*j*k ) {
	  cerr << "colArray wrong " << j << "," << k << ":" << arr[j][k] << ";" << endl;
	}
      }
    }
  }
  delete manager;
  cout << "END\n";
  return 0;
}