[HGCAL] MultiCluster to CaloParticle associator

DataFormats/ParticleFlowReco/interface/HGCalMultiCluster.h

@@ -31,5 +31,7 @@ namespace reco {
   private:
     edm::PtrVector<reco::BasicCluster> myclusters;
   };
+
+  typedef std::vector<HGCalMultiCluster> HGCalMultiClusterCollection;
 }  // namespace reco
 #endif

SimCalorimetry/HGCalAssociatorProducers/interface/AssociatorTools.h

@@ -4,12 +4,14 @@
 #include "SimDataFormats/CaloAnalysis/interface/CaloParticle.h"
 #include <vector>
 
-void removeCPFromPU(const std::vector<CaloParticle>& caloParticles, std::vector<size_t>& cPIndices) {
+static void removeCPFromPU(const std::vector<CaloParticle>& caloParticles,
+                           std::vector<size_t>& cPIndices,
+                           bool hardScatterOnly = true) {
   //Consider CaloParticles coming from the hard scatterer
   //excluding the PU contribution and save the indices.
   for (unsigned int cpId = 0; cpId < caloParticles.size(); ++cpId) {
-    if (caloParticles[cpId].g4Tracks()[0].eventId().event() != 0 or
-        caloParticles[cpId].g4Tracks()[0].eventId().bunchCrossing() != 0) {
+    if (hardScatterOnly && (caloParticles[cpId].g4Tracks()[0].eventId().event() != 0 or
+                            caloParticles[cpId].g4Tracks()[0].eventId().bunchCrossing() != 0)) {
       LogDebug("HGCalValidator") << "Excluding CaloParticles from event: "
                                  << caloParticles[cpId].g4Tracks()[0].eventId().event()
                                  << " with BX: " << caloParticles[cpId].g4Tracks()[0].eventId().bunchCrossing()
@@ -19,4 +21,5 @@ void removeCPFromPU(const std::vector<CaloParticle>& caloParticles, std::vector<
     cPIndices.emplace_back(cpId);
   }
 }
+
 #endif

SimCalorimetry/HGCalAssociatorProducers/plugins/LayerClusterAssociatorByEnergyScoreImpl.cc

@@ -28,7 +28,7 @@ hgcal::association LayerClusterAssociatorByEnergyScoreImpl::makeConnections(
   std::vector<size_t> cPIndices;
   //Consider CaloParticles coming from the hard scatterer
   //excluding the PU contribution and save the indices.
-  removeCPFromPU(caloParticles, cPIndices);
+  removeCPFromPU(caloParticles, cPIndices, hardScatterOnly_);
   auto nCaloParticles = cPIndices.size();
 
   // Initialize cPOnLayer. To be returned outside, since it contains the

SimCalorimetry/HGCalAssociatorProducers/plugins/MCToCPAssociatorEDProducer.cc

@@ -0,0 +1,94 @@
+//
+// Original Author:  Leonardo Cristella
+//         Created:  Tue Feb  2 10:52:11 CET 2021
+//
+//
+
+// system include files
+#include <memory>
+#include <string>
+
+// user include files
+#include "FWCore/Framework/interface/global/EDProducer.h"
+
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+
+#include "FWCore/Framework/interface/ESHandle.h"
+
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+
+#include "SimDataFormats/Associations/interface/MultiClusterToCaloParticleAssociator.h"
+
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+#include "SimDataFormats/CaloAnalysis/interface/CaloParticleFwd.h"
+#include "DataFormats/ParticleFlowReco/interface/HGCalMultiCluster.h"
+
+#include "FWCore/Utilities/interface/EDGetToken.h"
+
+//
+// class decleration
+//
+
+class MCToCPAssociatorEDProducer : public edm::global::EDProducer<> {
+public:
+  explicit MCToCPAssociatorEDProducer(const edm::ParameterSet &);
+  ~MCToCPAssociatorEDProducer() override;
+
+private:
+  void produce(edm::StreamID, edm::Event &, const edm::EventSetup &) const override;
+
+  edm::EDGetTokenT<CaloParticleCollection> CPCollectionToken_;
+  edm::EDGetTokenT<reco::HGCalMultiClusterCollection> MCCollectionToken_;
+  edm::EDGetTokenT<hgcal::MultiClusterToCaloParticleAssociator> associatorToken_;
+};
+
+MCToCPAssociatorEDProducer::MCToCPAssociatorEDProducer(const edm::ParameterSet &pset) {
+  produces<hgcal::SimToRecoCollectionWithMultiClusters>();
+  produces<hgcal::RecoToSimCollectionWithMultiClusters>();
+
+  CPCollectionToken_ = consumes<CaloParticleCollection>(pset.getParameter<edm::InputTag>("label_cp"));
+  MCCollectionToken_ = consumes<reco::HGCalMultiClusterCollection>(pset.getParameter<edm::InputTag>("label_mcl"));
+  associatorToken_ =
+      consumes<hgcal::MultiClusterToCaloParticleAssociator>(pset.getParameter<edm::InputTag>("associator"));
+}
+
+MCToCPAssociatorEDProducer::~MCToCPAssociatorEDProducer() {}
+
+//
+// member functions
+//
+
+// ------------ method called to produce the data  ------------
+void MCToCPAssociatorEDProducer::produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const {
+  using namespace edm;
+
+  edm::Handle<hgcal::MultiClusterToCaloParticleAssociator> theAssociator;
+  iEvent.getByToken(associatorToken_, theAssociator);
+
+  Handle<CaloParticleCollection> CPCollection;
+  iEvent.getByToken(CPCollectionToken_, CPCollection);
+
+  Handle<reco::HGCalMultiClusterCollection> MCCollection;
+  iEvent.getByToken(MCCollectionToken_, MCCollection);
+
+  // associate MutiCluster and CP
+  LogTrace("AssociatorValidator") << "Calling associateRecoToSim method"
+                                  << "\n";
+  hgcal::RecoToSimCollectionWithMultiClusters recSimColl =
+      theAssociator->associateRecoToSim(MCCollection, CPCollection);
+
+  LogTrace("AssociatorValidator") << "Calling associateSimToReco method"
+                                  << "\n";
+  hgcal::SimToRecoCollectionWithMultiClusters simRecColl =
+      theAssociator->associateSimToReco(MCCollection, CPCollection);
+
+  auto rts = std::make_unique<hgcal::RecoToSimCollectionWithMultiClusters>(recSimColl);
+  auto str = std::make_unique<hgcal::SimToRecoCollectionWithMultiClusters>(simRecColl);
+
+  iEvent.put(std::move(rts));
+  iEvent.put(std::move(str));
+}
+
+// define this as a plug-in
+DEFINE_FWK_MODULE(MCToCPAssociatorEDProducer);