]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliAODHeader.h
ITS cluster multiplicity and TPC standalone multiplicity in AODHeader
[u/mrichter/AliRoot.git] / STEER / AliAODHeader.h
CommitLineData
df9db588 1#ifndef AliAODHeader_H
2#define AliAODHeader_H
3/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
8//-------------------------------------------------------------------------
9ae2e5e6 9// AOD event header class
df9db588 10// Author: Markus Oldenburg, CERN
11//-------------------------------------------------------------------------
12
9ae2e5e6 13#include "AliVHeader.h"
df9db588 14#include "AliAODVertex.h"
a5f7aba4 15#include <TString.h>
f1c1d7c6 16#include "AliCentrality.h"
ce7adfe9 17#include "AliEventplane.h"
df9db588 18
1aa76f71 19class TGeoHMatrix;
27346f69 20class TString;
1aa76f71 21
f1c1d7c6 22
9ae2e5e6 23class AliAODHeader : public AliVHeader {
df9db588 24
25 public :
31fd97b2 26 AliAODHeader();
df9db588 27
abfce367 28 AliAODHeader(Int_t nRun, UShort_t nBunchX, UInt_t nOrbit, UInt_t nPeriod, const Char_t *title="");
31fd97b2 29 AliAODHeader(Int_t nRun,
30 UShort_t nBunchX,
31 UInt_t nOrbit,
89cf15db 32 UInt_t nPeriod,
31fd97b2 33 Int_t refMult,
34 Int_t refMultPos,
35 Int_t refMultNeg,
36 Double_t magField,
6b6f8d32 37 Double_t muonMagFieldScale,
38 Double_t cent,
ce7adfe9 39 Double_t eventplane,
6b6f8d32 40 Double_t n1Energy,
41 Double_t p1Energy,
42 Double_t n2Energy,
43 Double_t p2Energy,
a85132e7 44 Double_t *emEnergy,
31fd97b2 45 ULong64_t fTriggerMask,
46 UChar_t fTriggerCluster,
47 UInt_t fEventType,
fa8b0e56 48 const Char_t *title="",
49 Int_t nMuons=0,
50 Int_t nDimuons=0);
31fd97b2 51
df9db588 52 virtual ~AliAODHeader();
53 AliAODHeader(const AliAODHeader& evt);
54 AliAODHeader& operator=(const AliAODHeader& evt);
a5f7aba4 55
56 Int_t GetRunNumber() const { return fRunNumber;}
57 Int_t GetEventNumberESDFile() const { return fEventNumberESDFile;}
31fd97b2 58 UShort_t GetBunchCrossNumber() const { return fBunchCrossNumber; }
59 UInt_t GetOrbitNumber() const { return fOrbitNumber; }
89cf15db 60 UInt_t GetPeriodNumber() const { return fPeriodNumber; }
df9db588 61 ULong64_t GetTriggerMask() const { return fTriggerMask; }
62 UChar_t GetTriggerCluster() const { return fTriggerCluster; }
27346f69 63 TString GetFiredTriggerClasses()const { return fFiredTriggers;}
df9db588 64 UInt_t GetEventType() const { return fEventType; }
65 Double_t GetMagneticField() const { return fMagneticField; }
6b6f8d32 66 Double_t GetMuonMagFieldScale() const { return fMuonMagFieldScale; }
67
df9db588 68 Double_t GetCentrality() const { return fCentrality; }
ce7adfe9 69 Double_t GetEventplane() const { return fEventplane; }
6b6f8d32 70 Double_t GetZDCN1Energy() const { return fZDCN1Energy; }
71 Double_t GetZDCP1Energy() const { return fZDCP1Energy; }
72 Double_t GetZDCN2Energy() const { return fZDCN2Energy; }
73 Double_t GetZDCP2Energy() const { return fZDCP2Energy; }
a85132e7 74 Double_t GetZDCEMEnergy(Int_t i) const { return fZDCEMEnergy[i]; }
df9db588 75 Int_t GetRefMultiplicity() const { return fRefMult; }
76 Int_t GetRefMultiplicityPos() const { return fRefMultPos; }
77 Int_t GetRefMultiplicityNeg() const { return fRefMultNeg; }
fa8b0e56 78 Int_t GetNumberOfMuons() const { return fNMuons; }
79 Int_t GetNumberOfDimuons() const { return fNDimuons; }
ff254193 80
81 Double_t GetQTheta(UInt_t i) const;
82 UInt_t GetNQTheta() const { return (UInt_t)fNQTheta; }
613fc341 83
84 Double_t GetDiamondX() const {return fDiamondXY[0];}
85 Double_t GetDiamondY() const {return fDiamondXY[1];}
49fc2e2c 86 Double_t GetDiamondZ() const {return fDiamondZ;}
613fc341 87 Double_t GetSigma2DiamondX() const {return fDiamondCovXY[0];}
88 Double_t GetSigma2DiamondY() const {return fDiamondCovXY[2];}
49fc2e2c 89 Double_t GetSigma2DiamondZ() const {return fDiamondSig2Z;}
613fc341 90 void GetDiamondCovXY(Float_t cov[3]) const {
91 for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return;
92 }
a5f7aba4 93 UInt_t GetL0TriggerInputs() const {return fL0TriggerInputs;}
94 UInt_t GetL1TriggerInputs() const {return fL1TriggerInputs;}
95 UShort_t GetL2TriggerInputs() const {return fL2TriggerInputs;}
f1c1d7c6 96 AliCentrality* GetCentralityP() const { return fCentralityP; }
ce7adfe9 97 AliEventplane* GetEventplaneP() const { return fEventplaneP; }
a5f7aba4 98
6b6f8d32 99
df9db588 100 void SetRunNumber(Int_t nRun) { fRunNumber = nRun; }
a5f7aba4 101 void SetEventNumberESDFile(Int_t n) { fEventNumberESDFile=n; }
31fd97b2 102 void SetBunchCrossNumber(UShort_t nBx) { fBunchCrossNumber = nBx; }
9ae2e5e6 103 void SetOrbitNumber(UInt_t nOr) { fOrbitNumber = nOr; }
104 void SetPeriodNumber(UInt_t nPer) { fPeriodNumber = nPer; }
df9db588 105 void SetTriggerMask(ULong64_t trigMsk) { fTriggerMask = trigMsk; }
27346f69 106 void SetFiredTriggerClasses(TString trig) { fFiredTriggers = trig;}
df9db588 107 void SetTriggerCluster(UChar_t trigClus) { fTriggerCluster = trigClus; }
108 void SetEventType(UInt_t evttype) { fEventType = evttype; }
109 void SetMagneticField(Double_t magFld) { fMagneticField = magFld; }
6b6f8d32 110 void SetMuonMagFieldScale(Double_t magFldScl){ fMuonMagFieldScale = magFldScl; }
5e6a3170 111 void SetCentrality(AliCentrality* cent);
ce7adfe9 112 void SetEventplane(AliEventplane* eventplane);
6b6f8d32 113 void SetZDCN1Energy(Double_t n1Energy) { fZDCN1Energy = n1Energy; }
114 void SetZDCP1Energy(Double_t p1Energy) { fZDCP1Energy = p1Energy; }
115 void SetZDCN2Energy(Double_t n2Energy) { fZDCN2Energy = n2Energy; }
116 void SetZDCP2Energy(Double_t p2Energy) { fZDCP2Energy = p2Energy; }
a85132e7 117 void SetZDCEMEnergy(Double_t emEnergy1, Double_t emEnergy2)
118 { fZDCEMEnergy[0] = emEnergy1; fZDCEMEnergy[1] = emEnergy2;}
df9db588 119 void SetRefMultiplicity(Int_t refMult) { fRefMult = refMult; }
120 void SetRefMultiplicityPos(Int_t refMultPos) { fRefMultPos = refMultPos; }
121 void SetRefMultiplicityNeg(Int_t refMultNeg) { fRefMultNeg = refMultNeg; }
fa8b0e56 122 void SetNumberOfMuons(Int_t nMuons) { fNMuons = nMuons; }
123 void SetNumberOfDimuons(Int_t nDimuons) { fNDimuons = nDimuons; }
6b6f8d32 124
ff254193 125 void SetQTheta(Double_t *QTheta, UInt_t size = 5);
126 void RemoveQTheta();
127
613fc341 128 void SetDiamond(Float_t xy[2],Float_t cov[3]) {
11becff8 129 for(Int_t i=0;i<3;i++) {fDiamondCovXY[i] = cov[i];}
130 for(Int_t i=0;i<2;i++) {fDiamondXY[i] = xy[i] ;}
613fc341 131 }
49fc2e2c 132 void SetDiamondZ(Float_t z, Float_t sig2z){
133 fDiamondZ=z; fDiamondSig2Z=sig2z;
134 }
a5f7aba4 135 void SetL0TriggerInputs(UInt_t n) {fL0TriggerInputs=n;}
136 void SetL1TriggerInputs(UInt_t n) {fL1TriggerInputs=n;}
137 void SetL2TriggerInputs(UShort_t n) {fL2TriggerInputs=n;}
138 void SetESDFileName(TString name) {fESDFileName = name;}
df9db588 139 void Print(Option_t* option = "") const;
1aa76f71 140
141 void SetPHOSMatrix(TGeoHMatrix*matrix, Int_t i) {
142 if ((i >= 0) && (i < kNPHOSMatrix)) fPHOSMatrix[i] = matrix;
143 }
144 const TGeoHMatrix* GetPHOSMatrix(Int_t i) const {
145 return ((i >= 0) && (i < kNPHOSMatrix)) ? fPHOSMatrix[i] : NULL;
146 }
6b6f8d32 147
1aa76f71 148 void SetEMCALMatrix(TGeoHMatrix*matrix, Int_t i) {
149 if ((i >= 0) && (i < kNEMCALMatrix)) fEMCALMatrix[i] = matrix;
150 }
151 const TGeoHMatrix* GetEMCALMatrix(Int_t i) const {
152 return ((i >= 0) && (i < kNEMCALMatrix)) ? fEMCALMatrix[i] : NULL;
153 }
154
0c6c629b 155 UInt_t GetOfflineTrigger() { return fOfflineTrigger; }
f7ce7a37 156 void SetOfflineTrigger(UInt_t trigger) { fOfflineTrigger = trigger; }
157 UInt_t GetNumberOfITSClusters(Int_t ilay) const {return fITSClusters[ilay];}
158 void SetITSClusters(Int_t ilay, UInt_t nclus);
159 Int_t GetTPConlyRefMultiplicity() const {return fTPConlyRefMult;}
160 void SetTPConlyRefMultiplicity(Int_t mult) {fTPConlyRefMult = mult;}
161
5e6a3170 162 TString GetESDFileName() const {return fESDFileName;}
99dbf027 163 void Clear(Option_t* = "");
1aa76f71 164 enum {kNPHOSMatrix = 5};
165 enum {kNEMCALMatrix = 12};
6b6f8d32 166
df9db588 167 private :
6b6f8d32 168
169 Double32_t fMagneticField; // Solenoid Magnetic Field in kG
170 Double32_t fMuonMagFieldScale; // magnetic field scale of muon arm magnet
171 Double32_t fCentrality; // Centrality
ce7adfe9 172 Double32_t fEventplane; // Event plane angle
6b6f8d32 173 Double32_t fZDCN1Energy; // reconstructed energy in the neutron1 ZDC
174 Double32_t fZDCP1Energy; // reconstructed energy in the proton1 ZDC
175 Double32_t fZDCN2Energy; // reconstructed energy in the neutron2 ZDC
176 Double32_t fZDCP2Energy; // reconstructed energy in the proton2 ZDC
a85132e7 177 Double32_t fZDCEMEnergy[2]; // reconstructed energy in the electromagnetic ZDCs
ff254193 178 Int_t fNQTheta; // number of QTheta elements
179 Double32_t *fQTheta; // [fNQTheta] values to store Lee-Yang-Zeros
6b6f8d32 180 ULong64_t fTriggerMask; // Trigger Type (mask)
27346f69 181 TString fFiredTriggers; // String with fired triggers
6b6f8d32 182 Int_t fRunNumber; // Run Number
183 Int_t fRefMult; // reference multiplicity
184 Int_t fRefMultPos; // reference multiplicity of positive particles
185 Int_t fRefMultNeg; // reference multiplicity of negative particles
fa8b0e56 186 Int_t fNMuons; // number of muons in the forward spectrometer
187 Int_t fNDimuons; // number of dimuons in the forward spectrometer
9333290e 188 UInt_t fEventType; // Type of Event
189 UInt_t fOrbitNumber; // Orbit Number
190 UInt_t fPeriodNumber; // Period Number
191 UShort_t fBunchCrossNumber; // BunchCrossingNumber
6b6f8d32 192 UChar_t fTriggerCluster; // Trigger cluster (mask)
613fc341 193 Double32_t fDiamondXY[2]; // Interaction diamond (x,y) in RUN
194 Double32_t fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN
49fc2e2c 195 Double32_t fDiamondZ; // Interaction diamond (z) in RUN
196 Double32_t fDiamondSig2Z; // Interaction diamond sigma^2 (z) in RUN
1aa76f71 197 TGeoHMatrix* fPHOSMatrix[kNPHOSMatrix]; //PHOS module position and orientation matrices
198 TGeoHMatrix* fEMCALMatrix[kNEMCALMatrix]; //EMCAL supermodule position and orientation matrices
0c6c629b 199 UInt_t fOfflineTrigger; // fired offline triggers for this event
a5f7aba4 200 TString fESDFileName; // ESD file name to which this event belongs
201 Int_t fEventNumberESDFile; // Event number in ESD file
202 UInt_t fL0TriggerInputs; // L0 Trigger Inputs (mask)
203 UInt_t fL1TriggerInputs; // L1 Trigger Inputs (mask)
204 UShort_t fL2TriggerInputs; // L2 Trigger Inputs (mask)
f7ce7a37 205 UInt_t fITSClusters[6]; // Number of ITS cluster per layer
206 Int_t fTPConlyRefMult; // Reference multiplicty for standard TPC only tracks
f1c1d7c6 207 AliCentrality* fCentralityP; // Pointer to full centrality information
ce7adfe9 208 AliEventplane* fEventplaneP; // Pointer to full event plane information
f7ce7a37 209 ClassDef(AliAODHeader, 16);
df9db588 210};
5e6a3170 211inline
212void AliAODHeader::SetCentrality(AliCentrality* cent) {
213 if(cent){
214 if(fCentralityP)*fCentralityP = *cent;
215 else fCentralityP = new AliCentrality(*cent);
216 fCentrality = cent->GetCentralityPercentile("V0M");
217 }
218 else{
219 fCentrality = -999;
220 }
221}
ce7adfe9 222inline
223void AliAODHeader::SetEventplane(AliEventplane* eventplane) {
224 if(eventplane){
225 if(fEventplaneP)*fEventplaneP = *eventplane;
226 else fEventplaneP = new AliEventplane(*eventplane);
227 fEventplane = eventplane->GetEventplane("Q");
228 }
229 else{
230 fEventplane = -999;
231 }
232}
f7ce7a37 233
234inline
235void AliAODHeader::SetITSClusters(Int_t ilay, UInt_t nclus)
236{
237 if (ilay >= 0 && ilay < 6) fITSClusters[ilay] = nclus;
238}
239
240
df9db588 241#endif