PWGHF/hfe/AliSelectNonHFE.h

   1 #ifndef ALISELECTNONHFE_H
   2 #define ALISELECTNONHFE_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 ////////////////////////////////////////////////////////////////////////
   8 //                                                                    //
   9 //      Task for the Selection of Non-Heavy-Flavour-Electrons         //
  10 //                                                                    //
  11 //              Author: Elienos Pereira de Oliveira Filho                                 //
  12 //                                      (University of São Paulo)                                                //
  13 //                                                                    //
  14 ////////////////////////////////////////////////////////////////////////
  15
  16 #ifndef ROOT_TNamed
  17 #include <TNamed.h>
  18 #endif
  19
  20 class TH1F;
  21 class TH2F;
  22 class AliESDEvent;
  23 class AliESDtrack;
  24 class AliESDtrackCuts;
  25 class AliPIDResponse;
  26
  27
  28 class AliSelectNonHFE : public TNamed {
  29  public:
  30   AliSelectNonHFE();
  31   AliSelectNonHFE(const char *name, const Char_t *title);
  32   virtual ~AliSelectNonHFE();
  33   Int_t GetNLS() const {return fNLS;};
  34   Int_t GetNULS() const {return fNULS;};
  35   Int_t* GetPartnersLS() const {return fLSPartner;};
  36   Int_t* GetPartnersULS() const {return fULSPartner;};
  37   Bool_t IsLS() const {return fIsLS;};
  38   Bool_t IsULS() const {return fIsULS;};
  39   void FindNonHFE(Int_t iTrack, AliESDtrack *track1, AliESDEvent *fESD);
  40   void SetAlgorithm(TString Algorithm) {fAlgorithm = Algorithm;};
  41   void SetChi2OverNDFCut(Double_t Chi2OverNDFCut) {fChi2OverNDFCut = Chi2OverNDFCut;};
  42   void SetDCACut(Double_t dcaCut) {fdcaCut = dcaCut;};
  43   void SetHistAngleBack(TH1F *HistAngleBack) {fHistAngleBack = HistAngleBack;};
  44   void SetHistAngle(TH1F *HistAngle) {fHistAngle = HistAngle;};
  45   void SetHistDCABack(TH1F *HistDCABack) {fHistDCABack = HistDCABack;};
  46   void SetHistDCA(TH1F *HistDCA) {fHistDCA = HistDCA;};
  47   void SetHistMassBack(TH1F *HistMassBack) {fHistMassBack = HistMassBack;};
  48   void SetHistMass(TH1F *HistMass) {fHistMass = HistMass;};
  49   void SetInvariantMassCut(Double_t MassCut) {fMassCut = MassCut;};
  50   void SetOpeningAngleCut(Double_t AngleCut) {fAngleCut = AngleCut;};
  51   void SetPIDresponse(AliPIDResponse* PIDResponse) {fPIDResponse = PIDResponse;};
  52
  53         void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax)
  54         {
  55                 fTPCnSigmaMin = TPCnSigmaMin;
  56                 fTPCnSigmaMax = TPCnSigmaMax;
  57                 fTOFnSigmaMin = TOFnSigmaMin;
  58                 fTOFnSigmaMax = TOFnSigmaMax;
  59         };
  60
  61         void SetTrackCuts(Double_t TPCnSigmaMin, Double_t TPCnSigmaMax, Double_t TOFnSigmaMin, Double_t TOFnSigmaMax, AliESDtrackCuts* TrackCuts)
  62         {
  63                 fTPCnSigmaMin = TPCnSigmaMin;
  64                 fTPCnSigmaMax = TPCnSigmaMax;
  65                 fTOFnSigmaMin = TOFnSigmaMin;
  66                 fTOFnSigmaMax = TOFnSigmaMax;
  67                 fTrackCuts = TrackCuts;
  68         };
  69
  70  private:
  71   AliESDtrackCuts       *fTrackCuts;            //! Track quality
  72   TString               fAlgorithm;             //Algorithm choice: "MA" (Manual Algorithm) or "KF" (Kalman Filter Algorithm)
  73   Double_t              fAngleCut;              //Maximum opening angle between the tracks
  74   Double_t              fdcaCut;                //Maximum dca between the tracks
  75   Double_t              fTPCnSigmaMin;          //Minimum partner TPCnSigma value
  76   Double_t              fTPCnSigmaMax;                  //Maximum partner TPCnSigma value
  77   Double_t              fTOFnSigmaMin;          //Minimum partner TOFnSigma value
  78   Double_t              fTOFnSigmaMax;                  //Maximum partner TOFnSigma value
  79   Double_t              fMassCut;               //Maximum Invariant Mass Value for Non-HF-Electrons
  80   Double_t              fChi2OverNDFCut;        //Maximum value of Chi2 over NDF in the case of KF Algorithm
  81   Bool_t                fIsLS;                  //Unlike signal pairs Flag
  82   Bool_t                fIsULS;                 //like signal pairs Flag
  83   Int_t                 fNLS;                   //Number of Unlike signal pairs
  84   Int_t                 fNULS;                  //Number of like signal pairs
  85   Int_t                 *fLSPartner;            //! Pointer for the LS partners index
  86   Int_t                 *fULSPartner;           //! Pointer for the ULS partners index
  87   TH1F                  *fHistMass;             //! Invariant mass histogram for Unlike sign pairs
  88   TH1F                  *fHistMassBack;         //! Invariant mass histogram for like sign pairs
  89   TH1F                  *fHistDCA;              //! DCA histogram for Unlike sign pairs
  90   TH1F                  *fHistDCABack;          //! DCA histogram for like sign pairs
  91   TH1F                  *fHistAngle;            //! Opening Angle histogram for Unlike sign pairs
  92   TH1F                  *fHistAngleBack;        //! Opening Angle histogram for like sign pairs
  93   AliPIDResponse *fPIDResponse;     //! PID response object
  94
  95   AliSelectNonHFE(const AliSelectNonHFE&); // not implemented
  96   AliSelectNonHFE& operator=(const AliSelectNonHFE&); // not implemented
  97
  98   ClassDef(AliSelectNonHFE, 1); //!example of analysis
  99 };
 100
 101 #endif