CosthetaSec/dEdxVsCos.cxx

Go to the documentation of this file.

//This code can display dE/dx from two or more files vs costheta on one pad
 
#include <vector>
#include <string>
#include "TStyle.h"
#include "TString.h"
 
using namespace std;
 
vector<string>file;
void pushNo()
{
    file.push_back("costheta.root");
    file.push_back("check/costheta.root");
}
 
const int no = 4; //The number of graphs you want to draw
const int nbins = 80; //The number of cos(theta) bins 
const double cosMin = -0.93;  //histogram range for cos(theta)
const double cosMax = 0.93;
const double chiMin = -30;  //histogram range for chi mean
const double chiMax = 30;
const double chiSigmaMin = 0.5;  //histogram range for chi sigma
const double chiSigmaMax = 6.5;
const double fitMeanMin = 400;
const double fitMeanMax = 700;
const double fitSigmaMin = 20;
const double fitSigmaMax = 50;
 
void dEdxVsCos()
{
    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 cosBin[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("costhetacalib");
        branch.str("");
        branch<<branchheader[i]<<"chimean";
        tree -> SetBranchAddress(branch.str().c_str(),chimean[i]);
        branch.str("");
        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<<"cosBin";
        tree -> SetBranchAddress(branch.str().c_str(),cosBin[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,cosMin,cosMax,100,chiMin,chiMax);
    h[0] -> GetYaxis()->SetTitle("#chi_{mean}");
    //h[0] -> GetYaxis()->SetTitleOffset(.9);
    h[0] -> GetXaxis()->SetTitle("cos(#theta)");
    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,cosBin[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], file[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,cosMin,cosMax,100,chiSigmaMin,chiSigmaMax);
    h[1] -> GetYaxis()->SetTitle("#chi_{#sigma}");
    //h[1] -> GetYaxis()->SetTitleOffset(.9);
    h[1] -> GetXaxis()->SetTitle("cos(#theta)");
    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,cosBin[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], file[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,cosMin,cosMax,100,fitMeanMin,fitMeanMax);
    h[2] -> GetYaxis()->SetTitle("fitmean");
    //h[2] -> GetYaxis()->SetTitleOffset(.9);
    h[2] -> GetXaxis()->SetTitle("cos(#theta)");
    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,cosBin[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], file[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,cosMin,cosMax,100,fitSigmaMin,fitSigmaMax);
    h[3] -> GetYaxis()->SetTitle("fitsigma");
    //h[3] -> GetYaxis()->SetTitleOffset(.9);
    h[3] -> GetXaxis()->SetTitle("cos(#theta)");
    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,cosBin[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], file[i].c_str() , "p");
    }
    lgd[3] -> Draw();
 
    TFile* g = new TFile("dEdxVsCostheta.root","recreate");
    for(int i=0;i<no;i++)
    {
        c[i]->Write();
    }
    g->Close();
}