EVE/alice-macros/histo2d_all_events.C

   1 // Author: Stefano Carrazza 2010
   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 #if !defined(__CINT__) || defined(__MAKECINT__)
  10 #include <TGLViewer.h>
  11 #include <TGLWidget.h>
  12 #include <TH2.h>
  13 #include <TMath.h>
  14 #include <TTree.h>
  15 #include <TEveBrowser.h>
  16 #include <TEveCalo.h>
  17 #include <TEveCaloData.h>
  18 #include <TEveCaloLegoOverlay.h>
  19 #include <TEveLegoEventHandler.h>
  20 #include <TEveManager.h>
  21 #include <TEveProjectionManager.h>
  22 #include <TEveProjectionAxes.h>
  23 #include <TEveScene.h>
  24 #include <TEveTrans.h>
  25 #include <TEveViewer.h>
  26 #include <TEveWindow.h>
  27
  28 #include <AliESDEvent.h>
  29 #include <AliEveEventManager.h>
  30 #endif
  31
  32 double pi = TMath::Pi();
  33 TEveViewer *g_histo2d_all_events_v0 = 0;
  34 TEveViewer *g_histo2d_all_events_v1 = 0;
  35 TEveViewer *g_histo2d_all_events_v2 = 0;
  36 TEveViewer *g_histo2d_all_events_v3 = 0;
  37 TEveScene  *g_histo2d_all_events_s0 = 0;
  38 TEveScene  *g_histo2d_all_events_s1 = 0;
  39 TEveScene  *g_histo2d_all_events_s2 = 0;
  40 TEveScene  *g_histo2d_all_events_s3 = 0;
  41 TEveCaloLegoOverlay* g_histo2d_all_events_lego_overlay = 0;
  42 TEveWindowSlot* g_histo2d_all_events_slot = 0;
  43
  44 Double_t GetPhi(Double_t phi);
  45 TEveCaloLego* CreateHistoLego(TEveCaloData* data, TEveWindowSlot* slot);
  46 TEveCalo3D* Create3DView(TEveCaloData* data, TEveWindowSlot* slot);
  47 void CreateProjections(TEveCaloData* data, TEveCalo3D *calo3d, TEveWindowSlot* slot1, TEveWindowSlot* slot2);
  48
  49 TEveCaloDataHist* histo2d_all_events()
  50 {
  51
  52    TEveCaloDataHist* data_t;
  53
  54    if ( g_histo2d_all_events_slot == 0 ) {
  55       Info("histo2d_all_events", "Filling histogram...");
  56
  57       // Access to esdTree
  58       AliESDEvent* esd = AliEveEventManager::AssertESD();
  59       TTree* t = AliEveEventManager::GetMaster()->GetESDTree();
  60
  61       // Creating 2D histograms
  62       TH2F *histopos_t = new TH2F("histopos_t","Histo 2d positive",
  63                                  100,-1.5,1.5,80,-pi,pi);
  64       TH2F *histoneg_t = new TH2F("histoneg_t","Histo 2d negative",
  65                                  100,-1.5,1.5,80,-pi,pi);
  66
  67          // Getting current tracks for each event, filling histograms
  68          for ( int event = 0; event < t->GetEntries(); event++ ) {
  69             t->GetEntry(event);
  70                for ( int n = 0; n < esd->GetNumberOfTracks(); ++n ) {
  71
  72                   if ( esd->GetTrack(n)->GetSign() > 0 ) {
  73                      histopos_t->Fill(esd->GetTrack(n)->Eta(),
  74                                       GetPhi(esd->GetTrack(n)->Phi()),
  75                                       fabs(esd->GetTrack(n)->Pt()));
  76                   } else {
  77                      histoneg_t->Fill(esd->GetTrack(n)->Eta(),
  78                                       GetPhi(esd->GetTrack(n)->Phi()),
  79                                       fabs(esd->GetTrack(n)->Pt()));
  80                   }
  81                }
  82          }
  83
  84       data_t = new TEveCaloDataHist();
  85       data_t->AddHistogram(histoneg_t);
  86       data_t->RefSliceInfo(0).Setup("NegCg:", 0, kBlue);
  87       data_t->AddHistogram(histopos_t);
  88       data_t->RefSliceInfo(1).Setup("PosCg:", 0, kRed);
  89       data_t->GetEtaBins()->SetTitleFont(120);
  90       data_t->GetEtaBins()->SetTitle("h");
  91       data_t->GetPhiBins()->SetTitleFont(120);
  92       data_t->GetPhiBins()->SetTitle("f");
  93       data_t->IncDenyDestroy();
  94
  95       // Creating frames
  96       g_histo2d_all_events_slot = TEveWindow::CreateWindowInTab(gEve->GetBrowser()->GetTabRight());
  97       TEveWindowPack* packH = g_histo2d_all_events_slot->MakePack();
  98       packH->SetElementName("Projections");
  99       packH->SetHorizontal();
 100       packH->SetShowTitleBar(kFALSE);
 101
 102       g_histo2d_all_events_slot = packH->NewSlot();
 103       TEveWindowPack* pack0 = g_histo2d_all_events_slot->MakePack();
 104       pack0->SetShowTitleBar(kFALSE);
 105       TEveWindowSlot*  slotLeftTop   = pack0->NewSlot();
 106       TEveWindowSlot* slotLeftBottom = pack0->NewSlot();
 107
 108       g_histo2d_all_events_slot = packH->NewSlot();
 109       TEveWindowPack* pack1 = g_histo2d_all_events_slot->MakePack();
 110       pack1->SetShowTitleBar(kFALSE);
 111       TEveWindowSlot* slotRightTop    = pack1->NewSlot();
 112       TEveWindowSlot* slotRightBottom = pack1->NewSlot();
 113
 114       // Creating viewers and scenes
 115       TEveCalo3D* calo3d = Create3DView(data_t, slotLeftTop);
 116       CreateHistoLego(data_t, slotLeftBottom);
 117       CreateProjections(data_t, calo3d, slotRightTop, slotRightBottom);
 118
 119       gEve->Redraw3D(kTRUE);
 120
 121       Info("histo2d_all_events", "...Finished");
 122    }
 123
 124    return data_t;
 125 }
 126
 127 //______________________________________________________________________________
 128 Double_t GetPhi(Double_t phi)
 129 {
 130    if (phi > pi) {
 131       phi -= 2*pi;
 132    }
 133    return phi;
 134 }
 135
 136 //______________________________________________________________________________
 137 TEveCaloLego* CreateHistoLego(TEveCaloData* data, TEveWindowSlot* slot){
 138
 139    TEveCaloLego* lego;
 140
 141    // Viewer initialization, tab creation
 142    if ( g_histo2d_all_events_v0 == 0 ) {
 143
 144       TEveBrowser *browser = gEve->GetBrowser();
 145       slot->MakeCurrent();
 146       g_histo2d_all_events_v0 = gEve->SpawnNewViewer("2D Lego Histogram", "2D Lego Histogram");
 147       g_histo2d_all_events_s0 = gEve->SpawnNewScene("2D Lego Histogram", "2D Lego Histogram");
 148       g_histo2d_all_events_v0->AddScene(g_histo2d_all_events_s0);
 149       g_histo2d_all_events_v0->SetElementName("2D Lego Viewer");
 150       g_histo2d_all_events_s0->SetElementName("2D Lego Scene");
 151
 152       TGLViewer* glv = g_histo2d_all_events_v0->GetGLViewer();
 153       g_histo2d_all_events_lego_overlay = new TEveCaloLegoOverlay();
 154       glv->AddOverlayElement(g_histo2d_all_events_lego_overlay);
 155       glv->SetCurrentCamera(TGLViewer::kCameraPerspXOY);
 156
 157       // Plotting histogram lego
 158       lego = new TEveCaloLego(data);
 159       g_histo2d_all_events_s0->AddElement(lego);
 160
 161       // Move to real world coordinates
 162       lego->InitMainTrans();
 163       Float_t sc = TMath::Min(lego->GetEtaRng(), lego->GetPhiRng());
 164       lego->RefMainTrans().SetScale(sc, sc, sc);
 165
 166       // Set event handler to move from perspective to orthographic view.
 167       glv->SetEventHandler(new TEveLegoEventHandler(glv->GetGLWidget(), glv, lego));
 168
 169       g_histo2d_all_events_lego_overlay->SetCaloLego(lego);
 170    }
 171
 172    return lego;
 173 }
 174
 175 //______________________________________________________________________________
 176 TEveCalo3D* Create3DView(TEveCaloData* data, TEveWindowSlot* slot){
 177
 178    TEveCalo3D* calo3d;
 179
 180    if ( g_histo2d_all_events_v1 == 0 ) {
 181
 182       TEveBrowser *browser = gEve->GetBrowser();
 183       slot->MakeCurrent();
 184       g_histo2d_all_events_v1 = gEve->SpawnNewViewer("3D Histogram", "3D Histogram");
 185       g_histo2d_all_events_s1 = gEve->SpawnNewScene("3D Histogram", "3D Histogram");
 186       g_histo2d_all_events_v1->AddScene(g_histo2d_all_events_s1);
 187       g_histo2d_all_events_v1->SetElementName("3D Histogram Viewer");
 188       g_histo2d_all_events_s1->SetElementName("3D Histogram Scene");
 189
 190       calo3d = new TEveCalo3D(data);
 191
 192       calo3d->SetBarrelRadius(550);
 193       calo3d->SetEndCapPos(550);
 194       g_histo2d_all_events_s1->AddElement(calo3d);
 195    }
 196
 197    return calo3d;
 198 }
 199
 200 //______________________________________________________________________________
 201 void CreateProjections(TEveCaloData* data, TEveCalo3D *calo3d, TEveWindowSlot* slot1, TEveWindowSlot* slot2){
 202
 203    if ( g_histo2d_all_events_v2 == 0 ) {
 204
 205       TEveBrowser *browser = gEve->GetBrowser();
 206       slot1->MakeCurrent();
 207       g_histo2d_all_events_v2 = gEve->SpawnNewViewer("RPhi projection", "RPhi projection");
 208       g_histo2d_all_events_s2 = gEve->SpawnNewScene("RPhi projection", "RPhi projection");
 209       g_histo2d_all_events_v2->AddScene(g_histo2d_all_events_s2);
 210       g_histo2d_all_events_v2->SetElementName("RPhi Projection Viewer");
 211       g_histo2d_all_events_s2->SetElementName("RPhi Projection Scene");
 212
 213       TEveProjectionManager* mng1 = new TEveProjectionManager();
 214       mng1->SetProjection(TEveProjection::kPT_RPhi);
 215
 216       TEveProjectionAxes* axeg_histo2d_all_events_s1 = new TEveProjectionAxes(mng1);
 217       g_histo2d_all_events_s2->AddElement(axeg_histo2d_all_events_s1);
 218       TEveCalo2D* calo2d1 = (TEveCalo2D*) mng1->ImportElements(calo3d);
 219       g_histo2d_all_events_s2->AddElement(calo2d1);
 220
 221       g_histo2d_all_events_v2->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
 222    }
 223
 224    if ( g_histo2d_all_events_v3 == 0 ) {
 225
 226       TEveBrowser *browser = gEve->GetBrowser();
 227       slot2->MakeCurrent();
 228       g_histo2d_all_events_v3 = gEve->SpawnNewViewer("RhoZ projection", "RhoZ projection");
 229       g_histo2d_all_events_s3 = gEve->SpawnNewScene("RhoZ projection", "RhoZ projection");
 230       g_histo2d_all_events_v3->AddScene(g_histo2d_all_events_s3);
 231       g_histo2d_all_events_v3->SetElementName("RhoZ Projection Viewer");
 232       g_histo2d_all_events_s3->SetElementName("RhoZ Projection Viewer");
 233
 234       TEveProjectionManager* mng2 = new TEveProjectionManager();
 235       mng2->SetProjection(TEveProjection::kPT_RhoZ);
 236
 237       TEveProjectionAxes* axeg_histo2d_all_events_s2 = new TEveProjectionAxes(mng2);
 238       g_histo2d_all_events_s3->AddElement(axeg_histo2d_all_events_s2);
 239       TEveCalo2D* calo2d2 = (TEveCalo2D*) mng2->ImportElements(calo3d);
 240       g_histo2d_all_events_s3->AddElement(calo2d2);
 241
 242       g_histo2d_all_events_v3->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
 243    }
 244
 245    return;
 246 }