| 32 |
|
|
| 33 |
|
bool doesROOTFileExist(string filename); |
| 34 |
|
bool addoverunderflowbins=false; |
| 35 |
< |
string treename="PFevents"; |
| 35 |
> |
string treename="events"; |
| 36 |
|
|
| 37 |
|
class sample |
| 38 |
|
{ |
| 95 |
|
this->tfile = new TFile(m_filename.c_str()); |
| 96 |
|
// gErrorIgnoreLevel=currlevel; |
| 97 |
|
this->events=(TTree*)(this->tfile)->Get(treename.c_str()); |
| 98 |
+ |
// this->events->SetCacheSize(5000000);//only a 5 MB cache! |
| 99 |
|
if(Verbosity>0) dout << "The " << write_mc_or_data(is_data) << " file " << this->filename << " has been added successfully to the list of samples. " << endl; |
| 100 |
|
long long measured_nevents=(long)1; |
| 101 |
|
TH1F *weight_histo = (TH1F*)(this->tfile)->Get("weight_histo"); |
| 213 |
|
TLegend* allbglegend(string title,float x, float y); |
| 214 |
|
TLegend* allbglegend(string title, TH1F *data, float x, float y); |
| 215 |
|
|
| 216 |
< |
void PickUpFromThisFile(int isamp, string cut, vector<string> &output, vector<string> &pickupfile); |
| 217 |
< |
void PickUpEvents(string cut); |
| 216 |
> |
void PickUpFromThisFile(int isamp, string cut, vector<string> &output, vector<string> &pickupfile, bool QuietMode); |
| 217 |
> |
void PickUpEvents(string cut,string filename,bool QuietMode); |
| 218 |
|
string find_units(string&); |
| 219 |
|
|
| 220 |
|
void RemoveLastSample(); |
| 294 |
|
return histo; |
| 295 |
|
} |
| 296 |
|
|
| 297 |
+ |
|
| 298 |
|
TH1F* samplecollection::Draw(string m_histoname,string m_var, vector<float> binning, string m_xlabel, string m_ylabel, TCut Cut, bool m_is_data, float luminosity, vector<int> onlyindex, bool drawsignal=false) { |
| 299 |
|
if(Verbosity>0) dout << endl << endl; |
| 300 |
|
if(Verbosity>0) dout << "-------------------------------------------------------------------------------------" << endl; |
| 338 |
|
if(use_this_sample) { |
| 339 |
|
(this->collection)[isample].events->Draw(drawthis.str().c_str(),(essentialcut&&Cut)*cutWeight);//this weight is based on PU etc. not XS |
| 340 |
|
// cout << "\033[1;33m Drawing " << drawthis.str() << " with cut " << (const char*) (essentialcut&&Cut)*cutWeight << " for sample " << (this->collection)[isample].filename << "\033[0m (overflow: " << addoverunderflowbins << ")" << endl; |
| 341 |
+ |
// cout << "Drawing for : " << (this->collection)[isample].filename << endl; |
| 342 |
+ |
// for(int i=1;i<=tempdrawhisto->GetNbinsX();i++) cout << " Bin " << i << " [" << tempdrawhisto->GetBinLowEdge(i) << "," << tempdrawhisto->GetBinLowEdge(i)+tempdrawhisto->GetBinWidth(i) << "] : " << tempdrawhisto->GetBinContent(i) << " +/- " << tempdrawhisto->GetBinError(i) << endl; |
| 343 |
|
if(addoverunderflowbins) { |
| 344 |
|
//now also adding the overflow & underflow bins: |
| 345 |
|
tempdrawhisto->SetBinContent(tempdrawhisto->GetNbinsX(),tempdrawhisto->GetBinContent(tempdrawhisto->GetNbinsX()+1)+tempdrawhisto->GetBinContent(tempdrawhisto->GetNbinsX())); |
| 393 |
|
} |
| 394 |
|
stringstream h_histoname; |
| 395 |
|
h_histoname<<"h_"<<m_histoname; |
| 396 |
< |
float binningarray[binningx.size()+1]; |
| 393 |
< |
float binningyarray[binningy.size()+1]; |
| 394 |
< |
for(int i=0;i<(int)binningx.size();i++) {binningarray[i]=binningx[i];binningyarray[i]=binningy[i];} |
| 395 |
< |
TH2F *histo = new TH2F(m_histoname.c_str(),"",binningx.size()-1,binningarray,binningy.size()-1,binningyarray); |
| 396 |
> |
TH2F *histo = new TH2F(m_histoname.c_str(),"",binningx.size()-1,&binningx[0],binningy.size()-1,&binningy[0]); |
| 397 |
|
histo->Sumw2(); |
| 398 |
|
|
| 399 |
|
stringstream drawthis; |
| 409 |
|
if(do_only_selected_samples&&this->do_sample(isample,onlyindex)) { |
| 410 |
|
use_this_sample=true; |
| 411 |
|
} |
| 412 |
< |
TH2F *tempdrawhisto = new TH2F("tempdrawhisto","tempdrawhisto",binningx.size()-1,binningarray,binningy.size()-1,binningyarray); |
| 412 |
> |
TH2F *tempdrawhisto = new TH2F("tempdrawhisto","tempdrawhisto",binningx.size()-1,&binningx[0],binningy.size()-1,&binningy[0]); |
| 413 |
|
tempdrawhisto->Sumw2(); |
| 414 |
|
if(use_this_sample) { |
| 415 |
|
if(Verbosity>0) dout << "[samplecollection::Draw] : Added contribution from sample " << (this->collection)[isample].samplename << endl; |
| 471 |
|
TH1F *tempdrawhisto = new TH1F("tempdrawhisto","",binning.size()-1,binningarray); |
| 472 |
|
tempdrawhisto->Sumw2(); |
| 473 |
|
(this->collection)[isample].events->Draw(drawthis.str().c_str(),(essentialcut&&Cut)*cutWeight); |
| 474 |
< |
|
| 474 |
> |
// cout << "\033[1;33m Drawing " << drawthis.str() << " with cut " << (const char*) (essentialcut&&Cut)*cutWeight << " for sample " << (this->collection)[isample].filename << "\033[0m (overflow: " << addoverunderflowbins << ")" << endl; |
| 475 |
|
if(addoverunderflowbins) { |
| 476 |
|
//now also adding the overflow & underflow bins: |
| 477 |
|
tempdrawhisto->SetBinContent(tempdrawhisto->GetNbinsX(),tempdrawhisto->GetBinContent(tempdrawhisto->GetNbinsX()+1)+tempdrawhisto->GetBinContent(tempdrawhisto->GetNbinsX())); |
| 515 |
|
for(int index=0;index<(int)ordered_indices.size();index++) { |
| 516 |
|
thestack.Add(histogroups[ordered_indices[index]]); |
| 517 |
|
} |
| 518 |
+ |
|
| 519 |
|
return thestack; |
| 520 |
|
} |
| 521 |
|
|
| 586 |
|
|
| 587 |
|
int maxIgroup=0; |
| 588 |
|
for(int isample=0;isample<(int)(this->collection).size();isample++) { |
| 589 |
< |
if((this->collection)[isample].groupindex>maxIgroup) maxIgroup=(this->collection)[isample].groupindex+1; |
| 589 |
> |
if((this->collection)[isample].groupindex>maxIgroup) maxIgroup=(this->collection)[isample].groupindex; |
| 590 |
|
} |
| 591 |
+ |
maxIgroup++; |
| 592 |
|
|
| 593 |
|
TH1F *fakehistos[maxIgroup]; |
| 594 |
|
bool donealready[maxIgroup]; |
| 595 |
|
|
| 596 |
|
for(int i=0;i<maxIgroup;i++) donealready[i]=false; |
| 597 |
+ |
|
| 598 |
|
for(int isample=0;isample<(int)(this->collection).size();isample++) { |
| 599 |
|
if((this->collection)[isample].is_data||(this->collection)[isample].is_signal) continue; |
| 596 |
– |
|
| 600 |
|
if(!donealready[(this->collection)[isample].groupindex]) { |
| 601 |
+ |
assert((this->collection)[isample].groupindex<maxIgroup); |
| 602 |
|
fakehistos[(this->collection)[isample].groupindex] = new TH1F(GetNumericHistoName().c_str(),"",1,0,1); |
| 603 |
|
fakehistos[(this->collection)[isample].groupindex]->Fill(1); |
| 604 |
|
fakehistos[(this->collection)[isample].groupindex]->SetFillColor(this->GetColor(isample)); |
| 613 |
|
} |
| 614 |
|
|
| 615 |
|
TLegend* samplecollection::allbglegend(string title="",float posx=0.65, float posy=0.60) { |
| 612 |
– |
// Int_t currlevel=gErrorIgnoreLevel; |
| 613 |
– |
// gErrorIgnoreLevel=5000; |
| 616 |
|
TH1F *blub = new TH1F("nothing","nothing",1,0,1); |
| 615 |
– |
// gErrorIgnoreLevel=currlevel;//we know this possibly replaces a previous histo, but we don't care since it's fake anyway. |
| 617 |
|
TLegend *leg = this->allbglegend(title,blub,posx,posy); |
| 618 |
< |
delete blub; |
| 618 |
> |
PlottingSetup::FakeHistoHeap.push_back(blub); |
| 619 |
|
return leg; |
| 620 |
|
} |
| 621 |
|
|
| 650 |
|
|
| 651 |
|
} |
| 652 |
|
|
| 653 |
< |
void samplecollection::PickUpFromThisFile(int isamp, string cut, vector<string> &output, vector<string> &pickupfile) { |
| 654 |
< |
int lumi,eventNum,runNum; |
| 653 |
> |
void samplecollection::PickUpFromThisFile(int isamp, string cut, vector<string> &output, vector<string> &pickupfile, bool QuietMode) { |
| 654 |
> |
int lumi,runNum; |
| 655 |
> |
ULong64_t eventNum; |
| 656 |
> |
|
| 657 |
|
float jzb[30]; |
| 658 |
|
(this->collection)[isamp].events->SetBranchAddress("jzb",&jzb); |
| 659 |
|
(this->collection)[isamp].events->SetBranchAddress("lumi",&runNum); |
| 664 |
|
|
| 665 |
|
TTreeFormula *select = new TTreeFormula("select", cut.c_str()&&essentialcut, (this->collection)[isamp].events); |
| 666 |
|
int npickedup=0; |
| 667 |
+ |
stringstream filetext; |
| 668 |
|
for (Int_t entry = 0 ; entry < (this->collection)[isamp].events->GetEntries() ; entry++) { |
| 669 |
|
(this->collection)[isamp].events->LoadTree(entry); |
| 670 |
|
if (select->EvalInstance()) { |
| 671 |
|
(this->collection)[isamp].events->GetEntry(entry); |
| 672 |
< |
dout << runNum << ":" << lumi << ":" << eventNum << endl; |
| 672 |
> |
if(!QuietMode) dout << runNum << ":" << lumi << ":" << eventNum << endl; |
| 673 |
> |
filetext.str(""); |
| 674 |
> |
filetext << runNum << ":" << lumi << ":" << eventNum << endl; |
| 675 |
> |
output.push_back(filetext.str()); |
| 676 |
|
npickedup++; |
| 677 |
|
} |
| 678 |
|
} |
| 679 |
< |
dout << "Printed out a total of " << npickedup << " events" << endl; |
| 679 |
> |
if(!QuietMode) dout << "Printed out a total of " << npickedup << " events" << endl; |
| 680 |
|
} |
| 681 |
|
|
| 682 |
|
|
| 683 |
|
|
| 684 |
< |
void samplecollection::PickUpEvents(string cut) { |
| 684 |
> |
void samplecollection::PickUpEvents(string cut, string filename, bool QuietMode=false) { |
| 685 |
|
vector<string> output; |
| 686 |
|
vector<string> pickupfile; |
| 687 |
< |
for (int isamp=0;isamp<(int)this->collection.size();isamp++) |
| 687 |
> |
for (unsigned int isamp=0;isamp<this->collection.size();isamp++) |
| 688 |
|
{ |
| 689 |
|
if((this->collection)[isamp].is_data) { |
| 690 |
|
//we have a data sample ! |
| 691 |
< |
this->PickUpFromThisFile(isamp,cut,output,pickupfile); |
| 691 |
> |
this->PickUpFromThisFile(isamp,cut,output,pickupfile,QuietMode); |
| 692 |
|
} |
| 693 |
|
} |
| 694 |
+ |
|
| 695 |
|
//do something with output and of course the pickup file! |
| 696 |
+ |
ofstream myfile(filename.c_str()); |
| 697 |
+ |
for(unsigned int ievent=0;ievent<output.size();ievent++) myfile << output[ievent]; |
| 698 |
+ |
myfile.close(); |
| 699 |
+ |
|
| 700 |
|
} |
| 701 |
|
|
| 702 |
|
//________________________________________________________________________________________ |
