ROOT logo
/**************************************************************************
 * 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$ */

/* History of cvs commits:
 *
 * $Log$
 * Revision 1.6  2007/08/28 13:12:18  hristov
 * AliAlignObjAngles becomes AliAlignObjParams (Raffaele)
 *
 * Revision 1.5  2007/07/10 12:41:38  kharlov
 * Added a new class AliPHOSSurvet1 which read survey data from EDMS files
 *
 * Revision 1.4  2007/05/17 17:34:54  kharlov
 * Merging differences if v1.2 and 1.3
 *
 * Revision 1.3  2007/05/17 17:13:32  kharlov
 * Coding convensions satisfied (T.Pocheptsov)
 *
 * Revision 1.1  2007/04/19 15:47:20  kharlov
 * Add misalignment of strip units with AliPHOSSurvey class
 *
 */

// Objects of this class read txt file with survey (photogrammetry) data
// and convert the data into AliAlignObjParams of alignable PHOS volumes.
// It can be used as a base class, you need to override GetStripTransformation.
// AliPHOSSurvey inherits TObject only to use AliLog "functions".
// Author: Timur Pocheptsov (JINR)

#include <fstream>

#include <TClonesArray.h>
#include <TGeoManager.h>
#include <TString.h>
#include <TMath.h>

#include "AliSurveyObj.h"

#include "AliPHOSEMCAGeometry.h"
#include "AliAlignObjParams.h"
#include "AliPHOSGeometry.h"
#include "AliPHOSSurvey.h"
#include "AliLog.h"

ClassImp(AliPHOSSurvey)

//____________________________________________________________________________
AliPHOSSurvey::AliPHOSSurvey()
                : fStrNum(0),
                  fStripData(0)
{
  //Default constructor.
}

namespace {

  struct AliPHOSStripCoords {
    Double_t fX1; //x coordinate of the first strip point
    Double_t fZ1; //z coordinate of the first strip point
    Double_t fX2; //x coordinate of the second strip point
    Double_t fZ2; //z coordinate of the second strip point
  };

}

//____________________________________________________________________________
AliPHOSSurvey::AliPHOSSurvey(const TString &txtFileName)
                : fStrNum(0),
                  fStripData(0)
{
  //Read survey data from txt file.
  const AliPHOSGeometry *phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
  if (!phosGeom) {
    AliError("Cannot obtain AliPHOSGeometry instance.");
    return;
  }

  std::ifstream inputFile(txtFileName.Data());
  if (!inputFile) {
    AliError(("Cannot open the survey file " + txtFileName).Data());
    return;
  }

  AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
  fStrNum = emcaGeom->GetNStripX() * emcaGeom->GetNStripZ();

  Int_t dummyInt = 0;
  Double_t dummyY = 0.;
  Double_t *xReal = new Double_t[fStrNum * 2];//2
  Double_t *zReal = new Double_t[fStrNum * 2];//3

  for (Int_t i = 0; i < fStrNum * 2; ++i) {
    xReal[i] = 0.;
    zReal[i] = 0.;
  }
  
  for (Int_t i = 0; i < fStrNum * 2; ++i) {
    if (!inputFile) {
      AliError("Error while reading input file.");
      delete [] xReal;
      delete [] zReal;
      return;
    }
    inputFile>>dummyInt>>xReal[i]>>dummyY>>zReal[i];
    xReal[i] *= 0.1;
    zReal[i] *= 0.1;
  }

  InitStripData(xReal, zReal);

  delete [] zReal;
  delete [] xReal;
}

//____________________________________________________________________________
AliPHOSSurvey::~AliPHOSSurvey()
{
  delete [] fStripData;
}

