1 #ifndef AliAODCluster_H
2 #define AliAODCluster_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //-------------------------------------------------------------------------
9 // AOD cluster base class
10 // Author: Markus Oldenburg, CERN
11 //-------------------------------------------------------------------------
15 #include "AliVirtualParticle.h"
16 #include "AliAODVertex.h"
17 #include "AliAODTrack.h"
19 class AliAODCluster : public AliVirtualParticle {
23 enum AODClu_t {kUndef=-1, kPHOSNeutral,kPHOSCharged,
24 kEMCALPseudoCluster, kEMCALClusterv1,
25 kPMDNeutral, kPMDCharged};
28 kUnknown=0, kPhoton, kPi0, kNeutron, kKaon0, kEleCon, kCharged, kOther};
31 AliAODCluster(Int_t id,
35 Double_t covMatrix[10],
37 AliAODVertex *prodVertex,
38 AliAODTrack *primTrack,
41 AliAODCluster(Int_t id,
45 Float_t covMatrix[10],
47 AliAODVertex *prodVertex,
48 AliAODTrack *primTrack,
51 virtual ~AliAODCluster();
52 AliAODCluster(const AliAODCluster& trk);
53 AliAODCluster& operator=(const AliAODCluster& trk);
55 Double_t Chi2() const { return fChi2; }
57 virtual Double_t E() const { return fEnergy; }
58 // make a connection to the PID object, here!!!
59 virtual Double_t M() const { return -999.; }
61 // make a connection to the PID object, here!!!
62 virtual Double_t Y() const { return -999.; }
65 virtual const Double_t *PID() const { return fPID; }
67 template <class T> void GetPID(T *pid) const {
68 for(Int_t i=0; i<10; ++i) pid[i]=fPID[i];}
70 template <class T> void SetPID(const T *pid) {
71 if(pid) for(Int_t i=0; i<10; ++i) fPID[i]=pid[i];
72 else {for(Int_t i=1; i<10; fPID[i++]=0); fPID[0]=1.;}}
74 Int_t GetID() const { return fID; }
75 Int_t GetLabel() const { return fLabel; }
77 template <class T> Bool_t GetPosition(T *x) const {
78 x[0]=fPosition[0]; x[1]=fPosition[1]; x[2]=fPosition[2];
81 template <class T> void SetCovMatrix(const T *covMatrix) {
82 if(!fCovMatrix) fCovMatrix=new AliAODRedCov<4>();
83 fCovMatrix->SetCovMatrix(covMatrix);}
85 template <class T> Bool_t GetCovMatrix(T *covMatrix) const {
86 if(!fCovMatrix) return kFALSE;
87 fCovMatrix->GetCovMatrix(covMatrix); return kTRUE;}
89 void RemoveCovMatrix() {delete fCovMatrix; fCovMatrix=NULL;}
91 AliAODVertex *GetProdVertex() const { return (AliAODVertex*)fProdVertex.GetObject(); }
92 AliAODTrack *GetPrimTrack() const { return (AliAODTrack*)fPrimTrack.GetObject(); }
95 void Print(const Option_t *opt = "") const;
98 void SetID(Int_t id) { fID = id; }
99 void SetLabel(Int_t label) {fLabel = label; }
101 template <class T> void SetPosition(const T *x);
103 void SetChi2(Double_t chi2) { fChi2 = chi2; }
105 void SetProdVertex(TObject *vertex) { fProdVertex = vertex; }
106 void SetPrimTrack(TObject *ptrack) { fPrimTrack = ptrack; }
108 virtual Double_t Px() const {return 0.;}
109 virtual Double_t Py() const {return 0.;}
110 virtual Double_t Pz() const {return 0.;}
111 virtual Double_t Pt() const {return 0.;}
112 virtual Double_t P() const {return 0.;}
113 virtual Double_t OneOverPt() const {return 0.;}
114 virtual Double_t Phi() const {return 0.;}
115 virtual Double_t Theta() const {return 0.;}
116 virtual Double_t Eta() const {return 0.;}
117 virtual Short_t Charge() const {return 0;}
122 Double32_t fEnergy; // energy
123 Double32_t fPosition[3]; // position of the cluster
125 Double32_t fPID[10]; // [0.,1.,8] pointer to PID object
126 Double32_t fChi2; // chi2 of mometum fit
128 Int_t fID; // unique track ID, points back to the ESD track
129 Int_t fLabel; // particle label, points back to MC track
131 AliAODRedCov<4> *fCovMatrix; // covariance matrix (x, y, z, E)
132 TRef fProdVertex; // vertex of origin
133 TRef fPrimTrack; // primary track number associated with this cluster
138 ClassDef(AliAODCluster,1);