1 |
lethuill |
1.1 |
#ifndef MCAssociator_h
|
2 |
|
|
#define MCAssociator_h
|
3 |
|
|
|
4 |
|
|
#include <memory>
|
5 |
|
|
#include <string>
|
6 |
|
|
#include <iostream>
|
7 |
|
|
#include <map>
|
8 |
|
|
|
9 |
|
|
#include "../interface/TRootParticle.h"
|
10 |
|
|
#include "../interface/TRootMCParticle.h"
|
11 |
|
|
#include "../interface/TRootElectron.h"
|
12 |
|
|
|
13 |
|
|
#include "TClonesArray.h"
|
14 |
|
|
|
15 |
|
|
|
16 |
|
|
class MCAssociator{
|
17 |
|
|
|
18 |
|
|
public:
|
19 |
|
|
MCAssociator():verbosity_(0), mcParticles_(0), mcMap_() {;}
|
20 |
|
|
MCAssociator(int verbosity):verbosity_(verbosity), mcParticles_(0), mcMap_() {;}
|
21 |
|
|
~MCAssociator() {};
|
22 |
|
|
void setVerbosity(int verbosity) {verbosity_ = verbosity; };
|
23 |
|
|
void init(TClonesArray* mcParticles);
|
24 |
|
|
void process(TClonesArray* recoParticles);
|
25 |
|
|
void printParticleAssociation(TClonesArray* recoParticles);
|
26 |
|
|
|
27 |
|
|
private:
|
28 |
|
|
int verbosity_;
|
29 |
|
|
TClonesArray* mcParticles_;
|
30 |
|
|
std::map<int,int> mcMap_;
|
31 |
|
|
|
32 |
|
|
};
|
33 |
|
|
|
34 |
|
|
#endif
|