New ITS geometry class which uses direct queries to TGeoManager

diff --git a/ITS/AliITSgeomTGeo.cxx b/ITS/AliITSgeomTGeo.cxx
new file mode 100644 (file)
index 0000000..44bec70
--- /dev/null
+++ b/ITS/AliITSgeomTGeo.cxx
@@ -0,0 +1,327 @@
+/**************************************************************************
+ * 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.                  *
+ **************************************************************************/
+
+///////////////////////////////////////////////////////////////////////////
+//    AliITSgeomTGeo is a simple interface class to TGeoManager          //
+//    It is used in the simulation and reconstruction in order to        //
+//    query the TGeo ITS geometry                                        //
+//                                                                       //
+//    author - cvetan.cheshkov@cern.ch                                   //
+//    15/02/2007                                                         //
+///////////////////////////////////////////////////////////////////////////
+
+#include <TString.h>
+#include <TGeoManager.h>
+#include <TGeoPhysicalNode.h>
+
+#include "AliITSgeomTGeo.h"
+#include "AliLog.h"
+#include "AliAlignObj.h"
+
+ClassImp(AliITSgeomTGeo)
+
+const Int_t AliITSgeomTGeo::fgkNModules = 2198;
+const Int_t AliITSgeomTGeo::fgkNLadders[kNLayers] = {20,40,14,22,34,38};
+const Int_t AliITSgeomTGeo::fgkNDetectors[kNLayers] = {4,4,6,8,22,25};
+
+//______________________________________________________________________
+Int_t AliITSgeomTGeo::GetModuleIndex(Int_t lay,Int_t lad,Int_t det)
+{
+  // The method is taken from the old AliITSgeom class by Bjorn Nilsen
+  //
+  // This routine computes the module index number from the layer,
+  // ladder, and detector numbers. The number of ladders and detectors
+  // per layer is set statically
+  // see above for details.
+  // Inputs:
+  //    Int_t lay  The layer number. Starting from 1.
+  //    Int_t lad  The ladder number. Starting from 1.
+  //    Int_t det  The detector number. Starting from 1.
+  // Return:
+  //    the module index number, starting from zero.
+  //    -1 in case of error
+
+  if (lay < 1 || lay > kNLayers) {
+    AliErrorClass(Form("Invalid layer: %d (1 -> %d",lay,kNLayers));
+    return -1;
+  }
+
+  if (lad < 1 || lad > fgkNLadders[lay-1] ||
+      det < 1 || det > fgkNDetectors[lay-1]) {
+    AliErrorClass(Form("Invalid layer,ladder,detector combination: %d, %d, %d",lay,lad,det));
+    return -1;
+  }
+
+  Int_t index = fgkNDetectors[lay-1] * (lad-1) + (det-1);
+  for(Int_t iLayer=0;iLayer < (lay-1); iLayer++)
+    index += fgkNDetectors[iLayer]*fgkNLadders[iLayer];
+
+  return index;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetLayer(Int_t index,Int_t &lay,Int_t &index2) 
+{
+  // The method is taken from the old AliITSgeom class by Bjorn Nilsen
+  //
+  // This routine computes the layer number for a
+  // given the module index. The number of ladders and detectors
+  // per layer is defined statically,
+  // see above for details.
+  // Inputs:
+  //     Int_t index  The module index number, starting from zero.
+  // Outputs:
+  //     Int_t index2 The module index inside a layer, starting from zero.
+  //     Int_t lay    The layer number. Starting from 1.
+  // Return:
+  //     kTRUE in case of valid index
+  //     kFALSE in case of error
+
+  if (index < 0 || index >= fgkNModules) {
+    index2 = -1;
+    AliErrorClass(Form("Invalid module index: %d (0 -> %d)",index,fgkNModules));
+    return -1;
+  }
+
+  lay = 0;
+  index2 = 0;
+  do {
+    index2 += fgkNLadders[lay]*fgkNDetectors[lay];
+    lay++;
+  } while(index2 <= index);
+  index2 -= fgkNLadders[lay-1]*fgkNDetectors[lay-1];
+  index2 = index - index2;
+
+  return lay;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetModuleId(Int_t index,Int_t &lay,Int_t &lad,Int_t &det) 
+{
+  // The method is taken from the old AliITSgeom class by Bjorn Nilsen
+  //
+  // This routine computes the layer, ladder and detector number 
+  // given the module index number. The number of ladders and detectors
+  // per layer is defined statically,
+  // see above for details.
+  // Inputs:
+  //     Int_t index  The module index number, starting from zero.
+  // Outputs:
+  //     Int_t lay    The layer number. Starting from 1.
+  //     Int_t lad    The ladder number. Starting from 1.
+  //     Int_t det    The detector number. Starting from 1.
+  // Return:
+  //     kTRUE in case of valid index
+  //     kFALSE in case of error
+
+  if (index < 0 || index >= fgkNModules) {
+    lay = lad = det = -1;
+    AliErrorClass(Form("Invalid module index: %d (0 -> %d)",index,fgkNModules));
+    return kFALSE;
+  }
+
+  lay  = lad = det = 0;
+  Int_t index2 = 0;
+  do {
+    index2 += fgkNLadders[lay]*fgkNDetectors[lay];
+    lay++;
+  } while(index2 <= index);
+  index2 -= fgkNLadders[lay-1]*fgkNDetectors[lay-1];
+
+  do {
+    index2 += fgkNDetectors[lay-1];
+    lad++;
+  } while(index2 <= index);
+  index2 -= fgkNDetectors[lay-1];
+
+  det = index-index2+1;
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+const char* AliITSgeomTGeo::GetSymName(Int_t index) 
+{
+  // Get the TGeoPNEntry symbolic name
+  // for a given module identified by 'index'
+
+  if (index < 0 || index >= fgkNModules) {
+    AliErrorClass(Form("Invalid ITS module index: %d (0 -> %d) !",index,fgkNModules));
+    return NULL;
+  }
+
+  Int_t lay, index2;
+  if (!GetLayer(index,lay,index2)) return NULL;
+
+  return AliAlignObj::SymName((AliAlignObj::ELayerID)((lay-1)+AliAlignObj::kSPD1),index2);
+}
+
+//______________________________________________________________________
+TGeoHMatrix* AliITSgeomTGeo::GetMatrix(Int_t index) 
+{
+  // Get the transformation matrix for a given module 'index'
+  // by quering the TGeoManager
+
+  TGeoPNEntry *pne = GetPNEntry(index);
+  if (!pne) return NULL;
+
+  const char* path = pne->GetTitle();
+
+  if (!gGeoManager->cd(path)) {
+    AliErrorClass(Form("Volume path %s not valid!",path));
+    return NULL;
+  }
+
+  return gGeoManager->GetCurrentMatrix();
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetTranslation(Int_t index, Double_t t[3]) 
+{
+  // Get the translation vector for a given module 'index'
+  // by quering the TGeoManager
+
+  TGeoHMatrix *m = GetMatrix(index);
+  if (!m) return kFALSE;
+
+  Double_t *trans = m->GetTranslation();
+  for (Int_t i = 0; i < 3; i++) t[i] = trans[i];
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetRotation(Int_t index, Double_t r[9]) 
+{
+  // Get the rotation matrix for a given module 'index'
+  // by quering the TGeoManager
+
+  TGeoHMatrix *m = GetMatrix(index);
+  if (!m) return kFALSE;
+
+  Double_t *rot = m->GetRotationMatrix();
+  for (Int_t i = 0; i < 9; i++) r[i] = rot[i];
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetOrigMatrix(Int_t index, TGeoHMatrix &m)
+{
+  // Get the original (ideal geometry) TGeo matrix for
+  // a given module identified by 'index'.
+  // The method is slow, so it should be used
+  // with great care.
+
+  m.Clear();
+
+  const char *symname = GetSymName(index);
+  if (!symname) return kFALSE;
+
+  return AliAlignObj::GetOrigGlobalMatrix(symname,m);
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetOrigTranslation(Int_t index, Double_t t[3]) 
+{
+  // Get the original translation vector (ideal geometry)
+  // for a given module 'index' by quering the TGeoManager
+
+  TGeoHMatrix m;
+  if (!GetOrigMatrix(index,m)) return kFALSE;
+
+  Double_t *trans = m.GetTranslation();
+  for (Int_t i = 0; i < 3; i++) t[i] = trans[i];
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetOrigRotation(Int_t index, Double_t r[9]) 
+{
+  // Get the original rotation matrix (ideal geometry)
+  // for a given module 'index' by quering the TGeoManager
+
+  TGeoHMatrix m;
+  if (!GetOrigMatrix(index,m)) return kFALSE;
+
+  Double_t *rot = m.GetRotationMatrix();
+  for (Int_t i = 0; i < 9; i++) r[i] = rot[i];
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+const TGeoHMatrix* AliITSgeomTGeo::GetTracking2LocalMatrix(Int_t index)
+{
+  // Get the matrix which transforms from the tracking to local r.s.
+  // The method queries directly the TGeoPNEntry
+
+  TGeoPNEntry *pne = GetPNEntry(index);
+  if (!pne) return NULL;
+
+  const TGeoHMatrix *m = pne->GetMatrix();
+  if (!m)
+    AliErrorClass(Form("TGeoPNEntry (%s) contains no matrix !",pne->GetName()));
+
+  return m;
+}
+
+//______________________________________________________________________
+Bool_t AliITSgeomTGeo::GetTrackingMatrix(Int_t index, TGeoHMatrix &m)
+{
+  // Get the matrix which transforms from the tracking r.s. to
+  // the global one.
+  // Returns kFALSE in case of error.
+
+  m.Clear();
+
+  TGeoHMatrix *m1 = GetMatrix(index);
+  if (!m1) return kFALSE;
+
+  const TGeoHMatrix *m2 = GetTracking2LocalMatrix(index);
+  if (!m2) return kFALSE;
+
+  m = *m1;
+  m.Multiply(m2);
+
+  return kTRUE;
+}
+
+//______________________________________________________________________
+TGeoPNEntry* AliITSgeomTGeo::GetPNEntry(Int_t index)
+{
+  // Get a pointer to the TGeoPNEntry of a module
+  // identified by 'index'
+  // Returns NULL in case of invalid index,
+  // missing TGeoManager or invalid symbolic name
+
+  if (index < 0 || index >= fgkNModules) {
+    AliErrorClass(Form("Invalid ITS module index: %d (0 -> %d) !",index,fgkNModules));
+    return NULL;
+  }
+  
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    AliErrorClass("Can't get the matrix! gGeoManager doesn't exist or it is still opened!");
+    return NULL;
+  }
+
+  TGeoPNEntry* pne = gGeoManager->GetAlignableEntry(GetSymName(index));
+  if (!pne)
+    AliErrorClass(Form("The symbolic volume name %s does not correspond to a physical entry!",
+                      GetSymName(index)));
+
+  return pne;
+}

diff --git a/ITS/AliITSgeomTGeo.h b/ITS/AliITSgeomTGeo.h
new file mode 100644 (file)
index 0000000..a6cddd3
--- /dev/null
+++ b/ITS/AliITSgeomTGeo.h
@@ -0,0 +1,95 @@
+#ifndef ALIITSGEOMTGEO_H
+#define ALIITSGEOMTGEO_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/////////////////////////////////////////////////////////////////////////
+//  AliITSgeomTGeo is a simple interface class to TGeoManager          //
+//  It is used in the simulation and reconstruction in order to        //
+//  query the TGeo ITS geometry                                        //
+//                                                                     //
+//  author - cvetan.cheshkov@cern.ch                                   //
+//  15/02/2007                                                         //
+/////////////////////////////////////////////////////////////////////////
+
+#include <TObject.h>
+#include <TGeoMatrix.h>
+
+class TGeoPNEntry;
+
+class AliITSgeomTGeo : public TObject {
+
+ public:
+
+  AliITSgeomTGeo() { } // Default constructor
+  virtual ~AliITSgeomTGeo() { } // Destructor
+
+  // This function returns the number of detectors/ladder for a given layer 
+  static Int_t GetNDetectors(Int_t lay) {return fgkNDetectors[lay-1];}
+  // This function returns the number of ladders for a given layer
+  static Int_t GetNLadders(Int_t lay)   {return fgkNLadders[lay-1];}
+  // This function returns the number of layers
+  static Int_t GetNLayers()             {return kNLayers;}
+
+  // Two methods to map module index to layer,ladder,detector indeces
+  static Int_t  GetModuleIndex(Int_t lay,Int_t lad,Int_t det);
+  static Bool_t GetModuleId(Int_t index,Int_t &lay,Int_t &lad,Int_t &det);
+
+  static const char *GetSymName(Int_t index); // Get TGeoPNEntry symbolic name
+  static const char *GetSymName(Int_t lay,Int_t lad,Int_t det)
+    { return GetSymName(GetModuleIndex(lay,lad,det)); }
+ 
+  // This function returns a pointer to the TGeoHMatrix (local->global)
+  // of a given module index
+  static TGeoHMatrix* GetMatrix(Int_t index);
+  static TGeoHMatrix* GetMatrix(Int_t lay,Int_t lad,Int_t det)
+    { return GetMatrix(GetModuleIndex(lay,lad,det)); }
+
+  static Bool_t GetTranslation(Int_t index, Double_t t[3]);
+  static Bool_t GetTranslation(Int_t lay,Int_t lad,Int_t det, Double_t t[3])
+    { return GetTranslation(GetModuleIndex(lay,lad,det),t); }
+
+  static Bool_t GetRotation(Int_t index, Double_t r[9]);
+  static Bool_t GetRotation(Int_t lay,Int_t lad,Int_t det, Double_t r[9])
+    { return GetRotation(GetModuleIndex(lay,lad,det),r); }
+
+  // This function returns a pointer to the original TGeoHMatrix (local->global)
+  // for a specific module index
+  static Bool_t GetOrigMatrix(Int_t index, TGeoHMatrix &m);
+  static Bool_t GetOrigMatrix(Int_t lay,Int_t lad,Int_t det, TGeoHMatrix &m)
+    { return GetOrigMatrix(GetModuleIndex(lay,lad,det),m); }
+
+  static Bool_t GetOrigTranslation(Int_t index, Double_t t[3]);
+  static Bool_t GetOrigTranslation(Int_t lay,Int_t lad,Int_t det, Double_t t[3])
+    { return GetOrigTranslation(GetModuleIndex(lay,lad,det),t); }
+
+  static Bool_t GetOrigRotation(Int_t index, Double_t r[9]);
+  static Bool_t GetOrigRotation(Int_t lay,Int_t lad,Int_t det, Double_t r[9])
+    { return GetOrigRotation(GetModuleIndex(lay,lad,det),r); }
+
+  static const TGeoHMatrix* GetTracking2LocalMatrix(Int_t index);
+  static const TGeoHMatrix* GetTracking2LocalMatrix(Int_t lay,Int_t lad,Int_t det)
+    { return GetTracking2LocalMatrix(GetModuleIndex(lay,lad,det)); }
+
+  static Bool_t GetTrackingMatrix(Int_t index, TGeoHMatrix &m);
+  static Bool_t GetTrackingMatrix(Int_t lay,Int_t lad,Int_t det, TGeoHMatrix &m)
+    { return GetTrackingMatrix(GetModuleIndex(lay,lad,det),m); }
+
+ private:
+
+  enum {kNLayers = 6}; // The number of layers.
+
+  static Bool_t       GetLayer(Int_t index,Int_t &lay,Int_t &index2);
+  static TGeoPNEntry* GetPNEntry(Int_t index);
+
+  AliITSgeomTGeo(const AliITSgeomTGeo &geom);     // Copy constructor
+  AliITSgeomTGeo& operator=(const AliITSgeomTGeo &geom);// Assignment operator
+
+  static const Int_t  fgkNModules;            // The total number of modules
+  static const Int_t  fgkNLadders[kNLayers];  // Array of the number of ladders/layer(layer)
+  static const Int_t  fgkNDetectors[kNLayers];// Array of the number of detector/ladder(layer)
+
+  ClassDef(AliITSgeomTGeo, 0) // ITS geometry based on TGeo
+};
+
+#endif

diff --git a/ITS/ITSbaseLinkDef.h b/ITS/ITSbaseLinkDef.h
index 4cf30c0dd21225aae402cf3f2ebbaaa0f696c14c..4c43676b0900fc040c545c5e4419d2624fbc6e3b 100644 (file)
--- a/ITS/ITSbaseLinkDef.h
+++ b/ITS/ITSbaseLinkDef.h
@@ -22,6 +22,7 @@
 #pragma link C++ class  AliITSTransientDigit+;
 #pragma link C++ class  AliITSInitGeometry+;
 #pragma link C++ class  AliITSgeom+;
+#pragma link C++ class  AliITSgeomTGeo+;
 #pragma link C++ class  AliITSgeomMatrix-;
 #pragma link C++ class  AliITSgeomSPD+;
 #pragma link C++ class  AliITSgeomSDD+;

diff --git a/ITS/libITSbase.pkg b/ITS/libITSbase.pkg
index b35d0bc56355f04fb24e7f70a289e5cc30218668..612ca3702471eef23dbbcf87c7a3c56e6a966b60 100644 (file)
--- a/ITS/libITSbase.pkg
+++ b/ITS/libITSbase.pkg
@@ -38,7 +38,8 @@ SRCS =        AliITSgeom.cxx \
                AliITSRawStreamSPD.cxx \
                AliITSEventHeader.cxx \
                AliITSRawStreamSSDv1.cxx \
-               AliITSRawData.cxx  \
+               AliITSRawData.cxx \
+               AliITSgeomTGeo.cxx
 
 HDRS:=  $(SRCS:.cxx=.h)