示例#1
0
//=============================================================================
// Standard constructor, initializes variables
//=============================================================================
void makeplots(TString runmode = "d", TString drawopt = "norm") {
    gROOT->SetBatch(kTRUE);
    cout<<endl<<"If you want to use custom parameters, option 'b' for branch, 'c' for cuts,";
    cout<<", 'o' to specify the output, or 'f' to select the files and trees to include."<<endl;
    cout<<"Option 'C' saves canvases as .C files."<<endl;
    cout<<"The second parameter is the draw option. 'norm' by default."<<endl<<endl;
    cout<<"This program handles a specified number of variations and generates plots from the files corresponding to these.";
    cout<<" E.g., year, decay, and filetype would plot stuff from nYears*nDecays*nFileTypes files, labeled accordingly.";
    cout<<" This is in addition to however many branches and cuts you would like to perform."<<endl<<endl;
    TString placeholder;//this is to avoid adding strings in functions; assign right before use
    TString placeholder2;
    TString response;//this is for user input; assign right before use
    //default parameters
    TString outputlocation="./";
    TString filename="plots.pdf";
    int nFiles=8;
    vector<TFile*> f(nFiles);
    TString tuplelocation = "/afs/cern.ch/work/m/mwilkins/b_b-bar_cross-section/";
    placeholder = tuplelocation+"Strp20r1_SL_D0andDp_MD.root";
    f[0] = TFile::Open(placeholder);//2011 data B-D0
    placeholder = tuplelocation+"Strp20r1_SL_D0andDp_MD.root";
    f[1] = TFile::Open(placeholder);//2011 data B0D-
    placeholder = tuplelocation+"Bu_D0Xmunu_cocktail_12873441_MC2011_S20r1_noPID_Tuples.root";
    f[2] = TFile::Open(placeholder);//2011 MC B-D0
    placeholder = tuplelocation+"Bd_DpXmunu_cocktail_11874401_MC2011_S20r1_noPID_Tuples.root";
    f[3] = TFile::Open(placeholder);//2011 MC B0D-
    placeholder = tuplelocation+"B2DMuNuX_tuples_05082015.root";
    f[4] = TFile::Open(placeholder);//2015 data B-D0
    placeholder = tuplelocation+"B2DMuNuX_tuples_05082015.root";
    f[5] = TFile::Open(placeholder);//2015 data B0D-
    placeholder = tuplelocation+"Bu_D0Xmunu_cocktail_12873441_MC2015_S22_noPID_Tuples.root";
    f[6] = TFile::Open(placeholder);//2015 MC B-D0
    placeholder = tuplelocation+"Bd_DpXmunu_cocktail_11874401_MC2015_S22_noPID_Tuples.root";
    f[7] = TFile::Open(placeholder);//2015 MC B0D-
    vector<TString> tree(nFiles);
    tree[0] = "tupleb2D0Mu/tupleb2D0Mu";
    tree[1] = "tupleb2DpMu/tupleb2DpMu";
    tree[2] = "Tuple_b2D0MuX/DecayTree";
    tree[3] = "Tuple_b2DpMuX/DecayTree";
    tree[4] = "Tuple_b2D0MuX/DecayTree";
    tree[5] = "Tuple_b2DpMuX/DecayTree";
    tree[6] = "Tuple_b2D0MuX/DecayTree";
    tree[7] = "Tuple_b2DpMuX/DecayTree";
    int nLayers=5;//year, filetype, decay, branch, cuts
    vector<TString> Lresponse(nLayers);
    Lresponse[0]="year";
    Lresponse[1]="filetype";
    Lresponse[2]="decay";
    Lresponse[3]="branch";
    Lresponse[4]="cut";
    vector<int> nL(nLayers);
    nL[0]=2;
    nL[1]=2;
    nL[2]=2;
    nL[3]=1;
    nL[4]=1;
    vector< vector<TString> > L;//L[layer][layerelement]
    L.resize(nLayers);
    for(int i=0; i<nLayers; i++) {
        L[i].resize(nL[i]);
    }
    L[0][0]="2011";
    L[0][1]="2015";
    L[1][0]="data";
    L[1][1]="MC";
    L[2][0]="B^{-}->(D^{0}->K^{-} #pi^{+})#mu^{-}";
    L[2][1]="B^{0}->(D^{-}->K^{+} #pi^{-} #pi^{-})#mu^{+}";
    L[3][0]="";
    L[4][0]="";
    int bL=3;//index corresponding to branch layer
    int nBranches=1;
    int cL=4;//index corresponding to cut layer
    int nCuts=1;
    vector< vector<TString> > branch;//branch[file][branches]
    branch.resize(nFiles);//finished resizing below in loop with cut vector
    vector<int> nBins(nBranches);
    nBins[0]=131;//these are for nLongTracks
    vector<int> loBin(nBranches);
    loBin[0]=0;
    vector<int> hiBin(nBranches);
    hiBin[0]=262;
    vector< vector< vector<TString> > > cut;//cut[file][branch][cuts]
    cut.resize(nFiles);
    for(int i=0; i<nFiles; i++) { //resize and fill branch and cut vectors
        cut[i].resize(nBranches);
        branch[i].resize(nBranches);
        for(int j=0; j<nBranches; j++) {
            cut[i][j].resize(nCuts);
            branch[i][j]="nLongTracks";//only one branch; same for all these files
            for(int k=0; k<nCuts; k++) {
                if(i%2==0) { //since I've listed the files alternating by decay
                    cut[i][j][k]="(B_M>3500)&&(B_M<5000)&&(D_M>1849.84)&&(D_M<1879.84)";
                } else {
                    cut[i][j][k]="(B_M>3500)&&(B_M<5000)&&(D_M>1854.61)&&(D_M<1884.61)";
                }
            }
        }
    }
    int nComparisons=1;
    vector<TString> comparison(nComparisons);
    comparison[0]="decay";
    int nhpc=1;
    int nCanvases=1;
    for(int i=0; i<nLayers; i++) {
        if(i>=(nLayers-nComparisons)) {
            nhpc*=nL[i];//now we're in the realm of comparisons
        } else {
            nCanvases *= nL[i];
        }
    }
    vector<TString> list(0);//this holds the combinations of things used to describe the files;
    //                        keep empty as the combinations are appended in get custom parameters
    //-----------------------------get custom parameters-----------------------//
    if(runmode.Contains("o")) {
        cout<<"Where should the output be stored? (make sure the directory exists; include / at end)"<<endl;
        cin>>outputlocation;
        cout<<"What should the output file be called? (include extension, e.g., 'plots.pdf')"<<endl;
        cin>>filename;
    }
void signalFit()
{
  TFile *file = TFile::Open("treeEDBR_RSGravToZZ_kMpl01_M-4000.root");
  TTree *t0;
  file->GetObject("treeDumper/EDBRCandidates",t0);

  RooRealVar numCands("numCands", "number of candidates", 0., 100.);
  RooRealVar lep("lep", "leptonic number", 0., 100.);
  RooRealVar tau21("tau21", "n-subjettiness", 0., 1.);
  RooRealVar lumiWeight("lumiWeight", "pure weight", 0., 10.);
  RooRealVar candMass("candMass","M_{ZZ}", 100.,7100., "GeV");
  const char* elLo = "numCands>0 && lep<12 && tau21<0.5";
  const char* muLo = "numCands>0 && lep>12 && tau21<0.5";
  const char* elHi = "numCands>0 && lep<12 && tau21>0.5 && tau21<0.75";
  const char* muHi = "numCands>0 && lep>12 && tau21>0.5 && tau21<0.75";
  // The muon and electron channels are considered separately  
  RooDataSet elLods("elLods","elLods",RooArgSet(candMass,numCands,lep,tau21,lumiWeight),RooFit::Cut(elLo),RooFit::WeightVar(lumiWeight),RooFit::Import(*t0));
  RooDataSet muLods("muLods","muLods",RooArgSet(candMass,numCands,lep,tau21,lumiWeight),RooFit::Cut(muLo),RooFit::WeightVar(lumiWeight),RooFit::Import(*t0));
  RooDataSet elHids("elHids","elHids",RooArgSet(candMass,numCands,lep,tau21,lumiWeight),RooFit::Cut(elHi),RooFit::WeightVar(lumiWeight),RooFit::Import(*t0));
  RooDataSet muHids("muHids","muHids",RooArgSet(candMass,numCands,lep,tau21,lumiWeight),RooFit::Cut(muHi),RooFit::WeightVar(lumiWeight),RooFit::Import(*t0));
  RooRealVar mean("mean","mean of the Crystal Ball",4000.,100.,7100.);
  RooRealVar sigma("sigma","Crystal Ball sigma",100.,10.,1000.);
  RooRealVar alphaL("alphaL","alpha left",  1.,  0.,10.);
  RooRealVar alphaR("alphaR","alpha right",-1.,-10., 0.);
  RooRealVar nL("nL","n left", 3.,0.,10.);
  RooRealVar nR("nR","n right",3.,0.,10.);
  // Build Crystal Ball for the left tail
  RooCBShape cbL("cbL","Left Crystal Ball PDF",candMass,mean,sigma,alphaL,nL);
  // Build Crystal Ball for the right tail
  RooCBShape cbR("cbR","Right Crystal Ball PDF",candMass,mean,sigma,alphaR,nR);
  // Add the previous PDFs to build the Double Crystall Ball
  RooRealVar frac("frac","fraction",0.5);
  RooAddPdf cbLR("cbLR","Double Crystall Ball",RooArgList(cbL,cbR),frac);

  TCanvas *c1 = new TCanvas("c1","c1", 1200, 900);
  c1->Divide(2,2,0.00001,0.01);
  c1->cd(1);
  RooPlot* frame1 = candMass.frame(RooFit::Title("CMS Preliminary     #sqrt{s} = 13 TeV     #int L dt = 3 fb^{-1}") );
  muLods.plotOn(frame1); 
  cbLR.fitTo(muLods,RooFit::Range(2500.,5500.));
  cbLR.plotOn(frame1,RooFit::LineColor(kRed)); 
  gPad->SetGridx();
  gPad->SetGridy();
  frame1->Draw();
  frame1->GetYaxis()->SetTitle();
  frame1->GetXaxis()->SetTitleSize(0.045);
  frame1->GetXaxis()->SetLabelSize(0.045);
  TLegend *leg1 = new TLegend(0.13,0.7,0.4,0.88);
  leg1->AddEntry("h_muLods","MC signal","ep");
  leg1->AddEntry("cbLR_Norm[candMass]_Range[fit_nll_cbLR_muLods]_NormRange[fit_nll_cbLR_muLods]","Fit model","l");
  leg1->Draw(); 
  TPaveText *pt1 = new TPaveText(0.63,0.61,0.88,0.88,"NDC");
  pt1->SetFillColor(0);
  pt1->SetBorderSize(1);
  pt1->SetTextFont(42);
  pt1->AddText("muon channel");
  pt1->AddText("#tau_{21} < 0.5");
  pt1->AddText(Form("#chi^{2} / ndf  :  %.2f",frame1->chiSquare(6)));
  pt1->Draw();

  c1->cd(2);
  RooPlot* frame2 = candMass.frame(RooFit::Title("CMS Preliminary     #sqrt{s} = 13 TeV     #int L dt = 3 fb^{-1}") );
  elLods.plotOn(frame2); 
  cbLR.fitTo(elLods,RooFit::Range(2500.,5500.));
  cbLR.plotOn(frame2,RooFit::LineColor(kRed)); 
  gPad->SetGridx();
  gPad->SetGridy();
  frame2->Draw();
  frame2->GetYaxis()->SetTitle();
  frame2->GetXaxis()->SetTitleSize(0.045);
  frame2->GetXaxis()->SetLabelSize(0.045);
  TLegend *leg2 = new TLegend(0.13,0.7,0.4,0.88);
  leg2->AddEntry("h_elLods","MC signal","ep");
  leg2->AddEntry("cbLR_Norm[candMass]_Range[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods]_NormRange[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods]","Fit model","l");
  leg2->Draw(); 
  TPaveText *pt2 = new TPaveText(0.6,0.61,0.88,0.88,"NDC");
  pt2->SetFillColor(0);
  pt2->SetBorderSize(1);
  pt2->SetTextFont(42);
  pt2->AddText("electron channel");
  pt2->AddText("#tau_{21} < 0.5");
  pt2->AddText(Form("#chi^{2} / ndf  :  %.2f",frame2->chiSquare(6)));
  pt2->Draw();

  c1->cd(3);
  RooPlot* frame3 = candMass.frame(RooFit::Title("CMS Preliminary     #sqrt{s} = 13 TeV     #int L dt = 3 fb^{-1}") );
  muHids.plotOn(frame3); 
  cbLR.fitTo(muHids,RooFit::Range(2500.,5500.));
  cbLR.plotOn(frame3,RooFit::LineColor(kRed)); 
  gPad->SetGridx();
  gPad->SetGridy();
  frame3->Draw();
  frame3->GetYaxis()->SetTitle();
  frame3->GetXaxis()->SetTitleSize(0.045);
  frame3->GetXaxis()->SetLabelSize(0.045);
  TLegend *leg3 = new TLegend(0.13,0.7,0.4,0.88);
  leg3->AddEntry("h_muHids","MC signal","ep");
  leg3->AddEntry("cbLR_Norm[candMass]_Range[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods,fit_nll_cbLR_muHids]_NormRange[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods,fit_nll_cbLR_muHids]","Fit model","l");
  leg3->Draw(); 
  TPaveText *pt3 = new TPaveText(0.63,0.61,0.88,0.88,"NDC");
  pt3->SetFillColor(0);
  pt3->SetBorderSize(1);
  pt3->SetTextFont(42);
  pt3->AddText("muon channel");
  pt3->AddText("0.5 < #tau_{21} < 0.75");
  pt3->AddText(Form("#chi^{2} / ndf  :  %.2f",frame3->chiSquare(6)));
  pt3->Draw();

  c1->cd(4);
  RooPlot* frame4 = candMass.frame(RooFit::Title("CMS Preliminary     #sqrt{s} = 13 TeV     #int L dt = 3 fb^{-1}") );
  elHids.plotOn(frame4); 
  cbLR.fitTo(elHids,RooFit::Range(2500.,5500.));
  cbLR.plotOn(frame4,RooFit::LineColor(kRed)); 
  gPad->SetGridx();
  gPad->SetGridy();
  frame4->Draw();
  frame4->GetYaxis()->SetTitle();
  frame4->GetXaxis()->SetTitleSize(0.045);
  frame4->GetXaxis()->SetLabelSize(0.045);
  TLegend *leg4 = new TLegend(0.13,0.7,0.4,0.88);
  leg4->AddEntry("h_elHids","MC signal","ep");
  leg4->AddEntry("cbLR_Norm[candMass]_Range[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods,fit_nll_cbLR_muHids,fit_nll_cbLR_elHids]_NormRange[fit_nll_cbLR_muLods,fit_nll_cbLR_elLods,fit_nll_cbLR_muHids,fit_nll_cbLR_elHids]","Fit model","l");
  leg4->Draw(); 
  TPaveText *pt4 = new TPaveText(0.6,0.61,0.88,0.88,"NDC");
  pt4->SetFillColor(0);
  pt4->SetBorderSize(1);
  pt4->SetTextFont(42);
  pt4->AddText("electron channel");
  pt4->AddText("0.5 < #tau_{21} < 0.75");
  pt4->AddText(Form("#chi^{2} / ndf  :  %.2f",frame4->chiSquare(6)));
  pt4->Draw();

}