]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliAlignObj.cxx
Several modifications to the alignment object classes
[u/mrichter/AliRoot.git] / STEER / AliAlignObj.cxx
index ea47ca3a4c0b47a3c287306ad4816e84c3a9bce7..c27e61d4fa30089041adeb45118f5dc1e3908943 100644 (file)
  *   is fully consistent with the TGeo Rotation methods.                     *
  *****************************************************************************/
 
+#include <TGeoManager.h>
+#include <TGeoPhysicalNode.h>
+
 #include "AliAlignObj.h"
 #include "AliTrackPointArray.h"
 #include "AliLog.h"
+#include "AliAlignObjAngles.h"
  
 ClassImp(AliAlignObj)
 
@@ -41,7 +45,7 @@ Int_t AliAlignObj::fgLayerSize[kLastLayer - kFirstLayer] = {
   748, 950, // ITS SSD
   36, 36,   // TPC
   90, 90, 90, 90, 90, 90,  // TRD
-  1,        // TOF ??
+  1674,     // TOF
   1, 1,     // PHOS ??
   7,        // RICH ??
   1         // MUON ??
@@ -73,6 +77,19 @@ TString* AliAlignObj::fgVolPath[kLastLayer - kFirstLayer] = {
   0x0
 };
 
+AliAlignObj** AliAlignObj::fgAlignObjs[kLastLayer - kFirstLayer] = {
+  0x0,0x0,
+  0x0,0x0,
+  0x0,0x0,
+  0x0,0x0,
+  0x0,0x0,0x0,
+  0x0,0x0,0x0,
+  0x0,
+  0x0,0x0,
+  0x0,
+  0x0
+};
+
 //_____________________________________________________________________________
 AliAlignObj::AliAlignObj():
   fVolUID(0)
@@ -100,6 +117,21 @@ AliAlignObj &AliAlignObj::operator =(const AliAlignObj& theAlignObj)
   return *this;
 }
 
+//_____________________________________________________________________________
+AliAlignObj &AliAlignObj::operator*=(const AliAlignObj& theAlignObj)
+{
+  // multiplication operator
+  // The operator can be used to 'combine'
+  // two alignment objects
+  TGeoHMatrix m1;
+  GetMatrix(m1);
+  TGeoHMatrix m2;
+  theAlignObj.GetMatrix(m2);
+  m1.MultiplyLeft(&m2);
+  SetMatrix(m1);
+  return *this;
+}
+
 //_____________________________________________________________________________
 AliAlignObj::~AliAlignObj()
 {
@@ -238,6 +270,17 @@ UShort_t AliAlignObj::LayerToVolUID(ELayerID layerId, Int_t modId)
   return ((UShort_t(layerId) << 11) | UShort_t(modId));
 }
 
+//_____________________________________________________________________________
+UShort_t AliAlignObj::LayerToVolUID(Int_t   layerId, Int_t modId)
+{
+  // From detector (layer) index and module number (according to detector numbering)
+  // build fVolUID, unique numerical identity of that volume inside ALICE
+  // fVolUID is 16 bits, first 5 reserved for layerID (32 possible values),
+  // remaining 11 for module ID inside det (2048 possible values).
+  //
+  return ((UShort_t(layerId) << 11) | UShort_t(modId));
+}
+
 //_____________________________________________________________________________
 AliAlignObj::ELayerID AliAlignObj::VolUIDToLayer(UShort_t voluid, Int_t &modId)
 {
@@ -262,6 +305,176 @@ AliAlignObj::ELayerID AliAlignObj::VolUIDToLayer(UShort_t voluid)
   return ELayerID((voluid >> 11) & 0x1f);
 }
 
