PWGHF/hfe/AliHFEV0taginfo.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 #ifndef ALIHFEV0TAGINFO_H
  17 #define ALIHFEV0TAGINFO_H
  18
  19 #ifndef ROOT_TNamed
  20 #include <TNamed.h>
  21 #endif
  22
  23
  24 #include "AliPID.h"
  25 #include "AliESDv0KineCuts.h"
  26
  27 class AliHFEV0taginfo: public TNamed{
  28
  29     public:
  30
  31         AliHFEV0taginfo();
  32         AliHFEV0taginfo(const char* name);
  33         virtual ~AliHFEV0taginfo();
  34
  35         void Reset(); //resets the fTaggedTracks TList
  36         void TagV0Tracks(AliESDEvent *event); // loops over V0s in event and fills fTaggedTracks with V0 tracks
  37         AliPID::EParticleType GetV0Info(Int_t trackID); //check for V0 information from track ID
  38         void SetIsAODana() { fIsAODana = kTRUE; } // Setter AOD analysis
  39         void AddTrack(Int_t TrackID, Int_t pdgCode); //Translates the pdg code to AliPID enum and adds track to tagged list (currently only e, pion and proton tracks)
  40
  41         class AliHFEV0tag: public TObject{
  42             public:
  43                 AliHFEV0tag();
  44                 AliHFEV0tag(Int_t TrackID, AliPID::EParticleType Pinfo);
  45                 AliHFEV0tag(const AliHFEV0tag &ref);
  46                 AliHFEV0tag &operator=(const AliHFEV0tag &ref);
  47                 virtual ~AliHFEV0tag();
  48
  49
  50                 //TObject virtual functions
  51                 virtual Bool_t IsSortable() const { return kTRUE; };
  52                 virtual Bool_t IsEqual(const TObject *ref) const;
  53                 virtual Int_t Compare(const TObject *ref) const;
  54
  55                 //Setter
  56                 void SetTrack(const Int_t trackID, const AliPID::EParticleType Pinfo); //Set track ID
  57                 //Getter
  58                 Int_t GetTrackID() const { return fTrackID; };
  59                 AliPID::EParticleType GetPinfo() const { return fPinfo; };
  60             private:
  61                 Int_t fTrackID;
  62                 AliPID::EParticleType fPinfo;
  63
  64                 ClassDef(AliHFEV0taginfo::AliHFEV0tag,1);
  65         };
  66     private:
  67         TList *fTaggedTracks;
  68         Bool_t fIsAODana;
  69         AliESDv0KineCuts *fV0finder;
  70
  71         ClassDef(AliHFEV0taginfo,1)
  72 };
  73 #endif