changes for Vertex and Tracks classes

[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 4b8d1b09d7c034f7ca7ef4b5f8212572dc9589cc..88d8bb3b3e24d66d8e9c7d3958fa60a6e495c11f 100644 (file)
--- a/EVE/alice-macros/primary_vertex_tracks.C
+++ b/EVE/alice-macros/primary_vertex_tracks.C
@@ -1,28 +1,54 @@
-Reve::TrackList* primary_vertex_tracks()
-{
-  Reve::LoadMacro("esd_tracks.C");
-  AliESDEvent   *esd = Alieve::Event::AssertESD();
-  AliESDVertex *pv  = esd->GetPrimaryVertex();
+// $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.                                                 *
+ **************************************************************************/
+#if !defined(__CINT__) || defined(__MAKECINT__)
+#include <TMarker.h>
+#include <TEveManager.h>
+#include <TEveCompound.h>
+#include <TEveTrack.h>
+#include <TEveTrackPropagator.h>
+#include <TEveUtil.h>
+
+#include <AliESDEvent.h>
+#include <AliESDVertex.h>
+#include <AliESDtrack.h>
+#include <AliEveEventManager.h>
+#include <AliEveTrack.h>
 
-  Reve::TrackList* cont = new Reve::TrackList("Tracks for Primary Vertex"); 
-  cont->SetMainColor(Color_t(7));
-  Reve::TrackRnrStyle* rnrStyle = cont->GetRnrStyle();
-  rnrStyle->SetMagField( esd->GetMagneticField() );
+#include "esd_tracks.C"
+#endif
+
+TEveTrackList* primary_vertex_tracks()
+{
+  TEveUtil::LoadMacro("esd_tracks.C");
+  AliESDEvent   *esd = AliEveEventManager::AssertESD();
+  const AliESDVertex *pv  = esd->GetPrimaryVertex();
 
-  gReve->AddRenderElement(cont);
+  TEveTrackList* cont = new TEveTrackList("Tracks for Primary Vertex");
+  cont->SetMainColor(7);
+  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);
+    AliEveTrack* track = esd_make_track(at, cont);
     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;
 }