| 1 |
mlethuil |
1.1 |
#ifndef ParticleTreeDrawer_h
|
| 2 |
|
|
#define ParticleTreeDrawer_h
|
| 3 |
|
|
//
|
| 4 |
|
|
// class ParticleTreeDrawer
|
| 5 |
|
|
// Adapted from Luca Lista's plugin
|
| 6 |
|
|
//
|
| 7 |
|
|
#include "FWCore/Framework/interface/EDAnalyzer.h"
|
| 8 |
|
|
#include "FWCore/ParameterSet/interface/InputTag.h"
|
| 9 |
|
|
#include "FWCore/Framework/interface/ESHandle.h"
|
| 10 |
|
|
#include "SimGeneral/HepPDTRecord/interface/ParticleDataTable.h"
|
| 11 |
|
|
#include "DataFormats/Candidate/interface/CandidateFwd.h"
|
| 12 |
|
|
|
| 13 |
|
|
class ParticleTreeDrawer : public edm::EDAnalyzer {
|
| 14 |
|
|
public:
|
| 15 |
|
|
ParticleTreeDrawer();
|
| 16 |
|
|
void analyze( const edm::Event &, const edm::EventSetup & );
|
| 17 |
|
|
void setPrintOptions( bool printP4, bool printPtEtaPhi, bool printVertex, bool printStatus )
|
| 18 |
|
|
{ printP4_ = printP4; printPtEtaPhi_ = printPtEtaPhi; printVertex_ = printVertex; printStatus_ = printStatus; };
|
| 19 |
|
|
|
| 20 |
|
|
private:
|
| 21 |
|
|
edm::InputTag src_;
|
| 22 |
|
|
edm::ESHandle<ParticleDataTable> pdt_;
|
| 23 |
|
|
|
| 24 |
|
|
bool printP4_, printPtEtaPhi_, printVertex_, printStatus_;
|
| 25 |
|
|
void printDecay( const reco::Candidate &, const std::string & pre ) const;
|
| 26 |
|
|
void printP4( const reco::Candidate & ) const;
|
| 27 |
|
|
};
|
| 28 |
|
|
|
| 29 |
|
|
#endif
|
