Cosmetically sanitize ITS visualization code.

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

diff --git a/EVE/alice-macros/MUON_displayData.C b/EVE/alice-macros/MUON_displayData.C
index 4bcf2fdf3316bbd7de90d1f57b47ec24b8dd7dd8..185338fdf64ad31cfd86e2180b7ff480bd96ff0b 100644 (file)
--- a/EVE/alice-macros/MUON_displayData.C
+++ b/EVE/alice-macros/MUON_displayData.C
@@ -1,4 +1,4 @@
-// $Id: MUON_display.C 24485 2008-03-13 15:27:38Z mtadel $
+// $Id$
 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
 
 /**************************************************************************
@@ -11,14 +11,11 @@ class AliEveMUONData;
 class AliEveEventManager;
 
 AliEveMUONData     *g_muon_data       = 0;
-AliEveEventManager *g_muon_last_event = 0;
 
 Int_t  g_currentEvent = -1;
 Bool_t g_fromRaw      = kFALSE;
 
-AliMagFMaps *g_field = 0;
-
-void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE)
+void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE, Bool_t clustersFromESD = kTRUE)
 {
   //
   // display from real data, eventually with recreated digits
@@ -28,41 +25,42 @@ void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE)
   if (!AliMpSegmentation::Instance()) AliMpCDB::LoadMpSegmentation();
   if (!AliMpDDLStore::Instance())     AliMpCDB::LoadDDLStore();
 
-  if (g_field == 0) {
+  if (!TGeoGlobalMagField::Instance()->GetField()) {
     printf("Loading field map...\n");
-    g_field = new AliMagFMaps("Maps","Maps", 1, 1., 10., AliMagFMaps::k5kG);
-    AliTracker::SetFieldMap(g_field, kFALSE);
-    AliMUONTrackExtrap::SetField(AliTracker::GetFieldMap());
+    AliMagF* field = new AliMagF("Maps","Maps",2,1.,1., 10.,AliMagF::k5kG);
+    TGeoGlobalMagField::Instance()->SetField(field);
   }
+  // set the magnetic field for track extrapolations
+  AliMUONTrackExtrap::SetField();
 
   TTree* dt = 0;
   TTree* ct = 0;
   TTree* ht = 0;
 
-  if (gAliEveEvent == 0) {
+  if (AliEveEventManager::GetMaster() == 0) {
     printf("No alieve event: use alieve_init(...) \n");
     return;
   }
 
-  if (g_currentEvent == gAliEveEvent->GetEventId()) {
+  if (g_currentEvent == AliEveEventManager::GetMaster()->GetEventId()) {
     if (g_fromRaw == fromRaw) {
       printf("Same event... \n");
       return;
     } else {
       if (g_fromRaw) {
        printf("Same event with digits.\n");
-       gAliEveEvent->GotoEvent(g_currentEvent);
+       AliEveEventManager::GetMaster()->GotoEvent(g_currentEvent);
       } else {
        printf("Same event with raw.\n");
-       gAliEveEvent->GotoEvent(g_currentEvent);
+       AliEveEventManager::GetMaster()->GotoEvent(g_currentEvent);
       }
     }
   }
 
   g_fromRaw = fromRaw;
 
-  TString dataPath = TString(gAliEveEvent->GetTitle());
-  dataPath.Append("/rawmuon.root");
+  TString dataPath = TString(AliEveEventManager::GetMaster()->GetTitle());
+  dataPath.Append("/raw.root");
 
   AliRunLoader* rl =  AliEveEventManager::AssertRunLoader();
   g_muon_data = new AliEveMUONData;
@@ -85,16 +83,17 @@ void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE)
     }
   }
 
-  rl->LoadRecPoints("MUON");
-  ct = rl->GetTreeR("MUON", false);
-
-  TString esdDataPath = TString(gAliEveEvent->GetTitle());
+  TString esdDataPath = TString(AliEveEventManager::GetMaster()->GetTitle());
   esdDataPath.Append("/AliESDs.root");
-  g_muon_data->LoadRecPointsFromESD(esdDataPath.Data());
-
-  g_muon_last_event = gAliEveEvent;
-
-  g_currentEvent = g_muon_last_event->GetEventId();
+  if (clustersFromESD) {
+    g_muon_data->LoadRecPointsFromESD(esdDataPath.Data());
+  } else {
+    rl->LoadRecPoints("MUON");
+    ct = rl->GetTreeR("MUON", false);
+    g_muon_data->LoadRecPoints(ct);
+  }
+  
+  g_currentEvent = AliEveEventManager::GetMaster()->GetEventId();
 
   gStyle->SetPalette(1, 0);
 
@@ -102,7 +101,7 @@ void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE)
 
   TEveElementList* l = new TEveElementList("MUONChambers");
   l->SetTitle("MUON chambers");
-  l->SetMainColor(Color_t(2));
+  l->SetMainColor(2);
   gEve->AddElement(l);
 
   for (Int_t ic = 0; ic < 14; ic++)
@@ -114,7 +113,7 @@ void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE)
 
     mucha->SetDataSource(g_muon_data);
 
-    gEve->AddElement(mucha, l);
+    l->AddElement(mucha);
   }
 
   if (showTracks) {
@@ -130,8 +129,9 @@ void MUON_ESD_tracks()
 {
   AliESDEvent* esd = AliEveEventManager::AssertESD();
 
-  TEveTrackList* lt = new TEveTrackList("ESD-Tracks");
-  lt->SetMainColor(Color_t(6));
+  // TEveTrackList* lt = new TEveTrackList("ESD-Tracks");
+  AliEveMUONTrackList* lt = new TEveTrackList("ESD-Tracks");
+  lt->SetMainColor(6);
   //lt->SetMUON();
 
   gEve->AddElement(lt);
@@ -153,8 +153,8 @@ void MUON_ESD_tracks()
 
     track->MakeESDTrack(mt);
 
-    gEve->AddElement(track, lt);
+    lt->AddElement(track);
   }
-
+  lt->HackMomentumLimits();
 }