+//_____________________________________________________________________________
+Bool_t AliAlignObj::SetLocalPars(Double_t x, Double_t y, Double_t z,
+                                Double_t psi, Double_t theta, Double_t phi)
+{
+  // Set the translations and angles by using parameters
+  // defined in the local (in TGeo means) coordinate system
+  // of the alignable volume. In case that the TGeo was
+  // initialized, returns false and the object parameters are
+  // not set.
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    AliError("Can't set the alignment object parameters! gGeoManager doesn't exist or it is still opened!");
+    return kFALSE;
+  }
+
+  const char* volpath = GetVolPath();
+  TGeoPhysicalNode* node = (TGeoPhysicalNode*) gGeoManager->MakePhysicalNode(volpath);
+  if (!node) {
+    AliError(Form("Volume path %s not valid!",volpath));
+    return kFALSE;
+  }
+  if (node->IsAligned())
+    AliWarning(Form("Volume %s has been already misaligned!",volpath));
+
+  TGeoHMatrix m;
+  Double_t tr[3];
+  tr[0]=x; tr[1]=y; tr[2]=z;
+  m.SetTranslation(tr);
+  Double_t angles[3] = {psi, theta, phi};
+  Double_t rot[9];
+  AnglesToMatrix(angles,rot);
+  m.SetRotation(rot);
+
+  TGeoHMatrix align,gprime,gprimeinv;
+  gprime = *node->GetMatrix();
+  gprimeinv = gprime.Inverse();
+  m.Multiply(&gprimeinv);
+  m.MultiplyLeft(&gprime);
+
+  SetMatrix(m);
+
+  return kTRUE;
+}
+
+//_____________________________________________________________________________
+Bool_t AliAlignObj::ApplyToGeometry()
+{
+  // Apply the current alignment object
+  // to the TGeo geometry
+
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    AliError("Can't apply the alignment object! gGeoManager doesn't exist or it is still opened!");
+    return kFALSE;
+  }
+  
+  const char* volpath = GetVolPath();
+  TGeoPhysicalNode* node = (TGeoPhysicalNode*) gGeoManager->MakePhysicalNode(volpath);
+  if (!node) {
+    AliError(Form("Volume path %s not valid!",volpath));
+    return kFALSE;
+  }
+  if (node->IsAligned()) {
+    AliWarning(Form("Volume %s has been already misaligned!",volpath));
+    return kFALSE;
+  }
+
+  TGeoHMatrix align,gprime;
+  gprime = *node->GetMatrix();
+  GetMatrix(align);
+  gprime.MultiplyLeft(&align);
+  TGeoHMatrix *ginv = new TGeoHMatrix;
+  TGeoHMatrix *g = node->GetMatrix(node->GetLevel()-1);
+  *ginv = g->Inverse();
+  *ginv *= gprime;
+  AliAlignObj::ELayerID layerId; // unique identity for volume in the alobj
+  Int_t modId; // unique identity for volume in the alobj
+  GetVolUID(layerId, modId);
+  AliInfo(Form("Aligning volume %s of detector layer %d with local ID %d",volpath,layerId,modId));
+  node->Align(ginv);
+
+  return kTRUE;
+}
+
+//_____________________________________________________________________________
+Bool_t AliAlignObj::GetFromGeometry(const char *path, AliAlignObj &alobj)
+{
+  // Get the alignment object which correspond
+  // to the TGeo volume defined by the 'path'.
+  // The method is extremely slow due to the
+  // searching by string. Therefore it should
+  // be used with great care!!
+
+  // Reset the alignment object
+  alobj.SetPars(0,0,0,0,0,0);
+  alobj.SetVolPath(path);
+
+  if (!gGeoManager || !gGeoManager->IsClosed()) {
+    AliErrorClass("Can't get the alignment object! gGeoManager doesn't exist or it is still opened!");
+    return kFALSE;
+  }
+
+  if (!gGeoManager->GetListOfPhysicalNodes()) {
+    AliErrorClass("Can't get the alignment object! gGeoManager doesn't contain any aligned nodes!");
+    return kFALSE;
+  }
+
+  TObjArray* nodesArr = gGeoManager->GetListOfPhysicalNodes();
+  TGeoPhysicalNode* node = NULL;
+  for (Int_t iNode = 0; iNode < nodesArr->GetEntriesFast(); iNode++) {
+    node = (TGeoPhysicalNode*) nodesArr->UncheckedAt(iNode);
+    const char *nodePath = node->GetName();
+    if (strcmp(path,nodePath) == 0) break;
+  }
+  if (!node) {
+    AliWarningClass(Form("Volume path %s not found!",path));
+    return kFALSE;
+  }
+
+  TGeoHMatrix align,gprime,g,ginv,l;
+  gprime = *node->GetMatrix();
+  l = *node->GetOriginalMatrix();
+  g = *node->GetMatrix(node->GetLevel()-1);
+  g *= l;
+  ginv = g.Inverse();
+  align = gprime * ginv;
+  alobj.SetMatrix(align);
+
+  return kTRUE;
+}
+
+void  AliAlignObj::InitAlignObjFromGeometry()
+{
+  // Loop over all alignable volumes and extract
+  // the corresponding alignment objects from
+  // the TGeo geometry
+  
+  InitVolPaths();
+
+  for (Int_t iLayer = 0; iLayer < (AliAlignObj::kLastLayer - AliAlignObj::kFirstLayer); iLayer++) {
+    fgAlignObjs[iLayer] = new AliAlignObj*[AliAlignObj::LayerSize(iLayer)];
+    for (Int_t iModule = 0; iModule < AliAlignObj::LayerSize(iLayer); iModule++) {
+      UShort_t volid = AliAlignObj::LayerToVolUID(iLayer+ AliAlignObj::kFirstLayer,iModule);
+      fgAlignObjs[iLayer][iModule] = new AliAlignObjAngles("",volid,0,0,0,0,0,0);
+      const char *path = GetVolPath(volid);
+      if (!GetFromGeometry(path, *fgAlignObjs[iLayer][iModule]))
+       AliErrorClass(Form("Failed to extract the alignment object for the volume (ID=%d and path=%s) !",volid,path));
+    }
+  }
+  
+}
+
+//_____________________________________________________________________________
+AliAlignObj* AliAlignObj::GetAlignObj(ELayerID layerId, Int_t modId)
+{
+  if(modId<0 || modId>=fgLayerSize[layerId-kFirstLayer]){
+    AliWarningClass(Form("Module number %d not in the valid range (0->%d) !",modId,fgLayerSize[layerId-kFirstLayer]-1));
+    return NULL;
+  }
+  return fgAlignObjs[layerId-kFirstLayer][modId];
+}
+
+//_____________________________________________________________________________
+const char* AliAlignObj::GetVolPath(ELayerID layerId, Int_t modId)
+{
+  if(modId<0 || modId>=fgLayerSize[layerId-kFirstLayer]){
+    AliWarningClass(Form("Module number %d not in the valid range (0->%d) !",modId,fgLayerSize[layerId-kFirstLayer]-1));
+    return NULL;
+  }
+  return fgVolPath[layerId-kFirstLayer][modId].Data();
+}
+
 //_____________________________________________________________________________
 void AliAlignObj::InitVolPaths()
 {
@@ -283,7 +496,7 @@ void AliAlignObj::InitVolPaths()
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT12_1/I12B_"; //".../I12A_"
     TString str1 = "/I10B_";    //"/I10A_";
     TString str2 = "/I107_";    //"/I103_"
-    TString str3 = "/I101_1/ITS1_1";
+    //    TString str3 = "/I101_1/ITS1_1";
     TString volpath, volpath1, volpath2;
 
     for(Int_t c1 = 1; c1<=10; c1++){
@@ -297,7 +510,7 @@ void AliAlignObj::InitVolPaths()
        for(Int_t c3 =1; c3<=4; c3++){
          volpath2 = volpath1;
          volpath2 += c3;
-         volpath2 += str3;
+         //      volpath2 += str3;
          fgVolPath[kSPD1-kFirstLayer][modnum] = volpath2.Data();
          modnum++;
        }
@@ -311,7 +524,7 @@ void AliAlignObj::InitVolPaths()
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT12_1/I12B_";  //".../I12A_"
     TString str1 = "/I20B_";  //"/I20A"
     TString str2 = "/I1D7_";  //"/I1D3"
-    TString str3 = "/I1D1_1/ITS2_1";
+    //    TString str3 = "/I1D1_1/ITS2_1";
     TString volpath, volpath1, volpath2;
 
     for(Int_t c1 = 1; c1<=10; c1++){
@@ -325,7 +538,7 @@ void AliAlignObj::InitVolPaths()
        for(Int_t c3 =1; c3<=4; c3++){
          volpath2 = volpath1;
          volpath2 += c3;
-         volpath2 += str3;
+         //      volpath2 += str3;
          fgVolPath[kSPD2-kFirstLayer][modnum] = volpath2.Data();
          modnum++;
        }
@@ -338,7 +551,7 @@ void AliAlignObj::InitVolPaths()
     Int_t modnum=0;
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT34_1/I004_";
     TString str1 = "/I302_";
-    TString str2 = "/ITS3_1";
+    //    TString str2 = "/ITS3_1";
     TString volpath, volpath1;
 
     for(Int_t c1 = 1; c1<=14; c1++){
@@ -348,7 +561,7 @@ void AliAlignObj::InitVolPaths()
       for(Int_t c2 =1; c2<=6; c2++){
        volpath1 = volpath;
        volpath1 += c2;
-       volpath1 += str2;
+       //      volpath1 += str2;
        fgVolPath[kSDD1-kFirstLayer][modnum] = volpath1.Data();
        modnum++;
       }
@@ -360,7 +573,7 @@ void AliAlignObj::InitVolPaths()
     Int_t modnum=0;
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT34_1/I005_";
     TString str1 = "/I402_";
-    TString str2 = "/ITS4_1";
+    //    TString str2 = "/ITS4_1";
     TString volpath, volpath1;
 
     for(Int_t c1 = 1; c1<=22; c1++){
@@ -370,7 +583,7 @@ void AliAlignObj::InitVolPaths()
       for(Int_t c2 = 1; c2<=8; c2++){
        volpath1 = volpath;
        volpath1 += c2;
-       volpath1 += str2;
+       //      volpath1 += str2;
        fgVolPath[kSDD2-kFirstLayer][modnum] = volpath1.Data();
        modnum++;
       }
@@ -382,7 +595,7 @@ void AliAlignObj::InitVolPaths()
     Int_t modnum=0;
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT56_1/I565_";
     TString str1 = "/I562_";
-    TString str2 = "/ITS5_1";
+    //    TString str2 = "/ITS5_1";
     TString volpath, volpath1;
 
     for(Int_t c1 = 1; c1<=34; c1++){
@@ -392,7 +605,7 @@ void AliAlignObj::InitVolPaths()
       for(Int_t c2 = 1; c2<=22; c2++){
        volpath1 = volpath;
        volpath1 += c2;
-       volpath1 += str2;
+       //      volpath1 += str2;
        fgVolPath[kSSD1-kFirstLayer][modnum] = volpath1.Data();
        modnum++;
       }
@@ -404,7 +617,7 @@ void AliAlignObj::InitVolPaths()
     Int_t modnum=0;
     TString str0 = "ALIC_1/ITSV_1/ITSD_1/IT56_1/I569_";
     TString str1 = "/I566_";
-    TString str2 = "/ITS6_1";
+    //    TString str2 = "/ITS6_1";
     TString volpath, volpath1;
 
     for(Int_t c1 = 1; c1<=38; c1++){
@@ -414,12 +627,139 @@ void AliAlignObj::InitVolPaths()
       for(Int_t c2 = 1; c2<=25; c2++){
        volpath1 = volpath;
        volpath1 += c2;
-       volpath1 += str2;
+       //      volpath1 += str2;
        fgVolPath[kSSD2-kFirstLayer][modnum] = volpath1.Data();
        modnum++;
       }
     }
   }
 
+  /***************    TPC inner chambers' layer    ****************/
+  {
+    Int_t modnum = 0;
+    TString str1 = "ALIC_1/TPC_M_1/TPC_Drift_1/TPC_ENDCAP_1/TPC_SECT_";
+    TString str2 = "ALIC_1/TPC_M_1/TPC_Drift_1/TPC_ENDCAP_2/TPC_SECT_";
+    TString str_in = "/TPC_IROC_1";
+    TString volpath;
+    
+    for(Int_t cnt=1; cnt<=18; cnt++){
+      volpath = str1;
+      volpath += cnt;
+      volpath += str_in;
+      fgVolPath[kTPC1-kFirstLayer][modnum] = volpath.Data();
+      modnum++;
+    }
+    for(Int_t cnt=1; cnt<=18; cnt++){
+      volpath = str2;
+      volpath += cnt;
+      volpath += str_in;
+      fgVolPath[kTPC1-kFirstLayer][modnum] = volpath.Data();
+      modnum++;
+    }
+  }
+
+  /***************    TPC outer chambers' layer    ****************/
+  {
+    Int_t modnum = 0;
+    TString str1 = "ALIC_1/TPC_M_1/TPC_Drift_1/TPC_ENDCAP_1/TPC_SECT_";
+    TString str2 = "ALIC_1/TPC_M_1/TPC_Drift_1/TPC_ENDCAP_2/TPC_SECT_";
+    TString str_out = "/TPC_OROC_1";
+    TString volpath;
+    
+    for(Int_t cnt=1; cnt<=18; cnt++){
+      volpath = str1;
+      volpath += cnt;
+      volpath += str_out;
+      fgVolPath[kTPC2-kFirstLayer][modnum] = volpath.Data();
+      modnum++;
+    }
+    for(Int_t cnt=1; cnt<=18; cnt++){
+      volpath = str2;
+      volpath += cnt;
+      volpath += str_out;
+      fgVolPath[kTPC2-kFirstLayer][modnum] = volpath.Data();
+      modnum++;
+    }
+  }    
+
+  /*********************       TOF layer   ***********************/
+  {
+    Int_t nstrA=15;
+    Int_t nstrB=19;
+    Int_t nstrC=20;
+    Int_t nStripSec=nstrA+2*nstrB+2*nstrC;
+
+    for (Int_t modnum=0; modnum < 1674; modnum++) {
+
+      Int_t sector = modnum/nStripSec;
+      Char_t  string1[100];
+      Char_t  string2[100];
+
+      Int_t icopy=-1;
+
+      if(sector<3){
+       icopy=sector+1;
+       sprintf(string1,"/ALIC_1/B077_1/B075_%i/BTO3_1",icopy);
+      }
+      else if(sector<11){
+       icopy=sector-2;
+       sprintf(string1,"/ALIC_1/B077_1/B071_%i/BTO1_1",icopy);
+      }
+      else if(sector==11 || sector==12){
+       icopy=sector-10;
+       sprintf(string1,"/ALIC_1/B077_1/B074_%i/BTO2_1",icopy);
+      }
+      else {
+       icopy=sector-4;
+       sprintf(string1,"/ALIC_1/B077_1/B071_%i/BTO1_1",icopy);
+      }
+
+      Int_t strInSec=modnum%nStripSec;
+
+      if( strInSec < nstrC){
+       icopy= nstrC - (strInSec+1) + 1;
+       sprintf(string2,"FTOC_1/FLTC_0/FSTR_%i",icopy);
+      }
+      else if(strInSec< nstrC+nstrB){
  
+       icopy= nstrB - (strInSec-nstrC+1) + 1;
+       sprintf(string2,"FTOB_1/FLTB_0/FSTR_%i",icopy);
+
+      }
+      else if(strInSec< nstrC+nstrB+nstrA){   
+
+       icopy= strInSec-(nstrC+nstrB)+1;
+       sprintf(string2,"FTOA_0/FLTA_0/FSTR_%i",icopy); 
+      }
+      else if(strInSec< nstrC+2*nstrB+nstrA){ 
+
+       icopy= strInSec-(nstrC+nstrB+nstrA)+1;
+       sprintf(string2,"FTOB_2/FLTB_0/FSTR_%i",icopy);
+
+      }
+      else  { 
+
+       icopy= strInSec-(nstrC+2*nstrB+nstrA)+1;
+       sprintf(string2,"FTOC_2/FLTC_0/FSTR_%i",icopy);
+
+      }
+  
+      Char_t  path[100];
+      sprintf(path,"%s/%s",string1,string2); 
+      //      printf("%d  %s\n",modnum,path);
+      fgVolPath[kTOF-kFirstLayer][modnum] = path;
+    }
+  } 
+
+  /*********************      RICH layer   ***********************/
+  {
+    TString str = "ALIC_1/RICH_";
+    TString volpath;
+
+    for (Int_t modnum=0; modnum < 7; modnum++) {
+      volpath = str;
+      volpath += (modnum+1);
+      fgVolPath[kRICH-kFirstLayer][modnum] = volpath.Data();
+    }
+  }
 }