CMS 3D CMS Logo
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
NeutronHitsCollector.cc
Go to the documentation of this file.
1 // -*- C++ -*-
2 //
3 // Package: SimMuon/Neutron
4 // Class: NeutronHitsCollector
5 //
15 //
16 // Original Author: Vadim Khotilovich
17 // Created: Mon Aug 09 19:11:42 CST 2010
18 //
19 //
20 
21 #include "FWCore/Framework/interface/Frameworkfwd.h"
22 #include "FWCore/Framework/interface/EDProducer.h"
23 #include "FWCore/Framework/interface/ESHandle.h"
24 
25 #include "FWCore/Framework/interface/Event.h"
26 #include "FWCore/Framework/interface/MakerMacros.h"
27 
28 #include "FWCore/ParameterSet/interface/ParameterSet.h"
29 
30 #include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
31 #include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
32 
33 #include "SimDataFormats/Track/interface/SimTrackContainer.h"
34 #include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
35 
36 
37 
38 class NeutronHitsCollector : public edm::EDProducer
39 {
40 public:
41  explicit NeutronHitsCollector(const edm::ParameterSet&);
42  ~NeutronHitsCollector() {};
43 
44 private:
45  virtual void beginJob() override;
46  virtual void produce(edm::Event&, const edm::EventSetup&) override;
47  virtual void endJob() override;
48 
49  std::string neutron_label_csc;
50  std::string neutron_label_dt;
51  std::string neutron_label_rpc;
52 };
53 
54 
55 
56 NeutronHitsCollector::NeutronHitsCollector(const edm::ParameterSet& iConfig)
57 {
58  neutron_label_csc = iConfig.getUntrackedParameter<std::string>("neutronLabelCSC","");
59  neutron_label_dt = iConfig.getUntrackedParameter<std::string>("neutronLabelDT","");
60  neutron_label_rpc = iConfig.getUntrackedParameter<std::string>("neutronLabelRPC","");
61 
62  // The following list duplicates
63  // http://cmslxr.fnal.gov/lxr/source/SimG4Core/Application/plugins/OscarProducer.cc
64 
65  produces<edm::PSimHitContainer>("MuonDTHits");
66  produces<edm::PSimHitContainer>("MuonCSCHits");
67  produces<edm::PSimHitContainer>("MuonRPCHits");
68 
69  produces<edm::SimTrackContainer>().setBranchAlias("SimTracks");
70  produces<edm::SimVertexContainer>().setBranchAlias("SimVertices");
71 
72  produces<edm::PSimHitContainer>("TrackerHitsPixelBarrelLowTof");
73  produces<edm::PSimHitContainer>("TrackerHitsPixelBarrelHighTof");
74  produces<edm::PSimHitContainer>("TrackerHitsTIBLowTof");
75  produces<edm::PSimHitContainer>("TrackerHitsTIBHighTof");
76  produces<edm::PSimHitContainer>("TrackerHitsTIDLowTof");
77  produces<edm::PSimHitContainer>("TrackerHitsTIDHighTof");
78  produces<edm::PSimHitContainer>("TrackerHitsPixelEndcapLowTof");
79  produces<edm::PSimHitContainer>("TrackerHitsPixelEndcapHighTof");
80  produces<edm::PSimHitContainer>("TrackerHitsTOBLowTof");
81  produces<edm::PSimHitContainer>("TrackerHitsTOBHighTof");
82  produces<edm::PSimHitContainer>("TrackerHitsTECLowTof");
83  produces<edm::PSimHitContainer>("TrackerHitsTECHighTof");
84 
85  produces<edm::PSimHitContainer>("TotemHitsT1");
86  produces<edm::PSimHitContainer>("TotemHitsT2Gem");
87  produces<edm::PSimHitContainer>("TotemHitsRP");
88  produces<edm::PSimHitContainer>("FP420SI");
89  produces<edm::PSimHitContainer>("BSCHits");
90 
91  produces<edm::PCaloHitContainer>("EcalHitsEB");
92  produces<edm::PCaloHitContainer>("EcalHitsEE");
93  produces<edm::PCaloHitContainer>("EcalHitsES");
94  produces<edm::PCaloHitContainer>("HcalHits");
95  produces<edm::PCaloHitContainer>("CaloHitsTk");
96  produces<edm::PCaloHitContainer>("CastorPL");
97  produces<edm::PCaloHitContainer>("CastorFI");
98  produces<edm::PCaloHitContainer>("CastorBU");
99  produces<edm::PCaloHitContainer>("CastorTU");
100  produces<edm::PCaloHitContainer>("EcalTBH4BeamHits");
101  produces<edm::PCaloHitContainer>("HcalTB06BeamHits");
102  produces<edm::PCaloHitContainer>("ZDCHITS");
103  //produces<edm::PCaloHitContainer>("ChamberHits");
104  //produces<edm::PCaloHitContainer>("FibreHits");
105  //produces<edm::PCaloHitContainer>("WedgeHits");
106 
107 }
108 
109 
110 void NeutronHitsCollector::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
111 {
112  edm::PSimHitContainer::const_iterator hit;
113 
114  // ----- MuonCSCHits -----
115  //
116  std::auto_ptr<edm::PSimHitContainer> simCSC(new edm::PSimHitContainer);
117  if (neutron_label_csc.length()>0)
118  {
119  edm::Handle<edm::PSimHitContainer> MuonCSCHits;
120  iEvent.getByLabel(neutron_label_csc,MuonCSCHits);
121  for (hit = MuonCSCHits->begin(); hit != MuonCSCHits->end(); ++hit) simCSC->push_back(*hit);
122  }
123  iEvent.put(simCSC, "MuonCSCHits");
124 
125  // ----- MuonDTHits -----
126  //
127  std::auto_ptr<edm::PSimHitContainer> simDT(new edm::PSimHitContainer);
128  if (neutron_label_dt.length()>0)
129  {
130  edm::Handle<edm::PSimHitContainer> MuonDTHits;
131  iEvent.getByLabel(neutron_label_dt,MuonDTHits);
132  for (hit = MuonDTHits->begin(); hit != MuonDTHits->end(); ++hit) simDT->push_back(*hit);
133  }
134  iEvent.put(simDT, "MuonDTHits");
135 
136  // ----- MuonRPCHits -----
137  //
138  std::auto_ptr<edm::PSimHitContainer> simRPC(new edm::PSimHitContainer);
139  if (neutron_label_rpc.length()>0)
140  {
141  edm::Handle<edm::PSimHitContainer> MuonRPCHits;
142  iEvent.getByLabel(neutron_label_rpc,MuonRPCHits);
143  for (hit = MuonRPCHits->begin(); hit != MuonRPCHits->end(); ++hit) simRPC->push_back(*hit);
144  }
145  iEvent.put(simRPC, "MuonRPCHits");
146 
147  // Further, produce a bunch of empty collections
148 
149  // ----- calorimetry -----
150  //
151  std::auto_ptr<edm::PCaloHitContainer> calout1(new edm::PCaloHitContainer);
152  iEvent.put(calout1, "EcalHitsEB");
153  std::auto_ptr<edm::PCaloHitContainer> calout2(new edm::PCaloHitContainer);
154  iEvent.put(calout2, "EcalHitsEE");
155  std::auto_ptr<edm::PCaloHitContainer> calout3(new edm::PCaloHitContainer);
156  iEvent.put(calout3, "EcalHitsES");
157  std::auto_ptr<edm::PCaloHitContainer> calout4(new edm::PCaloHitContainer);
158  iEvent.put(calout4, "HcalHits");
159  std::auto_ptr<edm::PCaloHitContainer> calout5(new edm::PCaloHitContainer);
160  iEvent.put(calout5, "CaloHitsTk");
161  std::auto_ptr<edm::PCaloHitContainer> calout6(new edm::PCaloHitContainer);
162  iEvent.put(calout6, "CastorPL");
163  std::auto_ptr<edm::PCaloHitContainer> calout7(new edm::PCaloHitContainer);
164  iEvent.put(calout7, "CastorFI");
165  std::auto_ptr<edm::PCaloHitContainer> calout8(new edm::PCaloHitContainer);
166  iEvent.put(calout8, "CastorBU");
167  std::auto_ptr<edm::PCaloHitContainer> calout9(new edm::PCaloHitContainer);
168  iEvent.put(calout9, "CastorTU");
169  std::auto_ptr<edm::PCaloHitContainer> calout10(new edm::PCaloHitContainer);
170  iEvent.put(calout10, "EcalTBH4BeamHits");
171  std::auto_ptr<edm::PCaloHitContainer> calout11(new edm::PCaloHitContainer);
172  iEvent.put(calout11, "HcalTB06BeamHits");
173  std::auto_ptr<edm::PCaloHitContainer> calout12(new edm::PCaloHitContainer);
174  iEvent.put(calout12, "ZDCHITS");
175  //std::auto_ptr<edm::PCaloHitContainer> calout13(new edm::PCaloHitContainer);
176  //iEvent.put(calout13, "ChamberHits");
177  //std::auto_ptr<edm::PCaloHitContainer> calout14(new edm::PCaloHitContainer);
178  //iEvent.put(calout14, "FibreHits");
179  //std::auto_ptr<edm::PCaloHitContainer> calout15(new edm::PCaloHitContainer);
180  //iEvent.put(calout15, "WedgeHits");
181 
182  // ----- Tracker -----
183  //
184  std::auto_ptr<edm::PSimHitContainer> trout1(new edm::PSimHitContainer);
185  iEvent.put(trout1, "TrackerHitsPixelBarrelLowTof");
186  std::auto_ptr<edm::PSimHitContainer> trout2(new edm::PSimHitContainer);
187  iEvent.put(trout2, "TrackerHitsPixelBarrelHighTof");
188  std::auto_ptr<edm::PSimHitContainer> trout3(new edm::PSimHitContainer);
189  iEvent.put(trout3, "TrackerHitsTIBLowTof");
190  std::auto_ptr<edm::PSimHitContainer> trout4(new edm::PSimHitContainer);
191  iEvent.put(trout4, "TrackerHitsTIBHighTof");
192  std::auto_ptr<edm::PSimHitContainer> trout5(new edm::PSimHitContainer);
193  iEvent.put(trout5, "TrackerHitsTIDLowTof");
194  std::auto_ptr<edm::PSimHitContainer> trout6(new edm::PSimHitContainer);
195  iEvent.put(trout6, "TrackerHitsTIDHighTof");
196  std::auto_ptr<edm::PSimHitContainer> trout7(new edm::PSimHitContainer);
197  iEvent.put(trout7, "TrackerHitsPixelEndcapLowTof");
198  std::auto_ptr<edm::PSimHitContainer> trout8(new edm::PSimHitContainer);
199  iEvent.put(trout8, "TrackerHitsPixelEndcapHighTof");
200  std::auto_ptr<edm::PSimHitContainer> trout9(new edm::PSimHitContainer);
201  iEvent.put(trout9, "TrackerHitsTOBLowTof");
202  std::auto_ptr<edm::PSimHitContainer> trout10(new edm::PSimHitContainer);
203  iEvent.put(trout10, "TrackerHitsTOBHighTof");
204  std::auto_ptr<edm::PSimHitContainer> trout11(new edm::PSimHitContainer);
205  iEvent.put(trout11, "TrackerHitsTECLowTof");
206  std::auto_ptr<edm::PSimHitContainer> trout12(new edm::PSimHitContainer);
207  iEvent.put(trout12, "TrackerHitsTECHighTof");
208 
209  // ----- Forward stuff -----
210  //
211  std::auto_ptr<edm::PSimHitContainer> fwout1(new edm::PSimHitContainer);
212  iEvent.put(fwout1, "TotemHitsT1");
213  std::auto_ptr<edm::PSimHitContainer> fwout2(new edm::PSimHitContainer);
214  iEvent.put(fwout2, "TotemHitsT2Gem");
215  std::auto_ptr<edm::PSimHitContainer> fwout3(new edm::PSimHitContainer);
216  iEvent.put(fwout3, "TotemHitsRP");
217  std::auto_ptr<edm::PSimHitContainer> fwout4(new edm::PSimHitContainer);
218  iEvent.put(fwout4, "FP420SI");
219  std::auto_ptr<edm::PSimHitContainer> fwout5(new edm::PSimHitContainer);
220  iEvent.put(fwout5, "BSCHits");
221 
222  // ----- SimTracks & SimVertices -----
223  //
224  std::auto_ptr<edm::SimTrackContainer> simTr(new edm::SimTrackContainer);
225  iEvent.put(simTr);
226  std::auto_ptr<edm::SimVertexContainer> simVe(new edm::SimVertexContainer);
227  iEvent.put(simVe);
228 
229 
230 }
231 
232 
233 void NeutronHitsCollector::beginJob() {}
234 
235 
236 void NeutronHitsCollector::endJob() {}
237 
238 
239 //define this as a plug-in
240 DEFINE_FWK_MODULE(NeutronHitsCollector);
edm::ParameterSet::getUntrackedParameter
T getUntrackedParameter(std::string const &, T const &) const
edm::PCaloHitContainer
std::vector< PCaloHit > PCaloHitContainer
Definition: PCaloHitContainer.h:8
DEFINE_FWK_MODULE
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:17
AlCaHLTBitMon_QueryRunRegistry.string
string string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:255
NeutronHitsCollector::NeutronHitsCollector
NeutronHitsCollector(const edm::ParameterSet &)
Definition: NeutronHitsCollector.cc:56
iEvent
int iEvent
Definition: GenABIO.cc:230
edm::Event::put
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:113
edm::Event::getByLabel
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:402
NeutronHitsCollector::produce
virtual void produce(edm::Event &, const edm::EventSetup &) override
Definition: NeutronHitsCollector.cc:110
NeutronHitsCollector::beginJob
virtual void beginJob() override
Definition: NeutronHitsCollector.cc:233
edm::SimVertexContainer
std::vector< SimVertex > SimVertexContainer
Definition: SimVertexContainer.h:13
NeutronHitsCollector::endJob
virtual void endJob() override
Definition: NeutronHitsCollector.cc:236
edm::PSimHitContainer
std::vector< PSimHit > PSimHitContainer
Definition: PSimHitContainer.h:11
edm::SimTrackContainer
std::vector< SimTrack > SimTrackContainer
Definition: SimTrackContainer.h:12
Generated for CMSSW Reference Manual by doxygen 1.8.5