Adding AliAODTracklets (SPD tracklets) to the standard AOD.

[u/mrichter/AliRoot.git] / STEER / CreateAODfromESD.C

diff --git a/STEER/CreateAODfromESD.C b/STEER/CreateAODfromESD.C
index 097b0bd238181a85e90f77372c8e5e5e1ba26b06..ad1130dad9fe344718650a34be9b8f38bf12e481 100644 (file)
--- a/STEER/CreateAODfromESD.C
+++ b/STEER/CreateAODfromESD.C
@@ -10,6 +10,7 @@
 #include "AliAODVertex.h"
 #include "AliAODTrack.h"
 #include "AliAODCluster.h"
+#include "AliAODTracklets.h"
 
 #include "AliESDEvent.h"
 #include "AliESDtrack.h"
@@ -688,6 +689,20 @@ void CreateAODfromESD(const char *inFileName = "AliESDs.root",
 
     } // end of loop on calo clusters
 
+    // tracklets
+    const AliMultiplicity *mult = esd->GetMultiplicity();
+    if (mult) {
+      if (mult->GetNumberOfTracklets()>0) {
+       aod->GetTracklets()->CreateContainer(mult->GetNumberOfTracklets());
+
+       for (Int_t n=0; n<mult->GetNumberOfTracklets(); n++) {
+         aod->GetTracklets()->SetTracklet(n, mult->GetTheta(n), mult->GetPhi(n), mult->GetDeltaPhi(n), mult->GetLabel(n));
+       }
+      }
+    } else {
+      Printf("ERROR: AliMultiplicity could not be retrieved from ESD");
+    }
+
     delete [] usedTrack;
     delete [] usedV0;
     delete [] usedKink;