PWG3/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
  32 class AliHFEsignalCuts : public AliAnalysisCuts{
  33   public:
  34     AliHFEsignalCuts();
  35     AliHFEsignalCuts(const Char_t *name, const Char_t *title);
  36     AliHFEsignalCuts(const AliHFEsignalCuts &ref);
  37     AliHFEsignalCuts &operator=(const AliHFEsignalCuts &ref);
  38     virtual ~AliHFEsignalCuts();
  39
  40     virtual Bool_t IsSelected(TObject *o);
  41     virtual Bool_t IsSelected(TList * /*l*/) { return kTRUE; };
  42
  43     Bool_t IsCharmElectron(const TObject * const o) const;
  44     Bool_t IsBeautyElectron(const TObject * const o) const;
  45     Bool_t IsGammaElectron(const TObject * const o) const;
  46     Bool_t IsNonHFElectron(const TObject * const o) const;
  47
  48     //void SetMCEvent(AliMCEvent *mc) { fMC = mc; }
  49     void SetMCEvent(AliMCEvent *mc);
  50
  51   protected:
  52     Int_t GetMotherPDG(const AliVParticle * const track) const;
  53     Int_t GetTrackPDG(const AliVParticle * const track) const;
  54     Int_t GetElecSource(const AliVParticle * const track) const ;
  55
  56   private:
  57     AliMCEvent *fMC;   //! MC event
  58     AliHFEmcQA *fMCQA; //! MC QA
  59
  60     ClassDef(AliHFEsignalCuts, 2)
  61 };
  62 #endif
  63