eb2b7c388d6a853fc584eaac3a2200363ec4e3a2
[u/mrichter/AliRoot.git] / 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