Keep e+e- from pair production of primary gammas.
[u/mrichter/AliRoot.git] / PWG3 / muon / AliMCMuonPair.h
1 #ifndef ALIMCMUONPAIR_H
2 #define ALIMCMUONPAIR_H
3
4 #include <TLorentzVector.h>
5
6 #include "AliMCMuonTrack.h"
7 #include "AliAODMuonPair.h"
8
9 class AliMCMuonPair : public AliAODMuonPair {
10  public:
11
12   AliMCMuonPair();
13   AliMCMuonPair(AliMCMuonTrack *trk0, AliMCMuonTrack *trk1, Bool_t full=kFALSE);
14   virtual ~AliMCMuonPair();
15
16   AliMCMuonTrack* GetTrack(Int_t i) const { return (i<2 ? (AliMCMuonTrack*)(fTrk[i].GetObject()) : 0x0); }
17
18   TLorentzVector GetPGen() const { return fPGen; }
19   Int_t GetSource() const { return fSource; }
20
21  private:
22
23   void FindDimuonSourceFast();
24   void FindDimuonSourceFull();
25
26   Bool_t fIsFull;
27   TLorentzVector fPGen;
28   Int_t fSource;  // = 0, BBdiff
29                   // = 1, Bchain
30                   // = 2, DDdiff
31                   // = 3, Dchain
32                   // = 4, Resonance
33                   // = 5, UnCorr bkg
34
35   ClassDef(AliMCMuonPair, 1);
36 };
37
38 #endif