Some unused PDF are not compiled to reduce the memory consumption when loading liblha...

[u/mrichter/AliRoot.git] / MUON / AliMUONGMSSubprocessor.cxx

diff --git a/MUON/AliMUONGMSSubprocessor.cxx b/MUON/AliMUONGMSSubprocessor.cxx
index a36ce61d5a0cc6df0694f595a0e4990e28e0b895..0ac04540f2c0fe91c71112ac7d438a2d0e0279ab 100644 (file)
--- a/MUON/AliMUONGMSSubprocessor.cxx
+++ b/MUON/AliMUONGMSSubprocessor.cxx
@@ -14,20 +14,26 @@
  **************************************************************************/
 
 // $Id$
-//
+
+//-----------------------------------------------------------------------------
 // Class AliMUONGMSSubprocessor
 // -----------------------------
 // The shuttle subprocessor for GMS data
 // Author: Ivana Hrivnacova, IPN Orsay
 // 16/09/2006
+//-----------------------------------------------------------------------------
 
 #include "AliMUONGMSSubprocessor.h"
 #include "AliMUONPreprocessor.h"
+#include "AliMpConstants.h"
 
+#include "AliAlignObjMatrix.h"
 #include "AliCDBMetaData.h"
+#include "AliCDBEntry.h"
 
 #include <TTimeStamp.h>
 #include <TFile.h>
+#include <TArrayI.h>
 #include <TClonesArray.h>
 #include <TObjString.h>
 #include <Riostream.h>
@@ -43,16 +49,17 @@ const TString  AliMUONGMSSubprocessor::fgkMatrixArrayName = "GMSarray";
 //______________________________________________________________________________
 AliMUONGMSSubprocessor::AliMUONGMSSubprocessor(AliMUONPreprocessor* master) 
   : AliMUONVSubprocessor(master, "GMS", "Upload GMS matrices to OCDB"),
-    fTransformer(true)
+    fTransformer(0)
 {
 /// Constructor
-  fTransformer.ReadGeometryData("volpath.dat", "transform.dat");
 }
 
 //______________________________________________________________________________
 AliMUONGMSSubprocessor::~AliMUONGMSSubprocessor()
 {
 /// Destructor
+
+  delete fTransformer;
 }
 
 
@@ -61,13 +68,25 @@ AliMUONGMSSubprocessor::~AliMUONGMSSubprocessor()
 //
 
 
+//______________________________________________________________________________
+void  AliMUONGMSSubprocessor::Initialize(Int_t /*run*/, 
+                                         UInt_t /*startTime*/, UInt_t /*endTime*/)
+{
+/// Instantiate geometry transformer
+
+  if ( ! fTransformer ) {
+    fTransformer = new AliMUONGeometryTransformer();
+    fTransformer->CreateModules();
+  }  
+}                                           
+
 //______________________________________________________________________________
 UInt_t AliMUONGMSSubprocessor::ProcessFile(const TString& fileName)
 {
 /// Convert TGeoHMatrix to AliAlignObjMatrix and fill them into AliTestDataDCS object
 
   Master()->Log(Form("Processing GMS file %s", fileName.Data()));
-
+  
   // Open root file
   TFile f(fileName.Data());
   if ( ! f.IsOpen() ) {
@@ -80,14 +99,23 @@ UInt_t AliMUONGMSSubprocessor::ProcessFile(const TString& fileName)
   if ( ! array ) {
     Master()->Log(Form("TClonesArray not found in file %s",fileName.Data()));
     return 2;
-  }  
+  }    
+  
+  // Array to store correspondance between the moduleId 
+  // and its corresponding entry in the GMS array.
+  TArrayI moduleIdToGMSIndex;
+  moduleIdToGMSIndex.Set(AliMpConstants::NofGeomModules());
+  for (Int_t i=0; i<AliMpConstants::NofGeomModules(); i++){
+    moduleIdToGMSIndex[i]=-1;
+  }
   
   // Convert matrices into Alice alignment objects
   for (Int_t i=0; i<array->GetEntriesFast(); i++ ) {
     TGeoHMatrix* matrix = (TGeoHMatrix*)array->At(i);
-    fTransformer.AddMisAlignModule(matrix->GetUniqueID(), *matrix);
-  }  
-  TObject* data = const_cast< TClonesArray*>(fTransformer.GetMisAlignmentData());
+    fTransformer->AddMisAlignModule(matrix->GetUniqueID(), *matrix);
+    moduleIdToGMSIndex[matrix->GetUniqueID()]=i;
+  }
+  TObject* data = const_cast< TClonesArray*>(fTransformer->GetMisAlignmentData());
   
   //Now we have to store the final CDB file
   Master()->Log("Storing GMS");
