71int main(
int argc,
char* argv[]){
76 std::string sample =
"";
77 std::string truth_jetColl =
"";
78 std::string reco_jetColl =
"";
79 std::string string_kindofmass =
"";
80 std::string string_kindofmc =
"";
81 std::string output =
"";
82 std::string string_configfile_path =
"";
83 std::string string_debugtool =
"";
84 int eventsMax = 30000;
87 for (
int i=1; i< argc; i++){
89 std::string opt(argv[i]); std::vector< std::string > v;
90 std::istringstream iss(opt);
94 while (std::getline(iss, item, delim)){
98 if ( opt.find(
"--help") != std::string::npos ){
103 if ( opt.find(
"--sample=") != std::string::npos ) sample = v[1];
104 if ( opt.find(
"--truth_jetColl=") != std::string::npos ) truth_jetColl = v[1];
105 if ( opt.find(
"--reco_jetColl=") != std::string::npos ) reco_jetColl = v[1];
106 if ( opt.find(
"--MassDef=") != std::string::npos ) string_kindofmass = v[1];
107 if ( opt.find(
"--MCType=") != std::string::npos ) string_kindofmc = v[1];
108 if ( opt.find(
"--ConfigFile=") != std::string::npos ) string_configfile_path = v[1];
109 if ( opt.find(
"--DebugTool=") != std::string::npos ) string_debugtool = v[1];
110 if ( opt.find(
"--output=") != std::string::npos ) output = v[1];
111 if ( opt.find(
"--eventsMax=") != std::string::npos ) eventsMax = std::atoi(v[1].data());
116 std::cout <<
"No input xAOD file specified, exiting" << std::endl;
119 if (truth_jetColl==
""){
120 std::cout <<
"No truth jet collection specified, exiting" << std::endl;
123 if (reco_jetColl==
""){
124 std::cout <<
"No truth jet collection specified, exiting" << std::endl;
127 if (string_kindofmass==
""){
128 std::cout <<
"No kind of jet mass specified, exiting" << std::endl;
131 if (string_kindofmc==
""){
132 std::cout <<
"No kind of MC specified, exiting" << std::endl;
135 if (string_configfile_path==
""){
136 std::cout <<
"No ConfigFile specified, exiting" << std::endl;
139 if (string_debugtool==
""){
140 std::cout <<
"No debugtool specified, exiting" << std::endl;
144 std::cout <<
"Output not specified, exiting" << std::endl;
147 std::cout << sample << truth_jetColl << reco_jetColl << output << string_debugtool << std::endl;
149 TString kindofmass = string_kindofmass;
150 TString kindofmc = string_kindofmc;
152 bool want_to_debug =
false;
153 if (string_debugtool ==
"true"){
154 want_to_debug =
true;
155 }
else if (string_debugtool ==
"false"){
156 want_to_debug =
false;
161 std::unique_ptr< TFile > ifile( TFile::Open( sample.c_str(),
"READ" ) );
166 std::cout <<
"Failed to open file " << std::endl;
172 const std::string name_FFJetSmearingTool =
"FFJetSmearing_Example";
174 CHECK(ffjetsmearingtool.setProperty(
"MassDef", kindofmass));
175 CHECK(ffjetsmearingtool.setProperty(
"MCType", kindofmc));
176 CHECK(ffjetsmearingtool.setProperty(
"ConfigFile", string_configfile_path));
177 CHECK(ffjetsmearingtool.setProperty(
"OutputLevel", MSG::ERROR));
179 CHECK(ffjetsmearingtool.setProperty(
"OutputLevel", MSG::VERBOSE));
186 std::cout <<
"\nRecommended systematics:\n" << std::endl;
187 for (
auto sysItr = recommendedSysts.
begin(); sysItr != recommendedSysts.
end(); ++sysItr){
188 std::cout << sysItr->name().c_str() << std::endl;
191 std::vector<CP::SystematicSet> sysList;
195 const std::string name_JetCalibTools =
"JetCalibrationTool";
198 CHECK(m_JetCalibrationTool_handle.setProperty(
"JetCollection",
"AntiKt10UFOCSSKSoftDropBeta100Zcut10"));
199 CHECK(m_JetCalibrationTool_handle.setProperty(
"ConfigFile",
"JES_MC20PreRecommendation_R10_UFO_CSSK_SoftDrop_JMS_R21Insitu_26Nov2024.config"));
201 CHECK(m_JetCalibrationTool_handle.setProperty(
"CalibSequence",
"EtaJES_JMS"));
202 CHECK(m_JetCalibrationTool_handle.setProperty(
"IsData",
false));
203 CHECK(m_JetCalibrationTool_handle.setProperty(
"CalibArea",
"00-04-82"));
204 CHECK(m_JetCalibrationTool_handle.setProperty(
"DEVmode",
false));
217 for (
auto sys : recommendedSysts){
221 std::cout <<
"\n=============SYSTEMATICS CHECK NOW" << std::endl;
222 for (
auto sys : sysList){
224 std::cout <<
"\nRunning over the systematic " << sys.name().c_str() << std::endl;
225 static constexpr float MeVtoGeV = 1.e-3;
229 std::cout <<
"Error, Cannot configure calibration tool for systematics" << std::endl;
233 TH1F* reco_jet_mass_hist =
new TH1F(
"reco_jet_mass_hist",
"reco_jet_mass_hist", 300, 0, 600);
234 TH1F* matched_truth_jet_mass_hist =
new TH1F(
"matched_truth_jet_mass_hist",
"matched_truth_jet_mass_hist", 300, 0, 600);
235 TH1F* smeared_reco_jet_mass_hist =
new TH1F(
"smeared_reco_jet_mass_hist",
"smeared_reco_jet_mass_hist", 300, 0, 600);
237 TH1F* reco_jet_pt_hist =
new TH1F(
"reco_jet_pt_hist",
"reco_jet_pt_hist", 800, 0, 4000);
238 TH1F* matched_truth_jet_pt_hist =
new TH1F(
"matched_truth_jet_pt_hist",
"matched_truth_jet_pt_hist", 800, 0, 4000);
239 TH1F* smeared_reco_jet_pt_hist =
new TH1F(
"smeared_reco_jet_pt_hist",
"smeared_reco_jet_pt_hist", 800, 0, 4000);
241 TH1F* reco_jet_rapidity_hist =
new TH1F(
"reco_jet_rapidity_hist",
"reco_jet_rapidity_hist", 50, -2.5, 2.5);
242 TH1F* matched_truth_jet_rapidity_hist =
new TH1F(
"matched_truth_jet_rapidity_hist",
"matched_truth_jet_rapidity_hist", 50, -2.5, 2.5);
243 TH1F* smeared_reco_jet_rapidity_hist =
new TH1F(
"smeared_reco_jet_rapidity_hist",
"smeared_reco_jet_rapidity_hist", 50, -2.5, 2.5);
245 reco_jet_mass_hist->Sumw2();
246 matched_truth_jet_mass_hist->Sumw2();
247 smeared_reco_jet_mass_hist->Sumw2();
249 reco_jet_pt_hist->Sumw2();
250 matched_truth_jet_pt_hist->Sumw2();
251 smeared_reco_jet_pt_hist->Sumw2();
253 reco_jet_rapidity_hist->Sumw2();
254 matched_truth_jet_rapidity_hist->Sumw2();
255 smeared_reco_jet_rapidity_hist->Sumw2();
257 int upperlimit1 = 600;
258 int upperlimit2 = 1000;
260 int numBinsMass = 120;
263 std::unique_ptr<TH3F> hist_jet_mass_scale_change_3D;
264 hist_jet_mass_scale_change_3D = std::make_unique<TH3F>(
"hist_jet_mass_scale_change_3D",
"hist_jet_mass_scale_change_3D",numBinsPt,0,upperlimit2,numBinsMass,0,upperlimit1,numBinsMass,0,upperlimit1);
266 float lowerlimit3 = -0.5;
267 float upperlimit3 = 0.5;
268 int numBinsDiff = 100;
270 std::unique_ptr<TH3F> hist_jet_mass_resolution_change_3D;
271 hist_jet_mass_resolution_change_3D = std::make_unique<TH3F>(
"hist_jet_mass_resolution_change_3D",
"hist_jet_mass_resolution_change_3D",numBinsPt,0,upperlimit2,numBinsMass,0,upperlimit1,numBinsDiff,lowerlimit3,upperlimit3);
275 Long64_t nevents =
event->getEntries();
277 if (eventsMax < nevents){
281 for (Long64_t ievent = 0; ievent < nevents; ++ievent){
284 if ( event->getEntry( ievent ) < 0 ){
285 std::cout <<
"Failed to load entry " << ievent << std::endl;
290 if (ievent % 1000==0){
291 std::cout <<
"Event " << ievent <<
" of " << nevents << std::endl;
297 CHECK( event->retrieve(ei,
"EventInfo") );
299 std::cout <<
"===>>> start processing event " << ei->
eventNumber() <<
", run " << ei->
runNumber() <<
" - Events processed so far: " << ievent << std::endl;
303 CHECK( event->retrieve(jets_truth, truth_jetColl) );
304 std::cout <<
"Number of truth jets: " << jets_truth->
size() << std::endl;
307 for (
const xAOD::Jet* jet_truth : *jets_truth){
309 std::cout <<
"Truth Jet: pt = " << jet_truth->pt()*MeVtoGeV <<
", mass = " << jet_truth->m()*MeVtoGeV <<
", eta = " << jet_truth->eta() << std::endl;
314 CHECK( event->retrieve(jets_reco, reco_jetColl) );
315 std::cout <<
"Number of reco jets: " << jets_reco->
size() << std::endl;
318 for (
const xAOD::Jet* jet_reco : *jets_reco){
320 std::cout <<
"Reco Jet: pt = " << jet_reco->pt()*MeVtoGeV <<
", mass = " << jet_reco->m()*MeVtoGeV <<
", eta = " << jet_reco->eta() << std::endl;
325 jet_truth_matched.makePrivateStore();
329 CHECK( event->retrieve( jets, reco_jetColl ) );
335 std::cout <<
"Start the loop over the jets" << std::endl;
338 bool lead_jet =
true;
341 if ( m_JetCalibrationTool_handle.
applyCalibration( *(jets_shallowCopy.first) ) != StatusCode::SUCCESS ){
342 std::cout <<
"JetCalibration tool reported a EL::StatusCode::FAILURE" << std::endl;
348 for (
xAOD::Jet* jet_reco : *(jets_shallowCopy.first) ){
350 double jetpt = jet_reco->pt() * MeVtoGeV;
351 double jetm = jet_reco->m() * MeVtoGeV;
352 double jetrapidity = jet_reco->rapidity();
354 if (jetpt < 200 or jetpt > 3000)
continue;
356 if (jetm > 600)
continue;
358 if (abs(jetrapidity) > 2)
continue;
360 if ( ffjetsmearingtool.
getMatchedTruthJet(*jet_reco, jet_truth_matched) != StatusCode::SUCCESS ){
364 double aux_original_jet_mass = jet_reco->m() * MeVtoGeV;
366 if (lead_jet ==
true && aux_original_jet_mass > 0){
368 reco_jet_mass_hist->Fill(jet_reco->m() * MeVtoGeV);
369 reco_jet_pt_hist->Fill(jet_reco->pt() * MeVtoGeV);
370 reco_jet_rapidity_hist->Fill(jet_reco->rapidity());
372 matched_truth_jet_mass_hist->Fill(jet_truth_matched.
m() * MeVtoGeV);
373 matched_truth_jet_pt_hist->Fill(jet_truth_matched.
pt() * MeVtoGeV);
374 matched_truth_jet_rapidity_hist->Fill(jet_truth_matched.
rapidity());
378 std::cout <<
"FFJetSmearingTool reported a EL::StatusCode::FAILURE" << std::endl;
382 smeared_reco_jet_mass_hist->Fill(jet_reco->m() * MeVtoGeV);
383 smeared_reco_jet_pt_hist->Fill(jet_reco->pt() * MeVtoGeV);
384 smeared_reco_jet_rapidity_hist->Fill(jet_reco->rapidity());
386 hist_jet_mass_scale_change_3D->Fill(jet_reco->pt() * MeVtoGeV, aux_original_jet_mass, jet_reco->m() * MeVtoGeV);
387 hist_jet_mass_resolution_change_3D->Fill(jet_reco->pt() * MeVtoGeV, aux_original_jet_mass, TMath::Abs( (jet_reco->m()*MeVtoGeV) - (aux_original_jet_mass) )/aux_original_jet_mass);
397 TString output_path =
"output/" + sys.name() + output ;
398 TFile *jetmass_histograms =
new TFile(output_path,
"recreate");
400 reco_jet_mass_hist->Write();
401 matched_truth_jet_mass_hist->Write();
402 smeared_reco_jet_mass_hist->Write();
404 reco_jet_pt_hist->Write();
405 matched_truth_jet_pt_hist->Write();
406 smeared_reco_jet_pt_hist->Write();
408 reco_jet_rapidity_hist->Write();
409 matched_truth_jet_rapidity_hist->Write();
410 smeared_reco_jet_rapidity_hist->Write();
414 TH2F* hist_jet_mass_scale_change_2D =
new TH2F(
"hist_jet_mass_scale_change_2D",
"hist_jet_mass_scale_change_2D",numBinsPt,0,upperlimit2,numBinsMass,0,upperlimit1);
415 for (
int i=1; i<=hist_jet_mass_scale_change_3D->GetNbinsX(); i++){
416 for (
int j=1; j<= hist_jet_mass_scale_change_3D->GetNbinsY(); j++){
417 TH1F* hold =
new TH1F(
"",
"",numBinsMass,0,upperlimit1);
418 for (
int k=1; k<= hist_jet_mass_scale_change_3D->GetNbinsZ(); k++){
419 hold->SetBinContent(k,hist_jet_mass_scale_change_3D->GetBinContent(i,j,k));
421 hist_jet_mass_scale_change_2D->SetBinContent(i,j,hold->GetMean()/hist_jet_mass_scale_change_3D->GetYaxis()->GetBinCenter(j));
426 TH2F* hist_jet_mass_resolution_change_2D =
new TH2F(
"hist_jet_mass_resolution_change_2D",
"hist_jet_mass_resolution_change_2D",numBinsPt,0,upperlimit2,numBinsMass,0,upperlimit1);
427 for (
int i=1; i<=hist_jet_mass_resolution_change_3D->GetNbinsX(); i++){
428 for (
int j=1; j<= hist_jet_mass_resolution_change_3D->GetNbinsY(); j++){
429 TH1F* hold =
new TH1F(
"",
"",numBinsDiff,lowerlimit3,upperlimit3);
430 for (
int k=1; k<= hist_jet_mass_resolution_change_3D->GetNbinsZ(); k++){
431 hold->SetBinContent(k,hist_jet_mass_resolution_change_3D->GetBinContent(i,j,k));
433 hist_jet_mass_resolution_change_2D->SetBinContent(i,j,hold->GetMean()/hist_jet_mass_resolution_change_3D->GetYaxis()->GetBinCenter(j));
443 TCanvas *c1 =
new TCanvas(
"c1",
"c1",w,
h);
444 c1->SetWindowSize(w+(w-c1->GetWw()),
h+(
h-c1->GetWh()));
447 TPad *canvas_1 =
new TPad(
"canvas_1",
"canvas_1",0.0,0.0,1.0,1.0);
448 canvas_1->SetRightMargin(0.10);
449 canvas_1->SetFillStyle(4000);
453 hist_jet_mass_scale_change_2D->GetXaxis()->SetTitleOffset(1.5);
454 hist_jet_mass_scale_change_2D->GetXaxis()->SetNdivisions(5);
455 hist_jet_mass_scale_change_2D->GetYaxis()->SetTitleOffset(1.5);
456 hist_jet_mass_scale_change_2D->GetZaxis()->SetTitleOffset(1.5);
457 hist_jet_mass_scale_change_2D->GetYaxis()->SetTitle(
"Initial Reco Mass [GeV]");
458 hist_jet_mass_scale_change_2D->GetXaxis()->SetTitle(
"Reco p_{T} [GeV]");
459 hist_jet_mass_scale_change_2D->GetZaxis()->SetTitle(
"Average Mass smearing (Initial_reco_mass/smeared_reco_mass)");
460 hist_jet_mass_scale_change_2D->GetZaxis()->SetRangeUser(0.95,1.05);
461 hist_jet_mass_scale_change_2D->GetZaxis()->SetLabelSize(0.035);
462 hist_jet_mass_scale_change_2D->Draw(
"colz");
465 TString output_path_scale_debug =
"output/debug_plots/scale_variations/" + sys.name() +
"_scaleDebug.pdf" ;
466 c1->Print(output_path_scale_debug);
468 delete hist_jet_mass_scale_change_2D;
471 TCanvas *c2 =
new TCanvas(
"c2",
"c2",w,
h);
472 c2->SetWindowSize(w+(w-c2->GetWw()),
h+(
h-c2->GetWh()));
475 TPad *canvas_2 =
new TPad(
"canvas_2",
"canvas_2",0.0,0.0,1.0,1.0);
476 canvas_2->SetRightMargin(0.10);
477 canvas_2->SetFillStyle(4000);
481 hist_jet_mass_resolution_change_2D->GetXaxis()->SetTitleOffset(1.5);
482 hist_jet_mass_resolution_change_2D->GetXaxis()->SetNdivisions(5);
483 hist_jet_mass_resolution_change_2D->GetYaxis()->SetTitleOffset(1.5);
484 hist_jet_mass_resolution_change_2D->GetZaxis()->SetTitleOffset(1.5);
485 hist_jet_mass_resolution_change_2D->GetYaxis()->SetTitle(
"Initial Reco Mass [GeV]");
486 hist_jet_mass_resolution_change_2D->GetXaxis()->SetTitle(
"Reco p_{T} [GeV]");
487 hist_jet_mass_resolution_change_2D->GetZaxis()->SetTitle(
"Average Mass smearing (Initial_reco_mass/smeared_reco_mass)");
488 hist_jet_mass_resolution_change_2D->GetZaxis()->SetRangeUser(-0.1,0.1);
489 hist_jet_mass_resolution_change_2D->GetZaxis()->SetLabelSize(0.035);
490 hist_jet_mass_resolution_change_2D->Draw(
"colz");
493 TString output_path_resolution_debug =
"output/debug_plots/resolution_variations/" + sys.name() +
"_resolutionDebug.pdf" ;
494 c2->Print(output_path_resolution_debug);
496 delete hist_jet_mass_resolution_change_2D;
499 jetmass_histograms->Close();
500 delete jetmass_histograms;
502 delete reco_jet_mass_hist;
503 delete matched_truth_jet_mass_hist;
504 delete smeared_reco_jet_mass_hist;
506 delete reco_jet_pt_hist;
507 delete matched_truth_jet_pt_hist;
508 delete smeared_reco_jet_pt_hist;
510 delete reco_jet_rapidity_hist;
511 delete matched_truth_jet_rapidity_hist;
512 delete smeared_reco_jet_rapidity_hist;