int main (int argc, char *argv[])
{
printBoxedMessage("Starting plot generation");
// ####################
// ## Init tools ##
// ####################
string signalCategory = "T2bw-025";
// Create a sonic Screwdriver
SonicScrewdriver screwdriver;
// ##########################
// ## Create Variables ##
// ##########################
screwdriver.AddVariable("METoverSqrtHT", "MET / #sqrt{H_{T}}", "", 32,0,32, &(myEvent.METoverSqrtHT), "");
screwdriver.AddVariable("MET", "MET", "GeV", 15,50,500, &(myEvent.MET), "logY=true");
screwdriver.AddVariable("MT", "MT", "GeV", 20,0,400, &(myEvent.MT), "logY=true");
screwdriver.AddVariable("leadingBPt", "p_{T}(leading b jet)", "GeV", 20,0,200, &(myEvent.leadingBPt), "logY=true");
screwdriver.AddVariable("mStop", "m_{#tilde{t}}", "GeV", 28,112.5,812.5, &(myEvent.mStop), "");
screwdriver.AddVariable("mNeutralino", "m_{#chi^{0}}", "GeV", 16,-12.5,387.5, &(myEvent.mNeutralino), "noOverflowInLastBin");
// #########################################################
// ## Create ProcessClasses (and associated datasets) ##
// #########################################################
screwdriver.AddProcessClass("1ltop", "1l top", "background",kRed-7);
#ifdef USING_TTBAR_POWHEG
screwdriver.AddDataset("ttbar_powheg", "1ltop", 0, 0);
#endif
#ifdef USING_TTBAR_MADGRAPH
screwdriver.AddDataset("ttbar_madgraph_1l", "1ltop", 0, 0);
#endif
screwdriver.AddDataset("singleTop_st", "1ltop", 0, 0);
screwdriver.AddProcessClass("ttbar_2l", "t#bar{t} #rightarrow l^{+}l^{-}", "background",kCyan-3);
#ifdef USING_TTBAR_MADGRAPH
screwdriver.AddDataset("ttbar_madgraph_2l", "ttbar_2l", 0, 0);
#endif
screwdriver.AddProcessClass("W+jets", "W+jets", "background", kOrange-2);
screwdriver.AddDataset("W+jets", "W+jets", 0, 0);
screwdriver.AddProcessClass("rare", "rare", "background", kMagenta-5);
screwdriver.AddDataset("rare", "rare", 0, 0);
screwdriver.AddProcessClass(signalCategory, signalCategory, "signal", kViolet-1);
screwdriver.AddDataset(signalCategory, signalCategory, 0, 0);
screwdriver.AddProcessClass("signal_250_100", signalCategory+" (250/100)", "signal",COLORPLOT_BLUE );
screwdriver.AddProcessClass("signal_450_100", signalCategory+" (450/100)", "signal",COLORPLOT_GREEN2 );
screwdriver.AddProcessClass("signal_400_175", signalCategory+" (400/175)", "signal",COLORPLOT_GREEN2 );
screwdriver.AddProcessClass("signal_650_100", signalCategory+" (650/100)", "signal",COLORPLOT_GREEN );
// ##########################
// ## Create Regions ##
// ##########################
screwdriver.AddRegion("presel", "Preselection", &goesInPreselectionMTtail);
/*
screwdriver.AddRegion("veryOffShell_loose", "Cut-and-count;Very off-shell (loose)", &Selector_veryOffShell_loose);
screwdriver.AddRegion("offShell_loose", "Cut-and-count;Off-shell (loose)", &Selector_offShell_loose );
screwdriver.AddRegion("lowDeltaM_tight", "Cut-and-count;Low #DeltaM (tight)", &Selector_lowDeltaM_tight );
screwdriver.AddRegion("highDeltaM", "Cut-and-count;High #DeltaM", &Selector_highDeltaM );
*/
screwdriver.AddRegion("offshell", "Cut-and-count;Off-shell", &Selector_offShell);
screwdriver.AddRegion("lowMasses", "Cut-and-count;Low masses", &Selector_lowMasses);
screwdriver.AddRegion("highMasses", "Cut-and-count;High masses", &Selector_highMasses);
// ##########################
// ## Create Channels ##
// ##########################
screwdriver.AddChannel("singleLepton", "e/#mu-channels", &goesInSingleLeptonChannel);
// ########################################
// ## Create histograms and ##
// ## schedule type of plots to produce ##
// ########################################
screwdriver.SetLumi(20000);
// Create histograms
screwdriver.Create1DHistos();
screwdriver.Add2DHisto("mStop","mNeutralino");
screwdriver.SetGlobalBoolOption ("1DSuperimposed", "includeSignal", true );
screwdriver.SetGlobalStringOption("1DStack", "includeSignal", "stack");
screwdriver.SetGlobalFloatOption ("1DStack", "factorSignal", 1.0 );
screwdriver.SetGlobalStringOption("DataMCComparison", "includeSignal", "stack");
screwdriver.SetGlobalFloatOption ("DataMCComparison", "factorSignal", 1.0 );
// Schedule plots
screwdriver.SchedulePlots("1DSuperimposed");
screwdriver.SchedulePlots("1DStack");
screwdriver.SchedulePlots("2D");
screwdriver.SchedulePlots("2DSuperimposed");
// Config plots
screwdriver.SetGlobalStringOption("Plot", "infoTopRight", "CMS Internal");
screwdriver.SetGlobalStringOption("Plot", "infoTopLeft", "#sqrt{s} = 8 TeV, L = 20 fb^{-1}");
screwdriver.SetGlobalBoolOption("Plot", "exportPdf", true);
screwdriver.SetGlobalBoolOption("Plot", "exportEps", false);
screwdriver.SetGlobalBoolOption("Plot", "exportPng", false);
// ########################################
// ## Run over the datasets ##
// ########################################
vector<string> datasetsList;
screwdriver.GetDatasetList(&datasetsList);
cout << " > Reading datasets... " << endl;
cout << endl;
for (unsigned int d = 0 ; d < datasetsList.size() ; d++)
{
string currentDataset = datasetsList[d];
string currentProcessClass = screwdriver.GetProcessClass(currentDataset);
// Open the tree
TFile f((string(FOLDER_BABYTUPLES)+currentDataset+".root").c_str());
TTree* theTree = (TTree*) f.Get("babyTuple");
intermediatePointers pointers;
InitializeBranchesForReading(theTree,&myEvent,&pointers);
sampleName = currentDataset;
sampleType = screwdriver.GetProcessClassType(currentProcessClass);
if (currentDataset == signalCategory)
{
theTree->SetBranchAddress("mStop", &(myEvent.mStop));
theTree->SetBranchAddress("mNeutralino", &(myEvent.mNeutralino));
}
else
{
myEvent.mStop = -1;
myEvent.mNeutralino = -1;
}
// ########################################
// ## Run over the events ##
// ########################################
int nEntries = theTree->GetEntries();
for (int i = 0 ; i < nEntries ; i++)
//for (int i = 0 ; i < min(200000, (int) theTree->GetEntries()); i++)
{
if (i % (theTree->GetEntries() / 50) == 0)
printProgressBar(i,nEntries,currentDataset);
// Get the i-th entry
ReadEvent(theTree,i,&pointers,&myEvent);
// Split 1-lepton ttbar and 2-lepton ttbar
string currentProcessClass_ = currentProcessClass;
if ((currentDataset == "ttbar_powheg") && (myEvent.numberOfGenLepton == 2))
currentProcessClass_ = "ttbar_2l";
screwdriver.AutoFillProcessClass(currentProcessClass_,getWeight());
if ((myEvent.mStop == 250) && (myEvent.mNeutralino == 100))
screwdriver.AutoFillProcessClass("signal_250_100",getWeight());
if ((myEvent.mStop == 450) && (myEvent.mNeutralino == 100))
screwdriver.AutoFillProcessClass("signal_450_100",getWeight());
if ((myEvent.mStop == 400) && (myEvent.mNeutralino == 175))
screwdriver.AutoFillProcessClass("signal_400_175",getWeight());
if ((myEvent.mStop == 650) && (myEvent.mNeutralino == 100))
screwdriver.AutoFillProcessClass("signal_650_100",getWeight());
}
printProgressBar(nEntries,nEntries,currentDataset);
cout << endl;
f.Close();
}
// ###################################
// ## Make plots and write them ##
// ###################################
cout << endl;
cout << " > Making plots..." << endl;
screwdriver.MakePlots();
cout << " > Saving plots..." << endl;
screwdriver.WritePlots("../plots/cutAndCount_performances/"+signalCategory+"/");
printBoxedMessage("Plot generation completed");
// #############################
// ## Post-plotting tests ##
// #############################
printBoxedMessage("Now computing misc tests ... ");
/*
vector<string> cutAndCountRegions =
{
"veryOffShell_loose",
"offShell_loose",
"lowDeltaM_tight",
"highDeltaM"
};
float SF_1ltop_and_Wjets = 2;
float SF_allOthers = 1.3;
vector<float> globalBackgroundUncertainty =
{
0.2,
0.2,
0.2,
0.4
};
*/
vector<string> cutAndCountRegions =
{
"presel",
"offshell",
"lowMasses",
"highMasses"
};
float SF_1ltop_and_Wjets = 2;
float SF_allOthers = 1.3;
vector<float> globalBackgroundUncertainty =
{
0.2,
0.2,
0.2
};
TableBackgroundSignal(&screwdriver,cutAndCountRegions,"singleLepton").Print();
TableBackgroundSignal(&screwdriver,cutAndCountRegions,"singleLepton").PrintLatex();
// ##########################
// ## Compute FOM maps ##
// ##########################
vector<TH2F*> signalMaps;
vector<TH2F*> FOMdiscoveryMaps;
vector<TH2F*> FOMexclusionMaps;
vector<TH2F*> efficiencies;
int nBinsX = -1;
int nBinsY = -1;
TH2F* signalMapPresel = screwdriver.get2DHistoClone("mStop","mNeutralino",signalCategory,"presel","singleLepton");
TH2F* backgroundPresel = screwdriver.get2DCompositeHistoClone("mStop","mNeutralino","2DSumBackground","presel","singleLepton","");
if (nBinsX == -1) nBinsX = signalMapPresel->GetNbinsX();
if (nBinsY == -1) nBinsY = signalMapPresel->GetNbinsY();
// Store background eff in (mStop,mLSP) = (200,300)
int backgroundBin = signalMapPresel->FindBin(200,300);
float backgroundYieldPresel = backgroundPresel->Integral(0,nBinsX+1,0,nBinsY+1);
for (unsigned int i = 0 ; i < cutAndCountRegions.size() ; i++)
{
signalMaps.push_back(screwdriver.get2DHistoClone("mStop","mNeutralino",signalCategory,cutAndCountRegions[i],"singleLepton"));
signalMaps[i]->SetName((string("signalMap_")+cutAndCountRegions[i]).c_str());
float B = screwdriver.GetYieldAndError("1ltop", cutAndCountRegions[i],"singleLepton").value() * SF_1ltop_and_Wjets
+ screwdriver.GetYieldAndError("ttbar_2l", cutAndCountRegions[i],"singleLepton").value() * SF_allOthers
+ screwdriver.GetYieldAndError("W+jets", cutAndCountRegions[i],"singleLepton").value() * SF_1ltop_and_Wjets
+ screwdriver.GetYieldAndError("rare", cutAndCountRegions[i],"singleLepton").value() * SF_allOthers;
// Apply scale factor from background prediction
float f_B = globalBackgroundUncertainty[i];
//float f_B = 0.15;
if (B < 1.0) B = 1.0;
efficiencies.push_back((TH2F*) signalMaps[i]->Clone());
efficiencies[i]->SetName((string("eff_")+cutAndCountRegions[i]).c_str());
efficiencies[i]->Divide(signalMapPresel);
efficiencies[i]->SetBinContent(backgroundBin,B/backgroundYieldPresel);
FOMdiscoveryMaps.push_back((TH2F*) signalMaps[i]->Clone());
FOMdiscoveryMaps[i]->SetName((string("FOMdisco_")+cutAndCountRegions[i]).c_str());
FOMexclusionMaps.push_back((TH2F*) signalMaps[i]->Clone());
FOMexclusionMaps[i]->SetName((string("FOMexclu_")+cutAndCountRegions[i]).c_str());
for (int x = 1 ; x <= nBinsX ; x++)
for (int y = 1 ; y <= nBinsY ; y++)
{
float S = signalMaps[i]->GetBinContent(x,y);
float FOMdiscovery = figureOfMerit(S,B,"discovery",false,f_B);
FOMdiscoveryMaps[i]->SetBinContent(x,y,FOMdiscovery);
float FOMexclusion = figureOfMerit(S,B,"exclusion",false,f_B);
FOMexclusionMaps[i]->SetBinContent(x,y,FOMexclusion);
}
}
// ################################
// ## Compute "best" FOM map ##
// ################################
TH2F* bestDiscoFOMMap = (TH2F*) signalMaps[0]->Clone(); bestDiscoFOMMap->SetName("bestDiscoFOM");
TH2F* bestDiscoSetMap = (TH2F*) signalMaps[0]->Clone(); bestDiscoSetMap->SetName("bestDiscoSet");
TH2F* bestDiscoSigEff = (TH2F*) signalMaps[0]->Clone(); bestDiscoSigEff->SetName("bestDiscoSigEff");
TH2F* bestDiscoBkgEff = (TH2F*) signalMaps[0]->Clone(); bestDiscoBkgEff->SetName("bestDiscoBkgEff");
TH2F* bestExcluFOMMap = (TH2F*) signalMaps[0]->Clone(); bestExcluFOMMap->SetName("bestExcluFOM");
TH2F* bestExcluSetMap = (TH2F*) signalMaps[0]->Clone(); bestExcluSetMap->SetName("bestExcluSet");
TH2F* bestExcluSigEff = (TH2F*) signalMaps[0]->Clone(); bestExcluSigEff->SetName("bestExcluSigEff");
TH2F* bestExcluBkgEff = (TH2F*) signalMaps[0]->Clone(); bestExcluBkgEff->SetName("bestExcluBkgEff");
for (int x = 1 ; x <= nBinsX ; x++)
for (int y = 1 ; y <= nBinsY ; y++)
{
float bestDiscoFOM = -1.0;
int bestDiscoSet = 0;
float bestDiscoSigEff_ = -1.0;
float bestDiscoBkgEff_ = -1.0;
for (unsigned int i = 0 ; i < cutAndCountRegions.size() ; i++)
{
float DiscoFOM = FOMdiscoveryMaps[i]->GetBinContent(x,y);
if (bestDiscoFOM < DiscoFOM)
{
bestDiscoFOM = DiscoFOM;
if (bestDiscoFOM > 0) bestDiscoSet = i+1;
bestDiscoSigEff_ = efficiencies[i]->GetBinContent(x,y);
bestDiscoBkgEff_ = efficiencies[i]->GetBinContent(backgroundBin);
}
}
bestDiscoFOMMap->SetBinContent(x,y,bestDiscoFOM);
bestDiscoSetMap->SetBinContent(x,y,bestDiscoSet);
bestDiscoSigEff->SetBinContent(x,y,bestDiscoSigEff_);
bestDiscoBkgEff->SetBinContent(x,y,bestDiscoBkgEff_);
float bestExcluFOM = -1.0;
int bestExcluSet = 0;
float bestExcluSigEff_ = -1.0;
float bestExcluBkgEff_ = -1.0;
for (unsigned int i = 0 ; i < cutAndCountRegions.size() ; i++)
{
float ExcluFOM = FOMexclusionMaps[i]->GetBinContent(x,y);
if (bestExcluFOM < ExcluFOM)
{
bestExcluFOM = ExcluFOM;
if (bestExcluFOM > 0) bestExcluSet = i+1;
bestExcluSigEff_ = efficiencies[i]->GetBinContent(x,y);
bestExcluBkgEff_ = efficiencies[i]->GetBinContent(backgroundBin);
}
}
bestExcluFOMMap->SetBinContent(x,y,bestExcluFOM);
bestExcluSetMap->SetBinContent(x,y,bestExcluSet);
bestExcluSigEff->SetBinContent(x,y,bestExcluSigEff_);
bestExcluBkgEff->SetBinContent(x,y,bestExcluBkgEff_);
}
// #########################
// ## Save those maps ##
// #########################
float lineOffset = 0.0;
string label;
if (signalCategory == "T2tt" ) { lineOffset = 172; label = "T2tt;"; }
if (signalCategory == "T2bw-025") { lineOffset = 320; label = "T2bw (x = 0.25);"; }
if (signalCategory == "T2bw-050") { lineOffset = 160; label = "T2bw (x = 0.50);"; }
if (signalCategory == "T2bw-075") { lineOffset = 105; label = "T2bw (x = 0.75);"; }
TFile fOutput(("../plots/cutAndCount_performances/"+signalCategory+"/custom.root").c_str(),"RECREATE");
string pathExport = "../plots/cutAndCount_performances/"+signalCategory+"/";
gStyle->SetPaintTextFormat("4.0f");
formatAndWriteMapPlot(&screwdriver,bestDiscoSetMap,bestDiscoSetMap->GetName(),label+"Best set of cuts;(for discovery)",pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestExcluSetMap,bestExcluSetMap->GetName(),label+"Best set of cuts;(for exclusion)",pathExport,lineOffset);
gStyle->SetPaintTextFormat("4.1f");
for (unsigned int i = 0 ; i < cutAndCountRegions.size() ; i++)
{
FOMdiscoveryMaps[i]->SetMaximum(5.0);
formatAndWriteMapPlot(&screwdriver,FOMdiscoveryMaps[i],FOMdiscoveryMaps[i]->GetName(),string("Discovery FOM for ")+cutAndCountRegions[i], pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver, efficiencies[i], efficiencies[i]->GetName(),string("Efficiencies for " )+cutAndCountRegions[i], pathExport,lineOffset);
}
bestDiscoFOMMap->SetMaximum(5.0);
bestExcluFOMMap->SetMaximum(5.0);
formatAndWriteMapPlot(&screwdriver,bestDiscoFOMMap,bestDiscoFOMMap->GetName(),label+"Best FOM;(for discovery)" ,pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestDiscoSigEff,bestDiscoSigEff->GetName(),label+"Best signal efficiency;(for discovery)",pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestDiscoBkgEff,bestDiscoBkgEff->GetName(),label+"Best backgr efficiency;(for discovery)",pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestExcluFOMMap,bestExcluFOMMap->GetName(),label+"Best FOM;(for exclusion)" ,pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestExcluSigEff,bestExcluSigEff->GetName(),label+"Best signal efficiency;(for exclusion)",pathExport,lineOffset);
formatAndWriteMapPlot(&screwdriver,bestExcluBkgEff,bestExcluBkgEff->GetName(),label+"Best backgr efficiency;(for exclusion)",pathExport,lineOffset);
fOutput.Close();
printBoxedMessage("Program done.");
return (0);
}
int main (int argc, char *argv[])
{
printBoxedMessage("Starting plot generation");
// ####################
// ## Init tools ##
// ####################
// Create a sonic Screwdriver
SonicScrewdriver s;
// ##########################
// ## Create Variables ##
// ##########################
s.AddVariable("MET", "MET", "GeV", 16,50,530, &(myEvent.MET), "logY");
s.AddVariable("METPhi", "METPhi", "GeV", 16,50,530, &(myEvent.METPhi), "");
s.AddVariable("MT", "M_{T}", "GeV", 40,0,400, &(myEvent.MT), "");
/*
s.AddVariable("MTpeak", "M_{T}", "GeV", 20,0,100, &(myEvent.MT), "noOverflowInLastBin");
s.AddVariable("MTtail", "M_{T}", "GeV", 30,100,400, &(myEvent.MT), "logY,noUnderflowInFirstBin");
s.AddVariable("deltaPhiMETJets","#Delta#Phi(MET,j_{1,2})", "rad", 16,0,3.2, &(myEvent.deltaPhiMETJets), "");
s.AddVariable("MT2W", "M_{T2}^{W}", "GeV", 20,0,500, &(myEvent.MT2W), "");
s.AddVariable("HTratio", "H_{T}^{ratio}", "", 20,0,1.2, &(myEvent.HTRatio), "");
s.AddVariable("HadronicChi2", "Hadronic #chi^{2}", "", 20,0,20, &(myEvent.hadronicChi2), "logY");
s.AddVariable("leadingBPt", "p_{T}(leading b-jet)", "GeV", 20,0,400, &(myEvent.leadingBPt), "");
s.AddVariable("leadingJetPt", "p_{T}(leading jet)", "GeV", 20,0,600, &(myEvent.leadingJetPt), "");
s.AddVariable("leptonPt", "p_{T}(lepton)", "GeV", 28,20,300, &(myEvent.leadingLeptonPt), "");
s.AddVariable("Mlb", "M'(lb)", "GeV", 20,0,500, &(myEvent.Mlb), "");
s.AddVariable("Mlb_hemi", "M(lb)", "GeV", 20,0,500, &(myEvent.Mlb_hemi), "");
s.AddVariable("M3b", "M3b", "GeV", 20,0,1000, &(myEvent.M3b), "");
s.AddVariable("deltaRLeptonB", "#DeltaR(l,leading b)", "", 20,0,5, &(myEvent.deltaRLeptonLeadingB), "");
s.AddVariable("HTLeptonPtMET", "HT + MET + p_{T}(lepton)","GeV", 20,100,2100, &(myEvent.HTPlusLeptonPtPlusMET),"");
*/
s.AddVariable("HT", "H_{T}", "", 46,120,1500, &(myEvent.HT), "");
s.AddVariable("METoverSqrtHT", "MET / #sqrt{H_{T}}", "", 32,0,32, &(myEvent.METoverSqrtHT), "");
float leadingNonBPtN4;
s.AddVariable("leadingNonBPt", "p_{T}(leading nonb-jet)", "GeV", 20,0,500, &(leadingNonBPtN4), "");
float leadingNonBPtN5;
s.AddVariable("leadingNonBPtN5", "p_{T}(leading nonb-jet) (Njet>4)", "GeV", 20,0,500, &(leadingNonBPtN5), "noUnderflowInFirstBin");
int nJets, nBtag;
s.AddVariable("nJets", "Number of selected jets", "", 11,0,10, &(nJets), "");
s.AddVariable("nBtag", "Number of selected b-tagged jets", "", 5, 0,4, &(nBtag), "");
s.AddVariable("numberOfPrimaryVertices", "Number of primary vertices", "", 41, 0,40, &(myEvent.numberOfPrimaryVertices), "");
float METPtReso;
float METdPhiReso;
s.AddVariable("METPtReso", "MET p_{T} rel. resolution", "GeV", 50, 0,5, &(METPtReso), "");
s.AddVariable("METdPhiReso", "MET #Delta #Phi rel. resolution", "rad", 50, 0,2, &(METdPhiReso), "");
// #########################################################
// ## Create ProcessClasses (and associated datasets) ##
// #########################################################
s.AddProcessClass("1ltop", "1l top", "background",kRed-7);
s.AddDataset("ttbar_powheg", "1ltop", 0, 0);
s.AddDataset("singleTop_st", "1ltop", 0, 0);
s.AddProcessClass("ttbar_2l", "t#bar{t} #rightarrow l^{+}l^{-}", "background",kCyan-3);
s.AddProcessClass("W+jets", "W+jets", "background",kOrange-2);
s.AddDataset("W+jets", "W+jets", 0, 0);
s.AddProcessClass("rare", "rare", "background",kMagenta-5);
s.AddDataset("rare", "rare", 0, 0);
/*
s.AddProcessClass("T2tt", "T2tt (450, 50)", "signal", COLORPLOT_GREEN );
s.AddDataset("T2tt", "T2tt", 0, 0);
s.AddProcessClass("T2bw", "T2bw, x = 0.50 (450, 50)", "signal", COLORPLOT_MAGENTA );
s.AddDataset("T2bw-050", "T2bw", 0, 0);
*/
/*
s.AddProcessClass("data", "data", "data",COLORPLOT_BLACK);
s.AddDataset("SingleElec", "data", 0, 0);
s.AddDataset("SingleMuon", "data", 0, 0);
s.AddDataset("DoubleElec", "data", 0, 0);
s.AddDataset("DoubleMuon", "data", 0, 0);
s.AddDataset("MuEl", "data", 0, 0);
*/
// ##########################
// ## Create Regions ##
// ##########################
s.AddRegion("preselection", "Preselection", &goesInPreselection );
s.AddRegion("preselection_MTtail", "Preselection;M_{T} tail", &goesInPreselectionMTtail );
// ##########################
// ## Create Channels ##
// ##########################
s.AddChannel("singleLepton", "e/#mu-channels", &goesInSingleLeptonChannel);
// ########################################
// ## Create histograms and ##
// ## schedule type of plots to produce ##
// ########################################
// Create histograms
s.Create1DHistos();
s.Add2DHisto("MET","HT");
s.Add2DHisto("MT","METPtReso");
s.Add2DHisto("MT","METdPhiReso");
s.Add2DHisto("MT","numberOfPrimaryVertices");
s.Add2DHisto("numberOfPrimaryVertices","METdPhiReso");
s.Add2DHisto("numberOfPrimaryVertices","METPtReso");
// Schedule plots
//s.SchedulePlots("1DDataMCComparison");
s.SchedulePlots("1DStack");
s.SchedulePlots("1DSuperimposed");
s.SchedulePlots("2D");
s.SchedulePlots("1DFrom2DProjection","varX=MT,varY=METPtReso,projectionType=mean,labelY=MET Pt rel. resolution");
s.SchedulePlots("1DFrom2DProjection","varX=MT,varY=METdPhiReso,projectionType=mean,labelY=MET #Delta #Phi rel. resolution");
s.SchedulePlots("1DFrom2DProjection","varX=MT,varY=numberOfPrimaryVertices,projectionType=mean,labelY=Number of vertices");
s.SchedulePlots("1DFrom2DProjection","varX=numberOfPrimaryVertices,varY=METdPhiReso,projectionType=mean,labelY=MET #Delta #Phi rel. resolution");
s.SchedulePlots("1DFrom2DProjection","varX=numberOfPrimaryVertices,varY=METPtReso,projectionType=mean,labelY=MET Pt rel. resolution");
// Config plots
s.SetGlobalBoolOption ("1DSuperimposed", "includeSignal", true );
s.SetGlobalStringOption("DataMCComparison", "includeSignal", "superimposed");
s.SetGlobalFloatOption ("DataMCComparison", "factorSignal", 100.0 );
s.SetGlobalStringOption("1DStack", "includeSignal", "superimposed");
s.SetGlobalFloatOption ("1DStack", "factorSignal", 100.0 );
s.SetGlobalFloatOption ("DataMCRatio", "min", 0.0 );
s.SetGlobalFloatOption ("DataMCRatio", "max", 2.0 );
s.SetGlobalStringOption("Plot", "infoTopRight", "CMS Preliminary");
s.SetGlobalStringOption("Plot", "infoTopLeft", "#sqrt{s} = 8 TeV, L = 19.5 fb^{-1}");
s.SetGlobalBoolOption ("Plot", "exportPdf", true);
s.SetGlobalBoolOption ("Plot", "exportEps", false);
s.SetGlobalBoolOption ("Plot", "exportPng", false);
// ########################################
// ## Run over the datasets ##
// ########################################
vector<string> datasetsList;
s.GetDatasetList(&datasetsList);
cout << " > Reading datasets... " << endl;
cout << endl;
for (unsigned int d = 0 ; d < datasetsList.size() ; d++)
{
string currentDataset = datasetsList[d];
string currentProcessClass = s.GetProcessClass(currentDataset);
sampleName = currentDataset;
sampleType = s.GetProcessClassType(currentProcessClass);
// Open the tree
TFile f((string(FOLDER_BABYTUPLES)+currentDataset+".root").c_str());
TTree* theTree = (TTree*) f.Get("babyTuple");
intermediatePointers pointers;
InitializeBranchesForReading(theTree,&myEvent,&pointers);
if (sampleType == "signal")
{
theTree->SetBranchAddress("mStop", &(myEvent.mStop));
theTree->SetBranchAddress("mNeutralino", &(myEvent.mNeutralino));
}
else
{
myEvent.mStop = -1;
myEvent.mNeutralino = -1;
}
// ########################################
// ## Run over the events ##
// ########################################
bool ttbarDatasetToBeSplitted = false;
if (findSubstring(currentDataset,"ttbar"))
ttbarDatasetToBeSplitted = true;
int nEntries = theTree->GetEntries();
for (int i = 0 ; i < nEntries ; i++)
{
//if (i > float(nEntries) * 0.1) break;
if (i % (nEntries / 50) == 0) printProgressBar(i,nEntries,currentDataset);
// Get the i-th entry
ReadEvent(theTree,i,&pointers,&myEvent);
nJets = myEvent.nJets;
nBtag = myEvent.nBTag;
float weight = getWeight();
// Split 1-lepton ttbar and 2-lepton ttbar
string currentProcessClass_ = currentProcessClass;
if (ttbarDatasetToBeSplitted && (myEvent.numberOfGenLepton == 2))
currentProcessClass_ = "ttbar_2l";
#ifdef LEADING_NON_B_PT_ALREADY_COMPUTED
leadingNonBPtN4= myEvent.leadingNonBPt;
#else
leadingNonBPtN4= leadingNonBPt();
#endif
leadingNonBPtN5=-1;
if (nJets>4) leadingNonBPtN5= leadingNonBPtN4;
if ((currentProcessClass_ == "T2tt") && ((myEvent.mStop != 450) || (myEvent.mNeutralino != 50))) continue;
if ((currentProcessClass_ == "T2bw") && ((myEvent.mStop != 450) || (myEvent.mNeutralino != 50))) continue;
computeGenInfo();
METPtReso = myEvent.MET / myEvent.genMET;
METdPhiReso = abs(Phi_mpi_pi(myEvent.METPhi - myEvent.leadingLepton.Phi())) / myEvent.genMETdPhi;
s.AutoFillProcessClass(currentProcessClass_,weight);
}
printProgressBar(nEntries,nEntries,currentDataset);
cout << endl;
f.Close();
}
/*
// #############################
// ## Apply scale factors ##
// #############################
Table scaleFactors = Table("../prediction/scaleFactors/preselection.tab");
Figure SF_pre = scaleFactors.Get("value","SF_pre");
Figure SF_post = scaleFactors.Get("value","SF_post");
Figure SF_0btag = scaleFactors.Get("value","SF_0btag");
Figure SF_vetopeak = scaleFactors.Get("value","SF_vetopeak");
Figure SF_MTtail_1ltop = scaleFactors.Get("value","SF_MTtail_1ltop");
Figure SF_MTtail_Wjets = scaleFactors.Get("value","SF_MTtail_Wjets");
s.ApplyScaleFactor("ttbar_2l", "preselection_MTtail_withCorr", "singleLepton", SF_pre);
s.ApplyScaleFactor("1ltop", "preselection_MTtail_withCorr", "singleLepton", SF_post * SF_MTtail_1ltop);
s.ApplyScaleFactor("W+jets", "preselection_MTtail_withCorr", "singleLepton", SF_post * SF_MTtail_Wjets);
*/
// ###################################
// ## Make plots and write them ##
// ###################################
cout << endl;
cout << " > Making plots..." << endl;
s.MakePlots();
cout << " > Saving plots..." << endl;
s.WritePlots("./plots/METresolution/");
printBoxedMessage("Plot generation completed");
// #############################
// ## Post-plotting tests ##
// #############################
printBoxedMessage("Program done.");
return (0);
}
