#include "style.h"

#include "plot_histos.h"
#include "Global.h"

#include <iostream>
#include <string>

#include "TFile.h"
#include "TTree.h"
#include "TApplication.h"
#include "TROOT.h"

using namespace std;
void draw_multifiles_NToC(int n, int* nsigma, string* files);

int main_a(int argc, char* argv[]) {

  TApplication theApp("Analysis", &argc, argv);

  GetMyStyle(); // defined in style.h

  TFile* tf = new TFile("/project/dzero/balm/efficiency/rootfiles/AATrack_b_jpsiX_p1402-p14.04.00_100k.root");
  TTree* tt = (TTree*) tf->Get("Global");

  cout <<"tree at "<<tt<<endl;

  if(tt) {
    Global* a = new Global(tt); // create this on the heap to avoid stack overflow!
    
    cout <<"Doing nsigma = 1, 2, 4, 8\n\nGo!\n\n"<<endl;
    
    const int n = 4;
    int nsigma[n] = {1,2,4,8};
    string files[n];
    files[0] = "RatioNToC_0.root";
    files[1] = "RatioNToC_1.root";
    files[2] = "RatioNToC_2.root";
    files[3] = "RatioNToC_3.root";

    //for(int i=0; i<n; i++) a->Loop(nsigma[i], files[i].c_str());

    draw_multifiles_NToC(n, nsigma, files);
  }

  theApp.Run(kTRUE);
  //sleep(5);
  return 0;
}	

void draw_multifiles_NToC(int n, int* nsigma, string* files) {

  cout <<"=== Starting draw_multifiles_NToC in RatioNeutralsToCharged_x.cpp"<<endl;

  TCanvas* c_eff = new TCanvas("multi_eff","Selection Efficiency",0,0,500,500);

  for(int i=0; i<n; i++) {

    // Get filename, open file
    stringstream file_name;
    file_name <<"rootfiles/SelEff_MC/"<<files[i];
    cout <<"i="<<i<<": Reading file "<<file_name.str()<<endl;
    TFile* tf = new TFile(file_name.str().c_str(),"READ");
    if(tf && tf->IsOpen()) {

      // Read histograms
      TH1F* h_tmp_a = new TH1F( *(TH1F*) tf->Get("jpsi_lxy") );   // all
      TH1F* h_tmp_n = (TH1F*) tf->Get("jpsi_n_lxy"); // neutral
      TH1F* h_tmp_c = (TH1F*) tf->Get("jpsi_c_lxy"); // charged

      
      if(!h_tmp_a || !h_tmp_n || !h_tmp_c) {
	cout <<"Couldn't get histogram ouf of file "<<file_name.str()<<endl;
	cout <<"a="<<h_tmp_a<<"  n="<<h_tmp_n<<"  c="<<h_tmp_c<<endl;
	continue;
      }

      cout << "drawing"<<endl;
      c_eff->cd();
      h_tmp_a->Draw();

    } // if tf->IsOpen
    else cout <<"Couldn't open file."<<endl;
    
    //sleep(2);//2 secs

  } // for i

  c_eff->Update();
  cout <<"=== draw_multifiles_NToC in RatioNeutralsToCharged_x.cpp done."<<endl;
  return;
} // end draw multifilesNToC