Adding an HLT filter based on PPS local track multiplicity conditions for selection of events for PPS PCL

HLTrigger/special/plugins/HLTPPSCalFilter.cc

@@ -0,0 +1,120 @@
+// <author>Mariana Araujo</author>
+// <email>mariana.araujo@cern.ch</email>
+// <created>2020-12-30</created>
+// <description>
+// HLT filter module to select events with min and max multiplicity
+// in each tracker of the PPS for PCL
+// Adapted from a preexisting filter code by Laurent Forthomme
+// </description>
+
+#include "FWCore/Framework/interface/global/EDFilter.h"
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/EventSetup.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+#include "FWCore/Utilities/interface/InputTag.h"
+
+#include "DataFormats/Common/interface/Ref.h"
+#include "DataFormats/Common/interface/Handle.h"
+#include "DataFormats/Common/interface/DetSet.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+
+#include "DataFormats/HLTReco/interface/TriggerTypeDefs.h"
+#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
+
+#include "DataFormats/CTPPSDetId/interface/CTPPSPixelDetId.h"
+
+#include "DataFormats/CTPPSReco/interface/CTPPSPixelLocalTrack.h"  // pixel
+
+#include <unordered_map>
+
+class HLTPPSCalFilter : public edm::global::EDFilter<> {
+public:
+  explicit HLTPPSCalFilter(const edm::ParameterSet&);
+  ~HLTPPSCalFilter() override = default;
+
+  static void fillDescriptions(edm::ConfigurationDescriptions&);
+  bool filter(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
+
+private:
+  const edm::InputTag pixelLocalTrackInputTag_;  // Input tag identifying the pixel detector
+  const edm::EDGetTokenT<edm::DetSetVector<CTPPSPixelLocalTrack>> pixelLocalTrackToken_;
+
+  const int minTracks_;
+  const int maxTracks_;
+
+  const bool do_express_;
+};
+
+void HLTPPSCalFilter::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
+  edm::ParameterSetDescription desc;
+
+  desc.add<edm::InputTag>("pixelLocalTrackInputTag", edm::InputTag("ctppsPixelLocalTracks"))
+      ->setComment("input tag of the pixel local track collection");
+
+  desc.add<int>("minTracks", 1)->setComment("minimum number of tracks in pot");
+  desc.add<int>("maxTracks", -1)->setComment("maximum number of tracks in pot");
+
+  desc.add<bool>("do_express", true)->setComment("toggle on filter type; true for Express, false for Prompt");
+
+  desc.add<int>("triggerType", trigger::TriggerTrack);
+
+  descriptions.add("hltPPSCalFilter", desc);
+}
+
+HLTPPSCalFilter::HLTPPSCalFilter(const edm::ParameterSet& iConfig)
+    : pixelLocalTrackInputTag_(iConfig.getParameter<edm::InputTag>("pixelLocalTrackInputTag")),
+      pixelLocalTrackToken_(consumes<edm::DetSetVector<CTPPSPixelLocalTrack>>(pixelLocalTrackInputTag_)),
+      minTracks_(iConfig.getParameter<int>("minTracks")),
+      maxTracks_(iConfig.getParameter<int>("maxTracks")),
+      do_express_(iConfig.getParameter<bool>("do_express")) {}
+
+bool HLTPPSCalFilter::filter(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
+  // Helper tool to count valid tracks
+  const auto valid_trks = [](const auto& trk) { return trk.isValid(); };
+
+  // map for assigning filter pass / fail
+  std::unordered_map<uint32_t, bool> pixel_filter_;
+
+  // pixel map definition
+  pixel_filter_[CTPPSPixelDetId(0, 0, 3)] = false;
+  pixel_filter_[CTPPSPixelDetId(0, 2, 3)] = false;
+  pixel_filter_[CTPPSPixelDetId(1, 0, 3)] = false;
+  pixel_filter_[CTPPSPixelDetId(1, 2, 3)] = false;
+
+  // filter on pixels (2017+) selection
+  edm::Handle<edm::DetSetVector<CTPPSPixelLocalTrack>> pixelTracks;
+  iEvent.getByToken(pixelLocalTrackToken_, pixelTracks);
+
+  for (const auto& rpv : *pixelTracks) {
+    if (pixel_filter_.count(rpv.id) == 0) {
+      continue;
+    }
+    // assume pass condition if there is at least one track
+    pixel_filter_.at(rpv.id) = true;
+
+    // count number of valid tracks
+    const auto ntrks = std::count_if(rpv.begin(), rpv.end(), valid_trks);
+    // fail condition if ntrks not within minTracks, maxTracks values
+    if (minTracks_ > 0 && ntrks < minTracks_)
+      pixel_filter_.at(rpv.id) = false;
+    if (maxTracks_ > 0 && ntrks > maxTracks_)
+      pixel_filter_.at(rpv.id) = false;
+  }
+
+  // compilation of filter conditions
+  if (do_express_) {
+    return (pixel_filter_.at(CTPPSPixelDetId(0, 0, 3)) && pixel_filter_.at(CTPPSPixelDetId(0, 2, 3))) ||
+           (pixel_filter_.at(CTPPSPixelDetId(1, 0, 3)) && pixel_filter_.at(CTPPSPixelDetId(1, 2, 3)));
+  } else {
+    return (pixel_filter_.at(CTPPSPixelDetId(0, 0, 3)) || pixel_filter_.at(CTPPSPixelDetId(0, 2, 3))) ||
+           (pixel_filter_.at(CTPPSPixelDetId(1, 0, 3)) || pixel_filter_.at(CTPPSPixelDetId(1, 2, 3)));
+  }
+
+  return false;
+}
+
+// define as a framework module
+#include "FWCore/Framework/interface/MakerMacros.h"
+DEFINE_FWK_MODULE(HLTPPSCalFilter);