PHOS/AliPHOSSurvey.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 /* History of cvs commits:
  19  *
  20  * $Log$
  21  * Revision 1.6  2007/08/28 13:12:18  hristov
  22  * AliAlignObjAngles becomes AliAlignObjParams (Raffaele)
  23  *
  24  * Revision 1.5  2007/07/10 12:41:38  kharlov
  25  * Added a new class AliPHOSSurvet1 which read survey data from EDMS files
  26  *
  27  * Revision 1.4  2007/05/17 17:34:54  kharlov
  28  * Merging differences if v1.2 and 1.3
  29  *
  30  * Revision 1.3  2007/05/17 17:13:32  kharlov
  31  * Coding convensions satisfied (T.Pocheptsov)
  32  *
  33  * Revision 1.1  2007/04/19 15:47:20  kharlov
  34  * Add misalignment of strip units with AliPHOSSurvey class
  35  *
  36  */
  37
  38 // Objects of this class read txt file with survey (photogrammetry) data
  39 // and convert the data into AliAlignObjParams of alignable PHOS volumes.
  40 // It can be used as a base class, you need to override GetStripTransformation.
  41 // AliPHOSSurvey inherits TObject only to use AliLog "functions".
  42 // Author: Timur Pocheptsov (JINR)
  43
  44 #include <fstream>
  45
  46 #include <TClonesArray.h>
  47 #include <TGeoManager.h>
  48 #include <TString.h>
  49 #include <TMath.h>
  50
  51 #include "AliSurveyObj.h"
  52
  53 #include "AliPHOSEMCAGeometry.h"
  54 #include "AliAlignObjParams.h"
  55 #include "AliPHOSGeometry.h"
  56 #include "AliPHOSSurvey.h"
  57 #include "AliLog.h"
  58
  59 ClassImp(AliPHOSSurvey)
  60
  61 //____________________________________________________________________________
  62 AliPHOSSurvey::AliPHOSSurvey()
  63                 : fStrNum(0),
  64                   fStripData(0)
  65 {
  66   //Default constructor.
  67 }
  68
  69 namespace {
  70
  71   struct AliPHOSStripCoords {
  72     Double_t fX1; //x coordinate of the first strip point
  73     Double_t fZ1; //z coordinate of the first strip point
  74     Double_t fX2; //x coordinate of the second strip point
  75     Double_t fZ2; //z coordinate of the second strip point
  76   };
  77
  78 }
  79
  80 //____________________________________________________________________________
  81 AliPHOSSurvey::AliPHOSSurvey(const TString &txtFileName)
  82                 : fStrNum(0),
  83                   fStripData(0)
  84 {
  85   //Read survey data from txt file.
  86   const AliPHOSGeometry *phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
  87   if (!phosGeom) {
  88     AliError("Cannot obtain AliPHOSGeometry instance.");
  89     return;
  90   }
  91
  92   std::ifstream inputFile(txtFileName.Data());
  93   if (!inputFile) {
  94     AliError(("Cannot open the survey file " + txtFileName).Data());
  95     return;
  96   }
  97
  98   AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
  99   fStrNum = emcaGeom->GetNStripX() * emcaGeom->GetNStripZ();
 100
 101   Int_t dummyInt = 0;
 102   Double_t dummyY = 0.;
 103   Double_t *xReal = new Double_t[fStrNum * 2];//2
 104   Double_t *zReal = new Double_t[fStrNum * 2];//3
 105
 106   for (Int_t i = 0; i < fStrNum * 2; ++i) {
 107     xReal[i] = 0.;
 108     zReal[i] = 0.;
 109   }
 110
 111   for (Int_t i = 0; i < fStrNum * 2; ++i) {
 112     if (!inputFile) {
 113       AliError("Error while reading input file.");
 114       delete [] xReal;
 115       delete [] zReal;
 116       return;
 117     }
 118     inputFile>>dummyInt>>xReal[i]>>dummyY>>zReal[i];
 119     xReal[i] *= 0.1;
 120     zReal[i] *= 0.1;
 121   }
 122
 123   InitStripData(xReal, zReal);
 124
 125   delete [] zReal;
 126   delete [] xReal;
 127 }
 128
 129 //____________________________________________________________________________
 130 AliPHOSSurvey::~AliPHOSSurvey()
 131 {
 132   delete [] fStripData;
 133 }
 134
 135 //____________________________________________________________________________
 136 void AliPHOSSurvey::CreateAliAlignObjParams(TClonesArray &array)
 137 {
 138   //Create AliAlignObjParams.
 139   const AliPHOSGeometry * phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
 140   if (!phosGeom) {
 141     AliError("Cannot obtain AliPHOSGeometry instance.");
 142     return;
 143   }
 144
 145   if (!gGeoManager) {
 146     AliWarning("Cannot create local transformations for strip units - gGeoManager does not exist.");
 147     AliInfo("Null shifts and rotations will be created instead.");
 148     return CreateNullObjects(array, phosGeom);
 149   }
 150
 151   AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
 152   Int_t arrayInd = array.GetEntries(), iIndex = 0;
 153   AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
 154   UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);
 155
 156   for (Int_t module = 1; module <= phosGeom->GetNModules(); ++module) {
 157     for (Int_t i = 0, stripNum = 0; i < emcaGeom->GetNStripX(); ++i) {
 158       for (Int_t j = 0; j < emcaGeom->GetNStripZ(); ++j) {
 159         TString stripName(TString::Format("PHOS/Module%d/Strip_%d_%d", module, i, j));
 160         AliPHOSStripDelta t(GetStripTransformation(stripNum++, module));
 161         new(array[arrayInd])
 162           AliAlignObjParams(
 163                             stripName.Data(), volid,
 164                             t.fXShift, t.fYShift, t.fZShift,
 165                             -t.fPsi, -t.fTheta, -t.fPhi,
 166                             false
 167                            );
 168         ++arrayInd;
 169       }
 170     }
 171   }
 172 }
 173
 174 //____________________________________________________________________________
 175 void AliPHOSSurvey::CreateNullObjects(TClonesArray &array, const AliPHOSGeometry *phosGeom)const
 176 {
 177   //Create null shifts and rotations.
 178   const AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
 179   Int_t arrayInd = array.GetEntries(), iIndex = 0;
 180   AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
 181   UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);
 182
 183   for (Int_t module = 1; module <= phosGeom->GetNModules(); ++module)
 184     for (Int_t i = 0; i < emcaGeom->GetNStripX(); ++i)
 185       for (Int_t j = 0; j < emcaGeom->GetNStripZ(); ++j) {
 186         TString stripName(TString::Format("PHOS/Module%d/Strip_%d_%d", module, i, j));
 187         new(array[arrayInd]) AliAlignObjParams(stripName.Data(), volid, 0., 0., 0., 0., 0., 0., true);
 188         ++arrayInd;
 189       }
 190 }
 191
 192 //____________________________________________________________________________
 193 AliPHOSSurvey::AliPHOSStripDelta AliPHOSSurvey::GetStripTransformation(Int_t stripIndex, Int_t module)const
 194 {
 195   //Strip 'stripIndex' transformation.
 196   AliPHOSStripDelta t = {0., 0., 0., 0., 0., 0.};
 197   if (module != 3 || !fStripData)
 198     return t;
 199   return fStripData[stripIndex];
 200 }
 201
 202 //____________________________________________________________________________
 203 void AliPHOSSurvey::InitStripData(const Double_t *xReal, const Double_t *zReal)
 204 {
 205   const AliPHOSGeometry *phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
 206   AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
 207   const Float_t *strip = emcaGeom->GetStripHalfSize();
 208   const Float_t *cell  = emcaGeom->GetAirCellHalfSize();
 209
 210   AliPHOSStripCoords *idealStrips = new AliPHOSStripCoords[fStrNum];//1
 211   for (Int_t ix = 0, stripNumber = 0; ix < emcaGeom->GetNStripX(); ++ix) {
 212     for (Int_t iz = 0; iz < emcaGeom->GetNStripZ(); ++iz) {
 213       AliPHOSStripCoords &str = idealStrips[stripNumber++];
 214       str.fX1 = ix * 2 * strip[0];
 215       str.fX2 = str.fX1 + 14 * cell[0];
 216       str.fZ1 = iz * 2 * strip[2];
 217       str.fZ2 = str.fZ1 + 2 * cell[2];
 218     }
 219   }
 220
 221   AliPHOSStripCoords *realStrips = new AliPHOSStripCoords[fStrNum];//4
 222   for (Int_t j = 0, stripNumber = 0; j < emcaGeom->GetNStripX() * 2; j += 2) {
 223     for (Int_t i = 0; i < emcaGeom->GetNStripZ(); ++i) {
 224       AliPHOSStripCoords &str = realStrips[stripNumber++];
 225       str.fX1 = xReal[i + j * emcaGeom->GetNStripZ()];
 226       str.fZ1 = zReal[i + j * emcaGeom->GetNStripZ()];
 227       str.fX2 = xReal[i + (j + 1) * emcaGeom->GetNStripZ()];
 228       str.fZ2 = zReal[i + (j + 1) * emcaGeom->GetNStripZ()];
 229     }
 230   }
 231
 232   fStripData = new AliPHOSStripDelta[fStrNum];
 233
 234   for (Int_t i = 0; i < fStrNum; ++i) {
 235     const AliPHOSStripCoords &real = realStrips[i];
 236     const AliPHOSStripCoords &ideal = idealStrips[i];
 237     AliPHOSStripDelta &t = fStripData[i];
 238     t.fTheta = TMath::ATan((real.fZ2 - real.fZ1)  / (real.fX2 - real.fX1)) -
 239                TMath::ATan((ideal.fZ2 - ideal.fZ1) / (ideal.fX2 - ideal.fX1));
 240     t.fTheta *= TMath::RadToDeg();
 241     t.fXShift = (real.fX1 + real.fX2) / 2 - (ideal.fX1 + ideal.fX2) / 2;
 242     t.fZShift = (real.fZ1 + real.fZ2) / 2 - (ideal.fZ1 + ideal.fZ2) / 2;
 243     t.fYShift = 0., t.fPsi = 0., t.fPhi = 0.;
 244   }
 245
 246   delete [] realStrips;
 247   delete [] idealStrips;
 248 }