13 |
|
#define MITPHYSICS_MODS_MERGELEPTONSMOD_H |
14 |
|
|
15 |
|
#include "MitAna/TreeMod/interface/BaseMod.h" |
16 |
< |
#include "MitAna/DataTree/interface/Collections.h" |
16 |
> |
#include "MitAna/DataTree/interface/ElectronFwd.h" |
17 |
> |
#include "MitAna/DataTree/interface/MuonFwd.h" |
18 |
> |
#include "MitAna/DataTree/interface/ParticleFwd.h" |
19 |
|
|
20 |
|
namespace mithep |
21 |
|
{ |
24 |
|
public: |
25 |
|
MergeLeptonsMod(const char *name="MergeLeptonsMod", |
26 |
|
const char *title="Merging leptons module"); |
25 |
– |
~MergeLeptonsMod() {} |
27 |
|
|
28 |
< |
const char *GetElectronsName() const { return fElName; } |
29 |
< |
const char *GetMergedName() const { return fMergedName; } |
30 |
< |
const char *GetMuonsName() const { return fMuName; } |
31 |
< |
void SetElectronsName(const char *n) { fElName=n; } |
32 |
< |
void SetMergedName(const char *n) { fMergedName=n; } |
33 |
< |
void SetMuonsName(const char *n) { fMuName=n; } |
28 |
> |
const char *GetElectronsName() const { return fElName; } |
29 |
> |
const char *GetMergedName() const { return fMergedName; } |
30 |
> |
const char *GetMuonsName() const { return fMuName; } |
31 |
> |
const char *GetOutputName() const { return GetMergedName(); } |
32 |
> |
void SetElectronsName(const char *n) { fElName=n; } |
33 |
> |
void SetMergedName(const char *n) { fMergedName=n; } |
34 |
> |
void SetMuonsName(const char *n) { fMuName=n; } |
35 |
> |
void SetOutputName(const char *n) { SetMergedName(n); } |
36 |
|
|
37 |
|
protected: |
38 |
|
void Process(); |
39 |
|
|
40 |
< |
TString fElName; //name of electrons collection |
41 |
< |
TString fMuName; //name of muons collection |
42 |
< |
TString fMergedName; //name of merged collection |
43 |
< |
const ElectronCol *fElIn; //!pointer to electron collection (in) |
44 |
< |
const MuonCol *fMuIn; //!pointer to muon collection (in) |
45 |
< |
ParticleOArr *fColOut; //!pointer to merged collection (out) |
40 |
> |
TString fElName; //name of electrons collection (input) |
41 |
> |
TString fMuName; //name of muons collection (input) |
42 |
> |
TString fMergedName; //name of merged collection (output) |
43 |
> |
const ElectronCol *fElIn; //!pointer to electron collection |
44 |
> |
const MuonCol *fMuIn; //!pointer to muon collection |
45 |
> |
ParticleOArr *fColOut; //!pointer to merged collection |
46 |
|
|
47 |
|
ClassDef(MergeLeptonsMod, 1) // Merging leptons module |
48 |
|
}; |