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