// $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>

#include "esd_tracks.C"
#endif

TEveTrackList* primary_vertex_tracks()
{
  TEveUtil::LoadMacro("esd_tracks.C");
  AliESDEvent   *esd = AliEveEventManager::AssertESD();
  const AliESDVertex *pv  = esd->GetPrimaryVertex();

  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]);
    AliEveTrack* track = esd_make_track(at, cont);
    track->SetLineWidth(4);
    track->SetLineColor(cont->GetMainColor());
    track->SetLineStyle(7);
    gEve->AddElement(track, cont);
  }

  cont->MakeTracks();
  gEve->Redraw3D();

  return cont;
}