Costheta/dEdxVsCosNP.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("costheta.root");
}
 
const int no = 5; //The number of graphs you want to draw
const int nbins = 80; //The number of cos(theta) bins 
const double cosMin = -1.0;  //histogram range for cos(theta)
const double cosMax = 1.0;
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 = 530;
//const double fitMeanMax = 585;
const double fitMeanMin = 30;
const double fitMeanMax = 2585;
const double fitSigmaMin = 20;
const double fitSigmaMax = 50;
const double slopeMin = 0.8;
const double slopeMax = 1.2;
 
void dEdxVsCosNP()
{
    pushNo();
    const int FileNo = file.size();
    string branchheader[FileNo] = {"pos_","neg_"};
    string str_names[FileNo] = {"positron","electron"};
    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], slope[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], str_names[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], str_names[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], str_names[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], str_names[i].c_str() , "p");
    }
    lgd[3] -> Draw();
 
    //--------------------- slope -----------------------------//
    c[4] = new TCanvas("slope","slope",800,600);
    c[4] -> SetGridx();
    c[4] -> SetGridy();
    h[4] = new TH2F("slope","slope",100,cosMin,cosMax,100,slopeMin,slopeMax);
    h[4] -> GetYaxis()->SetTitle("slope");
    //h[4] -> GetYaxis()->SetTitleOffset(.9);
    h[4] -> GetXaxis()->SetTitle("cos(#theta)");
    h[4] -> Draw();
    lgd[4] = new TLegend(.65, .75, .89, .89);
    lgd[4]->SetFillColor(10);
    lgd[4]->SetBorderSize(1);
 
 
    for(int j=0; j<FileNo; j++){
      for(int k=0; k<nbins; k++){
    if(fitmean[j][k]) slope[j][k] = fitmean[j][k]/fitmean[j][nbins-k-1];
    else slope[j][k] = 0;
      }
    }
 
    for(int i=0;i<FileNo;i++)
    {
        gr[4][i] = new TGraphErrors(nbins/2,cosBin[i],slope[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_names[i].c_str() , "p");
    }
    lgd[4] -> Draw();
 
    // output file
    TFile* g = new TFile("dEdxVsCosNP.root","recreate");
    for(int i=0;i<no;i++)
    {
        c[i]->Write();
    }
    g->Close();
}