PWG/muon/AliDimuInfoStoreMC.h

   1 #ifndef ALIDIMUINFOSTOREMC_H
   2 #define ALIDIMUINFOSTOREMC_H
   3
   4 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8
   9 //***********************************************************
  10 // Class AliDimuInfoStoreMC
  11 // class used to extract and store info of MC particles
  12 // Author: X-M. Zhang, zhang@clermont.in2p3.fr
  13 //                     zhangxm@iopp.ccnu.edu.cn
  14 //***********************************************************
  15
  16 #include <TLorentzVector.h>
  17 #include "AliDimuInfoStoreRD.h"
  18
  19 class AliMuonInfoStoreMC;
  20 class AliDimuInfoStoreMC : public AliDimuInfoStoreRD {
  21  public:
  22
  23   AliDimuInfoStoreMC();
  24   AliDimuInfoStoreMC(AliMuonInfoStoreMC *trk0, AliMuonInfoStoreMC *trk1, Bool_t full=kFALSE);
  25   AliDimuInfoStoreMC(const AliDimuInfoStoreMC &src);
  26   AliDimuInfoStoreMC& operator=(const AliDimuInfoStoreMC &src);
  27   virtual ~AliDimuInfoStoreMC();
  28
  29   AliMuonInfoStoreMC* Muon(Int_t i) const { return (i<2 ? (AliMuonInfoStoreMC*)(fMuonRef[i].GetObject()) : 0x0); }
  30
  31   TLorentzVector LorentzP() const { return fLorentzP; }
  32   Int_t Source() const { return fSource; }
  33
  34   static const char* StdBranchName() { return fgkStdBranchName.Data(); }
  35   static Int_t SourcesN()            { return fgkSourcesN;             }
  36
  37
  38  private:
  39
  40   void FindDimuonSourceFast();
  41   void FindDimuonSourceFull();
  42
  43   static const TString fgkStdBranchName;  // Standard branch name
  44   static const Int_t   fgkSourcesN;       // num. of dimuon sources
  45
  46   Bool_t fIsFull;  // flag of using full analysis (for Pb-Pb)
  47   TLorentzVector fLorentzP;  // lorentz momentum of MC particle
  48   Int_t fSource;  // = 0, BBdiff
  49                   // = 1, Bchain
  50                   // = 2, DDdiff
  51                   // = 3, Dchain
  52                   // = 4, Resonance
  53                   // = 5, UnCorr bkg
  54
  55   ClassDef(AliDimuInfoStoreMC, 5);
  56 };
  57
  58 #endif