Handle missing rec-point tree in a uniform way: return null pointer to TEvePointSet.
[u/mrichter/AliRoot.git] / EVE / alice-macros / tpc_clusters.C
index f06b741..bbb0571 100644 (file)
@@ -30,19 +30,22 @@ TEvePointSet* tpc_clusters(TEveElement* cont=0, Float_t maxR=270)
 
   AliEveEventManager::AssertGeometry();
 
-  TEvePointSet* clusters = new TEvePointSet(kMaxCl);
-  clusters->SetOwnIds(kTRUE);
-
   AliRunLoader* rl = AliEveEventManager::AssertRunLoader();
   rl->LoadRecPoints("TPC");
 
-  AliTPCClustersRow *clrow=new AliTPCClustersRow();
+  TTree *cTree = rl->GetTreeR("TPC", false);
+  if (cTree == 0)
+    return 0;
+
+  AliTPCClustersRow *clrow = new AliTPCClustersRow();
   clrow->SetClass("AliTPCclusterMI");
   clrow->SetArray(kMaxCl);
-
-  TTree *cTree = rl->GetTreeR("TPC", false);
   cTree->SetBranchAddress("Segment", &clrow);
 
+  TEvePointSet* clusters = new TEvePointSet(kMaxCl);
+  clusters->SetOwnIds(kTRUE);
+
+
   Float_t maxRsqr = maxR*maxR;
   Int_t nentr=(Int_t)cTree->GetEntries();
   for (Int_t i=0; i<nentr; i++)
@@ -52,8 +55,6 @@ TEvePointSet* tpc_clusters(TEveElement* cont=0, Float_t maxR=270)
     TClonesArray *cl = clrow->GetArray();
     Int_t ncl = cl->GetEntriesFast();
 
-    printf("%d: %d\n", i, ncl);
-
     while (ncl--)
     {
       AliCluster *c = (AliCluster*) cl->UncheckedAt(ncl);
@@ -73,7 +74,7 @@ TEvePointSet* tpc_clusters(TEveElement* cont=0, Float_t maxR=270)
 
   if (clusters->Size() == 0 && gEve->GetKeepEmptyCont() == kFALSE)
   {
-    Warning("tpc_clusters", "No TPC clusters");
+    Warning("tpc_clusters.C", "No TPC clusters");
     delete clusters;
     return 0;
   }