beamtime/cern-oct11/go4/AnalysisMacros/plotSpectra.cxx (r4864/r2323)

Go to the documentation of this file.

#ifndef __CINT__
#include "Riostream.h"
#include "TRint.h"
#include "TROOT.h"
#include "TStyle.h"
#include "Riostream.h"


#include "TH2.h"
#include "TH1.h"
#include "TF1.h"

#include "TGraph.h"
#include "TMultiGraph.h"
#include "TCanvas.h"
#include "TProfile.h"
#include "TStopwatch.h"

#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TMath.h"
#include "TLegend.h"
#include "TColor.h"
#include "TText.h"
#include "TKey.h"
#include "TLatex.h"
#include "TCanvas.h"
#include "TLine.h"

#endif

using namespace std;

void plotSpectra( Bool_t debug) {
  TString ifname;
  TString folder = "data2011/Analysis/";
  TString title, newtitle;
  const Int_t settings = 13;
  TString file[settings] = {
    "MS_HVScan_1",
    "MS_HVScan_2",
    "MS_HVScan_3",
    "MS_HVScan_4",
    "MS_HVScan_5",
    "MS_RadiatorScan_1",
    "MS_RadiatorScan_2",
    "MS_RadiatorScan_3",
    "MS_RadiatorScan_4",
    "MS_RadiatorScan_5",
    "MS_RadiatorScan_6",
    "MS_RadiatorScan_7",
    "MS_RadiatorScan_8"
  };
  Int_t lcolor[settings] = {    1,    2,    3,    4,    6,    1,    2,    3,    4,    6,    7,    8,    9  };
  Int_t lstyle[settings] = {    1,    1,    2,    2,    2,    1,    1,    1,    1,    1,    1,    1,    1  };
  const Int_t usedSusibos = 8;
  const Int_t recoSets = 9;
  Int_t usedSuId[usedSusibos] = {         2,         12,        6,        18,         4,         5,        16,        17  };
  TString recoSetName[recoSets] = {"ClusterInteg","3PadInteg","1PadInteg","ClusterIntegWindow","3PadIntegWindow","1PadIntegWindow","ClusterAmpl","3PadAmpl","1PadAmpl"};
  Double_t y_max[recoSets] =      {          0.03,       0.03,       0.05,         0.12,        0.12,         0.12,         0.07,        0.07,        0.07};


  TH1D* hElectronSpectra[usedSusibos][recoSets][settings];
  TH1D* hPionSpectra[usedSusibos][recoSets][settings];

  Int_t xRes(12000), yRes(4000);
  if (debug) {
    xRes = 1200;
    yRes = 400;
  }

  TCanvas *e = new TCanvas("e","e",xRes,yRes);
  e->Divide(usedSusibos,recoSets);

  TCanvas *p = new TCanvas("p","p",xRes,yRes);
  p->Divide(usedSusibos,recoSets);

  TCanvas *c = new TCanvas("c","c",xRes/10,yRes);
  c->Divide(2,recoSets);



  //for (Int_t set = 0; set < settings; set++) {/*settings*/
  for (Int_t iset = 0; iset < settings; iset++) {
    ifname = file[iset];
    TFile inputFile(folder + ifname.Append(".root"));
    if (!inputFile.IsOpen()) {
      cout << "Error: " << TString(folder + TString(ifname(0,ifname.Sizeof()-5)).Append(".root")).Data() << " not found." << endl;
      return;
    } 


   
    // -------------------------------------------------------------------   clone spectra from readtree_spadic() output file
    for (Int_t suid = 0; suid < usedSusibos; suid++){
      for (Int_t rs = 0; rs < recoSets; rs++) {
        
        title.Form("preCompSpectra%sEl_%i",recoSetName[rs].Data(),usedSuId[suid]);
        newtitle.Form("Spectra_%s_%sEl_%i",file[iset].Data(),recoSetName[rs].Data(),usedSuId[suid]);
        hElectronSpectra[suid][rs][iset] = (TH1D*)inputFile.Get(title)->Clone(newtitle);
        if (rs < 3) {
          //hElectronSpectra[suid][rs][iset]->Rebin(5);
          hElectronSpectra[suid][rs][iset]->GetXaxis()->SetRangeUser(0,2000);
          hElectronSpectra[suid][rs][iset]->GetYaxis()->SetRangeUser(1e-6,y_max[rs]);
        }
        else {
          hElectronSpectra[suid][rs][iset]->GetXaxis()->SetRangeUser(0,1000);
          hElectronSpectra[suid][rs][iset]->GetYaxis()->SetRangeUser(1e-6,y_max[rs]);
        }
        hElectronSpectra[suid][rs][iset]->SetLineColor(lcolor[iset]);
        hElectronSpectra[suid][rs][iset]->SetLineStyle(lstyle[iset]);
      
        e->cd(rs * usedSusibos + suid + 1);
        if (iset==0 || iset == 5)
          hElectronSpectra[suid][rs][iset]->DrawCopy();
        else
          hElectronSpectra[suid][rs][iset]->DrawCopy("same");
        e->Update();

        c->cd(rs * 2 + 1);
        if (iset==0 || iset == 5)
          hElectronSpectra[suid][rs][iset]->DrawCopy();
        else
          hElectronSpectra[suid][rs][iset]->DrawCopy("same");
        c->Update();


        title.Form("preCompSpectra%sPi_%i",recoSetName[rs].Data(),usedSuId[suid]);
        newtitle.Form("Spectra_%s_%sPi_%i",file[iset].Data(),recoSetName[rs].Data(),usedSuId[suid]);
        hPionSpectra[suid][rs][iset] = (TH1D*)inputFile.Get(title)->Clone(newtitle);
        if (rs < 3) {
          //hPionSpectra[suid][rs][iset]->Rebin(5);
          hPionSpectra[suid][rs][iset]->GetXaxis()->SetRangeUser(0,2000);
          hPionSpectra[suid][rs][iset]->GetYaxis()->SetRangeUser(1e-6,y_max[rs]);
        }
        else {
          hPionSpectra[suid][rs][iset]->GetXaxis()->SetRangeUser(0,1000);
          hPionSpectra[suid][rs][iset]->GetYaxis()->SetRangeUser(1e-6,y_max[rs]);
        }
        hPionSpectra[suid][rs][iset]->SetLineColor(lcolor[iset]);
        hPionSpectra[suid][rs][iset]->SetLineStyle(lstyle[iset]);

        p->cd(rs * usedSusibos + suid + 1);
        if (iset==0 || iset == 5)
          hPionSpectra[suid][rs][iset]->DrawCopy();
        else
          hPionSpectra[suid][rs][iset]->DrawCopy("same");
        p->Update();

        c->cd(rs * 2 + 2);
        if (iset==0 || iset == 5)
          hPionSpectra[suid][rs][iset]->DrawCopy();
        else
          hPionSpectra[suid][rs][iset]->DrawCopy("same");
        c->Update();

      }
    }
    if (iset == 4) {
      e->SaveAs(folder + "Pics/HV_Scan_ESpectra" + ".png");
      p->SaveAs(folder + "Pics/HV_Scan_PSpectra" + ".png");
    }
    if (iset == 12) {
      e->SaveAs(folder + "Pics/Radiator_Scan_ESpectra" + ".png");
      p->SaveAs(folder + "Pics/Radiator_Scan_PSpectra" + ".png");
    }
  }  
  //e->SaveAs(folder + "Pics/" + TString(ifname(0,ifname.Sizeof()-5)) + "_ESpectra" + ".png");
  //p->SaveAs(folder + "Pics/" + TString(ifname(0,ifname.Sizeof()-5)) + "_PSpectra" + ".png");
}