//-*-c++-*-///////////////////////////////////////////////////////
//                                                              //
// Analysis class for simulation code -- record data output     //
//   by Exaos Lee (exaos@yahoo.com.cn)                          //
//                                                              //
//////////////////////////////////////////////////////////////////
// $Name$
// $Id$
// $Log$
//////////////////////////////////////////////////////////////////

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

using namespace std;

ERootAnalysis::ERootAnalysis()
{
  pDataFile = new TFile("ERootAnalysis.root","RECREATE");

  if( !pDataFile ) {
    std::cerr << "Error opening data file!" << endl;
    exit(-1);
  }

  dTree = new TTree();
}

ERootAnalysis::ERootAnalysis(TString fpre)
{
  fpre += ".root";
  pDataFile = new TFile(fpre,"RECREATE");

  if( !pDataFile ) {
    std::cerr << "Error opening data file!" << endl;
    exit(-1);
  }

  dTree = new TTree();
}

ERootAnalysis::~ERootAnalysis()
{
  if(pDataFile) {
    pDataFile->Write();
    pDataFile->Close();
  } else {
    std::cout << "No data recorded!" << endl;
  }
  
  if( !pH1FArr ) {
    free(pH1FArr);
  }
}

int ERootAnalysis::InitHist(int n)
{
  int errNum = 0;
  
  pH1FArr = calloc(n,sizeof(TH1F*));
  for(int i=0;i<n;i++) {
    TString hName = "h1f_"; hName += i;
    //TString hSpec = "";
    pH1FArr[i]=new TH1F(hName.Data(),"",1000,0,1000);
    if(!pH1FArr[i]) errNum+=-1;
  }

  return errNum; 
}