Updates for pile-up vertex (F. Prino)
[u/mrichter/AliRoot.git] / STEER / AliAODHeader.h
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 //-------------------------------------------------------------------------
9 //     AOD event header class
10 //     Author: Markus Oldenburg, CERN
11 //-------------------------------------------------------------------------
12
13 #include "AliVHeader.h"
14 #include "AliAODVertex.h"
15
16 class TGeoHMatrix;
17 class TString;
18
19 class AliAODHeader : public AliVHeader {
20
21  public :
22   AliAODHeader();
23  
24   AliAODHeader(Int_t nRun, UShort_t nBunchX, UInt_t nOrbit, UInt_t nPeriod, const Char_t *title="");
25   AliAODHeader(Int_t nRun, 
26                UShort_t nBunchX,
27                UInt_t nOrbit,
28                UInt_t nPeriod,
29                Int_t refMult,
30                Int_t refMultPos,
31                Int_t refMultNeg,
32                Double_t magField,
33                Double_t muonMagFieldScale,
34                Double_t cent,
35                Double_t n1Energy,
36                Double_t p1Energy,
37                Double_t n2Energy,
38                Double_t p2Energy,
39                Double_t *emEnergy,
40                ULong64_t fTriggerMask,
41                UChar_t   fTriggerCluster,
42                UInt_t    fEventType,
43                const Char_t *title="",
44                Int_t nMuons=0,
45                Int_t nDimuons=0);
46   
47   virtual ~AliAODHeader();
48   AliAODHeader(const AliAODHeader& evt); 
49   AliAODHeader& operator=(const AliAODHeader& evt);
50   
51   Int_t     GetRunNumber()          const { return fRunNumber; }
52   UShort_t  GetBunchCrossNumber()   const { return fBunchCrossNumber; }
53   UInt_t    GetOrbitNumber()        const { return fOrbitNumber; }
54   UInt_t    GetPeriodNumber()       const { return fPeriodNumber; }
55   ULong64_t GetTriggerMask()        const { return fTriggerMask; }
56   UChar_t   GetTriggerCluster()     const { return fTriggerCluster; }
57   TString   GetFiredTriggerClasses()const { return fFiredTriggers;}
58   UInt_t    GetEventType()          const { return fEventType; }
59   Double_t  GetMagneticField()      const { return fMagneticField; }
60   Double_t  GetMuonMagFieldScale()  const { return fMuonMagFieldScale; }
61   
62   Double_t  GetCentrality()         const { return fCentrality; }
63   Double_t  GetZDCN1Energy()        const { return fZDCN1Energy; }
64   Double_t  GetZDCP1Energy()        const { return fZDCP1Energy; }
65   Double_t  GetZDCN2Energy()        const { return fZDCN2Energy; }
66   Double_t  GetZDCP2Energy()        const { return fZDCP2Energy; }
67   Double_t  GetZDCEMEnergy(Int_t i) const { return fZDCEMEnergy[i]; }
68   Int_t     GetRefMultiplicity()    const { return fRefMult; }
69   Int_t     GetRefMultiplicityPos() const { return fRefMultPos; }
70   Int_t     GetRefMultiplicityNeg() const { return fRefMultNeg; }
71   Int_t     GetNumberOfMuons()      const { return fNMuons; }
72   Int_t     GetNumberOfDimuons()    const { return fNDimuons; }
73
74   Double_t  GetQTheta(UInt_t i) const;
75   UInt_t    GetNQTheta() const { return (UInt_t)fNQTheta; }
76
77   Double_t GetDiamondX() const {return fDiamondXY[0];}
78   Double_t GetDiamondY() const {return fDiamondXY[1];}
79   Double_t GetDiamondZ() const {return fDiamondZ;}
80   Double_t GetSigma2DiamondX() const {return fDiamondCovXY[0];}
81   Double_t GetSigma2DiamondY() const {return fDiamondCovXY[2];}
82   Double_t GetSigma2DiamondZ() const {return fDiamondSig2Z;}
83   void GetDiamondCovXY(Float_t cov[3]) const {
84     for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return;
85   }
86   
87   void SetRunNumber(Int_t nRun)                { fRunNumber = nRun; }
88   void SetBunchCrossNumber(UShort_t nBx)       { fBunchCrossNumber = nBx; }
89   void SetOrbitNumber(UInt_t nOr)              { fOrbitNumber = nOr; }
90   void SetPeriodNumber(UInt_t nPer)            { fPeriodNumber = nPer; }
91   void SetTriggerMask(ULong64_t trigMsk)       { fTriggerMask = trigMsk; }
92   void SetFiredTriggerClasses(TString trig)    { fFiredTriggers = trig;}
93   void SetTriggerCluster(UChar_t trigClus)     { fTriggerCluster = trigClus; }
94   void SetEventType(UInt_t evttype)            { fEventType = evttype; }
95   void SetMagneticField(Double_t magFld)       { fMagneticField = magFld; }
96   void SetMuonMagFieldScale(Double_t magFldScl){ fMuonMagFieldScale = magFldScl; }
97   
98   void SetCentrality(Double_t cent)            { fCentrality = cent; }
99   void SetZDCN1Energy(Double_t n1Energy)       { fZDCN1Energy = n1Energy; }
100   void SetZDCP1Energy(Double_t p1Energy)       { fZDCP1Energy = p1Energy; }
101   void SetZDCN2Energy(Double_t n2Energy)       { fZDCN2Energy = n2Energy; }
102   void SetZDCP2Energy(Double_t p2Energy)       { fZDCP2Energy = p2Energy; }
103   void SetZDCEMEnergy(Double_t emEnergy1, Double_t emEnergy2)      
104         { fZDCEMEnergy[0] = emEnergy1; fZDCEMEnergy[1] = emEnergy2;}
105   void SetRefMultiplicity(Int_t refMult)       { fRefMult = refMult; }
106   void SetRefMultiplicityPos(Int_t refMultPos) { fRefMultPos = refMultPos; }
107   void SetRefMultiplicityNeg(Int_t refMultNeg) { fRefMultNeg = refMultNeg; }
108   void SetNumberOfMuons(Int_t nMuons) { fNMuons = nMuons; }
109   void SetNumberOfDimuons(Int_t nDimuons) { fNDimuons = nDimuons; }
110   
111   void SetQTheta(Double_t *QTheta, UInt_t size = 5);  
112   void RemoveQTheta();
113
114   void SetDiamond(Float_t xy[2],Float_t cov[3]) { 
115     for(Int_t i=0;i<3;i++) {if(i<2) fDiamondXY[i]=xy[i]; fDiamondCovXY[i]=cov[i];}
116   }
117   void SetDiamondZ(Float_t z, Float_t sig2z){
118     fDiamondZ=z; fDiamondSig2Z=sig2z;
119   }
120
121   void Print(Option_t* option = "") const;
122
123   void    SetPHOSMatrix(TGeoHMatrix*matrix, Int_t i) {
124       if ((i >= 0) && (i < kNPHOSMatrix)) fPHOSMatrix[i] = matrix;
125   }
126   const TGeoHMatrix* GetPHOSMatrix(Int_t i) const {
127       return ((i >= 0) && (i < kNPHOSMatrix)) ? fPHOSMatrix[i] : NULL;
128   }
129   
130   void    SetEMCALMatrix(TGeoHMatrix*matrix, Int_t i) {
131       if ((i >= 0) && (i < kNEMCALMatrix)) fEMCALMatrix[i] = matrix;
132   }
133   const TGeoHMatrix* GetEMCALMatrix(Int_t i) const {
134       return ((i >= 0) && (i < kNEMCALMatrix)) ? fEMCALMatrix[i] : NULL;
135   }
136   
137   enum {kNPHOSMatrix = 5};
138   enum {kNEMCALMatrix = 12};
139   
140  private :
141   
142   Double32_t  fMagneticField;       // Solenoid Magnetic Field in kG
143   Double32_t  fMuonMagFieldScale;   // magnetic field scale of muon arm magnet
144   Double32_t  fCentrality;          // Centrality
145   Double32_t  fZDCN1Energy;         // reconstructed energy in the neutron1 ZDC
146   Double32_t  fZDCP1Energy;         // reconstructed energy in the proton1 ZDC
147   Double32_t  fZDCN2Energy;         // reconstructed energy in the neutron2 ZDC
148   Double32_t  fZDCP2Energy;         // reconstructed energy in the proton2 ZDC
149   Double32_t  fZDCEMEnergy[2];      // reconstructed energy in the electromagnetic ZDCs
150   Int_t       fNQTheta;             // number of QTheta elements
151   Double32_t *fQTheta;              // [fNQTheta] values to store Lee-Yang-Zeros
152   ULong64_t   fTriggerMask;         // Trigger Type (mask)
153   TString     fFiredTriggers;       // String with fired triggers
154   Int_t       fRunNumber;           // Run Number
155   Int_t       fRefMult;             // reference multiplicity
156   Int_t       fRefMultPos;          // reference multiplicity of positive particles
157   Int_t       fRefMultNeg;          // reference multiplicity of negative particles
158   Int_t       fNMuons;              // number of muons in the forward spectrometer
159   Int_t       fNDimuons;            // number of dimuons in the forward spectrometer
160   UInt_t      fEventType;           // Type of Event
161   UInt_t      fOrbitNumber;         // Orbit Number
162   UInt_t      fPeriodNumber;        // Period Number
163   UShort_t    fBunchCrossNumber;    // BunchCrossingNumber
164   UChar_t     fTriggerCluster;      // Trigger cluster (mask)
165   Double32_t      fDiamondXY[2];    // Interaction diamond (x,y) in RUN
166   Double32_t      fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN
167   Double32_t      fDiamondZ;        // Interaction diamond (z) in RUN
168   Double32_t      fDiamondSig2Z;    // Interaction diamond sigma^2 (z) in RUN
169   TGeoHMatrix*    fPHOSMatrix[kNPHOSMatrix];   //PHOS module position and orientation matrices
170   TGeoHMatrix*    fEMCALMatrix[kNEMCALMatrix]; //EMCAL supermodule position and orientation matrices
171
172   ClassDef(AliAODHeader,11);
173 };
174
175 #endif