Adding new macro, loadFromOCDB.C, for loading magnetic field, mapping and

reconstruction parameters from OCDB
(Philippe P.)

diff --git a/MUON/loadFromOCDB.C b/MUON/loadFromOCDB.C
new file mode 100644 (file)
index 0000000..dd739b8
--- /dev/null
+++ b/MUON/loadFromOCDB.C
@@ -0,0 +1,187 @@
+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ *                                                                        *
+ * Author: The ALICE Off-line Project.                                    *
+ * Contributors are mentioned in the code where appropriate.              *
+ *                                                                        *
+ * Permission to use, copy, modify and distribute this software and its   *
+ * documentation strictly for non-commercial purposes is hereby granted   *
+ * without fee, provided that the above copyright notice appears in all   *
+ * copies and that both the copyright notice and this permission notice   *
+ * appear in the supporting documentation. The authors make no claims     *
+ * about the suitability of this software for any purpose. It is          *
+ * provided "as is" without express or implied warranty.                  *
+ **************************************************************************/
+
+// $Id$
+
+/// \ingroup macros
+/// \file loadFromOCDB.C
+/// \brief Load magnetic field, mapping and reconstruction parameters from OCDB
+/// \author Philippe Pillot, SUBATECH
+
+#include <TString.h>
+#include <TObjArray.h>
+#include <TGeoGlobalMagField.h>
+#include <TGrid.h>
+
+#include "AliCDBManager.h"
+#include "AliCDBEntry.h"
+#include "AliGRPManager.h"
+
+#include "AliMpCDB.h"
+
+#include "AliMUONRecoParam.h"
+
+// prototypes
+Bool_t setupOCDB(const TString ocdbPath = "local://$ALICE_ROOT/OCDB", Int_t runNumber = 0);
+Bool_t checkOCDB();
+Bool_t loadField();
+Bool_t loadMapping();
+AliMUONRecoParam* loadRecoParam();
+
+//----------------------------------------------------------------------------------------
+AliMUONRecoParam* loadFromOCDB(const TString ocdbPath = "local://$ALICE_ROOT/OCDB", Int_t runNumber = 0,
+                              Bool_t ldField = kTRUE, Bool_t ldMapping = kTRUE, Bool_t ldRecoParam = kTRUE)
+{
+  /// Set the default OCDB path and run number (overload the existing ones if already set).
+  /// Load magnetic field, mapping and reconstruction parameters from OCDB as requested by the flags.
+  /// Return the AliMUONRecoParam object if loaded (0x0 otherwise).
+  /// Exit in case of any failure.
+  
+  AliMUONRecoParam* recoParam = 0x0;
+  
+  if (!setupOCDB(ocdbPath, runNumber)) exit(-1);
+  
+  if (ldField && !loadField()) exit(-1);
+  
+  if (ldMapping && !loadMapping()) exit(-1);
+  
+  if (ldRecoParam && !(recoParam = loadRecoParam())) exit(-1);
+  
+  return recoParam;
+  
+}
+
+//----------------------------------------------------------------------------------------
+Bool_t setupOCDB(const TString ocdbPath, Int_t runNumber)
+{
+  /// Set the default OCDB storage and run number (overload the existing ones if already set).
+  
+  Info("setupOCDB","setting OCDB path...");
+  
+  if (runNumber < 0) {
+    Error("setupOCDB", "runNumber must be a positive value");
+    return kFALSE;
+  }
+  
+  if (ocdbPath.BeginsWith("alien://")) TGrid::Connect("alien://");
+  
+  AliCDBManager* man = AliCDBManager::Instance();
+  
+  man->SetDefaultStorage(ocdbPath.Data());
+  if (!man->IsDefaultStorageSet()) return kFALSE;
+  
+  man->SetRun(runNumber);
+  
+  return kTRUE;
+  
+}
+
+//----------------------------------------------------------------------------------------
+Bool_t checkOCDB()
+{
+  /// Check that OCDB path and run number are properly set
+  
+  AliCDBManager* man = AliCDBManager::Instance();
+  if (!man->IsDefaultStorageSet() || man->GetRun() < 0) {
+    Error("checkOCDB", "OCDB path and runNumber must be properly set");
+    return kFALSE;
+  }
+  
+  return kTRUE;
+  
+}
+
+//----------------------------------------------------------------------------------------
+Bool_t loadField()
+{
+  /// Load magnetic field (existing field will be deleted).
+  /// OCDB path is supposed to be set.
+  
+  Info("loadField","Loading field map from GRP...");
+  
+  if (!checkOCDB()) return kFALSE;
+  
+  AliGRPManager grpMan;
+  
+  // in case it has already been set
+  TGeoGlobalMagField::Instance()->Unlock();
+  
+  if (!grpMan.ReadGRPEntry() || !grpMan.SetMagField()) {
+    Error("loadField", "failed to load magnetic field from OCDB");
+    return kFALSE;
+  }
+  
+  return kTRUE;
+  
+}
+
+//----------------------------------------------------------------------------------------
+Bool_t loadMapping()
+{
+  /// Load mapping (existing mapping will be unloaded).
+  /// OCDB path is supposed to be set.
+  
+  Info("loadMapping","Loading mapping from OCDB...");
+  
+  if (!checkOCDB()) return kFALSE;
+  
+  // in case it has already been set
+  AliMpCDB::UnloadAll();
+  
+  if (!AliMpCDB::LoadAll(kTRUE)) {
+    Error("loadMapping","failed to load mapping from OCDB");
+    return kFALSE;
+  }
+  
+  return kTRUE;
+  
+}
+
+//----------------------------------------------------------------------------------------
+AliMUONRecoParam* loadRecoParam()
+{
+  /// Load and return reconstruction parameters.
+  /// OCDB path is supposed to be set.
+  
+  Info("loadRecoParam","Loading RecoParam from OCDB...");
+  
+  if (!checkOCDB()) return kFALSE;
+  
+  AliMUONRecoParam* recoParam = 0x0;
+  AliCDBEntry* entry = AliCDBManager::Instance()->Get("MUON/Calib/RecoParam");
+  
+  if(entry) {
+    
+    // load recoParam according OCDB content (single or array)
+    if (!(recoParam = dynamic_cast<AliMUONRecoParam*>(entry->GetObject()))) {
+      
+      TObjArray* recoParamArray = static_cast<TObjArray*>(entry->GetObject());
+      
+      for(Int_t i = 0; i < recoParamArray->GetEntriesFast(); i++) {
+       recoParam = static_cast<AliMUONRecoParam*>(recoParamArray->UncheckedAt(i));
+       if (recoParam->IsDefault()) break;
+       recoParam = 0x0;
+      }
+      
+    }
+    
+  }
+  
+  if (!recoParam) Error("loadRecoParam", "failed to load RecoParam from OCDB");
+  
+  return recoParam;
+  
+}
+

diff --git a/MUON/loadmacros.C b/MUON/loadmacros.C
index 99fd52e2fcc5e95a7d50e88409bd7751a53b6443..947811afd3013151e499eb9bbfeee583cf9f0f99 100644 (file)
--- a/MUON/loadmacros.C
+++ b/MUON/loadmacros.C
@@ -77,6 +77,7 @@ void loadmacros ()
   loadmacro("DecodeRecoCocktail");         // Hermine, Alessandro     
   loadmacro("fastMUONGen");                // Hermine, Alessandro
   loadmacro("fastMUONSim");                // Hermine, Alessandro
+  loadmacro("loadFromOCDB");               // Philippe P.
   loadmacro("MakeMUONFullMisAlignment");   // Javier, Ivana
   loadmacro("MakeMUONResMisAlignment");    // Javier, Ivana
   loadmacro("MakeMUONZeroMisAlignment");   // Javier, Ivana
@@ -114,4 +115,5 @@ void loadmacros ()
   loadmacro("runSimulation");              // Laurent
   loadmacro("TestMUONPreprocessor");       // Laurent
   loadmacro("TestRecPoints");              // Diego
+  loadmacro("TestRecPoints");              // Diego
 }