File indexing completed on 2024-12-08 23:44:27
0001 #ifndef Validation_HGCalValidation_TICLCandidateValidator_h
0002 #define Validation_HGCalValidation_TICLCandidateValidator_h
0003
0004 #include <iostream>
0005 #include <vector>
0006 #include <unordered_map>
0007
0008 #include "FWCore/Framework/interface/ConsumesCollector.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0010 #include "FWCore/Framework/interface/EventSetup.h"
0011 #include "FWCore/Framework/interface/Event.h"
0012 #include "FWCore/Utilities/interface/EDGetToken.h"
0013 #include "FWCore/Framework/interface/Frameworkfwd.h"
0014
0015 #include "DataFormats/TrackReco/interface/Track.h"
0016 #include "DataFormats/HGCalReco/interface/Trackster.h"
0017 #include "DataFormats/HGCalReco/interface/TICLCandidate.h"
0018
0019 #include "SimDataFormats/Associations/interface/TICLAssociationMap.h"
0020
0021 #include "DQMServices/Core/interface/DQMStore.h"
0022
0023 namespace ticl {
0024 using TracksterToTracksterMap =
0025 ticl::AssociationMap<ticl::mapWithSharedEnergyAndScore, std::vector<ticl::Trackster>, std::vector<ticl::Trackster>>;
0026 }
0027
0028 struct TICLCandidateValidatorHistograms {
0029 dqm::reco::MonitorElement* h_tracksters_in_candidate;
0030 dqm::reco::MonitorElement* h_candidate_raw_energy;
0031 dqm::reco::MonitorElement* h_candidate_regressed_energy;
0032 dqm::reco::MonitorElement* h_candidate_pT;
0033 dqm::reco::MonitorElement* h_candidate_charge;
0034 dqm::reco::MonitorElement* h_candidate_pdgId;
0035 dqm::reco::MonitorElement* h_candidate_partType;
0036
0037 std::vector<dqm::reco::MonitorElement*> h_den_chg_energy_candidate;
0038 std::vector<dqm::reco::MonitorElement*> h_num_chg_energy_candidate_track;
0039 std::vector<dqm::reco::MonitorElement*> h_num_chg_energy_candidate_pdgId;
0040 std::vector<dqm::reco::MonitorElement*> h_num_chg_energy_candidate_energy;
0041 std::vector<dqm::reco::MonitorElement*> h_den_chg_pt_candidate;
0042 std::vector<dqm::reco::MonitorElement*> h_num_chg_pt_candidate_track;
0043 std::vector<dqm::reco::MonitorElement*> h_num_chg_pt_candidate_pdgId;
0044 std::vector<dqm::reco::MonitorElement*> h_num_chg_pt_candidate_energy;
0045 std::vector<dqm::reco::MonitorElement*> h_den_chg_eta_candidate;
0046 std::vector<dqm::reco::MonitorElement*> h_num_chg_eta_candidate_track;
0047 std::vector<dqm::reco::MonitorElement*> h_num_chg_eta_candidate_pdgId;
0048 std::vector<dqm::reco::MonitorElement*> h_num_chg_eta_candidate_energy;
0049 std::vector<dqm::reco::MonitorElement*> h_den_chg_phi_candidate;
0050 std::vector<dqm::reco::MonitorElement*> h_num_chg_phi_candidate_track;
0051 std::vector<dqm::reco::MonitorElement*> h_num_chg_phi_candidate_pdgId;
0052 std::vector<dqm::reco::MonitorElement*> h_num_chg_phi_candidate_energy;
0053
0054 std::vector<dqm::reco::MonitorElement*> h_den_neut_energy_candidate;
0055 std::vector<dqm::reco::MonitorElement*> h_num_neut_energy_candidate_pdgId;
0056 std::vector<dqm::reco::MonitorElement*> h_num_neut_energy_candidate_energy;
0057 std::vector<dqm::reco::MonitorElement*> h_den_neut_pt_candidate;
0058 std::vector<dqm::reco::MonitorElement*> h_num_neut_pt_candidate_pdgId;
0059 std::vector<dqm::reco::MonitorElement*> h_num_neut_pt_candidate_energy;
0060 std::vector<dqm::reco::MonitorElement*> h_den_neut_eta_candidate;
0061 std::vector<dqm::reco::MonitorElement*> h_num_neut_eta_candidate_pdgId;
0062 std::vector<dqm::reco::MonitorElement*> h_num_neut_eta_candidate_energy;
0063 std::vector<dqm::reco::MonitorElement*> h_den_neut_phi_candidate;
0064 std::vector<dqm::reco::MonitorElement*> h_num_neut_phi_candidate_pdgId;
0065 std::vector<dqm::reco::MonitorElement*> h_num_neut_phi_candidate_energy;
0066
0067 std::vector<dqm::reco::MonitorElement*> h_den_fake_chg_energy_candidate;
0068 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_energy_candidate_track;
0069 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_energy_candidate_pdgId;
0070 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_energy_candidate_energy;
0071 std::vector<dqm::reco::MonitorElement*> h_den_fake_chg_pt_candidate;
0072 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_pt_candidate_track;
0073 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_pt_candidate_pdgId;
0074 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_pt_candidate_energy;
0075 std::vector<dqm::reco::MonitorElement*> h_den_fake_chg_eta_candidate;
0076 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_eta_candidate_track;
0077 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_eta_candidate_pdgId;
0078 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_eta_candidate_energy;
0079 std::vector<dqm::reco::MonitorElement*> h_den_fake_chg_phi_candidate;
0080 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_phi_candidate_track;
0081 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_phi_candidate_pdgId;
0082 std::vector<dqm::reco::MonitorElement*> h_num_fake_chg_phi_candidate_energy;
0083
0084 std::vector<dqm::reco::MonitorElement*> h_den_fake_neut_energy_candidate;
0085 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_energy_candidate_pdgId;
0086 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_energy_candidate_energy;
0087 std::vector<dqm::reco::MonitorElement*> h_den_fake_neut_pt_candidate;
0088 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_pt_candidate_pdgId;
0089 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_pt_candidate_energy;
0090 std::vector<dqm::reco::MonitorElement*> h_den_fake_neut_eta_candidate;
0091 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_eta_candidate_pdgId;
0092 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_eta_candidate_energy;
0093 std::vector<dqm::reco::MonitorElement*> h_den_fake_neut_phi_candidate;
0094 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_phi_candidate_pdgId;
0095 std::vector<dqm::reco::MonitorElement*> h_num_fake_neut_phi_candidate_energy;
0096
0097 std::vector<dqm::reco::MonitorElement*> h_chg_tracksters_in_candidate;
0098 std::vector<dqm::reco::MonitorElement*> h_chg_candidate_regressed_energy;
0099 std::vector<dqm::reco::MonitorElement*> h_chg_candidate_charge;
0100 std::vector<dqm::reco::MonitorElement*> h_chg_candidate_pdgId;
0101 std::vector<dqm::reco::MonitorElement*> h_chg_candidate_partType;
0102
0103 std::vector<dqm::reco::MonitorElement*> h_neut_tracksters_in_candidate;
0104 std::vector<dqm::reco::MonitorElement*> h_neut_candidate_regressed_energy;
0105 std::vector<dqm::reco::MonitorElement*> h_neut_candidate_charge;
0106 std::vector<dqm::reco::MonitorElement*> h_neut_candidate_pdgId;
0107 std::vector<dqm::reco::MonitorElement*> h_neut_candidate_partType;
0108 };
0109
0110 class TICLCandidateValidator {
0111 public:
0112 typedef dqm::legacy::DQMStore DQMStore;
0113 typedef dqm::legacy::MonitorElement MonitorElement;
0114
0115 TICLCandidateValidator() {};
0116 TICLCandidateValidator(edm::EDGetTokenT<std::vector<TICLCandidate>> TICLCandidates,
0117 edm::EDGetTokenT<std::vector<TICLCandidate>> simTICLCandidatesToken,
0118 edm::EDGetTokenT<std::vector<reco::Track>> recoTracksToken,
0119 edm::EDGetTokenT<std::vector<ticl::Trackster>> trackstersToken,
0120 edm::EDGetTokenT<ticl::TracksterToTracksterMap> associatorMapRtSToken,
0121 edm::EDGetTokenT<ticl::TracksterToTracksterMap> associatorMapStRToken,
0122 bool isTICLv5);
0123 ~TICLCandidateValidator();
0124
0125 using Histograms = TICLCandidateValidatorHistograms;
0126
0127 void bookCandidatesHistos(DQMStore::IBooker& ibook, Histograms& histograms, std::string baseDir) const;
0128
0129 void fillCandidateHistos(const edm::Event& event,
0130 const Histograms& histograms,
0131 edm::Handle<ticl::TracksterCollection> simTrackstersCP_h) const;
0132
0133 private:
0134 edm::EDGetTokenT<std::vector<TICLCandidate>> TICLCandidatesToken_;
0135 edm::EDGetTokenT<std::vector<TICLCandidate>> simTICLCandidatesToken_;
0136 edm::EDGetTokenT<std::vector<reco::Track>> recoTracksToken_;
0137 edm::EDGetTokenT<std::vector<ticl::Trackster>> trackstersToken_;
0138 edm::EDGetTokenT<ticl::TracksterToTracksterMap> associatorMapRtSToken_;
0139 edm::EDGetTokenT<ticl::TracksterToTracksterMap> associatorMapStRToken_;
0140 bool isTICLv5_ = false;
0141 };
0142
0143 #endif
