macros for ITS upgrade visualization

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

diff --git a/EVE/alice-macros/emcal_digits.C b/EVE/alice-macros/emcal_digits.C
index 836bced5b048add02253b888e75c087da787e749..bed86d74ba8fd4257d537ad76e107a779ab1b679 100644 (file)
--- a/EVE/alice-macros/emcal_digits.C
+++ b/EVE/alice-macros/emcal_digits.C
@@ -6,43 +6,65 @@
  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
  * full copyright notice.                                                 *
  **************************************************************************/
+#ifndef __CINT__
+#include <TBranch.h>
+#include <TClonesArray.h>
+#include <TGeoNode.h>
+#include <TGeoBBox.h>
+#include <TGeoManager.h>
+#include <TStyle.h>
+#include <TTree.h>
+#include <TEveManager.h>
+#include <TEveQuadSet.h>
+#include <TEveTrans.h>
+
+#include <AliRunLoader.h>
+#include <AliCluster.h>
+#include <AliEMCALGeometry.h>
+#include <AliEMCALDigit.h>
+#include <AliLog.h>
+#include <AliEveEventManager.h>
+#endif
 
 void emcal_digits()
 {
-  AliRunLoader* rl =  AliEveEventManager::AssertRunLoader();
-
-  rl->LoadgAlice();
-  AliEMCAL         * emcal = (AliEMCAL*) rl->GetAliRun()->GetDetector("EMCAL");
-  AliEMCALGeometry * geom  = emcal->GetGeometry();
-
-  rl->LoadDigits("EMCAL");
-  TTree* dt = rl->GetTreeD("EMCAL", kFALSE);
+  AliEveEventManager::AssertGeometry();
 
-  gGeoManager = gEve->GetDefaultGeometry();
   TGeoNode* node = gGeoManager->GetTopVolume()->FindNode("XEN1_1");
+  if (!node) return;
 
-  TGeoBBox* bbbox = (TGeoBBox*) node->GetDaughter(0) ->GetVolume()->GetShape();
-  bbbox->Dump();
-  TGeoBBox* sbbox = (TGeoBBox*) node->GetDaughter(10)->GetVolume()->GetShape();
-  sbbox->Dump();
+  Int_t nModules = node->GetNdaughters();
 
   TEveElementList* l = new TEveElementList("EMCAL");
   l->SetTitle("Tooltip");
   gEve->AddElement(l);
 
+  TGeoBBox* bbbox = (TGeoBBox*) node->GetDaughter(0) ->GetVolume()->GetShape();
   TEveFrameBox* frame_big = new TEveFrameBox();
+  frame_big->SetFrameColorRGBA(200,200,0,50);
   frame_big->SetAABoxCenterHalfSize(0, 0, 0, bbbox->GetDX(), bbbox->GetDY(), bbbox->GetDZ());
 
-  TEveFrameBox* frame_sml = new TEveFrameBox();
-  frame_sml->SetAABoxCenterHalfSize(0, 0, 0, sbbox->GetDX(), sbbox->GetDY(), sbbox->GetDZ());
+  TEveFrameBox* frame_sml = 0x0;
+
+  if (nModules==12) {
+    TGeoBBox* sbbox = (TGeoBBox*) node->GetDaughter(10)->GetVolume()->GetShape();
+    frame_sml = new TEveFrameBox();
+    frame_sml->SetFrameColorRGBA(200,200,0,50);
+    frame_sml->SetAABoxCenterHalfSize(0, 0, 0, sbbox->GetDX(), sbbox->GetDY(), sbbox->GetDZ());
+  }
 
   gStyle->SetPalette(1, 0);
   TEveRGBAPalette* pal = new TEveRGBAPalette(0, 512);
   pal->SetLimits(0, 1024);
 
   TEveQuadSet* smodules[12];
+  memset(smodules,0,12*sizeof(TEveQuadSet*));
 
-  for (Int_t sm=0; sm<12; ++sm)
+
+  AliEMCALGeometry * geom  = AliEMCALGeometry::GetInstance();  
+  if (!geom) geom = AliEMCALGeometry::GetInstance("","");
+
+  for (Int_t sm=0; sm<nModules; ++sm)
   {
     TEveQuadSet* q = new TEveQuadSet(Form("SM %d", sm+1));
     q->SetOwnIds(kTRUE);
@@ -59,13 +81,18 @@ void emcal_digits()
     smodules[sm] = q;
   }
 
+  AliRunLoader* rl =  AliEveEventManager::AssertRunLoader();
+
+  rl->LoadDigits("EMCAL");
+  TTree* dt = rl->GetTreeD("EMCAL", kFALSE);
+  if (!dt) return;
+
   TClonesArray *digits = 0;
   dt->SetBranchAddress("EMCAL", &digits);
   dt->GetEntry(0);
   Int_t nEnt = digits->GetEntriesFast();
   AliEMCALDigit * dig;
 
-  Int_t iEvent  = -1 ;
   Float_t amp   = -1 ;
   Float_t time  = -1 ;
   Int_t id      = -1 ;
@@ -86,7 +113,8 @@ void emcal_digits()
       amp  = dig->GetAmp(); //amplitude in cell (digit)
       time = dig->GetTime();//time of creation of digit after collision
 
-      cout<<"Cell ID "<<id<<" Amp "<<amp<<endl;//" time "<<time<<endl;
+//      AliDebugGeneral("emcal_digits", 5, Form("Cell ID %3d, Amplitude: %f", id, amp));
+      // cout<<"Cell ID "<<id<<" Amp "<<amp<<endl;//" time "<<time<<endl;
 
       //Geometry methods
       geom->GetCellIndex(id,iSupMod,iTower,iIphi,iIeta);
@@ -95,23 +123,32 @@ void emcal_digits()
                                        iIphi, iIeta,iphi,ieta);
       //Gives label of cell in eta-phi position per each supermodule
 
