Skip to content

Commit

Permalink
store the hltobjects for Mass50 triggers
Browse files Browse the repository at this point in the history 
Add back a comment that I accidentally removed

Fix code-format

Fix per #44746 (comment)

Fix per #44746 (comment)

Only store the  trigger objects which pass the mass cut, fix per #44746 (comment)

code format

Change the default value of InputTag l1EGCand to hltEgammaCandidates, per #44746 (comment)

Fix l1EGCand customisation for HIon HLT menu using HLTEgammaCombMassFilter, per #44746 (comment)

fix

Fix per #44746 (comment)
  • Loading branch information
@pinchunchou
pinchunchou committed Jun 28, 2024
1 parent 68816ef commit 4f67362
Show file tree
Hide file tree
Showing 3 changed files with 76 additions and 13 deletions.
8 changes: 8 additions & 0 deletions HLTrigger/Configuration/python/customizeHLTforCMSSW.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -399,6 +399,13 @@ def customizeHLTfor45063(process):

return process

def customizeHLTfor44778(process):
for modLabel in ['hltDoubleEle10Mass50PPOnAAFilter', 'hltDoubleEle15Mass50PPOnAAFilter']:
if hasattr(process, modLabel):
mod = getattr(process, modLabel)
mod.l1EGCand = cms.InputTag('hltEgammaCandidatesPPOnAA')
return process

# CMSSW version specific customizations
def customizeHLTforCMSSW(process, menuType="GRun"):

Expand All @@ -411,5 +418,6 @@ def customizeHLTforCMSSW(process, menuType="GRun"):
process = customizeHLTfor44576(process)
process = customizeHLTfor45063(process)
process = customizeHLTfor45206(process)
process = customizeHLTfor44778(process)

