3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //-------------------------------------------------------------------------
9 // AOD event header class
10 // Author: Markus Oldenburg, CERN
11 //-------------------------------------------------------------------------
13 #include "AliVHeader.h"
14 #include "AliAODVertex.h"
16 #include "AliCentrality.h"
22 class AliAODHeader : public AliVHeader {
27 AliAODHeader(Int_t nRun, UShort_t nBunchX, UInt_t nOrbit, UInt_t nPeriod, const Char_t *title="");
28 AliAODHeader(Int_t nRun,
36 Double_t muonMagFieldScale,
43 ULong64_t fTriggerMask,
44 UChar_t fTriggerCluster,
46 const Char_t *title="",
50 virtual ~AliAODHeader();
51 AliAODHeader(const AliAODHeader& evt);
52 AliAODHeader& operator=(const AliAODHeader& evt);
54 Int_t GetRunNumber() const { return fRunNumber;}
55 Int_t GetEventNumberESDFile() const { return fEventNumberESDFile;}
56 UShort_t GetBunchCrossNumber() const { return fBunchCrossNumber; }
57 UInt_t GetOrbitNumber() const { return fOrbitNumber; }
58 UInt_t GetPeriodNumber() const { return fPeriodNumber; }
59 ULong64_t GetTriggerMask() const { return fTriggerMask; }
60 UChar_t GetTriggerCluster() const { return fTriggerCluster; }
61 TString GetFiredTriggerClasses()const { return fFiredTriggers;}
62 UInt_t GetEventType() const { return fEventType; }
63 Double_t GetMagneticField() const { return fMagneticField; }
64 Double_t GetMuonMagFieldScale() const { return fMuonMagFieldScale; }
66 Double_t GetCentrality() const { return fCentrality; }
67 Double_t GetZDCN1Energy() const { return fZDCN1Energy; }
68 Double_t GetZDCP1Energy() const { return fZDCP1Energy; }
69 Double_t GetZDCN2Energy() const { return fZDCN2Energy; }
70 Double_t GetZDCP2Energy() const { return fZDCP2Energy; }
71 Double_t GetZDCEMEnergy(Int_t i) const { return fZDCEMEnergy[i]; }
72 Int_t GetRefMultiplicity() const { return fRefMult; }
73 Int_t GetRefMultiplicityPos() const { return fRefMultPos; }
74 Int_t GetRefMultiplicityNeg() const { return fRefMultNeg; }
75 Int_t GetNumberOfMuons() const { return fNMuons; }
76 Int_t GetNumberOfDimuons() const { return fNDimuons; }
78 Double_t GetQTheta(UInt_t i) const;
79 UInt_t GetNQTheta() const { return (UInt_t)fNQTheta; }
81 Double_t GetDiamondX() const {return fDiamondXY[0];}
82 Double_t GetDiamondY() const {return fDiamondXY[1];}
83 Double_t GetDiamondZ() const {return fDiamondZ;}
84 Double_t GetSigma2DiamondX() const {return fDiamondCovXY[0];}
85 Double_t GetSigma2DiamondY() const {return fDiamondCovXY[2];}
86 Double_t GetSigma2DiamondZ() const {return fDiamondSig2Z;}
87 void GetDiamondCovXY(Float_t cov[3]) const {
88 for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return;
90 UInt_t GetL0TriggerInputs() const {return fL0TriggerInputs;}
91 UInt_t GetL1TriggerInputs() const {return fL1TriggerInputs;}
92 UShort_t GetL2TriggerInputs() const {return fL2TriggerInputs;}
93 AliCentrality* GetCentralityP() const { return fCentralityP; }
96 void SetRunNumber(Int_t nRun) { fRunNumber = nRun; }
97 void SetEventNumberESDFile(Int_t n) { fEventNumberESDFile=n; }
98 void SetBunchCrossNumber(UShort_t nBx) { fBunchCrossNumber = nBx; }
99 void SetOrbitNumber(UInt_t nOr) { fOrbitNumber = nOr; }
100 void SetPeriodNumber(UInt_t nPer) { fPeriodNumber = nPer; }
101 void SetTriggerMask(ULong64_t trigMsk) { fTriggerMask = trigMsk; }
102 void SetFiredTriggerClasses(TString trig) { fFiredTriggers = trig;}
103 void SetTriggerCluster(UChar_t trigClus) { fTriggerCluster = trigClus; }
104 void SetEventType(UInt_t evttype) { fEventType = evttype; }
105 void SetMagneticField(Double_t magFld) { fMagneticField = magFld; }
106 void SetMuonMagFieldScale(Double_t magFldScl){ fMuonMagFieldScale = magFldScl; }
107 void SetCentrality(AliCentrality* cent);
108 void SetZDCN1Energy(Double_t n1Energy) { fZDCN1Energy = n1Energy; }
109 void SetZDCP1Energy(Double_t p1Energy) { fZDCP1Energy = p1Energy; }
110 void SetZDCN2Energy(Double_t n2Energy) { fZDCN2Energy = n2Energy; }
111 void SetZDCP2Energy(Double_t p2Energy) { fZDCP2Energy = p2Energy; }
112 void SetZDCEMEnergy(Double_t emEnergy1, Double_t emEnergy2)
113 { fZDCEMEnergy[0] = emEnergy1; fZDCEMEnergy[1] = emEnergy2;}
114 void SetRefMultiplicity(Int_t refMult) { fRefMult = refMult; }
115 void SetRefMultiplicityPos(Int_t refMultPos) { fRefMultPos = refMultPos; }
116 void SetRefMultiplicityNeg(Int_t refMultNeg) { fRefMultNeg = refMultNeg; }
117 void SetNumberOfMuons(Int_t nMuons) { fNMuons = nMuons; }
118 void SetNumberOfDimuons(Int_t nDimuons) { fNDimuons = nDimuons; }
120 void SetQTheta(Double_t *QTheta, UInt_t size = 5);
123 void SetDiamond(Float_t xy[2],Float_t cov[3]) {
124 for(Int_t i=0;i<3;i++) {fDiamondCovXY[i] = cov[i];}
125 for(Int_t i=0;i<2;i++) {fDiamondXY[i] = xy[i] ;}
127 void SetDiamondZ(Float_t z, Float_t sig2z){
128 fDiamondZ=z; fDiamondSig2Z=sig2z;
130 void SetL0TriggerInputs(UInt_t n) {fL0TriggerInputs=n;}
131 void SetL1TriggerInputs(UInt_t n) {fL1TriggerInputs=n;}
132 void SetL2TriggerInputs(UShort_t n) {fL2TriggerInputs=n;}
133 void SetESDFileName(TString name) {fESDFileName = name;}
134 void Print(Option_t* option = "") const;
136 void SetPHOSMatrix(TGeoHMatrix*matrix, Int_t i) {
137 if ((i >= 0) && (i < kNPHOSMatrix)) fPHOSMatrix[i] = matrix;
139 const TGeoHMatrix* GetPHOSMatrix(Int_t i) const {
140 return ((i >= 0) && (i < kNPHOSMatrix)) ? fPHOSMatrix[i] : NULL;
143 void SetEMCALMatrix(TGeoHMatrix*matrix, Int_t i) {
144 if ((i >= 0) && (i < kNEMCALMatrix)) fEMCALMatrix[i] = matrix;
146 const TGeoHMatrix* GetEMCALMatrix(Int_t i) const {
147 return ((i >= 0) && (i < kNEMCALMatrix)) ? fEMCALMatrix[i] : NULL;
150 UInt_t GetOfflineTrigger() { return fOfflineTrigger; }
151 void SetOfflineTrigger(UInt_t trigger) { fOfflineTrigger = trigger; }
152 TString GetESDFileName() const {return fESDFileName;}
153 void Clear(Option_t* = "");
154 enum {kNPHOSMatrix = 5};
155 enum {kNEMCALMatrix = 12};
159 Double32_t fMagneticField; // Solenoid Magnetic Field in kG
160 Double32_t fMuonMagFieldScale; // magnetic field scale of muon arm magnet
161 Double32_t fCentrality; // Centrality
162 Double32_t fZDCN1Energy; // reconstructed energy in the neutron1 ZDC
163 Double32_t fZDCP1Energy; // reconstructed energy in the proton1 ZDC
164 Double32_t fZDCN2Energy; // reconstructed energy in the neutron2 ZDC
165 Double32_t fZDCP2Energy; // reconstructed energy in the proton2 ZDC
166 Double32_t fZDCEMEnergy[2]; // reconstructed energy in the electromagnetic ZDCs
167 Int_t fNQTheta; // number of QTheta elements
168 Double32_t *fQTheta; // [fNQTheta] values to store Lee-Yang-Zeros
169 ULong64_t fTriggerMask; // Trigger Type (mask)
170 TString fFiredTriggers; // String with fired triggers
171 Int_t fRunNumber; // Run Number
172 Int_t fRefMult; // reference multiplicity
173 Int_t fRefMultPos; // reference multiplicity of positive particles
174 Int_t fRefMultNeg; // reference multiplicity of negative particles
175 Int_t fNMuons; // number of muons in the forward spectrometer
176 Int_t fNDimuons; // number of dimuons in the forward spectrometer
177 UInt_t fEventType; // Type of Event
178 UInt_t fOrbitNumber; // Orbit Number
179 UInt_t fPeriodNumber; // Period Number
180 UShort_t fBunchCrossNumber; // BunchCrossingNumber
181 UChar_t fTriggerCluster; // Trigger cluster (mask)
182 Double32_t fDiamondXY[2]; // Interaction diamond (x,y) in RUN
183 Double32_t fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN
184 Double32_t fDiamondZ; // Interaction diamond (z) in RUN
185 Double32_t fDiamondSig2Z; // Interaction diamond sigma^2 (z) in RUN
186 TGeoHMatrix* fPHOSMatrix[kNPHOSMatrix]; //PHOS module position and orientation matrices
187 TGeoHMatrix* fEMCALMatrix[kNEMCALMatrix]; //EMCAL supermodule position and orientation matrices
188 UInt_t fOfflineTrigger; // fired offline triggers for this event
189 TString fESDFileName; // ESD file name to which this event belongs
190 Int_t fEventNumberESDFile; // Event number in ESD file
191 UInt_t fL0TriggerInputs; // L0 Trigger Inputs (mask)
192 UInt_t fL1TriggerInputs; // L1 Trigger Inputs (mask)
193 UShort_t fL2TriggerInputs; // L2 Trigger Inputs (mask)
194 AliCentrality* fCentralityP; // Pointer to full centrality information
195 ClassDef(AliAODHeader, 14);
198 void AliAODHeader::SetCentrality(AliCentrality* cent) {
200 if(fCentralityP)*fCentralityP = *cent;
201 else fCentralityP = new AliCentrality(*cent);
202 fCentrality = cent->GetCentralityPercentile("V0M");