Handle missing rec-point tree in a uniform way: return null pointer to TEvePointSet.

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

diff --git a/EVE/alice-macros/primary_vertex_tracks.C b/EVE/alice-macros/primary_vertex_tracks.C
index 0d59ac79f6dcbebc905f18ea1994a808f8e4c365..260db0e3858fc4a89e29085b9f8e6e9e4e51b56e 100644 (file)
--- a/EVE/alice-macros/primary_vertex_tracks.C
+++ b/EVE/alice-macros/primary_vertex_tracks.C
@@ -1,28 +1,38 @@
-Reve::TrackList* primary_vertex_tracks()
+// $Id$
+// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
+
+/**************************************************************************
+ * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
+ * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
+ * full copyright notice.                                                 *
+ **************************************************************************/
+
+TEveTrackList* primary_vertex_tracks()
 {
-  Reve::LoadMacro("esd_tracks.C");
-  AliESD       *esd = Alieve::Event::AssertESD();
+  TEveUtil::LoadMacro("esd_tracks.C");
+  AliESDEvent   *esd = AliEveEventManager::AssertESD();
   AliESDVertex *pv  = esd->GetPrimaryVertex();
 
-  Reve::TrackList* cont = new Reve::TrackList("Tracks for Primary Vertex"); 
+  TEveTrackList* cont = new TEveTrackList("Tracks for Primary Vertex");
   cont->SetMainColor(Color_t(7));
-  Reve::TrackRnrStyle* rnrStyle = cont->GetRnrStyle();
-  rnrStyle->SetMagField( esd->GetMagneticField() );
-
-  gReve->AddRenderElement(cont);
+  TEveTrackPropagator* rnrStyle = cont->GetPropagator();
+  rnrStyle->SetMagField( 0.1*esd->GetMagneticField() );
+  rnrStyle->fRnrFV = kTRUE;
+  rnrStyle->fFVAtt->SetMarkerColor(2);
+  gEve->AddElement(cont);
 
   for (Int_t n=0; n<pv->GetNIndices(); n++)
   {
     AliESDtrack* at = esd->GetTrack(pv->GetIndices()[n]);
-    Reve::Track* track = esd_make_track(rnrStyle, n, at, at);
+    TEveTrack* track = esd_make_track(rnrStyle, n, at, at);
     track->SetLineWidth(4);
+    track->SetLineColor(cont->GetMainColor());
     track->SetLineStyle(7);
-    gReve->AddRenderElement(cont, track);
+    gEve->AddElement(track, cont);
   }
 
   cont->MakeTracks();
-  cont->MakeMarkers();
-  gReve->Redraw3D();
+  gEve->Redraw3D();
 
   return cont;
 }