return process
71 changes: 58 additions & 13 deletions HLTrigger/Egamma/plugins/HLTEgammaCombMassFilter.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ HLTEgammaCombMassFilter::HLTEgammaCombMassFilter(const edm::ParameterSet& iConfi
firstLegLastFilterTag_ = iConfig.getParameter<edm::InputTag>("firstLegLastFilter");
secondLegLastFilterTag_ = iConfig.getParameter<edm::InputTag>("secondLegLastFilter");
minMass_ = iConfig.getParameter<double>("minMass");
l1EGTag_ = iConfig.getParameter<edm::InputTag>("l1EGCand");
firstLegLastFilterToken_ = consumes<trigger::TriggerFilterObjectWithRefs>(firstLegLastFilterTag_);
secondLegLastFilterToken_ = consumes<trigger::TriggerFilterObjectWithRefs>(secondLegLastFilterTag_);
}
Expand All @@ -35,6 +36,7 @@ void HLTEgammaCombMassFilter::fillDescriptions(edm::ConfigurationDescriptions& d
makeHLTFilterDescription(desc);
desc.add<edm::InputTag>("firstLegLastFilter", edm::InputTag("firstFilter"));
desc.add<edm::InputTag>("secondLegLastFilter", edm::InputTag("secondFilter"));
desc.add<edm::InputTag>("l1EGCand", edm::InputTag("hltEgammaCandidates"));
desc.add<double>("minMass", -1.0);
descriptions.add("hltEgammaCombMassFilter", desc);
}
Expand All @@ -43,32 +45,75 @@ void HLTEgammaCombMassFilter::fillDescriptions(edm::ConfigurationDescriptions& d
bool HLTEgammaCombMassFilter::hltFilter(edm::Event& iEvent,
const edm::EventSetup& iSetup,
trigger::TriggerFilterObjectWithRefs& filterproduct) const {
//right, issue 1, we dont know if this is a TriggerElectron, TriggerPhoton, TriggerCluster (should never be a TriggerCluster btw as that implies the 4-vectors are not stored in AOD)
using namespace trigger;
if (saveTags()) {
filterproduct.addCollectionTag(l1EGTag_);
}

//trigger::TriggerObjectType firstLegTrigType;
std::vector<math::XYZTLorentzVector> firstLegP4s;

//trigger::TriggerObjectType secondLegTrigType;
std::vector<math::XYZTLorentzVector> secondLegP4s;

math::XYZTLorentzVector pairP4;

getP4OfLegCands(iEvent, firstLegLastFilterToken_, firstLegP4s);
getP4OfLegCands(iEvent, secondLegLastFilterToken_, secondLegP4s);

bool accept = false;
for (auto& firstLegP4 : firstLegP4s) {
for (auto& secondLegP4 : secondLegP4s) {
math::XYZTLorentzVector pairP4 = firstLegP4 + secondLegP4;
std::set<math::XYZTLorentzVector, LorentzVectorComparator> addedLegP4s;

for (size_t i = 0; i < firstLegP4s.size(); i++) {
for (size_t j = 0; j < secondLegP4s.size(); j++) {
// Skip if it's the same object
if (firstLegP4s[i] == secondLegP4s[j])
continue;

math::XYZTLorentzVector pairP4 = firstLegP4s[i] + secondLegP4s[j];
double mass = pairP4.M();
if (mass >= minMass_)
if (mass >= minMass_) {
accept = true;

// Add first leg object if not already added
if (addedLegP4s.insert(firstLegP4s[i]).second) {
addObjectToFilterProduct(iEvent, filterproduct, firstLegLastFilterToken_, i);
}

// Add second leg object if not already added
if (addedLegP4s.insert(secondLegP4s[j]).second) {
addObjectToFilterProduct(iEvent, filterproduct, secondLegLastFilterToken_, j);
}
}
}
}

return accept;
}

void HLTEgammaCombMassFilter::addObjectToFilterProduct(
const edm::Event& iEvent,
trigger::TriggerFilterObjectWithRefs& filterproduct,
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs>& token,
size_t index) {
edm::Handle<trigger::TriggerFilterObjectWithRefs> PrevFilterOutput;
iEvent.getByToken(token, PrevFilterOutput);

// Get all types of objects
std::vector<edm::Ref<reco::RecoEcalCandidateCollection> > phoCandsPrev;
PrevFilterOutput->getObjects(trigger::TriggerPhoton, phoCandsPrev);
std::vector<edm::Ref<reco::RecoEcalCandidateCollection> > clusCandsPrev;
PrevFilterOutput->getObjects(trigger::TriggerCluster, clusCandsPrev);
std::vector<edm::Ref<reco::ElectronCollection> > eleCandsPrev;
PrevFilterOutput->getObjects(trigger::TriggerElectron, eleCandsPrev);

// Check which type of object corresponds to the given index
if (index < phoCandsPrev.size()) {
filterproduct.addObject(trigger::TriggerPhoton, phoCandsPrev[index]);
} else if (index < phoCandsPrev.size() + clusCandsPrev.size()) {
filterproduct.addObject(trigger::TriggerCluster, clusCandsPrev[index - phoCandsPrev.size()]);
} else if (index < phoCandsPrev.size() + clusCandsPrev.size() + eleCandsPrev.size()) {
filterproduct.addObject(trigger::TriggerElectron, eleCandsPrev[index - phoCandsPrev.size() - clusCandsPrev.size()]);
} else {
edm::LogWarning("HLTEgammaCombMassFilter") << "Could not find object at index " << index;
}
}

void HLTEgammaCombMassFilter::getP4OfLegCands(const edm::Event& iEvent,
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs>& filterToken,
std::vector<math::XYZTLorentzVector>& p4s) {
Expand All @@ -84,16 +129,16 @@ void HLTEgammaCombMassFilter::getP4OfLegCands(const edm::Event& iEvent,
filterOutput->getObjects(trigger::TriggerElectron, eleCands);

if (!phoCands.empty()) { //its photons
for (auto& phoCand : phoCands) {
for (auto const& phoCand : phoCands) {
p4s.push_back(phoCand->p4());
}
} else if (!clusCands.empty()) {
//try trigger cluster (should never be this, at the time of writing (17/1/11) this would indicate an error)
for (auto& clusCand : clusCands) {
for (auto const& clusCand : clusCands) {
p4s.push_back(clusCand->p4());
}
} else if (!eleCands.empty()) {
for (auto& eleCand : eleCands) {
for (auto const& eleCand : eleCands) {
p4s.push_back(eleCand->p4());
}
}
Expand Down
10 changes: 10 additions & 0 deletions HLTrigger/Egamma/plugins/HLTEgammaCombMassFilter.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,13 +26,23 @@ class HLTEgammaCombMassFilter : public HLTFilter {
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs>& filterToken,
std::vector<math::XYZTLorentzVector>& p4s);
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
static void addObjectToFilterProduct(const edm::Event& iEvent,
trigger::TriggerFilterObjectWithRefs& filterproduct,
const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs>& token,
size_t index);

private:
edm::InputTag firstLegLastFilterTag_;
edm::InputTag secondLegLastFilterTag_;
edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> firstLegLastFilterToken_;
edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> secondLegLastFilterToken_;
double minMass_;
edm::InputTag l1EGTag_;
struct LorentzVectorComparator {
bool operator()(const math::XYZTLorentzVector& lhs, const math::XYZTLorentzVector& rhs) const {
return lhs.pt() < rhs.pt();
}
};
};

#endif

0 comments on commit 4f67362

Please sign in to comment.