| 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
|
