PWGHF/hfe/AliHFEsignalCuts.h

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15 //
  16 // Signal cuts
  17 // Checks whether a particle (reconstructed or MC) is coming from MC Signal
  18 // For more information see implementation file
  19 //
  20 #ifndef ALIHFESIGNALCUTS_H
  21 #define ALIHFESIGNALCUTS_H
  22
  23 #ifndef ALIANALYSISCUTS_H
  24 #include "AliAnalysisCuts.h"
  25 #endif
  26
  27 class TList;
  28 class AliMCEvent;
  29 class AliVParticle;
  30 class AliHFEmcQA;
  31 class TClonesArray;
  32
  33 class AliHFEsignalCuts : public AliAnalysisCuts{
  34   public:
  35     AliHFEsignalCuts();
  36     AliHFEsignalCuts(const Char_t *name, const Char_t *title);
  37     AliHFEsignalCuts(const AliHFEsignalCuts &ref);
  38     AliHFEsignalCuts &operator=(const AliHFEsignalCuts &ref);
  39     virtual ~AliHFEsignalCuts();
  40
  41     virtual Bool_t IsSelected(TObject *o);
  42     virtual Bool_t IsSelected(TList * /*l*/) { return kTRUE; };
  43
  44     Bool_t IsCharmElectron(const TObject * const o) const;
  45     Bool_t IsBeautyElectron(const TObject * const o) const;
  46     Bool_t IsGammaElectron(const TObject * const o) const;
  47     Bool_t IsNonHFElectron(const TObject * const o) const;
  48     Bool_t IsJpsiElectron(const TObject * const o) const;
  49     Bool_t IsB2JpsiElectron(const TObject * const o) const;
  50     Bool_t IsKe3Electron(const TObject * const o) const;
  51
  52     //void SetMCEvent(AliMCEvent *mc) { fMC = mc; }
  53     void SetMCEvent(AliMCEvent *mc);
  54     void SetMCAODInfo(TClonesArray *mcarray);
  55
  56   protected:
  57     Int_t GetMotherPDG(const AliVParticle * const track) const;
  58     Int_t GetTrackPDG(const AliVParticle * const track) const;
  59     Int_t GetElecSource(const AliVParticle * const track) const ;
  60
  61   private:
  62     AliMCEvent *fMC;                   //! MC event
  63     TClonesArray *fAODArrayMCInfo;     //! MC info particle AOD
  64     AliHFEmcQA *fMCQA;                 //! MC QA
  65
  66     ClassDef(AliHFEsignalCuts, 3)
  67 };
  68 #endif
  69