Add positility to import geometry.root at Init

diff --git a/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.cxx b/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.cxx
index b966ee9cb7fd0c3510a9acd50e1210d5951d0b76..3fec2acebff4bf99d972124c41b75364c7b31db2 100644 (file)
--- a/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.cxx
+++ b/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.cxx
@@ -83,8 +83,8 @@ AliAnalysisTaskEMCALClusterize::AliAnalysisTaskEMCALClusterize(const char *name)
 , fCellMatchdEta(),       fCellMatchdPhi()
 , fMaxEvent(1000000000),  fDoTrackMatching(kFALSE)
 , fSelectCell(kFALSE),    fSelectCellMinE(0.005),     fSelectCellMinFrac(0.001)
-, fRemoveLEDEvents(kFALSE), fRemoveExoticEvents(kFALSE)
-
+, fRemoveLEDEvents(kFALSE),        fRemoveExoticEvents(kFALSE)
+, fImportGeometryFromFile(kFALSE), fImportGeometryFilePath("")
 {
   //ctor
   for(Int_t i = 0; i < 10;    i++)  fGeomMatrix[i] =  0;
@@ -125,6 +125,7 @@ AliAnalysisTaskEMCALClusterize::AliAnalysisTaskEMCALClusterize()
 , fMaxEvent(1000000000),    fDoTrackMatching(kFALSE)
 , fSelectCell(kFALSE),      fSelectCellMinE(0.005),     fSelectCellMinFrac(0.001)
 , fRemoveLEDEvents(kFALSE), fRemoveExoticEvents(kFALSE)
+, fImportGeometryFromFile(kFALSE), fImportGeometryFilePath("")
 {
   // Constructor
   for(Int_t i = 0; i < 10;    i++)  fGeomMatrix[i] =  0;
@@ -240,7 +241,7 @@ Bool_t AliAnalysisTaskEMCALClusterize::AccessOCDB()
   return kTRUE;
 }
 
-//_______________________________________________________________________
+//_____________________________________________________
 void AliAnalysisTaskEMCALClusterize::CheckAndGetEvent()
 {
   // Get the input event, it can depend in embedded events what you want to get
@@ -681,6 +682,12 @@ void AliAnalysisTaskEMCALClusterize::Init()
     
   }
   
+  // Init geometry, I do not like much to do it like this ...
+  if(fImportGeometryFromFile && !gGeoManager) {
+    printf("AliAnalysisTaskEMCALClusterize::Init() - Import geometry.root file\n");
+    TGeoManager::Import(Form("%s/geometry.root", fImportGeometryFilePath.Data())) ; // default need file "geometry.root" in local dir!!!!
+  }
+  
 }  
 
 //_______________________________________________________

diff --git a/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.h b/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.h
index e8becb9549b3191aee4078551e42e192a0bab201..618d9d0619c02ec53c576d9bec4406aabd3139e1 100644 (file)
--- a/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.h
+++ b/PWG4/CaloCalib/AliAnalysisTaskEMCALClusterize.h
@@ -62,6 +62,11 @@ class AliAnalysisTaskEMCALClusterize : public AliAnalysisTaskSE {
   void           SwitchOffLoadOwnGeometryMatrices()             { fLoadGeomMatrices = kFALSE   ; } 
   void           SetGeometryMatrixInSM(TGeoHMatrix* m, Int_t i) { fGeomMatrix[i]    = m        ; }
 
+  void           SetImportGeometryFromFile(Bool_t  im, 
+                                           TString pa = ".")    { 
+                                                                  fImportGeometryFromFile = im ; 
+                                                                  fImportGeometryFilePath = pa ; }    
+  
   //AOD methods
   void           SetAODBranchName(TString &name)                { fOutputAODBranchName = name  ; }
   void           FillAODFile(Bool_t yesno)                      { fFillAODFile         = yesno ; }
@@ -152,7 +157,11 @@ class AliAnalysisTaskEMCALClusterize : public AliAnalysisTaskSE {
   Bool_t                 fRemoveLEDEvents;         // Remove LED events, use only for LHC11a 
   Bool_t                 fRemoveExoticEvents;      // Remove exotic events
   
-  ClassDef(AliAnalysisTaskEMCALClusterize, 16);
+  Bool_t                 fImportGeometryFromFile;  // Import geometry settings in geometry.root file
+  TString                fImportGeometryFilePath;  // path fo geometry.root file
+
+  
+  ClassDef(AliAnalysisTaskEMCALClusterize, 17);
 
 };