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 #include "AliAODv0KineCuts.h"
  27
  28 class AliVEvent;
  29
  30 class AliHFEV0taginfo: public TNamed{
  31
  32     public:
  33
  34         AliHFEV0taginfo();
  35         AliHFEV0taginfo(const char* name);
  36         AliHFEV0taginfo(const AliHFEV0taginfo &ref);
  37         AliHFEV0taginfo& operator=(const AliHFEV0taginfo &ref);
  38         virtual ~AliHFEV0taginfo();
  39
  40         void Reset(); //resets the fTaggedTracks TList
  41         void TagV0Tracks(AliVEvent *event); // loops over V0s in event and fills fTaggedTracks with V0 tracks
  42         AliPID::EParticleType GetV0Info(Int_t trackID); //check for V0 information from track ID
  43         void SetIsAODana() { fIsAODana = kTRUE; } // Setter AOD analysis
  44         void SetIsESDana() { fIsAODana = kFALSE; } // Setter ESD analysis
  45         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)
  46
  47         class AliHFEV0tag: public TObject{
  48             public:
  49                 AliHFEV0tag();
  50                 AliHFEV0tag(Int_t TrackID, AliPID::EParticleType Pinfo);
  51                 AliHFEV0tag(const AliHFEV0tag &ref);
  52                 AliHFEV0tag &operator=(const AliHFEV0tag &ref);
  53                 virtual ~AliHFEV0tag();
  54
  55                 //TObject virtual functions
  56                 virtual Bool_t IsSortable() const { return kTRUE; };
  57                 virtual Bool_t IsEqual(const TObject *ref) const;
  58                 virtual Int_t Compare(const TObject *ref) const;
  59
  60                 //Setter
  61                 void SetTrack(const Int_t trackID, const AliPID::EParticleType Pinfo); //Set track ID
  62                 //Getter
  63                 Int_t GetTrackID() const { return fTrackID; };
  64                 AliPID::EParticleType GetPinfo() const { return fPinfo; };
  65             private:
  66                 Int_t fTrackID;
  67                 AliPID::EParticleType fPinfo;
  68
  69                 ClassDef(AliHFEV0taginfo::AliHFEV0tag,1);
  70         };
  71     private:
  72         Bool_t fIsAODana;
  73         TList *fTaggedTracks;
  74         AliESDv0KineCuts *fV0finder;
  75         AliAODv0KineCuts *fAODV0finder;
  76
  77         ClassDef(AliHFEV0taginfo,1)
  78 };
  79 #endif