Fix the way how dependence on esd_tracks.C is handled.
[u/mrichter/AliRoot.git] / EVE / alice-macros / esd_hlt_tracks.C
1 // $Id$
2 //**************************************************************************
3 //* This file is property of and copyright by the ALICE HLT Project        * 
4 //* ALICE Experiment at CERN, All rights reserved.                         *
5 //*                                                                        *
6 //* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no>        *
7 //*                  for The ALICE HLT Project.                            *
8 //*                                                                        *
9 //* Permission to use, copy, modify and distribute this software and its   *
10 //* documentation strictly for non-commercial purposes is hereby granted   *
11 //* without fee, provided that the above copyright notice appears in all   *
12 //* copies and that both the copyright notice and this permission notice   *
13 //* appear in the supporting documentation. The authors make no claims     *
14 //* about the suitability of this software for any purpose. It is          *
15 //* provided "as is" without express or implied warranty.                  *
16 //**************************************************************************
17
18 /**
19  * Display ESD Tracks from the HLTesdTree in AliEVE.
20  *
21  * Usage:
22  * <pre>
23  *   alieve $ALICE_ROOT/EVE/alice-macros/event_next.C \
24  *          $ALICE_ROOT/EVE/macros/alieve_init.C \
25  *          $ALICE_ROOT/EVE/alice-macros/geom_simple.C \
26  *          $ALICE_ROOT/EVE/alice-macros/esd_hlt_tracks.C
27  * </pre>
28  * Display is changed to next event by executing event_next()
29  * from the root prompt.
30  * <pre>
31  *   event_next(); esd_tracks(); esd_hlt_tracks();
32  * </pre>
33  *
34  * @ingroup alihlt_tpc
35  * @author Matthias.Richter@ift.uib.no
36  * @date   2008-11-22
37  */
38 TEveTrackList* esd_hlt_tracks()
39 {
40   if (!TClass::GetClass("AliEveEventManager")) {
41     Error("hlt_tpc_clusters.C", "EVE library not loaded, please start alieve correctly");
42     return NULL;
43   }
44
45   AliEveEventManager* eveManager=AliEveEventManager::GetMaster();
46   if (!eveManager) {
47     Error("esd_hlt_tracks.C", "EVE manager not initialized");
48     return NULL;
49   }
50
51   TEveUtil::LoadMacro("esd_tracks.C");
52
53   int eventId=eveManager->GetEventId();
54   TFile* esdFile=eveManager->GetESDFile();
55   if (!esdFile) {
56     Warning("esd_hlt_tracks.C", "can not get esd file from EVE manager");
57     return NULL;
58   }
59
60   TObject* pObj=NULL;
61   TTree* pHLTTree=NULL;
62   esdFile->GetObject("HLTesdTree", pObj);
63   if (!pObj || (pHLTTree=dynamic_cast<TTree*>(pObj))==NULL) {
64     Info("esd_hlt_tracks.C", "no HLT ESD tree in ESD file");
65     return NULL;
66   }
67   if (pHLTTree->GetEntries()<=eventId) {
68     Warning("esd_hlt_tracks.C", "skiping event %d: out of range %d", eventId, pHLTTree->GetEntries());
69     return NULL;
70   }
71
72   AliESDEvent* esd=new AliESDEvent;
73   esd->ReadFromTree(pHLTTree);
74   pHLTTree->GetEntry(eventId);
75
76   TEveTrackList* cont = new TEveTrackList("HLT ESD Tracks");
77   cont->SetMainColor(kCyan+3);
78   esd_track_propagator_setup(cont->GetPropagator(),
79                              0.1*esd->GetMagneticField(), 520);
80
81   eveManager->AddElement(cont);
82
83   Int_t count = 0;
84   for (Int_t n = 0; n < esd->GetNumberOfTracks(); ++n)
85   {
86     ++count;
87     TEveTrack* track = esd_make_track(esd->GetTrack(n), cont);
88
89     cont->AddElement(track);
90   }
91   cont->SetTitle(Form("N=%d", count));
92   cont->MakeTracks();
93
94   gEve->Redraw3D();
95
96   return cont;
97 }