void poisson(std::map< TString, std::map <unsigned int, TH1F*> > histo_, std::vector<TString> plotList_, const std::string decayChannel, TFile& outputfile, int luminosity, const unsigned int verbose, bool smear, bool useReweightedTop, double avReweight, bool useZprime, double zPrimeLumiWeight){
unsigned int kCombined=111;
if(verbose>1){
std::cout << "running function poisson" << std::endl;
std::cout << "-----------------------------" << std::endl;
std::cout << "luminosity: " << luminosity << std::endl;
std::cout << "decayChannel: " << decayChannel << std::endl;
std::cout << "smear: " << smear << std::endl;
std::cout << "useReweightedTop: " << useReweightedTop << std::endl;
std::cout << "avReweight: " << avReweight << std::endl;
std::cout << "useZprime: " << useZprime << std::endl;
std::cout << "zPrimeLumiWeight: " << zPrimeLumiWeight << std::endl;
sleep(5);
}
// upper bound for looping events: in/exclude zprime
int kLast =kSAToptW;
int kZprime=kLast+1;
if(useZprime) kLast=kZprime;
// go to output file
outputfile.cd();
// ---
// combine plot from all files
// ---
if(verbose>0) std::cout << "start to combine plots " << std::endl;
// loop plots
for(unsigned int plot=0; plot<plotList_.size(); ++plot){
if(verbose>1){
std::cout << "plot " << plotList_[plot] << std::endl;
std::cout << "considered samples: " << std::endl;
}
// indicate first plot existing in sample
bool first=true;
// loop samples
for(int sample = kSig; sample<=kLast; sample++){
// check if plot exists
if((histo_.count(plotList_[plot])>0)&&(histo_[plotList_[plot]].count(sample))){
// take care of non existing plots
if(!(useZprime&&sample==kZprime&&plotList_[plot].Contains("PUControlDistributions"))){
// do lumiweighting
if (useZprime &&sample==kZprime) histo_[plotList_[plot]][sample]->Scale(zPrimeLumiWeight);
else histo_[plotList_[plot]][sample]->Scale(lumiweight(sample, luminosity, sysNo, decayChannel));
// for reweighted ttbar: weight to keep same normalization
if(useReweightedTop&&(sample==kSig||sample==kBkg)) histo_[plotList_[plot]][sample]->Scale(1.0/avReweight);
// first subsample (should be ttbar signal)
// -> create histogram map entry
if(first){
histo_[plotList_[plot]][kCombined] = (TH1F*)(histo_[plotList_[plot]][sample]->Clone());
first=false;
}
// add other subsample
else{
histo_[plotList_[plot]][kCombined]->Add((TH1F*)(histo_[plotList_[plot]][sample]->Clone()),1.0);
}
if(verbose>1){
if(useZprime&&sample==kZprime) std::cout << "z prime, weight " << zPrimeLumiWeight << std::endl;
else if(useReweightedTop&&(sample==kSig||sample==kBkg)) std::cout << "reweighted "+sampleLabel(sample,decayChannel)+", weight " << 1.0/avReweight << std::endl;
else std::cout << sampleLabel(sample,decayChannel) << ", weight " << lumiweight(sample, luminosity, sysNo, decayChannel) << std::endl;
}
}
}
}
}
// ---
// poisson smearing
// ---
if(verbose>0) std::cout << "start to do the poisson smearing plots " << std::endl;
// loop plots
for(unsigned int plot=0; plot<plotList_.size(); ++plot){
// check if plot exists
if((histo_.count(plotList_[plot])>0)&&(histo_[plotList_[plot]].count(kCombined))){
if(verbose>1) std::cout << "plot " << plotList_[plot] << std::endl;
// do the poisson smearing
if(smear){
TRandom3 rnd(0);
for(int ibin=0; ibin<=histo_[plotList_[plot]][kCombined]->GetNbinsX()+1; ++ibin){
int evts=histo_[plotList_[plot]][kCombined]->GetBinContent(ibin);
evts=rnd.Poisson(histo_[plotList_[plot]][kCombined]->GetBinContent(ibin));
histo_[plotList_[plot]][kCombined]->SetBinContent(ibin, evts);
histo_[plotList_[plot]][kCombined]->SetBinError(ibin, sqrt(evts));
}
}
// do the saving
// use the same directory as in the input files
TString directory=getStringEntry(plotList_[plot], 1);
TString plotName=getStringEntry(plotList_[plot], 2);
if(verbose>1) std::cout << "check existence of directory " << directory << std::endl;
if((outputfile.GetDirectory(directory)!=0)&&(verbose>1)) std::cout << "already existing" << std::endl;
if(outputfile.GetDirectory(directory)==0){
if(verbose>1) std::cout << "not existing - will create it" << directory << std::endl;
outputfile.mkdir(directory);
}
outputfile.cd(directory);
if(verbose>1) std::cout << "save plot" << directory << std::endl;
// save plot
histo_[plotList_[plot]][kCombined]->Write(plotName);
}
}
}
void combineAllPlots(int sysTag, int sample, TString decayChannel, int verbose, TString inputFolderName, TString outputFolder){
// ---
// check input
// ---
// a) check if sample input is valid
if(!(sample==kQCD||sample==kDiBos||sample==kSTop||sample==kTTVjets)){
std::cout << "chosen input sample enumerator " << sample << " is invalid" << std::endl;
std::cout << "please check the list of allowed sample inputs" << std::endl;
exit(0);
}
// b) check if decayChannel input is valid
if(!(decayChannel=="electron"||decayChannel=="muon")){
std::cout << "chosen decay channel " << decayChannel << " is invalid" << std::endl;
std::cout << "has to be electron or muon" << std::endl;
exit(0);
}
// ---
// assign enumerator for subsamples
// corresponding to "sample"
// ---
// (i) collect first and last enumerator corresponding
// to enum samples in basicFunctions.h
// a) single top
int first=kSTops;
int last =kSAToptW;
// b) QCD electron channel
// (muon channel excluded above)
if(sample==kQCD){
first=kQCDEM1;
last =kQCDBCE3;
}
// c) diboson samples
else if(sample==kDiBos){
first=kWW;
last =kZZ;
}
// TTV samples
else if(sample==kTTVjets){
first=kTTGjets;
last =kTTWjets;
}
// (ii) list subsamples in vector
std::vector<int> subSamples_;
for(int subsample=first; subsample<=last; ++subsample){
if((sysTag!=sysTopScaleUp&&sysTag!=sysTopScaleDown)||(subsample!=kSAToptW&&subsample!=kSToptW)) subSamples_.push_back(subsample);
else{ // for single top scale samples Tw is splitted in 3 subsamples
if(subsample==kSToptW){
subSamples_.push_back(kSToptW1);
subSamples_.push_back(kSToptW2);
subSamples_.push_back(kSToptW3);
}
else if(subsample==kSAToptW){
subSamples_.push_back(kSAToptW1);
subSamples_.push_back(kSAToptW2);
subSamples_.push_back(kSAToptW3);
}
}
}
// ---
// input & output
// ---
// folder were subsamples can be found
TString inputFolder = groupSpace+inputFolderName;
// folder and name of the (combined) output file
TString outputFilename= (outputFolder=="" ? inputFolder : outputFolder);
outputFilename+="/"+TopFilename(sample, sysTag, std::string(decayChannel));
// ---
// container for all subsample files and
// their corresponding lumiweights
// ---
std::vector< std::pair< TFile*, double > > files_;
// loop subsamples
for(unsigned int subsample=0; subsample<subSamples_.size(); ++subsample){
// get subsample file name
TString fileName = inputFolder+"/"+TopFilename(subSamples_[subsample], sysTag, std::string(decayChannel));
//std::cout << fileName << ": " << lumiweight(subSamples_[subsample], 1, sysTag, std::string(decayChannel)) << std::endl;
// check existence & availability of file
if((fileName!="no")&&(fileName!="")){
TFile* file = TFile::Open(fileName);
if(file&&!(file->IsZombie())){
// N.B.: a luminosity of 1 pb is used, lumi normalization is done later in the main file
files_.push_back(std::make_pair(file, lumiweight(subSamples_[subsample], 1, sysTag, std::string(decayChannel))));
}
}
}
// check if all files are found
if(subSamples_.size()!=files_.size()){
std::cout << "ERROR: not all subsamples could be found" << std::endl;
exit(0);
}
// ---
// print out all information about current combination
// ---
if(verbose>0){
std::cout << "combining MC subsamples for: " << std::endl;
std::cout << " - " << sampleLabel(sample,(const std::string)decayChannel) << std::endl;
std::cout << " - " << decayChannel << " channel " << std::endl;
std::cout << " - " << "systematic variation " << sysLabel(sysTag) << std::endl;
std::cout << " - " << "input folder:" << inputFolder << std::endl;
std::cout << " - " << "input subSamples(weights without luminosity):";
for(unsigned int subsample=0; subsample<subSamples_.size(); ++subsample){
std::cout << std::endl << " " << sampleLabel(subSamples_[subsample],(const std::string)decayChannel) << " ( " << files_[subsample].second << " )";
}
std::cout << std::endl;
std::cout << " - " << " output file created: " << outputFilename << std::endl << std::endl;
// wait for 1 second
sleep(1);
}
// ---
// create output file
// ---
TFile* output_file = TFile::Open(outputFilename, "recreate");
// open input files and weight them
addDir("",files_,output_file, verbose);
// close output files
output_file->Close();
// close and delete input files
for(std::vector< std::pair< TFile*, double > >::const_iterator file=files_.begin(); file!=files_.end(); ++file){
file->first->Close();
delete file->first;
}
}
void treeComparison(double luminosity = 19712, bool save = true, int verbose=1, TString inputFolderName= "RecentAnalysisRun8TeV_doubleKinFit", TString dataFile= "/afs/naf.desy.de/group/cms/scratch/tophh/RecentAnalysisRun8TeV_doubleKinFit/elecDiffXSecData2012ABCDAll.root:/afs/naf.desy.de/group/cms/scratch/tophh/RecentAnalysisRun8TeV_doubleKinFit/muonDiffXSecData2012ABCDAll.root", const std::string decayChannel = "combined", bool withRatioPlot=true, TString test="prob")
{
// test= "prob" or "PV"
// data/MC -> MC/data
bool invert=true;
// linear fit in ratio?
bool linFit=true;
// ===================================
// Define plotting order
// ===================================
std::vector<int> samples_;
samples_.push_back(kSig);
samples_.push_back(kBkg);
samples_.push_back(kSTop);
samples_.push_back(kWjets);
samples_.push_back(kZjets);
samples_.push_back(kDiBos);
samples_.push_back(kQCD);
samples_.push_back(kData);
// ============================
// Set Root Style
// ============================
TStyle myStyle("HHStyle","HHStyle");
setHHStyle(myStyle);
myStyle.SetStripDecimals(true);
myStyle.cd();
gROOT->SetStyle("HHStyle");
gROOT->ForceStyle();
TGaxis::SetMaxDigits(2);
// user specific configuration
TString testQuantity=""; // name of separator in tree
TString treePath=""; // path of tree
// values for splitting topPt (Val0<=plot1<Val1<=plot2<Val2)
std::vector< double > Val_;
TString treeExt="";
if(test=="PV"){
testQuantity="nPV";
treePath="compositedKinematicsKinFit/tree";
Val_.push_back(0. );
Val_.push_back(8. );
Val_.push_back(13.);
Val_.push_back(17.);
Val_.push_back(22.);
Val_.push_back(50.);
treeExt="Fit";
}
if(test=="prob"){
testQuantity="chi2";
treePath="analyzeTopRecoKinematicsKinFit/tree";
Val_.push_back(0. );
Val_.push_back(1.386); // chi2<1.386 ~prob>0.50
Val_.push_back(2.1); // chi2<2.1 ~prob>0.35
Val_.push_back(3.219); // chi2<3.219 ~prob>0.20
Val_.push_back(7.824); // chi2<7.824 ~prob>0.02
Val_.push_back(99999.);
treeExt="";
}
// default configurations
unsigned int systematicVariation=sysNo;
TString ttbarMC="Madgraph";
bool scaleTtbarToMeasured=true;
// adjust luminosity and data files for combined control plots
double luminosityEl=constLumiElec;
double luminosityMu=constLumiMuon;
if(!dataFile.Contains(":")){
std::cout << "wrong input filenames, should be dataFileEl:dataFileMu, but is ";
std::cout << dataFile << std::endl;
exit(0);
}
TString dataFileEl=getStringEntry(dataFile,1 , ":");
TString dataFileMu=getStringEntry(dataFile,42, ":");
// file container
std::map<unsigned int, TFile*> files_, filesMu_, filesEl_;
// file vector storage
std::vector< std::map<unsigned int, TFile*> > fileList_;
// get analysis files
TString inputFolder="/afs/naf.desy.de/group/cms/scratch/tophh/"+inputFolderName;
if(verbose>0) std::cout << "loading files from " << inputFolder << std::endl;
if(decayChannel!="combined"){
TString dataFiletemp= decayChannel=="muon" ? dataFileMu : dataFileEl;
files_ = getStdTopAnalysisFiles(inputFolder, systematicVariation, dataFiletemp, decayChannel, ttbarMC);
fileList_.push_back(files_);
}
else{
filesMu_ = getStdTopAnalysisFiles(inputFolder, systematicVariation, dataFileMu, "muon" , ttbarMC);
filesEl_ = getStdTopAnalysisFiles(inputFolder, systematicVariation, dataFileEl, "electron", ttbarMC);
fileList_.push_back(filesMu_);
fileList_.push_back(filesEl_);
}
// topPt histogram template
if(verbose>0) std::cout << "creating temp histo" << std::endl;
TH1F* temp= (TH1F*)(((TH1F*)(fileList_.at(0)[kSig]->Get("analyzeTopRecoKinematicsKinFit/topPt"))->Clone()));
temp->Rebin(20);
temp->Reset("icms");
temp->SetTitle("");
temp->GetXaxis()->SetTitle("p_{T}^{t} #left[GeV#right]");
temp->GetYaxis()->SetTitle("Top quarks");
//axesStyle(*temp, "p_{T}^{t} #left[GeV#right]", "norm. Top quarks", 0., 0.15);
temp->GetXaxis()->SetRangeUser(0.,500.);
//temp->GetYaxis()->SetRangeUser(0.,0.2 );
temp->SetStats(kFALSE);
temp->SetLineWidth(3);
temp->SetMarkerSize(1.25);
int binMax=temp->GetNbinsX()+1;
// container for all histos
std::map< TString, std::map <unsigned int, TH1F*> > histo_;
// determine number of channels
unsigned int nchannels = decayChannel=="combined" ? 2 : 1;
// loop decay channels
if(verbose>0) std::cout << "looping channels" << std::endl;
for(unsigned int channel=0; channel<nchannels; ++channel){
std::map<unsigned int, TFile*> tempfiles_=fileList_.at(channel);
std::string tempChannel= decayChannel!="combined" ? decayChannel : (channel==0 ? "muon" : "electron");
if(verbose>1) std::cout << " - " << tempChannel << std::endl;
TString channelExt=getTStringFromInt(channel);
// loop samples
for(unsigned int sample=kSig; sample<=kSAToptW; ++sample){
bool note=false;
// check if sample is relevant
if(isValidsample(sample, systematicVariation)){
if(verbose>1){
std::cout << " -> processing " << sampleLabel(sample, tempChannel);
std::cout << " (file " << tempfiles_[sample]->GetName() << ")" << std::endl;
}
// calculate luminosity event weight
double lumi=decayChannel!="combined" ? luminosity : (channel==0 ? luminosityMu : luminosityEl);
double lumiwgt=lumiweight(sample, lumi, systematicVariation, tempChannel);
if(verbose>1) std::cout << " (lumiweight=" << lumiwgt << ")" << std::endl;
// get trees
TTree* tree = (TTree*)(tempfiles_[sample]->Get(treePath));
if(!tree){
std::cout << " !ERROR: tree not found!" << std::endl;
exit(0);
}
else if(verbose>1) std::cout << " (tree found, contains " << tree->GetEntries() << " entries)" << std::endl;
// container for values read from tree
std::map< TString, float > value_;
// initialize map entries with 0
value_["weight" ]=1.;
value_["testQuantity" ]=0.;
value_["topPtLep"]=0.;
value_["topPtHad"]=0.;
// initialize branches
tree->SetBranchStatus ("*", 0);
tree->SetBranchStatus ("weight" , 1);
tree->SetBranchStatus ("topPtLep"+treeExt, 1);
tree->SetBranchStatus ("topPtHad"+treeExt, 1);
tree->SetBranchStatus (testQuantity , 1);
tree->SetBranchAddress(testQuantity ,(&value_["testQuantity" ]));
tree->SetBranchAddress("weight" ,(&value_["weight" ]));
tree->SetBranchAddress("topPtLep"+treeExt,(&value_["topPtLep"]));
tree->SetBranchAddress("topPtHad"+treeExt,(&value_["topPtHad"]));
// initialize result plots
histo_["topPt"+channelExt ][sample]=(TH1F*)(temp->Clone());
for(int plot=1; plot<(int)Val_.size(); ++plot){
histo_["topPtProb"+getTStringFromInt(plot)+channelExt][sample]=(TH1F*)(temp->Clone());
}
if(verbose>1) std::cout << " -> looping tree" << std::endl;
// loop all events to fill plots
for(unsigned int event=0; event<tree->GetEntries(); ++event){
// get event
tree->GetEntry(event);
// check if values are reasonable
if(!((value_["weight"]>0&&value_["weight"]<10)||(test!="PV"&&value_["weight"]==0.))){
if(!note){ std::cout << "!!! WARNING - some weights are strange (e.g." << value_["weight"] << ") !!!"<< std::endl; note=true; }
value_["weight"]=1.0;
}
// get relevant quantities
double weight=value_["weight"]*lumiwgt;
double filterQuantity =value_["testQuantity" ];
double topPtLep=value_["topPtLep"];
double topPtHad=value_["topPtHad"];
if(verbose>2){
std::cout << " event #" << event+1 << "/" << tree->GetEntries() << ":" << std::endl;
std::cout << " weight=" << weight << ", " << "testQuantity" << "=" << filterQuantity << ", topPtLep=" << topPtLep << ", topPtHad=" << topPtHad << std::endl;
}
// fill histo for all
histo_["topPt"+channelExt][sample]->Fill(topPtLep, weight);
histo_["topPt"+channelExt][sample]->Fill(topPtHad, weight);
// fill histo for different ranges of the filterQuantity
for(int plot=1; plot<(int)Val_.size(); ++plot){
TString nameNr=getTStringFromInt(plot);
if(filterQuantity>=Val_[plot-1]&&filterQuantity<Val_[plot]){
histo_["topPtProb"+nameNr+channelExt][sample]->Fill(topPtLep, weight);
histo_["topPtProb"+nameNr+channelExt][sample]->Fill(topPtHad, weight);
}
} // end for loop separation values
} // end for loop tree events
} // end if is valid sample
} // end for loop samples
} // end for loop decay channels
// create final plots
// -> combine decay channels and MC samples
unsigned int kAllMC=42;
// loop samples
if(verbose>0) std::cout << "combining decay channels and MC samples" << std::endl;
for(unsigned int sample=kSig; sample<=kSAToptW; ++sample){
// check if sample is relevant
if(isValidsample(sample, systematicVariation)){
// loop decay channels
for(unsigned int channel=0; channel<nchannels; ++channel){
std::string tempChannel= decayChannel!="combined" ? decayChannel : (channel==0 ? "muon" : "electron");
if(verbose>1) std::cout << " -> processing " << sampleLabel(sample, tempChannel) << "(" << tempChannel << ")" << std::endl;
TString channelExt=getTStringFromInt(channel);
// get plots for current channels
std::vector <TH1F*> tempHist_;
tempHist_.push_back((TH1F*)histo_["topPt"+channelExt ][sample]->Clone());
for(int plot=1; plot<(int)Val_.size(); ++plot){
TString nameNr=getTStringFromInt(plot);
tempHist_.push_back((TH1F*)histo_["topPtProb"+nameNr+channelExt][sample]->Clone());
}
std::vector <int> nevents_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
nevents_.push_back(tempHist_[plot]->Integral(0,binMax));
}
// add all channels for final histogram
if(channel==0){
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][sample]=(TH1F*)tempHist_[plot]->Clone();
}
}
else{
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][sample]->Add((TH1F*)tempHist_[plot]->Clone());
}
}
if(verbose>2){
std::cout << " (#events(" << tempChannel << ")=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << nevents_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << ")" << std::endl;
}
} // end channel for loop
std::vector <double> neventsComb_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsComb_.push_back(histo_["topPt"+nameNr][sample]->Integral(0,binMax));
}
if(verbose>1){
std::cout << " (#events=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsComb_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << ")" << std::endl;
}
// combine all MC samples
if(sample!=kData){
if(sample==kSig){
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][kAllMC]=(TH1F*)histo_["topPt"+nameNr][sample]->Clone();
}
}
else{
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][kAllMC]->Add((TH1F*)histo_["topPt"+nameNr][sample]->Clone());
}
}
// MC histogram style
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histogramStyle(*histo_["topPt"+nameNr][sample], sample, true);
//histo_["topPt"+nameNr][sample]->SetLineColor(histo_["topPt"+nameNr][sample]->GetFillColor());
histo_["topPt"+nameNr][sample]->SetLineWidth(1);
}
}
else{
// data histogram style
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histogramStyle(*histo_["topPt"+nameNr][sample], kData, false);
}
}
} // end if sample is valid
} // end sample for loop
// print some interesting numbers
// chosen slices
std::vector< double >neventsMC_;
if(verbose>0){
std::cout << "slices in " << testQuantity << ": all / ";
for(int plot=1; plot<(int)Val_.size(); ++plot){
std::cout << "[" << Val_[plot-1] << ".." << Val_[plot] << "]";
if(plot<(int)Val_.size()-1) std::cout << "/ ";
}
std::cout << std::endl;
}
// total number of MC events
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsMC_.push_back(histo_["topPt"+nameNr][kAllMC]->Integral(0,binMax));
}
if(verbose>0){
std::cout << "#events( MC )=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsMC_[plot];
if(plot<(int)Val_.size()-1) std::cout << ", ";
}
std::cout << std::endl;
}
// total number of data events
std::vector< double >neventsData_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsData_.push_back(histo_["topPt"+nameNr][kData]->Integral(0,binMax));
}
if(verbose>0){
std::cout << "#events(Data)=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsData_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << std::endl;
}
// data over MC ratio
if(verbose>0){
std::cout << "(data/MC ratio=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsData_[plot]/neventsMC_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << ")" << std::endl;
}
// scale ttbar to match total number of events
if(scaleTtbarToMeasured){
if(verbose>0) std::cout << "scale ttbar component to match #data events " << std::endl;
std::vector<double>neventsTop_, SFTop_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsTop_.push_back(histo_["topPt"+nameNr][kSig]->Integral(0,binMax)+histo_["topPt"+nameNr][kBkg]->Integral(0,binMax));
SFTop_.push_back((neventsTop_[plot]+(neventsData_[plot]-neventsMC_[plot]))/neventsTop_[plot]);
}
// scale combined and top MC plots
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
// subtract ttbar from all MC
histo_["topPt"+nameNr][kAllMC]->Add((TH1F*)(histo_["topPt"+nameNr][kSig]->Clone()) , -1.);
histo_["topPt"+nameNr][kAllMC]->Add((TH1F*)(histo_["topPt"+nameNr][kBkg]->Clone()) , -1.);
// scale ttbar
histo_["topPt"+nameNr][kSig]->Scale(SFTop_[plot]);
histo_["topPt"+nameNr][kBkg]->Scale(SFTop_[plot]);
// re-add ttbar MC
histo_["topPt"+nameNr][kAllMC]->Add((TH1F*)(histo_["topPt"+nameNr][kSig]->Clone()) );
histo_["topPt"+nameNr][kAllMC]->Add((TH1F*)(histo_["topPt"+nameNr][kBkg]->Clone()) );
}
// printout
std::vector<double>neventsMCscaled_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsMCscaled_.push_back(histo_["topPt"+nameNr][kAllMC]->Integral(0,binMax));
}
if(verbose>1){
std::cout << "#events(scaledMC)=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsMCscaled_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << std::endl;
std::cout << "(data/scaledMC ratio=";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsData_[plot]/neventsMCscaled_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << ")" << std::endl;
}
};
// combine single top subsamples
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][kSTop]=(TH1F*)histo_["topPt"+nameNr][kSAToptW]->Clone();
for(int sample=(int)kSTops; sample<(int)kSAToptW; ++sample){
// add to combined STop
histo_["topPt"+nameNr][kSTop]->Add((TH1F*)histo_["topPt"+nameNr][sample]->Clone());
} // end for loop single top subsamples
} // end for loop plots
// create MC histo stack plots
int lastSample=-1;
// loop samples
if(verbose>0) std::cout << "creating MC stack histos" << std::endl;
for(int sampleOri=(int)kDiBos; sampleOri>=(int)kSig; --sampleOri){
// use previous defined order
int sampleMod=samples_[sampleOri];
if(verbose>1) std::cout << "processing " << sampleLabel(sampleMod, decayChannel) << "(= " << sampleMod <<", last sample=" << lastSample << ")" << std::endl;
// exclude QCD and Diboson
if(sampleMod!=kQCD&&sampleMod!=kDiBos){
if(lastSample>-1){
if(verbose>1) std::cout << "adding " << sampleLabel(lastSample, decayChannel) << " to " << sampleLabel(sampleMod, decayChannel) << std::endl;
// loop plots
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
if(verbose>1) std::cout << "processing " << "topPt"+nameNr << std::endl;
// add to stack
if(!histo_.count("topPt"+nameNr)>0) std::cout << "WARNING: topPt"+nameNr+" does not exist in histo_!" << std::endl;
else if(!histo_["topPt"+nameNr].count(sampleMod )>0) std::cout << "WARNING: sample " << sampleMod << " does not exist in histo_[topPt"+nameNr+"]!" << std::endl;
else if(!histo_["topPt"+nameNr].count(lastSample)>0) std::cout << "WARNING: sample " << lastSample << " does not exist in histo_[topPt"+nameNr+"]!" << std::endl;
histo_["topPt"+nameNr][sampleMod]->Add((TH1F*)histo_["topPt"+nameNr][lastSample]->Clone());
} // end for loop plots
if(verbose>1) std::cout << "done" << std::endl;
} // if not last sample
if(verbose>1) std::cout << "lastSample set to " << sampleMod << std::endl;
lastSample=sampleMod;
} // end else if !QCD
} // end for loop original sample ordering
// printout
std::vector<double>neventsMCstack_;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
neventsMCstack_.push_back(histo_["topPt"+nameNr][kAllMC]->Integral(0,binMax));
}
if(verbose>1){
std::cout << "#events(stack MC) =";
for(int plot=0; plot<(int)Val_.size(); ++plot){
std::cout << neventsMCstack_[plot];
if(plot<(int)Val_.size()-1) std::cout << " / ";
}
std::cout << std::endl;
}
// all MC histogram style
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
histo_["topPt"+nameNr][kAllMC]->SetLineColor(kBlue);
histo_["topPt"+nameNr][kAllMC]->SetMarkerColor(kBlue);
histo_["topPt"+nameNr][kAllMC]->SetLineStyle(1);
}
std::vector<double> zeroerr_;
for(int bin=0; bin<histo_["topPt"][kAllMC]->GetNbinsX(); ++bin) zeroerr_.push_back(0);
// normalization
// -> not done at the moment, scaled wrt Ndata for each plot separately
// Create canvas
if(verbose>0) std::cout << "creating canvas" << std::endl;
std::vector<TCanvas*> plotCanvas_;
for(unsigned int sample=0; sample<Val_.size(); sample++){
addCanvas(plotCanvas_);
}
// create legends
if(verbose>0) std::cout << "creating legend" << std::endl;
std::vector< TLegend* > leg_;
TLegend *leg= new TLegend(0.73, 0.5, 0.91, 0.88);
legendStyle(*leg,"");
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr = plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
TString sVallow = plot==0 ? "" : getTStringFromDouble(Val_[plot-1], getRelevantDigits(Val_[plot-1]));
TString sValhigh = plot==0 ? "" : getTStringFromDouble(Val_[plot ], getRelevantDigits(Val_[plot ]));
TString legHeader= plot==0 ? "KinFit, all" : sVallow+"#leq"+testQuantity+"<"+sValhigh;
TLegend *templeg=(TLegend*)leg->Clone();
templeg ->SetHeader(legHeader);
templeg ->AddEntry(histo_["topPt"+nameNr][kData ], "data" , "P");
//templeg ->AddEntry(histo_["topPt"+nameNr][kAllMC], "all MC" , "L" );
for(unsigned int sample=kSig; sample<kData; sample++){
unsigned int sampleMod=samples_[sample];
if(sampleMod!=kQCD&&sampleMod!=kDiBos) templeg ->AddEntry(histo_["topPt"+nameNr][sampleMod], sampleLabel(sampleMod, decayChannel), "F" );
}
leg_.push_back((TLegend*)(templeg->Clone()));
}
int canvasNumber=0;
// do the plotting
if(verbose>0) std::cout << "plotting " << std::endl;
for(int plot=0; plot<(int)Val_.size(); ++plot){
TString nameNr= plot==0 ? "" : "TestSlice"+getTStringFromInt(plot);
TString title = plot==0 ? "topPtKinFit"+TString(decayChannel) : "topPt"+testQuantity+getTStringFromInt(plot)+TString(decayChannel);
plotCanvas_[canvasNumber]->cd(0);
plotCanvas_[canvasNumber]->SetTitle(title);
// drawing
// plots
histo_["topPt"+nameNr][kSig]->SetMaximum(1.3*histo_["topPt"+nameNr][kData]->GetMaximum());
histo_["topPt"+nameNr][kSig]->GetXaxis()->SetNoExponent(true);
histo_["topPt"+nameNr][kSig]->GetYaxis()->SetNoExponent(true);
histo_["topPt"+nameNr][kSig]->Draw("axis");
// loop samples
for(int sampleOri=(int)kSig; sampleOri<=(int)kDiBos; ++sampleOri){
// use previous defined order
int sampleMod=samples_[sampleOri];
if(verbose>2) std::cout << "processing sample " << sampleMod << " ("+sampleLabel(sampleMod, decayChannel)+")" << std::endl;
// draw other MC samples, excluding QCD
if(sampleMod!=kQCD&&sampleMod!=kDiBos){
if(verbose>2) std::cout << "-> drawing!" << sampleMod << std::endl;
histo_["topPt"+nameNr][sampleMod]->Draw("hist same");
}
} // end for loop ori samples
//histo_["topPt"+nameNr][kAllMC]->Draw("hist same");
histo_["topPt"+nameNr][kData ]->Draw("ep same");
// legend
leg_[plot]->Draw("same");
// add labels for decay channel, luminosity, energy and CMS preliminary (if applicable)
if (decayChannel=="muon" ) DrawDecayChLabel("#mu + Jets");
else if (decayChannel=="electron") DrawDecayChLabel("e + Jets");
else DrawDecayChLabel("e/#mu + Jets Combined");
DrawCMSLabels(true,luminosity);
// draw ratio
if(withRatioPlot){
// labels of ratio
TString ratioLabelNominator ="N_{MC}";
TString ratioLabelDenominator="N_{Data}";
double ratMin= invert ? 0.75 : 0.30;
double ratMax= invert ? 1.75 : 1.29;
std::vector<double> err_;
for(int bin=1; bin<histo_["topPt"+nameNr][kSig]->GetNbinsX(); ++bin){
double ratio = histo_["topPt"+nameNr][kData]->GetBinContent(bin)/histo_["topPt"+nameNr][kSig]->GetBinContent(bin);
if(invert) ratio=1./ratio;
double val=ratio*(histo_["topPt"+nameNr][kData]->GetBinError(bin)/histo_["topPt"+nameNr][kData]->GetBinContent(bin));
if(val<0||val>histo_["topPt"+nameNr][kData]->GetBinContent(bin)) val=1.;
err_.push_back(val);
}
int rval1 = drawRatio(histo_["topPt"+nameNr][kData], histo_["topPt"+nameNr][kSig], ratMin, ratMax, myStyle, verbose, err_, ratioLabelNominator, ratioLabelDenominator, "p e", kBlack, true, 0.5, 505, invert, true, linFit);
if (rval1!=0) std::cout << " Problem occured when creating ratio plot for " << nameNr << std::endl;
}
canvasNumber++;
}
// saving
if(verbose>0) std::cout << "saving" << std::endl;
if(save){
TString outfolder="./diffXSecFromSignal/plots/combined/2012/topPtTest/";
// eps and png
if(verbose==0) gErrorIgnoreLevel=kWarning;
saveCanvas(plotCanvas_, outfolder, "topPtTest"+testQuantity+TString(decayChannel), true, true, true);
}
}