@@ -95,12 +123,63 @@ UInt_t AliMUONGMSSubprocessor::ProcessFile(const TString& fileName)
   metaData.SetBeamPeriod(0);
   metaData.SetResponsible("");
   metaData.SetComment("This preprocessor fills GMS alignment objects.");
-
-  Bool_t result = Master()->Store("SHUTTLE", "GMS", data, &metaData, 0, 0);
-
+  
+  Bool_t result = Master()->Store("Align", "GMS", data, &metaData, 0, 0);
+  
+  // This section apply the GMS misalignments on top of the misalignments 
+  // of the GMS reference run and stores the new misalignment array in the OCDB
+  
+  // Get mis alignment from reference run for GMS
+  AliCDBEntry* cdbEntry = Master()->GetFromOCDB("Align", "Baseline");
+  if (cdbEntry) {
+    TClonesArray* refArray = (TClonesArray*)cdbEntry->GetObject();
+    if (refArray) {
+      // Create new misalignment array
+      TClonesArray* newArray = new TClonesArray("AliAlignObjMatrix", 200);
+      for (Int_t i=0; i<refArray->GetEntriesFast(); i++) {
+        AliAlignObjMatrix* refAOMat = (AliAlignObjMatrix*)refArray->At(i);      
+        TGeoHMatrix refMat;
+        refAOMat->GetMatrix(refMat);
+        // Need the module containing this module or detection element
+        TString sName = refAOMat->GetSymName(); //Format "/MUON/GMx" or "/MUON/GMx/DEy"
+        Int_t iGM = sName.Index("GM");
+        Int_t iLS = sName.Last('/');
+        if (iLS>iGM) { // This is a detection element
+          sName.Remove(iLS,sName.Sizeof());
+        }
+        sName.Remove(0,iGM+2);
+        Int_t iMod = sName.Atoi();
+        if (moduleIdToGMSIndex[iMod]>=0) {
+          TGeoHMatrix* gmsMat = (TGeoHMatrix*)array->At(moduleIdToGMSIndex[iMod]);
+          refMat.MultiplyLeft(gmsMat);
+        }
+        else {
+          Master()->Log(Form("Missing GMS entry for module %d",iMod));
+        }
+        AliAlignObjMatrix* newAOMat = new((*newArray)[i]) AliAlignObjMatrix(*refAOMat); // Copy the reference misalignment object to the new array ...
+        newAOMat->SetMatrix(refMat); // ... and set its matrix
+      }
+      
+      //Now we have also to store this CDB file
+      TString sMDComment(cdbEntry->GetMetaData()->GetComment());
+      sMDComment += " GMS";
+      Master()->Log("Storing MisAlignment");
+      metaData.SetComment(sMDComment);
+      
+      result = result && Master()->Store("Align", "Data", newArray, &metaData, 0, 1);
+    }
+    else {
+      Master()->Log("Empty entry?");
+    }
+  }
+  else {
+    Master()->Log("Could not get GMS reference misalignment from OCDB! Will not add a new MUON/Align/Data entry!");    
+  }
+  // Done with applying GMS misalignments on top of misalignment of reference run
+  
   // Clear MisAlignArray in transformer
-  fTransformer.ClearMisAlignmentData();
-
+  fTransformer->ClearMisAlignmentData();
+  
   return (result!=kTRUE);
 }