In muon-related macros in EVE:

[u/mrichter/AliRoot.git] / EVE / alice-macros / trd_clusters.C

diff --git a/EVE/alice-macros/trd_clusters.C b/EVE/alice-macros/trd_clusters.C
index 902c46a4a69e47680ebc936225d789c89ee83c3f..dda663ad2caa00ccec2f7789c62696898d5a99a5 100644 (file)
--- a/EVE/alice-macros/trd_clusters.C
+++ b/EVE/alice-macros/trd_clusters.C
@@ -1,4 +1,4 @@
-// $Id: tpc_clusters.C 23497 2008-01-23 20:43:14Z mtadel $
+// $Id$
 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
 
 /**************************************************************************
@@ -23,18 +23,23 @@ class TEveElement;
 TEvePointSet* trd_clusters(TEveElement *cont = 0)
 {
   const Int_t kMaxClusters = 18 * 6 * 24 *10;
-  AliEveEventManager::AssertGeometry();
 
-  TEvePointSet *clusters = new TEvePointSet(kMaxClusters);
-  clusters->SetOwnIds(kTRUE);
+  AliEveEventManager::AssertGeometry();
 
   AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
   rl->LoadRecPoints("TRD");
 
-  TObjArray *TRDcluster = 0x0;
   TTree *recPoints = rl->GetTreeR("TRD", kFALSE);
+  if (recPoints == 0)
+    return 0;
+
+  TObjArray *TRDcluster = 0x0;
   recPoints->SetBranchAddress("TRDcluster", &TRDcluster);
 
+  TEvePointSet *clusters = new TEvePointSet(kMaxClusters);
+  clusters->SetOwnIds(kTRUE);
+
+
   Int_t nentr=(Int_t)recPoints->GetEntries();
   for (Int_t i=0; i<nentr; i++) {
     if (!recPoints->GetEvent(i)) continue;
@@ -52,24 +57,24 @@ TEvePointSet* trd_clusters(TEveElement *cont = 0)
     TRDcluster->Clear();
   }
 
+  rl->UnloadRecPoints("TRD");
+
   if(clusters->Size() == 0 && gEve->GetKeepEmptyCont() == kFALSE) {
-    Warning("trd_clusters", "No TRD clusters");
+    Warning("trd_clusters.C", "No TRD clusters");
     delete clusters;
     return 0;
   }
 
-  clusters->SetMarkerStyle(2);
-  clusters->SetMarkerSize(0.2);
-  clusters->SetMarkerColor(4);
+  clusters->SetName("TRD Clusters");
+
+  clusters->SetTitle(Form("N=%d", clusters->Size()));
 
-  char form[1000];
-  sprintf(form,"TRD Clusters");
-  clusters->SetName(form);
+  const TString viz_tag("REC Clusters TRD");
+
+  clusters->ApplyVizTag(viz_tag, "Clusters");
 
-  char tip[1000];
-  sprintf(tip,"N=%d", clusters->Size());
-  clusters->SetTitle(tip);
   gEve->AddElement(clusters, cont);
+
   gEve->Redraw3D();
 
   return clusters;