//____________________________________________________________________________
void AliPHOSSurvey::CreateAliAlignObjParams(TClonesArray &array)
{
  //Create AliAlignObjParams.
  const AliPHOSGeometry * phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
  if (!phosGeom) {
    AliError("Cannot obtain AliPHOSGeometry instance.");
    return;
  }

  if (!gGeoManager) {
    AliWarning("Cannot create local transformations for strip units - gGeoManager does not exist.");
    AliInfo("Null shifts and rotations will be created instead.");
    return CreateNullObjects(array, phosGeom);
  }

  AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
  Int_t arrayInd = array.GetEntries(), iIndex = 0;
  AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
  UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);

  for (Int_t module = 1; module <= phosGeom->GetNModules(); ++module) {
    for (Int_t i = 0, stripNum = 0; i < emcaGeom->GetNStripX(); ++i) {
      for (Int_t j = 0; j < emcaGeom->GetNStripZ(); ++j) {
        TString stripName(TString::Format("PHOS/Module%d/Strip_%d_%d", module, i, j));
        AliPHOSStripDelta t(GetStripTransformation(stripNum++, module));
        new(array[arrayInd])
          AliAlignObjParams(
                            stripName.Data(), volid, 
                            t.fXShift, t.fYShift, t.fZShift, 
                            -t.fPsi, -t.fTheta, -t.fPhi, 
                            false
                           );
        ++arrayInd;
      }
    }
  }
}

//____________________________________________________________________________
void AliPHOSSurvey::CreateNullObjects(TClonesArray &array, const AliPHOSGeometry *phosGeom)const
{
  //Create null shifts and rotations.
  const AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
  Int_t arrayInd = array.GetEntries(), iIndex = 0;
  AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
  UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,iIndex);

  for (Int_t module = 1; module <= phosGeom->GetNModules(); ++module)
    for (Int_t i = 0; i < emcaGeom->GetNStripX(); ++i)
      for (Int_t j = 0; j < emcaGeom->GetNStripZ(); ++j) {
        TString stripName(TString::Format("PHOS/Module%d/Strip_%d_%d", module, i, j));
        new(array[arrayInd]) AliAlignObjParams(stripName.Data(), volid, 0., 0., 0., 0., 0., 0., true);
        ++arrayInd;
      }
}

//____________________________________________________________________________
AliPHOSSurvey::AliPHOSStripDelta AliPHOSSurvey::GetStripTransformation(Int_t stripIndex, Int_t module)const
{
  //Strip 'stripIndex' transformation.
  AliPHOSStripDelta t = {0., 0., 0., 0., 0., 0.};
  if (module != 3 || !fStripData)
    return t;
  return fStripData[stripIndex];
}

