[u/mrichter/AliRoot.git] / STEER / AliAODHeader.h

#ifndef AliAODHeader_H
#define AliAODHeader_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------------------------
//     AOD event header class
//     Author: Markus Oldenburg, CERN
//-------------------------------------------------------------------------

#include "AliVHeader.h"
#include "AliAODVertex.h"
#include <TString.h>

class TGeoHMatrix;
class TString;

class AliAODHeader : public AliVHeader {

 public :
  AliAODHeader();
 
  AliAODHeader(Int_t nRun, UShort_t nBunchX, UInt_t nOrbit, UInt_t nPeriod, const Char_t *title="");
  AliAODHeader(Int_t nRun, 
	       UShort_t nBunchX,
	       UInt_t nOrbit,
	       UInt_t nPeriod,
	       Int_t refMult,
	       Int_t refMultPos,
	       Int_t refMultNeg,
	       Double_t magField,
	       Double_t muonMagFieldScale,
	       Double_t cent,
	       Double_t n1Energy,
	       Double_t p1Energy,
	       Double_t n2Energy,
	       Double_t p2Energy,
	       Double_t *emEnergy,
	       ULong64_t fTriggerMask,
	       UChar_t   fTriggerCluster,
	       UInt_t    fEventType,
	       const Char_t *title="",
	       Int_t nMuons=0,
	       Int_t nDimuons=0);
  
  virtual ~AliAODHeader();
  AliAODHeader(const AliAODHeader& evt); 
  AliAODHeader& operator=(const AliAODHeader& evt);

  Int_t     GetRunNumber()          const { return fRunNumber;}
  Int_t     GetEventNumberESDFile() const { return fEventNumberESDFile;}
  UShort_t  GetBunchCrossNumber()   const { return fBunchCrossNumber; }
  UInt_t    GetOrbitNumber()        const { return fOrbitNumber; }
  UInt_t    GetPeriodNumber()       const { return fPeriodNumber; }
  ULong64_t GetTriggerMask()        const { return fTriggerMask; }
  UChar_t   GetTriggerCluster()     const { return fTriggerCluster; }
  TString   GetFiredTriggerClasses()const { return fFiredTriggers;}
  UInt_t    GetEventType()          const { return fEventType; }
  Double_t  GetMagneticField()      const { return fMagneticField; }
  Double_t  GetMuonMagFieldScale()  const { return fMuonMagFieldScale; }
  
  Double_t  GetCentrality()         const { return fCentrality; }
  Double_t  GetZDCN1Energy()        const { return fZDCN1Energy; }
  Double_t  GetZDCP1Energy()        const { return fZDCP1Energy; }
  Double_t  GetZDCN2Energy()        const { return fZDCN2Energy; }
  Double_t  GetZDCP2Energy()        const { return fZDCP2Energy; }
  Double_t  GetZDCEMEnergy(Int_t i) const { return fZDCEMEnergy[i]; }
  Int_t     GetRefMultiplicity()    const { return fRefMult; }
  Int_t     GetRefMultiplicityPos() const { return fRefMultPos; }
  Int_t     GetRefMultiplicityNeg() const { return fRefMultNeg; }
  Int_t     GetNumberOfMuons()      const { return fNMuons; }
  Int_t     GetNumberOfDimuons()    const { return fNDimuons; }

  Double_t  GetQTheta(UInt_t i) const;
  UInt_t    GetNQTheta() const { return (UInt_t)fNQTheta; }

  Double_t GetDiamondX() const {return fDiamondXY[0];}
  Double_t GetDiamondY() const {return fDiamondXY[1];}
  Double_t GetDiamondZ() const {return fDiamondZ;}
  Double_t GetSigma2DiamondX() const {return fDiamondCovXY[0];}
  Double_t GetSigma2DiamondY() const {return fDiamondCovXY[2];}
  Double_t GetSigma2DiamondZ() const {return fDiamondSig2Z;}
  void GetDiamondCovXY(Float_t cov[3]) const {
    for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return;
  }
  UInt_t   GetL0TriggerInputs() const {return fL0TriggerInputs;}  
  UInt_t   GetL1TriggerInputs() const {return fL1TriggerInputs;} 
  UShort_t GetL2TriggerInputs() const {return fL2TriggerInputs;} 

  
  void SetRunNumber(Int_t nRun)                { fRunNumber = nRun; }
  void SetEventNumberESDFile(Int_t n)          { fEventNumberESDFile=n; }
  void SetBunchCrossNumber(UShort_t nBx)       { fBunchCrossNumber = nBx; }
  void SetOrbitNumber(UInt_t nOr)              { fOrbitNumber = nOr; }
  void SetPeriodNumber(UInt_t nPer)            { fPeriodNumber = nPer; }
  void SetTriggerMask(ULong64_t trigMsk)       { fTriggerMask = trigMsk; }
  void SetFiredTriggerClasses(TString trig)    { fFiredTriggers = trig;}
  void SetTriggerCluster(UChar_t trigClus)     { fTriggerCluster = trigClus; }
  void SetEventType(UInt_t evttype)            { fEventType = evttype; }
  void SetMagneticField(Double_t magFld)       { fMagneticField = magFld; }
  void SetMuonMagFieldScale(Double_t magFldScl){ fMuonMagFieldScale = magFldScl; }
  
