Separate class for the V0 reconstruction parameters (M.Ivanov)

[u/mrichter/AliRoot.git] / STEER / AliESD.cxx

diff --git a/STEER/AliESD.cxx b/STEER/AliESD.cxx
index a00b62f88f72350d9d826ea1f36e0aa5c746c4a6..b6a64982ce1d8961ed6cb97ab7d368aa30c7aa7c 100644 (file)
--- a/STEER/AliESD.cxx
+++ b/STEER/AliESD.cxx
@@ -33,6 +33,10 @@ AliESD::AliESD():
   fTrigger(0),
   fRecoVersion(0),
   fMagneticField(0),
+  fZDCNEnergy(0),
+  fZDCPEnergy(0),
+  fZDCEMEnergy(0),
+  fZDCParticipants(0),
   fT0zVertex(0),
   fPrimaryVertex(),
   fTracks("AliESDtrack",15000),
@@ -40,8 +44,10 @@ AliESD::AliESD():
   fHLTHoughTracks("AliESDHLTtrack",15000),
   fMuonTracks("AliESDMuonTrack",30),
   fPmdTracks("AliESDPmdTrack",3000),
-  fV0s("AliESDv0",200),
+  fV0s("AliESDv0",200),  
   fCascades("AliESDcascade",20),
+  fKinks("AliESDkink",4000),
+  fV0MIs("AliESDV0MI",4000),
   fPHOSParticles(0), 
   fEMCALParticles(0), 
   fFirstPHOSParticle(-1), 
@@ -55,12 +61,62 @@ AliESD::~AliESD()
   // Standard destructor
   //
   fTracks.Delete();
+  fHLTConfMapTracks.Delete();
+  fHLTHoughTracks.Delete();
   fMuonTracks.Delete();
   fPmdTracks.Delete();
   fV0s.Delete();
   fCascades.Delete();
+  fKinks.Delete();
+  fV0MIs.Delete();
 }
 
+void AliESD::UpdateV0PIDs()
+{
+  //
+  //
+  //
+  Int_t nV0 = GetNumberOfV0MIs();
+  for (Int_t i=0;i<nV0;i++){
+    AliESDV0MI * v0 = GetV0MI(i);
+    AliESDtrack* tp = GetTrack(v0->GetIndex(0));
+    AliESDtrack* tm = GetTrack(v0->GetIndex(1));
+    if (!tm || !tp){
+      printf("BBBUUUUUUUGGGG\n");
+    }
+    Double_t pp[5],pm[5];
+    tp->GetESDpid(pp);
+    tm->GetESDpid(pm);
+    v0->UpdatePID(pp,pm);    
+  }
+}
+
+//______________________________________________________________________________
+void AliESD::Reset()
+{
+  fEventNumber=0;
+  fRunNumber=0;
+  fTrigger=0;
+  fRecoVersion=0;
+  fMagneticField=0;
+  fZDCNEnergy=0;
+  fZDCPEnergy=0;
+  fZDCEMEnergy=0;
+  fZDCParticipants=0;
+  fT0zVertex=0;
+  fPrimaryVertex.Reset();
+  fTracks.Clear();
+  fHLTConfMapTracks.Clear();
+  fHLTHoughTracks.Clear();
+  fMuonTracks.Clear();
+  fPmdTracks.Clear();
+  fV0s.Clear();
+  fCascades.Clear();
+  fPHOSParticles=0; 
+  fEMCALParticles=0; 
+  fFirstPHOSParticle=-1; 
+  fFirstEMCALParticle=-1;
+}
 
 //______________________________________________________________________________
 void AliESD::Print(Option_t *) const 
@@ -68,7 +124,7 @@ void AliESD::Print(Option_t *) const
   //
   // Print header information of the event
   //
-  Info("Print","ESD run information");
+  printf("ESD run information\n");
   printf("Event # %d Run # %d Trigger %ld Magnetic field %f \n",
         GetEventNumber(),
         GetRunNumber(),
@@ -89,4 +145,6 @@ void AliESD::Print(Option_t *) const
   printf("                 pmd       %d\n", GetNumberOfPmdTracks());
   printf("                 v0        %d\n", GetNumberOfV0s());
   printf("                 cascades  %d\n)", GetNumberOfCascades());
+  printf("                 kinks     %d\n)", GetNumberOfKinks());
+  printf("                 V0MIs     %d\n)", GetNumberOfV0MIs());
 }