| 8 |
|
// Authors: J.Bendavid |
| 9 |
|
//-------------------------------------------------------------------------------------------------- |
| 10 |
|
|
| 11 |
< |
#ifndef TREEFILLER_FILLERGSFTRACKS_H |
| 12 |
< |
#define TREEFILLER_FILLERGSFTRACKS_H |
| 11 |
> |
#ifndef MITPROD_TREEFILLER_FILLERGSFTRACKS_H |
| 12 |
> |
#define MITPROD_TREEFILLER_FILLERGSFTRACKS_H |
| 13 |
|
|
| 14 |
|
#include "FWCore/ParameterSet/interface/ParameterSet.h" |
| 15 |
– |
#include "MitAna/DataUtil/interface/TreeWriter.h" |
| 16 |
– |
#include "MitAna/DataTree/interface/Track.h" |
| 15 |
|
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h" |
| 16 |
|
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h" |
| 17 |
< |
#include "MitAna/DataTree/interface/Collections.h" |
| 20 |
< |
#include "MitProd/TreeFiller/interface/BaseFiller.h" |
| 21 |
< |
#include "MitProd/TreeFiller/interface/AssociationMaps.h" |
| 17 |
> |
#include "MitProd/TreeFiller/interface/FillerTracks.h" |
| 18 |
|
|
| 19 |
|
namespace mithep |
| 20 |
|
{ |
| 21 |
< |
class FillerGsfTracks : public BaseFiller |
| 21 |
> |
class FillerGsfTracks : public FillerTracks |
| 22 |
|
{ |
| 23 |
|
public: |
| 24 |
< |
FillerGsfTracks(const edm::ParameterSet &cfg, const char *name, |
| 29 |
< |
bool active=1, const SimParticleMap *sm=0); |
| 24 |
> |
FillerGsfTracks(const edm::ParameterSet &cfg, const char *name, bool active=1, bool ecalActive=0); |
| 25 |
|
~FillerGsfTracks(); |
| 26 |
|
|
| 27 |
< |
void BookDataBlock(TreeWriter &tws); |
| 28 |
< |
void FillDataBlock(const edm::Event &e, const edm::EventSetup &es); |
| 29 |
< |
const GsfTrackMap *GetTrackMap() const { return trackMap_; } |
| 35 |
< |
const TrackCol *GetTrackCol() const { return tracks_; } |
| 27 |
> |
void BookDataBlock(TreeWriter &tws); |
| 28 |
> |
void FillDataBlock(const edm::Event &e, const edm::EventSetup &es); |
| 29 |
> |
void InitLayerMap(); |
| 30 |
|
|
| 31 |
|
private: |
| 32 |
< |
std::string edmName_; |
| 39 |
< |
std::string mitName_; |
| 40 |
< |
std::string edmSimAssociationName_; |
| 41 |
< |
const mithep::SimParticleMap *simMap_; |
| 42 |
< |
mithep::Array<mithep::Track> *tracks_; |
| 43 |
< |
mithep::GsfTrackMap *trackMap_; |
| 44 |
< |
edm::Handle<reco::GsfTrackCollection> trackProduct_; |
| 32 |
> |
mithep::GsfTrackMap *trackMap_; //map wrt tracks |
| 33 |
|
}; |
| 34 |
|
} |
| 35 |
|
#endif |
