Fix writing of tracklets for first event

[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.cxx

diff --git a/TRD/AliTRDclusterizer.cxx b/TRD/AliTRDclusterizer.cxx
index 6631537c3beead687f7403aec6bf4c5732a252a4..97cf806675f63bdf3bdc24df23696d2d9f8d1fe9 100644 (file)
--- a/TRD/AliTRDclusterizer.cxx
+++ b/TRD/AliTRDclusterizer.cxx
@@ -360,8 +360,16 @@ Bool_t AliTRDclusterizer::OpenOutput(TTree *clusterTree)
     fClusterTree = clusterTree;
     fClusterTree->Branch("TRDcluster", "TObjArray", &ioArray, 32000, 0);
   }
+  return kTRUE;
+}
+
+//_____________________________________________________________________________
+Bool_t AliTRDclusterizer::OpenTrackletOutput()
+{
+  //
+  // Tracklet writing
+  //
 
-  // tracklet writing
   if (fReconstructor->IsWritingTracklets()){
     TString evfoldname = AliConfig::GetDefaultEventFolderName();
     fRunLoader         = AliRunLoader::GetRunLoader(evfoldname);
@@ -381,21 +389,18 @@ Bool_t AliTRDclusterizer::OpenOutput(TTree *clusterTree)
       dl = new AliDataLoader("TRD.Tracklets.root","tracklets", "tracklets");
       fRunLoader->GetLoader("TRDLoader")->AddDataLoader(dl);
     }
-    else {
-      fTrackletTree = dl->Tree();
-      if (!fTrackletTree)
-        {
-        dl->MakeTree();
-        fTrackletTree = dl->Tree();
-        }
-      TBranch *trkbranch = fTrackletTree->GetBranch("trkbranch");
-      if (!trkbranch)
-        fTrackletTree->Branch("trkbranch",leaves[0],"det/i:side/i:tracklets[256]/i");
-    }
+    fTrackletTree = dl->Tree();
+    if (!fTrackletTree)
+      {
+       dl->MakeTree();
+       fTrackletTree = dl->Tree();
+      }
+    TBranch *trkbranch = fTrackletTree->GetBranch("trkbranch");
+    if (!trkbranch)
+      fTrackletTree->Branch("trkbranch",leaves[0],"det/i:side/i:tracklets[256]/i");
   }
 
   return kTRUE;
-
 }
 
 //_____________________________________________________________________________