EVE/alice-macros/esd_muon_tracks.C

   1 // $Id$
   2
   3 /**************************************************************************
   4  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
   5  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
   6  * full copyright notice.                                                 *
   7  **************************************************************************/
   8
   9 /// \ingroup evemacros
  10 /// \file esd_muon_tracks.C
  11 /// \brief Macro to visualise ESD tracks from MUON spectrometer
  12 /// (both tracker and trigger).
  13 ///
  14 /// Use esd_muon_tracks(Bool_t showClusters, Bool_t showDigits) in order to run it
  15 ///
  16 /// Needs that alieve_init() is already called
  17 ///
  18 /// \author P. Pillot, L. Aphecetche; Subatech
  19
  20 #if !defined(__CINT__) || defined(__MAKECINT__)
  21
  22 #include "AliMUONESDInterface.h"
  23 #include "AliMUONTrack.h"
  24 #include "AliMUONTrackExtrap.h"
  25 #include "AliMUONTrackParam.h"
  26 #include "AliMUONConstants.h"
  27 #include "AliMUONCDB.h"
  28 #include "AliMUONGeometryTransformer.h"
  29 #include "AliMUONTriggerCircuit.h"
  30
  31 #include "AliMpCDB.h"
  32
  33 #include "AliESDEvent.h"
  34 #include "AliESDMuonTrack.h"
  35
  36 #include "EveBase/AliEveMagField.h"
  37 #include "EveBase/AliEveTrack.h"
  38 #include "EveBase/AliEveEventManager.h"
  39
  40 #include <TEveManager.h>
  41 #include <TEveUtil.h>
  42 #include <TEveTrack.h>
  43 #include <TEvePointSet.h>
  44 #include <TEveQuadSet.h>
  45 #include <TEveTrackPropagator.h>
  46 #include <TEveVSDStructs.h>
  47
  48 #include <TStyle.h>
  49 #include <TROOT.h>
  50
  51 #endif
  52
  53 //______________________________________________________________________________
  54 void esd_muon_track_propagator_setup(TEveTrackPropagator* trkProp, Bool_t tracker, Bool_t trigger)
  55 {
  56   // set magnetic field
  57   if (AliMUONTrackExtrap::IsFieldON())
  58   {
  59     trkProp->SetMagFieldObj(new AliEveMagField);
  60   }
  61   else
  62   {
  63     trkProp->SetMagField(0.0);
  64   }
  65   trkProp->SetStepper(TEveTrackPropagator::kRungeKutta);
  66
  67   // set propagation range
  68   trkProp->SetMaxR(1000);
  69   if (trigger) trkProp->SetMaxZ(-AliMUONConstants::DefaultChamberZ(13)+10.);
  70   else trkProp->SetMaxZ(-AliMUONConstants::MuonFilterZBeg());
  71
  72   // go through pathmarks
  73   trkProp->SetFitDaughters(kFALSE);
  74   trkProp->SetFitReferences(kTRUE);
  75   trkProp->SetFitDecay(kFALSE);
  76   trkProp->SetFitCluster2Ds(kFALSE);
  77
  78   // Render the ref pathmarks
  79   trkProp->SetRnrReferences(kTRUE);
  80   trkProp->RefPMAtt().SetMarkerSize(0.5);
  81   if (trigger) trkProp->RefPMAtt().SetMarkerColor(kGreen);
  82   else trkProp->RefPMAtt().SetMarkerColor(kAzure);
  83
  84   // Render first vertex
  85   if (tracker)
  86   {
  87     trkProp->SetRnrFV(kTRUE);
  88     if (trigger) trkProp->RefFVAtt().SetMarkerColor(kGreen);
  89     else trkProp->RefFVAtt().SetMarkerColor(kAzure);
  90   }
  91 }
  92
  93 //______________________________________________________________________________
  94 void add_esd_muon_tracks(AliESDEvent* esd, AliMUONESDInterface* data,
  95                          TEveTrackList* match, TEveTrackList* nomatch, TEveTrackList* ghost)
  96 {
  97   // load trigger circuit
  98   static AliMUONTriggerCircuit* gTriggerCircuit = 0x0;
  99   if (!gTriggerCircuit)
 100   {
 101     AliEveEventManager::AssertGeometry();
 102     AliMUONGeometryTransformer* fMUONGeometryTransformer = new AliMUONGeometryTransformer();
 103     fMUONGeometryTransformer->LoadGeometryData();
 104     gTriggerCircuit = new AliMUONTriggerCircuit(fMUONGeometryTransformer);
 105   }
 106
 107   Int_t nTrack(esd->GetNumberOfMuonTracks());
 108   TEveRecTrack recTrack;
 109   TEveTrack* track;
 110
 111   // add ESD tracks to the proper list
 112   for (Int_t n = 0; n < nTrack; ++n)
 113   {
 114     AliESDMuonTrack* emt = esd->GetMuonTrack(n);
 115
 116     // fill general info
 117     UInt_t trackId = emt->GetUniqueID();
 118     recTrack.fLabel = emt->GetLabel();
 119     recTrack.fIndex = (Int_t)trackId;
 120
 121     // fill tracker track specific info
 122     if ( emt->ContainTrackerData() )
 123     {
 124       recTrack.fStatus = emt->GetMatchTrigger();
 125       recTrack.fSign = emt->Charge();
 126       recTrack.fV.Set(emt->GetNonBendingCoorAtDCA(),emt->GetBendingCoorAtDCA(),emt->GetZ());
 127       recTrack.fP.Set(emt->PxAtDCA(),emt->PyAtDCA(),emt->PzAtDCA());
 128       recTrack.fBeta = ( emt->E() > 0 ) ? emt->P()/emt->E() : 0;
 129
 130       // get proper track list
 131       TEveTrackList* trackList = nomatch;
 132       if ( emt->GetMatchTrigger() > 0 ) trackList = match;
 133
 134       // produce eve track
 135       track = new AliEveTrack(&recTrack,trackList->GetPropagator());
 136       track->SetName(Form("%cmu",emt->Charge()>0 ? '+':'-'));
 137       track->SetStdTitle();
 138       track->SetSourceObject(emt); // WARNING: Change the UniqueID of the object!!
 139
 140       // add path mark
 141       TIter next(data->FindTrack(trackId)->GetTrackParamAtCluster());
 142       AliMUONTrackParam* param;
 143       while ( ( param = static_cast<AliMUONTrackParam*>(next()) ) )
 144       {
 145         TEveVector v(param->GetNonBendingCoor(),param->GetBendingCoor(),param->GetZ());
 146         TEveVector p(param->Px(),param->Py(),param->Pz());
 147         track->AddPathMark(TEvePathMark(TEvePathMark::kReference,v,p));
 148       }
 149
 150       // add trigger track if any
 151       if (emt->ContainTriggerData())
 152       {
 153         Double_t x11 = gTriggerCircuit->GetX11Pos(emt->LoCircuit(), emt->LoStripY());
 154         Double_t y11 = gTriggerCircuit->GetY11Pos(emt->LoCircuit(), emt->LoStripX());
 155         Double_t z11 = gTriggerCircuit->GetZ11Pos(emt->LoCircuit(), emt->LoStripX());
 156         Double_t y21 = gTriggerCircuit->GetY21Pos(emt->LoCircuit(), emt->LoStripX()+emt->LoDev()+1);
 157         Double_t z21 = gTriggerCircuit->GetZ21Pos(emt->LoCircuit(), emt->LoStripX()+emt->LoDev()+1);
 158         Double_t pz  = -emt->PUncorrected(); // max value
 159         TEveVector v(x11, y11, z11);
 160         TEveVector p(pz*x11/z11, pz*(y21-y11)/(z21-z11), pz);
 161         track->AddPathMark(TEvePathMark(TEvePathMark::kReference,v,p));
 162       }
 163
 164       // add the track to proper list
 165       track->SetAttLineAttMarker(trackList);
 166       trackList->AddElement(track);
 167     }
 168     else // fill ghost track specific info
 169     {
 170       recTrack.fStatus = 0;
 171       recTrack.fSign = emt->Charge();
 172       Double_t z11 = (emt->GetZUncorrected() < -1.) ? emt->GetZUncorrected() : (Double_t)AliMUONConstants::DefaultChamberZ(10);
 173       recTrack.fV.Set(emt->GetNonBendingCoorUncorrected(),emt->GetBendingCoorUncorrected(),z11);
 174       recTrack.fP.Set(-TMath::Tan(emt->GetThetaXUncorrected()),-TMath::Tan(emt->GetThetaYUncorrected()),-1.);
 175
 176       // produce eve track
 177       track = new AliEveTrack(&recTrack,ghost->GetPropagator());
 178       track->SetName("mu");
 179       track->SetTitle("Trigger only");
 180       track->SetSourceObject(emt);
 181
 182       // add the track to proper list
 183       track->SetAttLineAttMarker(ghost);
 184       ghost->AddElement(track);
 185     }
 186
 187   }
 188
 189 }
 190
 191 //______________________________________________________________________________
 192 void esd_muon_tracks(Bool_t showClusters, Bool_t showDigits)
 193 {
 194   // load ESD
 195   AliESDEvent* esd = AliEveEventManager::AssertESD();
 196   if (esd->GetNumberOfMuonTracks() == 0 && !gEve->GetKeepEmptyCont()) return;
 197
 198   // load field
 199   AliEveEventManager::AssertMagField();
 200   if (!AliMUONESDInterface::GetTracker()) AliMUONESDInterface::ResetTracker(AliMUONCDB::LoadRecoParam());
 201
 202   // load mapping
 203   AliMpCDB::LoadAll(kFALSE);
 204
 205   // convert ESD objects to MUON objects
 206   AliMUONESDInterface data;
 207   data.LoadEvent(*esd);
 208
 209   // track containers
 210   TEveElementList* trackCont = new TEveElementList("ESD MUON Tracks");
 211   trackCont->SetTitle(Form("N=%d", esd->GetNumberOfMuonTracks()));
 212
 213   TEveTrackList* match = new TEveTrackList("Matched");
 214   match->SetRnrPoints(kFALSE);
 215   match->SetRnrLine(kTRUE);
 216   match->SetLineColor(kGreen);
 217   esd_muon_track_propagator_setup(match->GetPropagator(), kTRUE, kTRUE);
 218   trackCont->AddElement(match);
 219
 220   TEveTrackList* nomatch = new TEveTrackList("Not matched");
 221   nomatch->SetRnrPoints(kFALSE);
 222   nomatch->SetRnrLine(kTRUE);
 223   nomatch->SetLineColor(kAzure);
 224   esd_muon_track_propagator_setup(nomatch->GetPropagator(), kTRUE, kFALSE);
 225   trackCont->AddElement(nomatch);
 226
 227   TEveTrackList* ghost = new TEveTrackList("Ghost");
 228   ghost->SetRnrPoints(kFALSE);
 229   ghost->SetRnrLine(kTRUE);
 230   ghost->SetLineColor(kAzure);
 231   esd_muon_track_propagator_setup(ghost->GetPropagator(), kFALSE, kTRUE);
 232   trackCont->AddElement(ghost);
 233
 234   // cluster container
 235   TEvePointSet* clusterList = 0x0;
 236   if (showClusters && (data.GetNClusters() > 0 || gEve->GetKeepEmptyCont()))
 237   {
 238     clusterList = new TEvePointSet(10000);
 239     clusterList->SetName("ESD MUON Clusters");
 240     clusterList->SetTitle(Form("N=%d",data.GetNClusters()));
 241     clusterList->SetPickable(kFALSE);
 242     clusterList->SetMarkerStyle(5);
 243     clusterList->SetMarkerColor(kYellow);
 244     clusterList->SetMarkerSize(2.5);
 245   }
 246
 247   // digit containers
 248   TEveElementList* digitCont = 0x0;
 249   TEveQuadSet* bending = 0x0;
 250   TEveQuadSet* nonBending = 0x0;
 251   if (showDigits && (data.GetNDigits() > 0 || gEve->GetKeepEmptyCont()))
 252   {
 253     digitCont = new TEveElementList("ESD MUON Digits");
 254     digitCont->SetTitle(Form("N=%d",data.GetNDigits()));
 255
 256     bending = new TEveQuadSet(TEveQuadSet::kQT_RectangleXY, kFALSE, 32);
 257     bending->SetName("Bending");
 258     bending->SetRenderMode(TEveDigitSet::kRM_Fill);
 259     bending->SetPickable(kFALSE);
 260     digitCont->AddElement(bending);
 261
 262     nonBending = new TEveQuadSet(TEveQuadSet::kQT_RectangleXY, kFALSE, 32);
 263     nonBending->SetName("Non bending");
 264     nonBending->SetRenderMode(TEveDigitSet::kRM_Line);
 265     nonBending->SetPickable(kFALSE);
 266     digitCont->AddElement(nonBending);
 267   }
 268
 269   // add tracks to the proper list and propagate them
 270   add_esd_muon_tracks(esd, &data, match, nomatch, ghost);
 271   match->SetTitle(Form("N=%d",match->NumChildren()));
 272   nomatch->SetTitle(Form("N=%d",nomatch->NumChildren()));
 273   ghost->SetTitle(Form("N=%d",ghost->NumChildren()));
 274   match->MakeTracks();
 275   nomatch->MakeTracks();
 276   ghost->MakeTracks();
 277
 278   // add cluster to the container
 279   if (clusterList)
 280   {
 281     TEveUtil::LoadMacro("muon_clusters.C+");
 282     TIter next(data.CreateClusterIterator());
 283     gROOT->ProcessLine(Form("add_muon_clusters((TIter*)%p, (TEvePointSet*)%p);",&next, clusterList));
 284   }
 285
 286   // add digits to the containers
 287   if (digitCont)
 288   {
 289     TEveUtil::LoadMacro("muon_digits.C+");
 290     TIter next(data.CreateDigitIterator());
 291     gROOT->ProcessLine(Form("add_muon_digits((TIter*)%p, (TEveQuadSet*)%p, (TEveQuadSet*)%p, kFALSE);",
 292                             &next, bending, nonBending));
 293
 294     // set containers' title
 295     bending->SetTitle(Form("N=%d",bending->GetPlex()->Size()));
 296     nonBending->SetTitle(Form("N=%d",nonBending->GetPlex()->Size()));
 297
 298     // automatic scaling
 299     gStyle->SetPalette(1);
 300     bending->AssertPalette();
 301     nonBending->AssertPalette();
 302   }
 303
 304   // add graphic containers
 305   gEve->DisableRedraw();
 306   gEve->AddElement(trackCont);
 307   if (clusterList) gEve->AddElement(clusterList);
 308   if (digitCont) gEve->AddElement(digitCont);
 309   gEve->EnableRedraw();
 310   gEve->Redraw3D();
 311 }