//____________________________________________________________________________
void AliPHOSSurvey::InitStripData(const Double_t *xReal, const Double_t *zReal)
{
  const AliPHOSGeometry *phosGeom = AliPHOSGeometry::GetInstance("IHEP", "IHEP");
  AliPHOSEMCAGeometry * emcaGeom = phosGeom->GetEMCAGeometry();
  const Float_t *strip = emcaGeom->GetStripHalfSize();
  const Float_t *cell  = emcaGeom->GetAirCellHalfSize(); 

  AliPHOSStripCoords *idealStrips = new AliPHOSStripCoords[fStrNum];//1
  for (Int_t ix = 0, stripNumber = 0; ix < emcaGeom->GetNStripX(); ++ix) {
    for (Int_t iz = 0; iz < emcaGeom->GetNStripZ(); ++iz) {
      AliPHOSStripCoords &str = idealStrips[stripNumber++];
      str.fX1 = ix * 2 * strip[0];
      str.fX2 = str.fX1 + 14 * cell[0];
      str.fZ1 = iz * 2 * strip[2];
      str.fZ2 = str.fZ1 + 2 * cell[2];
    }
  }

  AliPHOSStripCoords *realStrips = new AliPHOSStripCoords[fStrNum];//4
  for (Int_t j = 0, stripNumber = 0; j < emcaGeom->GetNStripX() * 2; j += 2) {
    for (Int_t i = 0; i < emcaGeom->GetNStripZ(); ++i) {
      AliPHOSStripCoords &str = realStrips[stripNumber++];
      str.fX1 = xReal[i + j * emcaGeom->GetNStripZ()];
      str.fZ1 = zReal[i + j * emcaGeom->GetNStripZ()];
      str.fX2 = xReal[i + (j + 1) * emcaGeom->GetNStripZ()];
      str.fZ2 = zReal[i + (j + 1) * emcaGeom->GetNStripZ()];
    }
  }

  fStripData = new AliPHOSStripDelta[fStrNum];
  
  for (Int_t i = 0; i < fStrNum; ++i) {
    const AliPHOSStripCoords &real = realStrips[i];
    const AliPHOSStripCoords &ideal = idealStrips[i];
    AliPHOSStripDelta &t = fStripData[i];
    t.fTheta = TMath::ATan((real.fZ2 - real.fZ1)  / (real.fX2 - real.fX1)) - 
               TMath::ATan((ideal.fZ2 - ideal.fZ1) / (ideal.fX2 - ideal.fX1));
    t.fTheta *= TMath::RadToDeg();
    t.fXShift = (real.fX1 + real.fX2) / 2 - (ideal.fX1 + ideal.fX2) / 2;
    t.fZShift = (real.fZ1 + real.fZ2) / 2 - (ideal.fZ1 + ideal.fZ2) / 2;
    t.fYShift = 0., t.fPsi = 0., t.fPhi = 0.;
  }

  delete [] realStrips;
  delete [] idealStrips;
}
 AliPHOSSurvey.cxx:1
 AliPHOSSurvey.cxx:2
 AliPHOSSurvey.cxx:3
 AliPHOSSurvey.cxx:4
 AliPHOSSurvey.cxx:5
 AliPHOSSurvey.cxx:6
 AliPHOSSurvey.cxx:7
 AliPHOSSurvey.cxx:8
 AliPHOSSurvey.cxx:9
 AliPHOSSurvey.cxx:10
 AliPHOSSurvey.cxx:11
 AliPHOSSurvey.cxx:12
 AliPHOSSurvey.cxx:13
 AliPHOSSurvey.cxx:14
 AliPHOSSurvey.cxx:15
 AliPHOSSurvey.cxx:16
 AliPHOSSurvey.cxx:17
 AliPHOSSurvey.cxx:18
 AliPHOSSurvey.cxx:19
 AliPHOSSurvey.cxx:20
 AliPHOSSurvey.cxx:21
 AliPHOSSurvey.cxx:22
 AliPHOSSurvey.cxx:23
 AliPHOSSurvey.cxx:24
 AliPHOSSurvey.cxx:25
 AliPHOSSurvey.cxx:26
 AliPHOSSurvey.cxx:27
 AliPHOSSurvey.cxx:28
 AliPHOSSurvey.cxx:29
 AliPHOSSurvey.cxx:30
 AliPHOSSurvey.cxx:31
 AliPHOSSurvey.cxx:32
 AliPHOSSurvey.cxx:33
 AliPHOSSurvey.cxx:34
 AliPHOSSurvey.cxx:35
 AliPHOSSurvey.cxx:36
 AliPHOSSurvey.cxx:37
 AliPHOSSurvey.cxx:38
 AliPHOSSurvey.cxx:39
 AliPHOSSurvey.cxx:40
 AliPHOSSurvey.cxx:41
 AliPHOSSurvey.cxx:42
 AliPHOSSurvey.cxx:43
 AliPHOSSurvey.cxx:44
 AliPHOSSurvey.cxx:45
 AliPHOSSurvey.cxx:46
 AliPHOSSurvey.cxx:47
 AliPHOSSurvey.cxx:48
 AliPHOSSurvey.cxx:49
 AliPHOSSurvey.cxx:50
 AliPHOSSurvey.cxx:51
 AliPHOSSurvey.cxx:52
 AliPHOSSurvey.cxx:53
 AliPHOSSurvey.cxx:54
 AliPHOSSurvey.cxx:55
 AliPHOSSurvey.cxx:56
 AliPHOSSurvey.cxx:57
 AliPHOSSurvey.cxx:58
 AliPHOSSurvey.cxx:59
 AliPHOSSurvey.cxx:60
 AliPHOSSurvey.cxx:61
 AliPHOSSurvey.cxx:62
 AliPHOSSurvey.cxx:63
 AliPHOSSurvey.cxx:64
 AliPHOSSurvey.cxx:65
 AliPHOSSurvey.cxx:66
 AliPHOSSurvey.cxx:67
 AliPHOSSurvey.cxx:68
 AliPHOSSurvey.cxx:69
 AliPHOSSurvey.cxx:70
 AliPHOSSurvey.cxx:71
 AliPHOSSurvey.cxx:72
 AliPHOSSurvey.cxx:73
 AliPHOSSurvey.cxx:74
 AliPHOSSurvey.cxx:75
 AliPHOSSurvey.cxx:76
 AliPHOSSurvey.cxx:77
 AliPHOSSurvey.cxx:78
 AliPHOSSurvey.cxx:79
 AliPHOSSurvey.cxx:80
 AliPHOSSurvey.cxx:81
 AliPHOSSurvey.cxx:82
 AliPHOSSurvey.cxx:83
 AliPHOSSurvey.cxx:84
 AliPHOSSurvey.cxx:85
 AliPHOSSurvey.cxx:86
 AliPHOSSurvey.cxx:87
 AliPHOSSurvey.cxx:88
 AliPHOSSurvey.cxx:89
 AliPHOSSurvey.cxx:90
 AliPHOSSurvey.cxx:91
 AliPHOSSurvey.cxx:92
 AliPHOSSurvey.cxx:93
 AliPHOSSurvey.cxx:94
 AliPHOSSurvey.cxx:95
 AliPHOSSurvey.cxx:96
 AliPHOSSurvey.cxx:97
 AliPHOSSurvey.cxx:98
 AliPHOSSurvey.cxx:99
 AliPHOSSurvey.cxx:100
 AliPHOSSurvey.cxx:101
 AliPHOSSurvey.cxx:102
 AliPHOSSurvey.cxx:103
 AliPHOSSurvey.cxx:104
 AliPHOSSurvey.cxx:105
 AliPHOSSurvey.cxx:106
 AliPHOSSurvey.cxx:107
 AliPHOSSurvey.cxx:108
 AliPHOSSurvey.cxx:109
 AliPHOSSurvey.cxx:110
 AliPHOSSurvey.cxx:111
 AliPHOSSurvey.cxx:112
 AliPHOSSurvey.cxx:113
 AliPHOSSurvey.cxx:114
 AliPHOSSurvey.cxx:115
 AliPHOSSurvey.cxx:116
 AliPHOSSurvey.cxx:117
 AliPHOSSurvey.cxx:118
 AliPHOSSurvey.cxx:119
 AliPHOSSurvey.cxx:120
 AliPHOSSurvey.cxx:121
 AliPHOSSurvey.cxx:122
 AliPHOSSurvey.cxx:123
 AliPHOSSurvey.cxx:124
 AliPHOSSurvey.cxx:125
 AliPHOSSurvey.cxx:126
 AliPHOSSurvey.cxx:127
 AliPHOSSurvey.cxx:128
 AliPHOSSurvey.cxx:129
 AliPHOSSurvey.cxx:130
 AliPHOSSurvey.cxx:131
 AliPHOSSurvey.cxx:132
 AliPHOSSurvey.cxx:133
 AliPHOSSurvey.cxx:134
 AliPHOSSurvey.cxx:135
 AliPHOSSurvey.cxx:136
 AliPHOSSurvey.cxx:137
 AliPHOSSurvey.cxx:138
 AliPHOSSurvey.cxx:139
 AliPHOSSurvey.cxx:140
 AliPHOSSurvey.cxx:141
 AliPHOSSurvey.cxx:142
 AliPHOSSurvey.cxx:143
 AliPHOSSurvey.cxx:144
 AliPHOSSurvey.cxx:145
 AliPHOSSurvey.cxx:146
 AliPHOSSurvey.cxx:147
 AliPHOSSurvey.cxx:148
 AliPHOSSurvey.cxx:149
 AliPHOSSurvey.cxx:150
 AliPHOSSurvey.cxx:151
 AliPHOSSurvey.cxx:152
 AliPHOSSurvey.cxx:153
 AliPHOSSurvey.cxx:154
 AliPHOSSurvey.cxx:155
 AliPHOSSurvey.cxx:156
 AliPHOSSurvey.cxx:157
 AliPHOSSurvey.cxx:158
 AliPHOSSurvey.cxx:159
 AliPHOSSurvey.cxx:160
 AliPHOSSurvey.cxx:161
 AliPHOSSurvey.cxx:162
 AliPHOSSurvey.cxx:163
 AliPHOSSurvey.cxx:164
 AliPHOSSurvey.cxx:165
 AliPHOSSurvey.cxx:166
 AliPHOSSurvey.cxx:167
 AliPHOSSurvey.cxx:168
 AliPHOSSurvey.cxx:169
 AliPHOSSurvey.cxx:170
 AliPHOSSurvey.cxx:171
 AliPHOSSurvey.cxx:172
 AliPHOSSurvey.cxx:173
 AliPHOSSurvey.cxx:174
 AliPHOSSurvey.cxx:175
 AliPHOSSurvey.cxx:176
 AliPHOSSurvey.cxx:177
 AliPHOSSurvey.cxx:178
 AliPHOSSurvey.cxx:179
 AliPHOSSurvey.cxx:180
 AliPHOSSurvey.cxx:181
 AliPHOSSurvey.cxx:182
 AliPHOSSurvey.cxx:183
 AliPHOSSurvey.cxx:184
 AliPHOSSurvey.cxx:185
 AliPHOSSurvey.cxx:186
 AliPHOSSurvey.cxx:187
 AliPHOSSurvey.cxx:188
 AliPHOSSurvey.cxx:189
 AliPHOSSurvey.cxx:190
 AliPHOSSurvey.cxx:191
 AliPHOSSurvey.cxx:192
 AliPHOSSurvey.cxx:193
 AliPHOSSurvey.cxx:194
 AliPHOSSurvey.cxx:195
 AliPHOSSurvey.cxx:196
 AliPHOSSurvey.cxx:197
 AliPHOSSurvey.cxx:198
 AliPHOSSurvey.cxx:199
 AliPHOSSurvey.cxx:200
 AliPHOSSurvey.cxx:201
 AliPHOSSurvey.cxx:202
 AliPHOSSurvey.cxx:203
 AliPHOSSurvey.cxx:204
 AliPHOSSurvey.cxx:205
 AliPHOSSurvey.cxx:206
 AliPHOSSurvey.cxx:207
 AliPHOSSurvey.cxx:208
 AliPHOSSurvey.cxx:209
 AliPHOSSurvey.cxx:210
 AliPHOSSurvey.cxx:211
 AliPHOSSurvey.cxx:212
 AliPHOSSurvey.cxx:213
 AliPHOSSurvey.cxx:214
 AliPHOSSurvey.cxx:215
 AliPHOSSurvey.cxx:216
 AliPHOSSurvey.cxx:217
 AliPHOSSurvey.cxx:218
 AliPHOSSurvey.cxx:219
 AliPHOSSurvey.cxx:220
 AliPHOSSurvey.cxx:221
 AliPHOSSurvey.cxx:222
 AliPHOSSurvey.cxx:223
 AliPHOSSurvey.cxx:224
 AliPHOSSurvey.cxx:225
 AliPHOSSurvey.cxx:226
 AliPHOSSurvey.cxx:227
 AliPHOSSurvey.cxx:228
 AliPHOSSurvey.cxx:229
 AliPHOSSurvey.cxx:230
 AliPHOSSurvey.cxx:231
 AliPHOSSurvey.cxx:232
 AliPHOSSurvey.cxx:233
 AliPHOSSurvey.cxx:234
 AliPHOSSurvey.cxx:235
 AliPHOSSurvey.cxx:236
 AliPHOSSurvey.cxx:237
 AliPHOSSurvey.cxx:238
 AliPHOSSurvey.cxx:239
 AliPHOSSurvey.cxx:240
 AliPHOSSurvey.cxx:241
 AliPHOSSurvey.cxx:242
 AliPHOSSurvey.cxx:243
 AliPHOSSurvey.cxx:244
 AliPHOSSurvey.cxx:245
 AliPHOSSurvey.cxx:246
 AliPHOSSurvey.cxx:247
 AliPHOSSurvey.cxx:248