Merge pull request #17223 from kskovpen/CMSSW_7_1_X

Add SIM muon filter
cms-sw · Feb 13, 2017 · 695f219 · 695f219
2 parents 7177f80 + 534872d
commit 695f219
Show file tree

Hide file tree

Showing 3 changed files with 129 additions and 0 deletions.
diff --git a/SimMuon/MCTruth/plugins/SimMuFilter.cc b/SimMuon/MCTruth/plugins/SimMuFilter.cc
@@ -0,0 +1,110 @@
+#include "FWCore/Framework/interface/stream/EDFilter.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+#include "SimDataFormats/Track/interface/SimTrackContainer.h"
+#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
+
+class SimMuFilter : public edm::stream::EDFilter<>
+{
+ public:
+
+   explicit SimMuFilter(const edm::ParameterSet&);
+   ~SimMuFilter();
+
+ private:
+
+   virtual void beginJob();
+   virtual void endJob();
+   virtual bool filter(edm::Event&, const edm::EventSetup&);
+
+ private:
+
+   edm::EDGetTokenT<std::vector<SimTrack> > simTracksToken_;
+   edm::EDGetTokenT<edm::PSimHitContainer> simHitsMuonRPCToken_;
+   edm::EDGetTokenT<edm::PSimHitContainer> simHitsMuonCSCToken_;
+   edm::EDGetTokenT<edm::PSimHitContainer> simHitsMuonDTToken_;
+
+   edm::Handle<std::vector<SimTrack> > simTracksHandle;
+   edm::Handle<edm::PSimHitContainer> simHitsMuonRPCHandle;
+   edm::Handle<edm::PSimHitContainer> simHitsMuonCSCHandle;
+   edm::Handle<edm::PSimHitContainer> simHitsMuonDTHandle;
+};
+
+SimMuFilter::SimMuFilter(const edm::ParameterSet& iConfig)
+{
+   simTracksToken_    = consumes<std::vector<SimTrack> >(iConfig.getParameter<edm::InputTag>("simTracksInput"));
+   simHitsMuonRPCToken_ = consumes<edm::PSimHitContainer>(iConfig.getParameter<edm::InputTag>("simHitsMuonRPCInput"));
+   simHitsMuonCSCToken_ = consumes<edm::PSimHitContainer>(iConfig.getParameter<edm::InputTag>("simHitsMuonCSCInput"));
+   simHitsMuonDTToken_ = consumes<edm::PSimHitContainer>(iConfig.getParameter<edm::InputTag>("simHitsMuonDTInput"));
+}
+
+SimMuFilter::~SimMuFilter()
+{
+}
+
+bool SimMuFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
+{
+   using namespace edm;
+
+   iEvent.getByToken(simTracksToken_,simTracksHandle);
+   iEvent.getByToken(simHitsMuonRPCToken_,simHitsMuonRPCHandle);
+   iEvent.getByToken(simHitsMuonCSCToken_,simHitsMuonCSCHandle);
+   iEvent.getByToken(simHitsMuonDTToken_,simHitsMuonDTHandle);
+
+   const std::vector<SimTrack> &simTracks = *simTracksHandle.product();
+
+   int nTracks = simTracks.size();
+
+   bool pass = 0;
+
+   for(int it=0;it<nTracks;it++)
+     {
+	SimTrack simTrk = simTracks[it];
+
+	int pdgId = simTrk.type();
+	float pt = simTrk.momentum().pt();
+
+	if( abs(pdgId) != 13 ) continue;
+	if( pt < 3. ) continue;
+
+	int nSimHitRPC = 0;
+	int nSimHitCSC = 0;
+	int nSimHitDT = 0;
+
+	for( PSimHitContainer::const_iterator simHitIt = simHitsMuonRPCHandle->begin();simHitIt!=simHitsMuonRPCHandle->end();simHitIt++ )
+	  {
+	     if( simHitIt->trackId() != simTrk.trackId() ) continue;
+
+	     nSimHitRPC++;
+	  }	     
+
+	for( PSimHitContainer::const_iterator simHitIt = simHitsMuonCSCHandle->begin();simHitIt!=simHitsMuonCSCHandle->end();simHitIt++ )
+	  {
+	     if( simHitIt->trackId() != simTrk.trackId() ) continue;
+
+	     nSimHitCSC++;
+	  }
+
+	for( PSimHitContainer::const_iterator simHitIt = simHitsMuonDTHandle->begin();simHitIt!=simHitsMuonDTHandle->end();simHitIt++ )
+	  {
+	     if( simHitIt->trackId() != simTrk.trackId() ) continue;
+
+	     nSimHitDT++;
+	  }
+
+	if( nSimHitRPC+nSimHitCSC+nSimHitDT > 0 ) pass = 1;
+     }
+
+   return pass;
+}
+
+void SimMuFilter::beginJob()
+{
+}
+
+void SimMuFilter::endJob()
+{
+}
+
+//define this as a plug-in
+DEFINE_FWK_MODULE(SimMuFilter);
+
diff --git a/SimMuon/MCTruth/python/SimMuFilter_cfi.py b/SimMuon/MCTruth/python/SimMuFilter_cfi.py
@@ -0,0 +1,10 @@
+import FWCore.ParameterSet.Config as cms
+
+SimMuFilter = cms.EDFilter('SimMuFilter',
+
+              simTracksInput = cms.InputTag("g4SimHits","","SIM"),
+              simHitsMuonRPCInput = cms.InputTag("g4SimHits","MuonRPCHits","SIM"),
+              simHitsMuonCSCInput = cms.InputTag("g4SimHits","MuonCSCHits","SIM"),
+              simHitsMuonDTInput = cms.InputTag("g4SimHits","MuonDTHits","SIM")
+)
+
diff --git a/SimMuon/MCTruth/test/SimMuFilter_cfg.py b/SimMuon/MCTruth/test/SimMuFilter_cfg.py
@@ -0,0 +1,9 @@
+import FWCore.ParameterSet.Config as cms
+
+from SimMuon.MCTruth.SimMuFilter_cfi import *
+
+process = cms.Process("SimFilter")
+
+process.filter = SimMuFilter
+
+process.p = cms.Path(process.filter)