]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVE/alice-macros/emcal_digits.C
doxy: code cleanup: comments and clarifications
[u/mrichter/AliRoot.git] / EVE / alice-macros / emcal_digits.C
index 54c23f11b6a4d2ddc3ad29bae9536e64ead3dd70..6a513c3e9650e5b7297cbb583f679aa1d0669e5e 100644 (file)
 #include <TStyle.h>
 #include <TEveTrans.h>
 #include <TClonesArray.h>
+#include <TTree.h>
+#include <TBranch.h>
 
 #include <EveBase/AliEveEventManager.h>
 
 #include <AliRunLoader.h>
 #include <AliCluster.h>
-#include <EMCAL/AliEMCALGeometry.h>
-#include <EMCAL/AliEMCALDigit.h>
+#include <AliEMCALGeometry.h>
+#include <AliEMCALDigit.h>
+#include <AliLog.h>
 
-#include <Riostream.h>
+// #include <Riostream.h>
 #endif
 
 void emcal_digits()
 {
-  AliRunLoader* rl =  AliEveEventManager::AssertRunLoader();
-
-  rl->LoadDigits("EMCAL");
-  TTree* dt = rl->GetTreeD("EMCAL", kFALSE);
-  if (!dt) return;
-
   AliEveEventManager::AssertGeometry();
 
   TGeoNode* node = gGeoManager->GetTopVolume()->FindNode("XEN1_1");
+  if (!node) return;
 
-  TGeoBBox* bbbox = (TGeoBBox*) node->GetDaughter(0) ->GetVolume()->GetShape();
-  TGeoBBox* sbbox = (TGeoBBox*) node->GetDaughter(10)->GetVolume()->GetShape();
+  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->SetFrameColorRGBA(200,200,0,50);
-  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*));
 
 
   AliEMCALGeometry * geom  = AliEMCALGeometry::GetInstance();  
   if (!geom) geom = AliEMCALGeometry::GetInstance("","");
 
-  for (Int_t sm=0; sm<12; ++sm)
+  for (Int_t sm=0; sm<nModules; ++sm)
   {
     TEveQuadSet* q = new TEveQuadSet(Form("SM %d", sm+1));
     q->SetOwnIds(kTRUE);
@@ -81,6 +85,12 @@ 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);
@@ -107,7 +117,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);
@@ -116,26 +127,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(TMath::Nint(amp));
-      q->QuadId(new AliEMCALDigit(*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;
     }
   }
 
   rl->UnloadDigits("EMCAL");
 
 
-  for (Int_t sm = 0; sm < 12; ++sm)
+  for (Int_t sm = 0; sm < nModules; ++sm)
   {
     smodules[iSupMod]->RefitPlex();
   }