| 1 |
//-------------------------------------------------------------------------------
|
| 2 |
// Use this utility to reweight MC to match the nvtx distribution in data.
|
| 3 |
// This requires a root file with a histogram named "hratio" whose bin contents
|
| 4 |
// specify the weight for each nvtx bin. This weight is determined by plotting
|
| 5 |
// the nvtx distribution for a data sample and a MC sample and taking the ratio.
|
| 6 |
//
|
| 7 |
// A sample file can be found at:
|
| 8 |
// /tas/benhoob/vtxreweight/vtxreweight_Spring11MC_153pb_Zselection.root
|
| 9 |
//
|
| 10 |
// This root file was made with this macro:
|
| 11 |
// /tas/benhoob/vtxreweight/make_vtxreweight_Spring11MC_153pb_Zselection.cc
|
| 12 |
//
|
| 13 |
// and was produced after applying a Z selection.
|
| 14 |
//
|
| 15 |
// You can make your own root file by modifying this macro
|
| 16 |
//
|
| 17 |
//
|
| 18 |
// usage:
|
| 19 |
//
|
| 20 |
// //include header
|
| 21 |
// #include "../Tools/vtxreweight.cc"
|
| 22 |
//
|
| 23 |
// //initialize
|
| 24 |
// bool verbose = true;
|
| 25 |
// char* vtxfile = "/tas/benhoob/vtxreweight/vtxreweight_Spring11MC_153pb_Zselection.root";
|
| 26 |
// set_vtxreweight_rootfile( vtxfile , verbose );
|
| 27 |
//
|
| 28 |
// //in the event loop
|
| 29 |
// float myvtxweight = vtxweight();
|
| 30 |
//
|
| 31 |
//
|
| 32 |
// PLEASE NOTE: ALWAYS CHECK THAT THE WEIGHTING HAS BEEN DONE PROPERLY
|
| 33 |
// BY COMPARING THE DATA NVTX DIST WITH THE WEIGHTED MC NVTX DISTRIBUTION,
|
| 34 |
// WHERE NVTX IS THE NUMBER OF DA VERTICES PASSING isGoodDAVertex()
|
| 35 |
//------------------------------------------------------------------------------
|
| 36 |
|
| 37 |
// $Id: vtxreweight.cc,v 1.4 2011/05/20 14:24:22 warren Exp $
|
| 38 |
|
| 39 |
// CINT is allowed to see this, but nothing else:
|
| 40 |
#include "vtxreweight.h"
|
| 41 |
|
| 42 |
#ifndef __CINT__
|
| 43 |
|
| 44 |
#include <assert.h>
|
| 45 |
#include <stdio.h>
|
| 46 |
#include <stdlib.h>
|
| 47 |
#include <string.h>
|
| 48 |
#include <unistd.h>
|
| 49 |
#include <sys/types.h>
|
| 50 |
#include <sys/wait.h>
|
| 51 |
#include <sys/stat.h>
|
| 52 |
#include <set>
|
| 53 |
#include <string>
|
| 54 |
|
| 55 |
#include "../CORE/CMS2.h"
|
| 56 |
#include "../CORE/eventSelections.h"
|
| 57 |
|
| 58 |
bool loaded_vtxreweight_hist = false;
|
| 59 |
|
| 60 |
float vtxweight( bool isData , int nvtx ){
|
| 61 |
|
| 62 |
if( isData ) return 1;
|
| 63 |
|
| 64 |
if( !loaded_vtxreweight_hist ){
|
| 65 |
cout << "vtxreweight.cc: You need to do" << endl;
|
| 66 |
cout << "set_vtxreweight_rootfile( filename )" << endl;
|
| 67 |
cout << "before calling vtxweight()" << endl;
|
| 68 |
cout << "a sample vtxreweight file can be found at" << endl;
|
| 69 |
cout << "/tas/benhoob/vtxreweight/vtxreweight_Spring11MC_23pbPR.root" << endl;
|
| 70 |
cout << "now, quitting" << endl;
|
| 71 |
exit(2);
|
| 72 |
}
|
| 73 |
|
| 74 |
if( nvtx == 0 ){
|
| 75 |
cout << "vtxreweight.cc: warning 0 good vertices found, returning 0 weight" << endl;
|
| 76 |
return 0;
|
| 77 |
}
|
| 78 |
|
| 79 |
if( nvtx > vtxreweight_hist->GetNbinsX() ) nvtx = vtxreweight_hist->GetNbinsX();
|
| 80 |
|
| 81 |
//cout << "nvtx " << nvtx << " weight " << vtxreweight_hist->GetBinContent(nvtx) << endl;
|
| 82 |
return vtxreweight_hist->GetBinContent(nvtx);
|
| 83 |
|
| 84 |
}
|
| 85 |
|
| 86 |
void set_vtxreweight_rootfile ( const char* filename , bool verbose ){
|
| 87 |
TFile* file = TFile::Open(filename);
|
| 88 |
|
| 89 |
if( file == 0 ){
|
| 90 |
cout << "vtxreweight.cc: error, couldn't open file : " << filename << endl;
|
| 91 |
cout << "Quitting!" << endl;
|
| 92 |
exit(0);
|
| 93 |
}
|
| 94 |
|
| 95 |
vtxreweight_hist = (TH1F*) file->Get("hratio");
|
| 96 |
|
| 97 |
if( vtxreweight_hist == 0 ){
|
| 98 |
cout << "vtxreweight.cc: error, couldn't retrieve hist hratio from file " << filename << endl;
|
| 99 |
cout << "Quitting" << endl;
|
| 100 |
exit(1);
|
| 101 |
}
|
| 102 |
|
| 103 |
if( verbose ){
|
| 104 |
cout << "Opened vtx reweighting file " << filename << endl;
|
| 105 |
|
| 106 |
cout << "|" << setw(10) << "nvtx" << setw(4)
|
| 107 |
<< "|" << setw(10) << "weight" << setw(4) << "|" << endl;
|
| 108 |
|
| 109 |
for(unsigned int ibin = 1 ; ibin <= (unsigned int)vtxreweight_hist->GetNbinsX() ; ++ibin ){
|
| 110 |
|
| 111 |
cout << "|" << setw(10) << ibin << setw(4)
|
| 112 |
<< "|" << setw(10) << vtxreweight_hist->GetBinContent(ibin) << setw(4) << "|" << endl;
|
| 113 |
|
| 114 |
}
|
| 115 |
}
|
| 116 |
loaded_vtxreweight_hist = true;
|
| 117 |
|
| 118 |
}
|
| 119 |
|
| 120 |
#endif // __CUNT__
|
| 121 |
|