  void SetCentrality(Double_t cent)            { fCentrality = cent; }
  void SetZDCN1Energy(Double_t n1Energy)       { fZDCN1Energy = n1Energy; }
  void SetZDCP1Energy(Double_t p1Energy)       { fZDCP1Energy = p1Energy; }
  void SetZDCN2Energy(Double_t n2Energy)       { fZDCN2Energy = n2Energy; }
  void SetZDCP2Energy(Double_t p2Energy)       { fZDCP2Energy = p2Energy; }
  void SetZDCEMEnergy(Double_t emEnergy1, Double_t emEnergy2)      
  	{ fZDCEMEnergy[0] = emEnergy1; fZDCEMEnergy[1] = emEnergy2;}
  void SetRefMultiplicity(Int_t refMult)       { fRefMult = refMult; }
  void SetRefMultiplicityPos(Int_t refMultPos) { fRefMultPos = refMultPos; }
  void SetRefMultiplicityNeg(Int_t refMultNeg) { fRefMultNeg = refMultNeg; }
  void SetNumberOfMuons(Int_t nMuons) { fNMuons = nMuons; }
  void SetNumberOfDimuons(Int_t nDimuons) { fNDimuons = nDimuons; }
  
  void SetQTheta(Double_t *QTheta, UInt_t size = 5);  
  void RemoveQTheta();

  void SetDiamond(Float_t xy[2],Float_t cov[3]) { 
    for(Int_t i=0;i<3;i++) {fDiamondCovXY[i] = cov[i];}
    for(Int_t i=0;i<2;i++) {fDiamondXY[i]    = xy[i] ;}
  }
  void SetDiamondZ(Float_t z, Float_t sig2z){
    fDiamondZ=z; fDiamondSig2Z=sig2z;
  }
  void SetL0TriggerInputs(UInt_t n)   {fL0TriggerInputs=n;}
  void SetL1TriggerInputs(UInt_t n)   {fL1TriggerInputs=n;}
  void SetL2TriggerInputs(UShort_t n) {fL2TriggerInputs=n;}
  void SetESDFileName(TString name)   {fESDFileName = name;}
  void Print(Option_t* option = "") const;

  void    SetPHOSMatrix(TGeoHMatrix*matrix, Int_t i) {
      if ((i >= 0) && (i < kNPHOSMatrix)) fPHOSMatrix[i] = matrix;
  }
  const TGeoHMatrix* GetPHOSMatrix(Int_t i) const {
      return ((i >= 0) && (i < kNPHOSMatrix)) ? fPHOSMatrix[i] : NULL;
  }
  
  void    SetEMCALMatrix(TGeoHMatrix*matrix, Int_t i) {
      if ((i >= 0) && (i < kNEMCALMatrix)) fEMCALMatrix[i] = matrix;
  }
  const TGeoHMatrix* GetEMCALMatrix(Int_t i) const {
      return ((i >= 0) && (i < kNEMCALMatrix)) ? fEMCALMatrix[i] : NULL;
  }
  
  UInt_t GetOfflineTrigger() { return fOfflineTrigger; }
  void SetOfflineTrigger(UInt_t trigger) { fOfflineTrigger = trigger; }
  TString GetESDFileName()   {return fESDFileName;}
  enum {kNPHOSMatrix = 5};
  enum {kNEMCALMatrix = 12};
  
 private :
  
  Double32_t  fMagneticField;       // Solenoid Magnetic Field in kG
  Double32_t  fMuonMagFieldScale;   // magnetic field scale of muon arm magnet
  Double32_t  fCentrality;          // Centrality
  Double32_t  fZDCN1Energy;         // reconstructed energy in the neutron1 ZDC
  Double32_t  fZDCP1Energy;         // reconstructed energy in the proton1 ZDC
  Double32_t  fZDCN2Energy;         // reconstructed energy in the neutron2 ZDC
  Double32_t  fZDCP2Energy;         // reconstructed energy in the proton2 ZDC
  Double32_t  fZDCEMEnergy[2];      // reconstructed energy in the electromagnetic ZDCs
  Int_t       fNQTheta;             // number of QTheta elements
  Double32_t *fQTheta;              // [fNQTheta] values to store Lee-Yang-Zeros
  ULong64_t   fTriggerMask;         // Trigger Type (mask)
  TString     fFiredTriggers;       // String with fired triggers
  Int_t       fRunNumber;           // Run Number
  Int_t       fRefMult;             // reference multiplicity
  Int_t       fRefMultPos;          // reference multiplicity of positive particles
  Int_t       fRefMultNeg;          // reference multiplicity of negative particles
  Int_t       fNMuons;              // number of muons in the forward spectrometer
  Int_t       fNDimuons;            // number of dimuons in the forward spectrometer
  UInt_t      fEventType;           // Type of Event
  UInt_t      fOrbitNumber;         // Orbit Number
  UInt_t      fPeriodNumber;        // Period Number
  UShort_t    fBunchCrossNumber;    // BunchCrossingNumber
  UChar_t     fTriggerCluster;      // Trigger cluster (mask)
  Double32_t      fDiamondXY[2];    // Interaction diamond (x,y) in RUN
  Double32_t      fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN
  Double32_t      fDiamondZ;        // Interaction diamond (z) in RUN
  Double32_t      fDiamondSig2Z;    // Interaction diamond sigma^2 (z) in RUN
  TGeoHMatrix*    fPHOSMatrix[kNPHOSMatrix];   //PHOS module position and orientation matrices
  TGeoHMatrix*    fEMCALMatrix[kNEMCALMatrix]; //EMCAL supermodule position and orientation matrices
  UInt_t      fOfflineTrigger;      // fired offline triggers for this event
  TString     fESDFileName;         // ESD file name to which this event belongs
  Int_t       fEventNumberESDFile;  // Event number in ESD file
  UInt_t      fL0TriggerInputs;     // L0 Trigger Inputs (mask)
  UInt_t      fL1TriggerInputs;     // L1 Trigger Inputs (mask)
  UShort_t    fL2TriggerInputs;     // L2 Trigger Inputs (mask)
  ClassDef(AliAODHeader, 13);
};

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