-      cout<< "SModule "<<iSupMod<<"; Tower "<<iTower
-         <<"; Eta "<<iIeta<<"; Phi "<<iIphi
-         <<"; Cell Eta "<<ieta<<"; Cell Phi "<<iphi<<endl;
+//      AliDebugGeneral("emcal_digits", 5, Form("SModule %3d; Tover %3d; Eta %3d; Phi %3d; Cell Eta %3d; Cell Phi %3d",
+//                    iSupMod, iTower, iIeta, iIphi, ieta, iphi));
+      // cout<< "SModule "<<iSupMod<<"; Tower "<<iTower
+      //     <<"; Eta "<<iIeta<<"; Phi "<<iIphi
+      //     <<"; Cell Eta "<<ieta<<"; Cell Phi "<<iphi<<endl;
 
       geom->RelPosCellInSModule(id, x, y, z);
-      cout << x <<" "<< y <<" "<< z <<endl;
+      // cout << x <<" "<< y <<" "<< z <<endl;
+//      AliDebugGeneral("emcal_digits", 5, Form("(x,y,z)=(%8.3f,%8.3f,%8.3f)", x, y, z));
 
       TEveQuadSet* q = smodules[iSupMod];
-      q->AddQuad(y, z);
-      q->QuadValue(amp);
-      q->QuadId(dig);
+      if (q) {
+       q->AddQuad(y, z);
+       q->QuadValue(TMath::Nint(amp));
+       q->QuadId(new AliEMCALDigit(*dig));
+      }
     } else {
-      cout<<"Digit pointer 0x0"<<endl;
+//      AliDebugGeneral("emcal_digits", 1, Form("Digit pointer 0x0"));
+      // cout<<"Digit pointer 0x0"<<endl;
     }
   }
 
-  for (Int_t sm = 0; sm < 12; ++sm)
+  rl->UnloadDigits("EMCAL");
+
+
+  for (Int_t sm = 0; sm < nModules; ++sm)
   {
     smodules[iSupMod]->RefitPlex();
   }