//-----------------------------------------------------------------
// Implementation of the ESD class
// This is the class to deal with during the phisical analysis of data
-//
+// This class is generated directly by the reconstruction methods
// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
//-----------------------------------------------------------------
#include "AliESD.h"
+#include "AliESDfriend.h"
ClassImp(AliESD)
-//_______________________________________________________________________
+//______________________________________________________________________________
AliESD::AliESD():
- fEventNumber(0),
+ fEventNumberInFile(0),
+ fBunchCrossNumber(0),
+ fOrbitNumber(0),
+ fPeriodNumber(0),
fRunNumber(0),
- fTrigger(0),
+ fTimeStamp(0),
+ fEventType(0),
+ fTriggerMask(0),
+ fTriggerCluster(0),
fRecoVersion(0),
fMagneticField(0),
+ fZDCN1Energy(0),
+ fZDCP1Energy(0),
+ fZDCN2Energy(0),
+ fZDCP2Energy(0),
+ fZDCEMEnergy(0),
+ fZDCParticipants(0),
+ fT0zVertex(0),
+ fSPDVertex(),
+ fPrimaryVertex(),
+ fSPDMult(),
+ fT0timeStart(0),
fTracks("AliESDtrack",15000),
- fCaloTracks("AliESDCaloTrack",500),
+ fHLTConfMapTracks("AliESDHLTtrack",25000),
+ fHLTHoughTracks("AliESDHLTtrack",15000),
fMuonTracks("AliESDMuonTrack",30),
- fV0s("AliESDv0",200),
- fCascades("AliESDcascade",20)
+ fPmdTracks("AliESDPmdTrack",3000),
+ fTrdTracks("AliESDTrdTrack",300),
+ fV0s("AliESDv0",200),
+ fCascades("AliESDcascade",20),
+ fKinks("AliESDkink",4000),
+ fCaloClusters("AliESDCaloCluster",10000),
+ fEMCALClusters(0),
+ fFirstEMCALCluster(-1),
+ fEMCALTriggerPosition(0x0),
+ fEMCALTriggerAmplitudes(0x0),
+ fPHOSClusters(0),
+ fFirstPHOSCluster(-1),
+ fPHOSTriggerPosition(0x0),
+ fPHOSTriggerAmplitudes(0x0),
+ fESDFMD(0x0),
+ fESDVZERO(0x0),
+ fErrorLogs("AliRawDataErrorLog",5)
+
+{
+ for (Int_t i=0; i<24; i++) {
+ fT0time[i] = 0;
+ fT0amplitude[i] = 0;
+ }
+ for (Int_t i=0; i<2; i++) fDiamondXY[i]=0.;
+ for (Int_t i=0; i<3; i++) fDiamondCovXY[i]=0.;
+}
+
+//______________________________________________________________________________
+AliESD::AliESD(const AliESD& esd):
+ TObject(esd),
+ fEventNumberInFile(esd.fEventNumberInFile),
+ fBunchCrossNumber(esd.fBunchCrossNumber),
+ fOrbitNumber(esd.fOrbitNumber),
+ fPeriodNumber(esd.fPeriodNumber),
+ fRunNumber(esd.fRunNumber),
+ fTimeStamp(esd.fTimeStamp),
+ fEventType(esd.fEventType),
+ fTriggerMask(esd.fTriggerMask),
+ fTriggerCluster(esd.fTriggerCluster),
+ fRecoVersion(esd.fRecoVersion),
+ fMagneticField(esd.fMagneticField),
+ fZDCN1Energy(esd.fZDCN1Energy),
+ fZDCP1Energy(esd.fZDCP1Energy),
+ fZDCN2Energy(esd.fZDCN2Energy),
+ fZDCP2Energy(esd.fZDCP2Energy),
+ fZDCEMEnergy(esd.fZDCEMEnergy),
+ fZDCParticipants(esd.fZDCParticipants),
+ fT0zVertex(esd.fT0zVertex),
+ fSPDVertex(esd.fSPDVertex),
+ fPrimaryVertex(esd.fPrimaryVertex),
+ fSPDMult(esd.fSPDMult),
+ fT0timeStart(esd.fT0timeStart),
+ fTracks(*((TClonesArray*)esd.fTracks.Clone())),
+ fHLTConfMapTracks(*((TClonesArray*)esd.fHLTConfMapTracks.Clone())),
+ fHLTHoughTracks(*((TClonesArray*)esd.fHLTHoughTracks.Clone())),
+ fMuonTracks(*((TClonesArray*)esd.fMuonTracks.Clone())),
+ fPmdTracks(*((TClonesArray*)esd.fPmdTracks.Clone())),
+ fTrdTracks(*((TClonesArray*)esd.fTrdTracks.Clone())),
+ fV0s(*((TClonesArray*)esd.fV0s.Clone())),
+ fCascades(*((TClonesArray*)esd.fCascades.Clone())),
+ fKinks(*((TClonesArray*)esd.fKinks.Clone())),
+ fCaloClusters(*((TClonesArray*)esd.fCaloClusters.Clone())),
+ fEMCALClusters(esd.fEMCALClusters),
+ fFirstEMCALCluster(esd.fFirstEMCALCluster),
+ fEMCALTriggerPosition(esd. fEMCALTriggerPosition),
+ fEMCALTriggerAmplitudes(esd.fEMCALTriggerAmplitudes),
+ fPHOSClusters(esd.fPHOSClusters),
+ fFirstPHOSCluster(esd.fFirstPHOSCluster),
+ fPHOSTriggerPosition(esd.fPHOSTriggerPosition),
+ fPHOSTriggerAmplitudes(esd.fPHOSTriggerAmplitudes),
+ fESDFMD(esd.fESDFMD),
+ fESDVZERO(esd.fESDVZERO),
+ fErrorLogs(*((TClonesArray*)esd.fErrorLogs.Clone()))
+{
+ for (Int_t i=0; i<24; i++) {
+ fT0time[i] = esd.fT0time[i];
+ fT0amplitude[i] = esd.fT0amplitude[i];
+ }
+ for (Int_t i=0; i<2; i++) fDiamondXY[i]=esd.fDiamondXY[i];
+ for (Int_t i=0; i<3; i++) fDiamondCovXY[i]=esd.fDiamondCovXY[i];
+}
+
+//______________________________________________________________________________
+AliESD::~AliESD()
+{
+ //
+ // Standard destructor
+ //
+ fTracks.Delete();
+ fHLTConfMapTracks.Delete();
+ fHLTHoughTracks.Delete();
+ fMuonTracks.Delete();
+ fPmdTracks.Delete();
+ fTrdTracks.Delete();
+ fV0s.Delete();
+ fCascades.Delete();
+ fKinks.Delete();
+ fCaloClusters.Delete();
+ delete fESDFMD;
+ delete fESDVZERO;
+// fEMCALTriggerPosition->Delete();
+// fEMCALTriggerAmplitudes->Delete();
+// fPHOSTriggerPosition->Delete();
+// fPHOSTriggerAmplitudes->Delete();
+// delete fEMCALTriggerPosition;
+// delete fEMCALTriggerAmplitudes;
+// delete fPHOSTriggerPosition;
+// delete fPHOSTriggerAmplitudes;
+ fErrorLogs.Delete();
+
+}
+
+//______________________________________________________________________________
+void AliESD::Reset()
{
- Int_t i;
- for (i=0; i<3; i++) fVtx[i]=0.;
- for (i=0; i<6; i++) fCovVtx[i]=0.;
+ fEventNumberInFile=0;
+ fBunchCrossNumber=0;
+ fOrbitNumber=0;
+ fPeriodNumber=0;
+ fRunNumber=0;
+ fTimeStamp = 0;
+ fEventType = 0;
+ fTriggerMask=0;
+ fTriggerCluster=0;
+ fRecoVersion=0;
+ fMagneticField=0;
+ fZDCN1Energy=0;
+ fZDCP1Energy=0;
+ fZDCN2Energy=0;
+ fZDCP2Energy=0;
+ fZDCEMEnergy=0;
+ fZDCParticipants=0;
+ fT0zVertex=0;
+ fT0timeStart = 0;
+ new (&fSPDVertex) AliESDVertex();
+ new (&fPrimaryVertex) AliESDVertex();
+ new (&fSPDMult) AliMultiplicity();
+ fTracks.Clear();
+ fHLTConfMapTracks.Clear();
+ fHLTHoughTracks.Clear();
+ fMuonTracks.Clear();
+ fPmdTracks.Clear();
+ fTrdTracks.Clear();
+ fV0s.Clear();
+ fCascades.Clear();
+ fCaloClusters.Clear();
+ fEMCALClusters=0;
+ fFirstEMCALCluster=-1;
+ fPHOSClusters=0;
+ fFirstPHOSCluster=-1;
+ if (fESDFMD) fESDFMD->Clear();
+// fEMCALTriggerPosition->Clear();
+// fEMCALTriggerAmplitudes->Clear();
+// fPHOSTriggerPosition->Clear();
+// fPHOSTriggerAmplitudes->Clear();
+ fErrorLogs.Clear();
}
-void AliESD::SetVertex(const Double_t *vtx, const Double_t *cvtx) {
- //Save the primary vertex position
- Int_t i;
- for (i=0; i<3; i++) fVtx[i]=vtx[i];
- if (cvtx)
- for (i=0; i<6; i++) fCovVtx[i]=cvtx[i];
+Bool_t AliESD::RemoveTrack(Int_t rm) {
+ // ---------------------------------------------------------
+ // Remove a track and references to it from ESD,
+ // if this track does not come from a reconstructed decay
+ // ---------------------------------------------------------
+ Int_t last=GetNumberOfTracks()-1;
+ if ((rm<0)||(rm>last)) return kFALSE;
+
+ Int_t used=0;
+
+ // Check if this track comes from a reconstructed decay
+ Int_t nv0=GetNumberOfV0s();
+ for (Int_t n=0; n<nv0; n++) {
+ AliESDv0 *v0=GetV0(n);
+
+ Int_t idx=v0->GetNindex();
+ if (rm==idx) return kFALSE;
+ if (idx==last) used++;
+
+ idx=v0->GetPindex();
+ if (rm==idx) return kFALSE;
+ if (idx==last) used++;
+ }
+
+ Int_t ncs=GetNumberOfCascades();
+ for (Int_t n=0; n<ncs; n++) {
+ AliESDcascade *cs=GetCascade(n);
+
+ Int_t idx=cs->GetIndex();
+ if (rm==idx) return kFALSE;
+ if (idx==last) used++;
+ }
+
+ Int_t nkn=GetNumberOfKinks();
+ for (Int_t n=0; n<nkn; n++) {
+ AliESDkink *kn=GetKink(n);
+
+ Int_t idx=kn->GetIndex(0);
+ if (rm==idx) return kFALSE;
+ if (idx==last) used++;
+
+ idx=kn->GetIndex(1);
+ if (rm==idx) return kFALSE;
+ if (idx==last) used++;
+ }
+
+
+ //Replace the removed track with the last track
+ TClonesArray &a=fTracks;
+ delete a.RemoveAt(rm);
+
+ if (rm==last) return kTRUE;
+
+ AliESDtrack *t=GetTrack(last);
+ t->SetID(rm);
+ new (a[rm]) AliESDtrack(*t);
+ delete a.RemoveAt(last);
+
+ if (!used) return kTRUE;
+
+
+ // Remap the indices of the daughters of reconstructed decays
+ for (Int_t n=0; n<nv0; n++) {
+ AliESDv0 *v0=GetV0(n);
+ if (v0->GetIndex(0)==last) {
+ v0->SetIndex(0,rm);
+ used--;
+ if (!used) return kTRUE;
+ }
+ if (v0->GetIndex(1)==last) {
+ v0->SetIndex(1,rm);
+ used--;
+ if (!used) return kTRUE;
+ }
+ }
+
+ for (Int_t n=0; n<ncs; n++) {
+ AliESDcascade *cs=GetCascade(n);
+ if (cs->GetIndex()==last) {
+ cs->SetIndex(rm);
+ used--;
+ if (!used) return kTRUE;
+ }
+ }
+
+ for (Int_t n=0; n<nkn; n++) {
+ AliESDkink *kn=GetKink(n);
+ if (kn->GetIndex(0)==last) {
+ kn->SetIndex(rm,0);
+ used--;
+ if (!used) return kTRUE;
+ }
+ if (kn->GetIndex(1)==last) {
+ kn->SetIndex(rm,1);
+ used--;
+ if (!used) return kTRUE;
+ }
+ }
+
+ return kTRUE;
}
-void AliESD::GetVertex(Double_t *vtx, Double_t *cvtx) const {
- //Get the primary vertex position
- Int_t i;
- for (i=0; i<3; i++) vtx[i]=fVtx[i];
- if (cvtx)
- for (i=0; i<6; i++) cvtx[i]=fCovVtx[i];
+Bool_t AliESD::Clean(Float_t *cleanPars) {
+ //
+ // Remove the data which are not needed for the physics analysis.
+ //
+ // If track's transverse parameter is larger than dmax
+ // OR
+ // track's longitudinal parameter is larger than zmax
+ // an attempt to remove this track from ESD is made.
+ //
+ // The track gets removed if it does not come
+ // from a reconstructed decay
+ //
+
+ Float_t dmax=cleanPars[0], zmax=cleanPars[1];
+
+ const AliESDVertex *vertex=GetVertex();
+ Bool_t vtxOK=vertex->GetStatus(), rc=kFALSE;
+
+ Int_t nTracks=GetNumberOfTracks();
+ for (Int_t i=nTracks-1; i>=0; i--) {
+ AliESDtrack *track=GetTrack(i);
+ Float_t xy,z; track->GetImpactParameters(xy,z);
+ if ((TMath::Abs(xy) > dmax) || (vtxOK && (TMath::Abs(z) > zmax))) {
+ if (RemoveTrack(i)) rc=kTRUE;
+ }
+ }
+
+ return rc;
}
-void AliESD::Print(Option_t *) const {
- //Print header information of the event
- Info("Print","ESD run information");
- printf("Event # %d Run # %d Trigger %ld Magnetic field %f \n",
- GetEventNumber(),
+Int_t AliESD::AddV0(const AliESDv0 *v) {
+ //
+ // Add V0
+ //
+ Int_t idx=fV0s.GetEntriesFast();
+ new(fV0s[idx]) AliESDv0(*v);
+ return idx;
+}
+
+//______________________________________________________________________________
+void AliESD::Print(Option_t *) const
+{
+ //
+ // Print header information of the event
+ //
+ printf("ESD run information\n");
+ printf("Event # in file %d Bunch crossing # %d Orbit # %d Period # %d Run # %d Trigger %lld Magnetic field %f \n",
+ GetEventNumberInFile(),
+ GetBunchCrossNumber(),
+ GetOrbitNumber(),
+ GetPeriodNumber(),
GetRunNumber(),
- GetTrigger(),
+ GetTriggerMask(),
GetMagneticField() );
+ printf("Vertex: (%.4f +- %.4f, %.4f +- %.4f, %.4f +- %.4f) cm\n",
+ fPrimaryVertex.GetXv(), fPrimaryVertex.GetXRes(),
+ fPrimaryVertex.GetYv(), fPrimaryVertex.GetYRes(),
+ fPrimaryVertex.GetZv(), fPrimaryVertex.GetZRes());
+ printf("Mean vertex in RUN: X=%.4f Y=%.4f cm\n",
+ GetDiamondX(),GetDiamondY());
+ printf("SPD Multiplicity. Number of tracklets %d \n",
+ fSPDMult.GetNumberOfTracklets());
printf("Event from reconstruction version %d \n",fRecoVersion);
printf("Number of tracks: \n");
- printf("charged %d\n",GetNumberOfTracks());
- printf(" calo %d\n", GetNumberOfCaloTracks());
+ printf(" charged %d\n", GetNumberOfTracks());
+ printf(" hlt CF %d\n", GetNumberOfHLTConfMapTracks());
+ printf(" hlt HT %d\n", GetNumberOfHLTHoughTracks());
printf(" muon %d\n", GetNumberOfMuonTracks());
+ printf(" pmd %d\n", GetNumberOfPmdTracks());
+ printf(" trd %d\n", GetNumberOfTrdTracks());
printf(" v0 %d\n", GetNumberOfV0s());
- printf(" cascades %d\n)", GetNumberOfCascades());
+ printf(" cascades %d\n", GetNumberOfCascades());
+ printf(" kinks %d\n", GetNumberOfKinks());
+ printf(" CaloClusters %d\n", GetNumberOfCaloClusters());
+ printf(" phos %d\n", GetNumberOfPHOSClusters());
+ printf(" emcal %d\n", GetNumberOfEMCALClusters());
+ printf(" FMD %s\n", (fESDFMD ? "yes" : "no"));
+ printf(" VZERO %s\n", (fESDVZERO ? "yes" : "no"));
+}
+
+void AliESD::SetESDfriend(const AliESDfriend *ev) {
+ //
+ // Attaches the complementary info to the ESD
+ //
+ if (!ev) return;
+
+ Int_t ntrk=ev->GetNumberOfTracks();
+
+ for (Int_t i=0; i<ntrk; i++) {
+ const AliESDfriendTrack *f=ev->GetTrack(i);
+ GetTrack(i)->SetFriendTrack(f);
+ }
+}
+
+void AliESD::GetESDfriend(AliESDfriend *ev) const {
+ //
+ // Extracts the complementary info from the ESD
+ //
+ if (!ev) return;
+
+ Int_t ntrk=GetNumberOfTracks();
+
+ for (Int_t i=0; i<ntrk; i++) {
+ AliESDtrack *t=GetTrack(i);
+ const AliESDfriendTrack *f=t->GetFriendTrack();
+ ev->AddTrack(f);
+
+ t->ReleaseESDfriendTrack();// Not to have two copies of "friendTrack"
+
+ }
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff