CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
DataMixingGeneralTrackWorker.cc
Go to the documentation of this file.
1 // File: DataMixingGeneralTrackWorker.cc
2 // Description: see DataMixingGeneralTrackWorker.h
3 // Author: Mike Hildreth, University of Notre Dame
4 //
5 //--------------------------------------------
6 
7 #include <map>
15 //
16 //
18 
19 using namespace std;
20 
21 namespace edm
22 {
23 
24  // Virtual constructor
25 
26  DataMixingGeneralTrackWorker::DataMixingGeneralTrackWorker() { }
27 
28  // Constructor
29  DataMixingGeneralTrackWorker::DataMixingGeneralTrackWorker(const edm::ParameterSet& ps)
30  {
31 
32  // get the subdetector names
33  // this->getSubdetectorNames(); //something like this may be useful to check what we are supposed to do...
34 
35  // declare the products to produce
36 
37  GeneralTrackLabelSig_ = ps.getParameter<edm::InputTag>("GeneralTrackLabelSig");
38 
39  GeneralTrackPileInputTag_ = ps.getParameter<edm::InputTag>("GeneralTrackPileInputTag");
40 
41  GeneralTrackCollectionDM_ = ps.getParameter<std::string>("GeneralTrackDigiCollectionDM");
42 
43  }
44 
45 
46  // Virtual destructor needed.
47  DataMixingGeneralTrackWorker::~DataMixingGeneralTrackWorker() {
48  }
49 
50 
51 
52  void DataMixingGeneralTrackWorker::addGeneralTrackSignals(const edm::Event &e) {
53 
54  // Create new track list; Rely on the fact that addSignals gets called first...
55 
56  NewTrackList_ = std::auto_ptr<reco::TrackCollection>(new reco::TrackCollection());
57 
58  // grab tracks, store copy
59 
60  //edm::Handle<reco::TrackCollection> generalTrkHandle;
61  //e.getByLabel("generalTracks", generalTrkHandle);
63  e.getByLabel(GeneralTrackLabelSig_, tracks);
64 
65  if (tracks.isValid()) {
66  for (reco::TrackCollection::const_iterator track = tracks->begin(); track != tracks->end(); ++track) {
67  NewTrackList_->push_back(*track);
68  }
69 
70  }
71 
72  } // end of addGeneralTrackSignals
73 
74 
75 
76  void DataMixingGeneralTrackWorker::addGeneralTrackPileups(const int bcr, const EventPrincipal *ep, unsigned int eventNr) {
77  LogDebug("DataMixingGeneralTrackWorker") <<"\n===============> adding pileups from event "<<ep->id()<<" for bunchcrossing "<<bcr;
78 
79 
80  boost::shared_ptr<Wrapper<reco::TrackCollection > const> inputPTR =
81  getProductByTag<reco::TrackCollection >(*ep, GeneralTrackPileInputTag_ );
82 
83  if(inputPTR ) {
84 
85  const reco::TrackCollection *tracks = const_cast< reco::TrackCollection * >(inputPTR->product());
86 
87  // grab tracks, store copy
88 
89 
90  for (reco::TrackCollection::const_iterator track = tracks->begin(); track != tracks->end(); ++track) {
91  NewTrackList_->push_back(*track);
92  }
93 
94  }
95 
96  }
97 
98 
99 
100  void DataMixingGeneralTrackWorker::putGeneralTrack(edm::Event &e) {
101 
102  // collection of Tracks to put in the event
103 
104  // put the collection of digis in the event
105  LogInfo("DataMixingGeneralTrackWorker") << "total # Merged Tracks: " << NewTrackList_->size() ;
106 
107  // put collection
108 
109  e.put( NewTrackList_, GeneralTrackCollectionDM_ );
110 
111  // clear local storage for this event
112  //NewTrackList_.clear();
113  }
114 
115 } //edm
#define LogDebug(id)
T getParameter(std::string const &) const
EventID const & id() const
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:10
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:94
bool getByLabel(InputTag const &tag, Handle< PROD > &result) const
Definition: Event.h:361
tuple tracks
Definition: testEve_cfg.py:39