rootgINukeVal.C File Reference

#include <iostream>
#include <fstream>
#include <sstream>
#include <list>
#include <TMath.h>
#include <TCanvas.h>
#include <TError.h>
#include <TTree.h>
#include <string>
#include <TCollection.h>
#include <TNtuple.h>
#include <vector>
#include <typeinfo>
#include <genieStyle.C>

Go to the source code of this file.

Classes
class	DataFile
class	FormatFile

Functions
string	trim (string in)
int	rootgINukeVal (char *fFile, char *dataDir=".", char *ROOTDir=".", char *saveDir=".")

Function Documentation

int rootgINukeVal	(	char *	fFile,
		char *	dataDir = ".",
		char *	ROOTDir = ".",
		char *	saveDir = "."
	)

Definition at line 458 of file rootgINukeVal.C.

{
  string tFile (fFile);
  string dDir (dataDir);
  int legendSize; //counting entries in legend
  float TextSize, y1; //adjusted font size and lower bound on legend
  FormatFile format (tFile);
  TCanvas* cans;
  set_root_env();
  size_t curRecord=1;
  bool doCurrent = false;
  doCurrent = format.process(curRecord);
  if(doCurrent == false){
    //This if statement shouldn't be necessary, but if it
    // isn't here, the while loop tries to execute anyways...
   cout<<"Format File not found"<<endl;
   return 0;
  }
  while(doCurrent == true){
    cout<<format.numOfType("GENIE")<<" root files and "<<format.numOfType("EXPERIMENTAL")<<" published data files "<<endl;

    int legendSize = format.numOfType("GENIE") + format.numOfType("EXPERIMENTAL");
    stringstream convert;
    //gStyle->SetErrorX(0);
    //gStyle->SetStyle(genieStyle);
    set_root_env();
    Double_t size = 0;
    Int_t num = 0;
    Int_t target = 0;
    Int_t A = 0;
    Double_t factor = 0;
    TH1F* htemp = 0;
    bool good = false;
    //gROOT->SetStyle("T2K");

    //the FormatFile class holds the graph limits as strings, converting here 
    Double_t xl, yl, xu, yu;
    convert.str(format.xl); convert.clear();
    convert >> xl;
    convert.str(format.yl); convert.clear();
    convert >> yl;
    convert.str(format.xu); convert.clear();
    convert >> xu;
    convert.str(format.yu); convert.clear();
    convert >> yu;

    //Creating a new canvas and setting some parameters 
    string canName;
    canName.assign(curRecord,'*');
    cans= new TCanvas(canName.c_str(),format.fetchGTitle().c_str());
    cans->cd();
    TPad* curP = gPad;
    if(format.logx==true){gPad->SetLogx(1);}
    if(format.logy==true){gPad->SetLogy(1);}
    
    TextSize = .035;
    y1 = 1.05-(.075*legendSize);//1, .075

    TLegend* leg1 = new TLegend(.52,y1,1,1,"");//.6
    leg1->SetTextSize(TextSize);
    //TLegend* leg1 = new TLegend(.6,.8,1,1,"");

    //Get the frame, set parameters, redraw frame
    TH1F* hf1 = (TH1F*) cans->DrawFrame(xl,yl,xu,yu);
    hf1->SetTitle(format.mtitle.c_str());
    if(format.type.compare("Angle")==0){
      hf1->GetXaxis()->SetTitle("cos(#theta)");
      hf1->GetYaxis()->SetTitle("#frac{d#sigma}{d#Omega} [#frac{mb}{sr}]");
    }
    else if(format.type.compare("Momentum")==0){
      hf1->GetXaxis()->SetTitle("Momentum [Mev]");
      hf1->GetYaxis()->SetTitle("#frac{d#sigma}{dp} [#frac{mb}{MeV}]");
    }
    else if(format.type.compare("XS")==0){
      hf1->GetXaxis()->SetTitle("Energy [MeV]");
      hf1->GetYaxis()->SetTitle("#sigma (mb)");
    }
    else{
      hf1->GetXaxis()->SetTitle("Energy [MeV]");
      hf1->GetYaxis()->SetTitle("#frac{d#sigma}{d#OmegadE} [#frac{mb}{sr#upointMev}]");
    }
    //hf1->GetXaxis()->SetNdivisions(-50202);
    hf1->GetYaxis()->CenterTitle();
    hf1->Draw();


    //Loop over each GINUKE file, scaling and drawing each
    int numRoots = format.numOfType("GENIE");
    int k;
    int markerStyle=20;

 
    for(k=0;k<numRoots;k++){
      DataFile* simData = format.makeDataFile(k,ROOTDir,"GENIE");
      if(format.type.compare("XS")==0){
        cans->cd();
        //Do things completely differently
        int curCol = simData->color;
        simData->dataTuple->SetMarkerColor(curCol);
        simData->dataTuple->SetMarkerStyle(markerStyle);
        markerStyle++;
        simData->dataTuple->SetLineStyle(2);//2
        simData->dataTuple->SetLineColor(curCol);
        cout<<"About to draw tuple"<<endl;
        //simData->dataTuple->Draw();
        cout<<simData->cut.c_str()<<endl;
        simData->dataTuple->Draw(simData->cut.c_str(),"","line psame L");
        leg1->AddEntry(simData->dataTuple,simData->title.c_str(),"P");
        cout<<"Tuple drawn"<<endl;
      }
      else{
        TCanvas* tempVas = new TCanvas("tempName","No title");
        tempVas->cd();
        //TPad* curP = gPad;
        string newCut = simData->cut;
        newCut = newCut +"&&"+simData->cols+"<="+format.xu;
        newCut = newCut + "&&"+simData->cols+">"+format.xl;
        simData->dataTree->Draw(simData->cols.c_str(),newCut.c_str(), "L");
        //tempVas is used in order to not clobber cans's htemp
        if(simData->valid()){

          //Grab the associated histogram from tempVas, apply scaling factor
          htemp = (TH1F*) gPad->GetPrimitive("htemp");
          TH1F* hist1;
          TH1F* hist1error;
          good=false;
          //Make sure that something actually existed in the cut
          if (htemp != 0x0) {
            good=true;
            hist1 = (TH1F*) htemp->Clone("hist1");
            size = hist1->GetBinWidth(1);
            hist1->Sumw2();
            hist1->Rebin(format.binFactor);
            num = simData->dataTree->GetEntries();
            simData->dataTree->SetBranchAddress("tgt",&target);
            simData->dataTree->GetEntry(1);
            A = (target/10) - (target/10000)*1000;
            factor = TMath::Power(3*1.4*TMath::Power(A,(1.0/3.0)),2)*10.0/(2*num*simData->dcth);
            hist1->Scale(factor,"width");
            int curCol = simData->color;
            hist1->SetMarkerColor(curCol);
            hist1->SetMarkerStyle(markerStyle);
            markerStyle++;
            hist1->SetLineStyle(2);
            hist1->SetLineColor(curCol);
          }
          else{
            cout<<"Nothing was found in the cut of "<<simData->filename<<endl;
          }
          if(good){
            //Draw a copy of histogram to cans on top of any other histograms already there
            cans->cd();
            //TPad* curP = gPad;
            leg1->AddEntry(hist1,simData->title.c_str());
            hist1->DrawCopy("e1 psame");//e1 psame
            hist1->Draw("hist l same");
            
            //make and draw an extra histogram to have solid error bars
            hist1error = (TH1F*) hist1->Clone("hist1error"); 
            hist1error->SetLineStyle(1);
            hist1error->DrawCopy("e1 same");
          }
        }
        else{
          cout<<"Something is wrong with data file "<<simData->filename<<endl;
          cans->cd();
        }
        tempVas->Close();
      }
    }

    //Draw all of the Experimental files, I'm pretty sure that this will only handle
    // the simple tree type generated from the .txt type data files.
    cans->cd();
    TPad* curP = gPad;
    int numFiles = format.numOfType("EXPERIMENTAL");

    /*
    int j = 0;
    
    for(j=0;j<3;j++){
      DataFile* experimental = format.makeDataFile(0,dDir,"EXPERIMENTAL");
      TGraphErrors* data1;
      if(format.type.compare("XS")==0){
        experimental->dataTuple->Draw(experimental->cut.c_str(),"","goff");
        data1 = new TGraphErrors(experimental->dataTuple->GetSelectedRows(),experimental->dataTuple->GetV1(), experimental->dataTuple->GetV2(),experimental->dataTuple->GetV3(),experimental->dataTuple->GetV4());
      }
      else{
        experimental->dataTree->Draw(experimental->cols.c_str(),"","goff");
        data1 = new TGraphErrors(experimental->dataTree->GetSelectedRows(),experimental->dataTree->GetV2(), experimental->dataTree->GetV1(),0,experimental->dataTree->GetV3());
      }
      //data1->SetLineStyle(0);
      data1->SetMarkerColor(experimental->color);
      data1->SetMarkerStyle(markerStyle);
      markerStyle++;
      leg1->AddEntry(data1,experimental->title.c_str(),"P");
      data1->Draw("p same");
    }
    */
    
    for(j=0;j<numFiles;j++){
      DataFile* experimental = format.makeDataFile(j,dDir,"EXPERIMENTAL");
      TGraphErrors* data1;
      if(format.type.compare("XS")==0){
        experimental->dataTuple->Draw(experimental->cut.c_str(),"","goff");
        data1 = new TGraphErrors(experimental->dataTuple->GetSelectedRows(),experimental->dataTuple->GetV1(), experimental->dataTuple->GetV2(),experimental->dataTuple->GetV3(),experimental->dataTuple->GetV4());
      }
      else{
        experimental->dataTree->Draw(experimental->cols.c_str(),"","goff");
        data1 = new TGraphErrors(experimental->dataTree->GetSelectedRows(),experimental->dataTree->GetV2(), experimental->dataTree->GetV1(),0,experimental->dataTree->GetV3());
      }
      data1->SetLineStyle(3);
      data1->SetMarkerColor(experimental->color);
      data1->SetMarkerStyle(markerStyle);
      markerStyle++;
      leg1->AddEntry(data1,experimental->title.c_str(),"P");
      data1->Draw("p same");
    }
    
    
    //Draw the legend    

    //set_root_env();
    leg1->SetLineWidth(2);
    leg1->Draw();

    //add_plot_label("THIS IS A TEST #alpha #Alpha#gamma", .5, .5, .05);

    //Save the record
    string saveName(saveDir);
    saveName = saveName+"/"+format.savename+".png";
    cans->SaveAs(saveName.c_str());

    //Try to find the next record
    curRecord++;
    cout<<"\nLooking for next record:"<<endl;
    doCurrent = format.process(curRecord);
  }
  cout<<"No more records found, exiting"<<endl;
  //delete [] entryArray;
  return 0;
}

string trim

(

string

in

)

Definition at line 65 of file rootgINukeVal.C.

                      {
  size_t f = in.find_first_not_of(' ',0);
  size_t l = in.find_last_not_of(' ',0);
  if(l-f-1 > f){
    in = in.substr(f,l-f-1);
  }
  return in;
}