5 |
|
// |
6 |
|
// Imlementation of a filler to fill EDM tracks into our mithep::Track data structure. |
7 |
|
// |
8 |
+ |
// |
9 |
|
// Authors: J.Bendavid, C.Loizides |
10 |
|
//-------------------------------------------------------------------------------------------------- |
11 |
|
|
16 |
|
#include "DataFormats/TrackReco/interface/TrackFwd.h" |
17 |
|
#include "MitAna/DataUtil/interface/TreeWriter.h" |
18 |
|
#include "MitAna/DataTree/interface/Collections.h" |
19 |
+ |
#include "MitAna/DataTree/interface/Track.h" |
20 |
|
#include "MitProd/TreeFiller/interface/BaseFiller.h" |
21 |
|
#include "MitProd/TreeFiller/interface/AssociationMaps.h" |
22 |
+ |
#include "MitAna/DataTree/interface/Names.h" |
23 |
|
|
24 |
|
namespace mithep |
25 |
|
{ |
26 |
|
class FillerTracks : public BaseFiller |
27 |
|
{ |
28 |
|
public: |
29 |
< |
FillerTracks(const edm::ParameterSet &cfg, const char *name, |
30 |
< |
bool active=1, const SimParticleMap* sm=0); |
31 |
< |
~FillerTracks(); |
32 |
< |
|
33 |
< |
void BookDataBlock(TreeWriter &tws); |
34 |
< |
void FillDataBlock(const edm::Event &e, |
35 |
< |
const edm::EventSetup &es); |
36 |
< |
const TrackMap *GetTrackMap() { return trackMap_; } |
37 |
< |
const TrackCol *GetTrackCol() { return tracks_; } |
38 |
< |
|
29 |
> |
FillerTracks(const edm::ParameterSet &cfg, const char *name, bool active=1, |
30 |
> |
const char *edmName="generalTracks", const char *mitName=Names::gkTrackBrn); |
31 |
> |
virtual ~FillerTracks(); |
32 |
> |
|
33 |
> |
void BookDataBlock(TreeWriter &tws); |
34 |
> |
void FillDataBlock(const edm::Event &e, const edm::EventSetup &es); |
35 |
> |
void InitLayerMap(); |
36 |
> |
|
37 |
> |
protected: |
38 |
> |
std::string edmName_; //edm name of tracks collection |
39 |
> |
std::string mitName_; //name of Tracks in OAK |
40 |
> |
std::string edmSimAssociationName_; //edm name of sim association map |
41 |
> |
std::string trackingMapName_; //name of imported map wrt simparts |
42 |
> |
std::string trackMapName_; //name of export map |
43 |
> |
const mithep::TrackingParticleMap *trackingMap_; //map wrt simulated particles |
44 |
> |
mithep::TrackArr *tracks_; //array of Tracks |
45 |
> |
std::map<uint32_t,mithep::Track::HitLayer> layerMap_; //hitlayer bitmap conversion map |
46 |
> |
|
47 |
|
private: |
48 |
< |
std::string edmName_; |
38 |
< |
std::string mitName_; |
39 |
< |
std::string edmSimAssociationName_; |
40 |
< |
const mithep::SimParticleMap *simMap_; |
41 |
< |
mithep::TrackArr *tracks_; |
42 |
< |
mithep::TrackMap* trackMap_; |
43 |
< |
edm::Handle<reco::TrackCollection> trackProduct_; |
48 |
> |
mithep::TrackMap *trackMap_; //map wrt tracks |
49 |
|
}; |
50 |
|
} |
51 |
|
#endif |