From Jennifer - get geo-manager via AliEveEventManager::AssertGeometry().

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

diff --git a/EVE/alice-macros/emcal_all.C b/EVE/alice-macros/emcal_all.C
index 6d0b267630b501791c9fb948e6ec17f83886f370..1801dac2c8ab8d7a6183504c176d9fd4620e99a0 100644 (file)
--- a/EVE/alice-macros/emcal_all.C
+++ b/EVE/alice-macros/emcal_all.C
@@ -32,44 +32,76 @@ void emcal_all(const UInt_t evtNum = 0, Bool_t digFile = 0,
               const UInt_t eventsToProcess = 5, TString dirName = "./", 
               const TString esdTreeName = "esdTree", const char *  pattern = ".")
 {
+
+  Int_t iLoader             = 1;
+  Int_t iESD                = 1;
+  Int_t iAOD                = 0;
+  Int_t iHits               = 1;
+  Int_t iDigits             = 1;
+  Int_t iClusters           = 1;
+
   AliRunLoader* rl =  AliEveEventManager::AssertRunLoader();
-  if (rl == 0x0)
-    cout<<"Can not instatiate the Run Loader"<<endl;
-  AliESDEvent* esd = AliEveEventManager::AssertESD();
+  // runloader check already in AssertRunLoader function 
+  AliESDEvent* esd = 0x0;
+  if(iESD) esd = AliEveEventManager::AssertESD();
+  // esd check already in AssertESD function 
   AliEMCALLoader *emcl = dynamic_cast<AliEMCALLoader*> (rl->GetDetectorLoader("EMCAL"));
   Int_t evtID = AliEveEventManager::GetMaster()->GetEventId();
   if(evtID != evtNum) AliEveEventManager::GetMaster()->GotoEvent(evtNum);
 
-  //Load Hits
-  rl->LoadHits("EMCAL");
-  //Load Digits
-  rl->LoadDigits("EMCAL");
-  //Load RecPoints
-  rl->LoadRecPoints("EMCAL");
-  
-  TTree* ht = rl->GetTreeH("EMCAL",false);
-  TTree* dt = rl->GetTreeD("EMCAL",false);
-  TTree *rt = rl->GetTreeR("EMCAL",false);
-
-  gGeoManager = gEve->GetDefaultGeometry();
+  TTree* ht = 0x0; 
+  TTree* dt = 0x0; 
+  TTree* rt = 0x0; 
+  if(iLoader)
+    {
+      //Load Hits
+      if(iHits) {
+       if(!rl->LoadHits("EMCAL"))
+         ht = rl->GetTreeH("EMCAL",false);
+       else {printf("Please make sure a have a EMCal.Hits.root file \n"); return;}
+      }
+      //Load Digits
+      if(iDigits) {
+       if(!rl->LoadDigits("EMCAL"))
+         dt = rl->GetTreeD("EMCAL",false);
+       else {printf("Please make sure a have a EMCal.Digits.root file \n"); return;}
+      }
+      //Load RecPoints
+      if(iClusters) {
+       if(!rl->LoadRecPoints("EMCAL"))
+         rt = rl->GetTreeR("EMCAL",false);
+       else {printf("Please make sure a have a EMCal.RecPoints.root file \n"); return;}
+      }
+    }
+
+  //  gGeoManager = gEve->GetDefaultGeometry();
+  AliEveEventManager::AssertGeometry();
   TGeoNode* node = gGeoManager->GetTopVolume()->FindNode("XEN1_1");
   TGeoHMatrix* m = gGeoManager->GetCurrentMatrix();
   emcal_data = new AliEveEMCALData(rl,node,m);
-  emcal_data->SetESD(esd);
-  // RunLoader information
-//  emcal_data->LoadHits(ht); // Does not work with my aliroot version ?
-//  emcal_data->LoadDigits(dt);
-//  emcal_data->LoadRecPoints(rt);
-
-  // To be uncommented if use of emcalLoader
-  rl->GetEvent(evtNum);
-  emcal_data->LoadHitsFromEMCALLoader(emcl);       
-//  emcal_data->LoadDigitsFromEMCALLoader(emcl);     
-//  emcal_data->LoadRecPointsFromEMCALLoader(emcl); 
-
-  // To be uncommented to read esds
-  emcal_data->LoadDigitsFromESD();
-  emcal_data->LoadRecPointsFromESD();
+  if(iESD) emcal_data->SetESD(esd);
+
+  // Get information from RunLoader
+  if(iLoader)
+    {
+      if(iHits)     emcal_data->LoadHits(ht); // Does not work with my aliroot version 
+      if(iDigits)   emcal_data->LoadDigits(dt);
+      if(iClusters) emcal_data->LoadRecPoints(rt);
+      
+      rl->GetEvent(evtNum);
+
+      if(iHits)     emcal_data->LoadHitsFromEMCALLoader(emcl);       
+      if(iDigits)   emcal_data->LoadDigitsFromEMCALLoader(emcl);     
+      if(iClusters) emcal_data->LoadRecPointsFromEMCALLoader(emcl); 
+    }
+
+  // Get information from ESDs
+  if(iESD)
+    {
+      rl->GetEvent(evtNum);
+      if(iDigits) emcal_data->LoadDigitsFromESD();
+      if(iClusters) emcal_data->LoadRecPointsFromESD();
+    }
 
   gStyle->SetPalette(1, 0);
 
@@ -82,11 +114,9 @@ void emcal_all(const UInt_t evtNum = 0, Bool_t digFile = 0,
 
   for (Int_t sm=0; sm<12; sm++)
     {
-      cout << "\n Form: " << Form("SM %d", sm) << endl;
       AliEveEMCALSModule* esm = new AliEveEMCALSModule(sm,Form("SM %d Element \n", sm),"test");
       //      esm->SetSModuleID(sm);
       esm->SetDataSource(emcal_data);
-      esm->ComputeBBox();
       esm->UpdateQuads();
       l->AddElement(esm);
     }