Eangle1DSec/dEdxVsP.cxx

Go to the documentation of this file.

//This code can display dE/dx from two files vs momentum on one pad.
 
#include "TStyle.h"
#include "TString.h"
#include <vector>
#include <string>
 
using namespace std;
 
vector<string>file;
vector<string> str_name;
void pushNo()
{
    file.push_back("dEdxVsP.root");
    file.push_back("dEdxVsP.root");
    str_name.push_back("positron");
    str_name.push_back("electron");
}
 
 
 
const int no = 6; //The number of graphs you want to draw
const int nbins = 50; //The number of momentum bins 
const double pMin = 0.1;  //histogram range for momentum
const double pMax = 3.1/2;
const double chiMin = -1;  //histogram range for chi mean
const double chiMax = 1;
const double chiSigmaMin = 0.5;  //histogram range for chi sigma
const double chiSigmaMax = 1.5;
const double fitMeanMin = 520; 
const double fitMeanMax = 560;
const double fitSigmaMin = 25;
const double fitSigmaMax = 50;
const double hitsMeanMin = 10;
const double hitsMeanMax = 30;
const double hitsSigmaMin = 0;
const double hitsSigmaMax = 10;
 
void dEdxVsP()
{
    pushNo();
    const int FileNo = file.size();
    string branchheader[FileNo] = {"pos_","neg_"};
    TH2F** h = new TH2F*[no];
    TCanvas** c = new TCanvas*[no];
    TLegend** lgd = new TLegend*[no];
    TGraphErrors*** gr = new TGraphErrors[no][FileNo];
    for(int i =0;i<no;i++) gr[i] = new TGraphErrors*[FileNo];
    gStyle->SetOptStat(0);
    gStyle->SetCanvasColor(10);
 
    TFile* f;
    TTree* tree;
 
    double chimean[FileNo][nbins],chimeanerr[FileNo][nbins],chisigma[FileNo][nbins];
    double fitmean[FileNo][nbins],fitmeanerr[FileNo][nbins],fitsigma[FileNo][nbins];
    double hits_mean[FileNo][nbins],hits_sigma[FileNo][nbins];
    double pBin[FileNo][nbins];
 
    stringstream ss;
    stringstream branch;
    for(int i=0;i<FileNo;i++)
    {
        ss.str("");
        ss<<file[i];
        f = new TFile(ss.str().c_str());
        tree = (TTree*) f->Get("momInfor");
        branch.str("");
        branch<<branchheader[i]<<"chimean"; 
        //cout<<branch.str().c_str()<<endl;
        tree -> SetBranchAddress(branch.str().c_str(),chimean[i]);
        branch.str("");
        //cout<<branch.str().c_str()<<endl;
        branch<<branchheader[i]<<"chimeanerr";
        tree -> SetBranchAddress(branch.str().c_str(),chimeanerr[i]);
        branch.str("");
        branch<<branchheader[i]<<"chisigma";
        tree -> SetBranchAddress(branch.str().c_str(),chisigma[i]);
        branch.str("");
        branch<<branchheader[i]<<"fitmean";
        tree -> SetBranchAddress(branch.str().c_str(),fitmean[i]);
        branch.str("");
        branch<<branchheader[i]<<"fitmeanerr";
        tree -> SetBranchAddress(branch.str().c_str(),fitmeanerr[i]);
        branch.str("");
        branch<<branchheader[i]<<"fitsigma";
        tree -> SetBranchAddress(branch.str().c_str(),fitsigma[i]);
        branch.str("");
        branch<<"hits_mean";
        tree -> SetBranchAddress(branch.str().c_str(),hits_mean[i]);
        branch.str("");
        branch<<"hits_sigma";
        tree -> SetBranchAddress(branch.str().c_str(),hits_sigma[i]);
        branch.str("");
        branch<<"pBin";
        tree -> SetBranchAddress(branch.str().c_str(),pBin[i]);
        tree -> GetEntry(0);
    }
 
    //--------------------- chimean ------------------------------//
    c[0] = new TCanvas("chimean","chimean",800,600);
    c[0] -> SetGridx();
    c[0] -> SetGridy();
    h[0] = new TH2F("chimean","chimean",100,pMin,pMax,100,chiMin,chiMax);
    h[0] -> GetYaxis()->SetTitle("#chi_{mean}");
    //h[0] -> GetYaxis()->SetTitleOffset(.9);
    h[0] -> GetXaxis()->SetTitle("momentum");
    h[0] -> Draw();
    lgd[0] = new TLegend(.65, .75, .89, .89);
    lgd[0]->SetFillColor(10);
    lgd[0]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)
    {
        gr[0][i] = new TGraphErrors(nbins,pBin[i],chimean[i],0,chimeanerr[i]);
        gr[0][i] -> SetMarkerStyle(20+i);
        gr[0][i] -> SetMarkerColor(i+1);
        gr[0][i] -> Draw("Psame");
        lgd[0] -> AddEntry(gr[0][i], str_name[i].c_str() , "p");
    }
    lgd[0] -> Draw();
 
    //--------------------- chisigma -----------------------------//
    c[1] = new TCanvas("chisigma","chisigma",800,600);
    c[1] -> SetGridx();
    c[1] -> SetGridy();
    h[1] = new TH2F("chisigma","chisigma",100,pMin,pMax,100,chiSigmaMin,chiSigmaMax);
    h[1] -> GetYaxis()->SetTitle("#chi_{#sigma}");
    //h[1] -> GetYaxis()->SetTitleOffset(.9);
    h[1] -> GetXaxis()->SetTitle("momentum");
    h[1] -> Draw();
    lgd[1] = new TLegend(.65, .75, .89, .89);
    lgd[1]->SetFillColor(10);
    lgd[1]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)
    {
        gr[1][i] = new TGraphErrors(nbins,pBin[i],chisigma[i],0,0);
        gr[1][i] -> SetMarkerStyle(20+i);
        gr[1][i] -> SetMarkerColor(i+1);
        gr[1][i] -> Draw("Psame");
        lgd[1] -> AddEntry(gr[1][i], str_name[i].c_str() , "p");
    }
    lgd[1] -> Draw();
 
    //--------------------- fitmean -----------------------------//
    c[2] = new TCanvas("fitmean","fitmean",800,600);
    c[2] -> SetGridx();
    c[2] -> SetGridy();
    h[2] = new TH2F("fitmean","fitmean",100,pMin,pMax,100,fitMeanMin,fitMeanMax);
    h[2] -> GetYaxis()->SetTitle("fitmean");
    //h[2] -> GetYaxis()->SetTitleOffset(.9);
    h[2] -> GetXaxis()->SetTitle("momentum");
    h[2] -> Draw();
    lgd[2] = new TLegend(.65, .75, .89, .89);
    lgd[2]->SetFillColor(10);
    lgd[2]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)
    {
        gr[2][i] = new TGraphErrors(nbins,pBin[i],fitmean[i],0,0);
        gr[2][i] -> SetMarkerStyle(20+i);
        gr[2][i] -> SetMarkerColor(i+1);
        gr[2][i] -> Draw("Psame");
        lgd[2] -> AddEntry(gr[2][i], str_name[i].c_str() , "p");
    }
    lgd[2] -> Draw();
 
    //--------------------- fitsigma -----------------------------//
    c[3] = new TCanvas("fitsigma","fitsigma",800,600);
    c[3] -> SetGridx();
    c[3] -> SetGridy();
    h[3] = new TH2F("fitsigma","fitsigma",100,pMin,pMax,100,fitSigmaMin,fitSigmaMax);
    h[3] -> GetYaxis()->SetTitle("fitsigma");
    //h[3] -> GetYaxis()->SetTitleOffset(.9);
    h[3] -> GetXaxis()->SetTitle("momentum");
    h[3] -> Draw();
    lgd[3] = new TLegend(.65, .75, .89, .89);
    lgd[3]->SetFillColor(10);
    lgd[3]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)
    {
        gr[3][i] = new TGraphErrors(nbins,pBin[i],fitsigma[i],0,0);
        gr[3][i] -> SetMarkerStyle(20+i);
        gr[3][i] -> SetMarkerColor(i+1);
        gr[3][i] -> Draw("Psame");
        lgd[3] -> AddEntry(gr[3][i], str_name[i].c_str() , "p");
    }
    lgd[3] -> Draw();
 
    //---------------------used hits for dE/dx calculation -----------------------------//
    c[4] = new TCanvas("hits_mean","hits_mean",800,600);
    c[4] -> SetGridx();
    c[4] -> SetGridy();
    h[4] = new TH2F("hits_mean","hits_mean",100,pMin,pMax,100,hitsMeanMin,hitsMeanMax);
    h[4] -> GetYaxis()->SetTitle("hits_mean");
    //h[4] -> GetYaxis()->SetTitleOffset(.9);
    h[4] -> GetXaxis()->SetTitle("momentum");
    h[4] -> Draw();
    lgd[4] = new TLegend(.65, .75, .89, .89);
    lgd[4]->SetFillColor(10);
    lgd[4]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)                   
    {                                                 
        gr[4][i] = new TGraphErrors(nbins,pBin[i],hits_mean[i],0,0); 
        gr[4][i] -> SetMarkerStyle(20+i);
        gr[4][i] -> SetMarkerColor(i+1);
        gr[4][i] -> Draw("Psame");
        lgd[4] -> AddEntry(gr[4][i], str_name[i].c_str() , "p");    
    }
    lgd[4] -> Draw();
 
    //---------------------sigma of used hits for dE/dx calculation -------------------------//
    c[5] = new TCanvas("hits_sigma","hits_sigma",800,600);
    c[5] -> SetGridx();
    c[5] -> SetGridy();
    h[5] = new TH2F("hits_sigma","hits_sigma",100,pMin,pMax,100,hitsSigmaMin,hitsSigmaMax);
    h[5] -> GetYaxis()->SetTitle("hits_sigma");
    //h[5] -> GetYaxis()->SetTitleOffset(.9);
    h[5] -> GetXaxis()->SetTitle("momentum");
    h[5] -> Draw();
    lgd[5] = new TLegend(.65, .75, .89, .89);
    lgd[5]->SetFillColor(10);
    lgd[5]->SetBorderSize(1);
 
    for(int i=0;i<FileNo;i++)                   
    {                                                 
        gr[5][i] = new TGraphErrors(nbins,pBin[i],hits_sigma[i],0,0); 
        gr[5][i] -> SetMarkerStyle(20+i);
        gr[5][i] -> SetMarkerColor(i+1);
        gr[5][i] -> Draw("Psame");
        lgd[5] -> AddEntry(gr[5][i], str_name[i].c_str() , "p");    
    }
    lgd[5] -> Draw();
 
    TFile* g = new TFile("dEdxVsP_show.root","recreate");
    for(int i=0;i<no;i++)    
    {
      c[i]->Write();
      TString str_path = "figs/mom_check_";
      TString str_epsname = c[i]->GetName();
      if(str_epsname=="fitmean" || str_epsname=="fitsigma") c[i]->SaveAs(str_path + str_epsname + ".eps");
    }
    g->Close();
}