Extacting the OCDB in a separate module. The detectors have write permission in the...
[u/mrichter/AliRoot.git] / ITS / AliITSSurveyToAlignSSD.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 /*
17 $Log$
18 Revision 1.1  2007/10/01 14:12:45  pchrist
19 Class creating the aligmnent object fro the surveyor measurements.
20
21 */ 
22
23 //
24 //  Creates the SSD align object
25 //
26 //
27 #include "Riostream.h"
28 #include "TFile.h"
29 #include "TSystem.h"
30 #include "TClonesArray.h"
31 #include "TGeoManager.h"
32 #include "TGeoMatrix.h"
33 #include "TGeoPhysicalNode.h"
34
35 #include "AliITSSurveyToAlignSSD.h"
36 #include "AliSurveyObj.h"
37 #include "AliSurveyPoint.h"
38 #include "AliAlignObjParams.h"
39
40 #include "AliLog.h"
41
42 #include "AliCDBManager.h"
43 #include "AliCDBEntry.h"
44 #include "AliCDBStorage.h"
45
46 ClassImp(AliITSSurveyToAlignSSD)
47
48 //________________________________________________________________________
49 AliITSSurveyToAlignSSD::AliITSSurveyToAlignSSD() :
50   TObject(),
51   fRun(0),
52   fFileLocal(0x0),
53   fFileGrid(0x0),
54   fSurveyPoints(0),
55   fSSDAlignObj(new TClonesArray("AliAlignObjParams",100)),
56   fSSDAlignObjParam(0),
57   fDebug(0) {
58   //
59   //  default constructor
60   //
61 }   
62
63 //________________________________________________________________________
64 AliITSSurveyToAlignSSD::AliITSSurveyToAlignSSD(Int_t reportId, Int_t runNumber) :
65   TObject(),
66   fRun(0),
67   fFileLocal(0x0),
68   fFileGrid(0x0),
69   fSurveyPoints(0),
70   fSSDAlignObj(new TClonesArray("AliAlignObjParams",100)),
71   fSSDAlignObjParam(0),
72   fDebug(0) {
73   //
74   // constructor - defines data files
75   //
76   fRun = runNumber;
77   fFileLocal = new Char_t[80];
78   fFileGrid = new Char_t[80];
79   Char_t path[50];
80
81   sprintf(path,gSystem->Getenv("ALICE_ROOT")); 
82   sprintf(fFileLocal,"%s/ITS/Survey_SSD_%d.txt",path,reportId);  
83 }
84
85 //_________________________________________________________________________
86 AliITSSurveyToAlignSSD::AliITSSurveyToAlignSSD(const AliITSSurveyToAlignSSD &align) :
87   TObject(),
88   fRun(0),
89   fFileLocal(0x0),
90   fFileGrid(0x0),
91   fSurveyPoints(0),
92   fSSDAlignObj(new TClonesArray("AliAlignObjParams",100)),
93   fSSDAlignObjParam(0),
94   fDebug(0) {
95   //
96   //  copy constructor - dummy
97   
98   fDebug = align.fDebug;
99 }
100
101 //__________________________________________________________________________
102 AliITSSurveyToAlignSSD & AliITSSurveyToAlignSSD::operator =(const AliITSSurveyToAlignSSD & align) {
103   //
104   // assignment operator - dummy
105   //
106   fDebug = align.fDebug;
107   return (*this);
108 }
109
110 //__________________________________________________________________________
111 AliITSSurveyToAlignSSD::~AliITSSurveyToAlignSSD() {
112   //
113   // destructor
114   //
115   if(fSurveyPoints) delete fSurveyPoints;
116   if(fSSDAlignObj) delete fSSDAlignObj;
117   if(fSSDAlignObjParam) delete fSSDAlignObjParam;
118 }
119
120 //______________________________________________________________________
121 void AliITSSurveyToAlignSSD::Run() { 
122   //
123   // runs the full chain
124   //
125   SetDebug(0);
126   Bool_t flag = LoadSurveyData();
127   if(!flag) {
128     cout<<"Missing points"<<endl;
129     return;
130   }
131   CreateAlignObj();
132   StoreAlignObj();
133 }
134
135 //__________________________________________________________________________
136 Bool_t AliITSSurveyToAlignSSD::LoadSurveyData() {
137   //
138   // for a time being it loads from the local file the surveyed point
139   // and has the ideal points hardwired. I am waiting until Ricardo
140   // completes his job
141   // 
142   
143   //Load survey data from the local file
144   AliSurveyObj * s1 = new AliSurveyObj();
145   if(s1->FillFromLocalFile(fFileLocal))
146     fSurveyPoints = s1->GetData();
147   else 
148     return kFALSE;
149   
150   cout<<"Number of SSD survey points: "<<fSurveyPoints->GetEntries()<<endl;  
151   
152   return kTRUE;
153 }
154
155 //_______________________________________________________________________
156 void AliITSSurveyToAlignSSD::CreateAlignObj(){
157   //
158   // This method creates AliAlignObj and fills it with Euler angles
159   // and shifts. The units are degrees and cm.
160   // 
161   TClonesArray &alobj = *fSSDAlignObj;
162
163   //load ideal geometry from the OCDB
164   AliCDBManager *cdb = AliCDBManager::Instance();
165   cdb->SetDefaultStorage("local://$ALICE_ROOT/OCDB");
166   cdb->SetRun(fRun);
167   AliCDBEntry* entry = cdb->Get("GRP/Geometry/Data");
168   AliGeomManager::SetGeometry(gGeoManager);
169
170   AliSurveyPoint* pt = 0;
171   TGeoHMatrix* hm = 0;
172   TGeoPNEntry* pne = 0;
173   Double_t* tr;
174   Int_t uid;
175   const char* symname;
176   Double_t sx, sy, sz;
177   Int_t ilayer, imodule;
178
179   for(Int_t imod = 0; imod < fSurveyPoints->GetEntries(); imod++) {
180     pt = (AliSurveyPoint*) fSurveyPoints->At(imod);
181     if(!pt) continue;
182     sx = pt->GetX();
183     sy = pt->GetY();
184     sz = pt->GetZ();
185
186     ilayer = (imod < 748) ? AliGeomManager::kSSD1 : AliGeomManager::kSSD2;
187     imodule = (imod < 748) ? imod : imod - 748;
188
189     uid = AliGeomManager::LayerToVolUID(ilayer,imodule);
190     symname = AliGeomManager::SymName(uid);
191     pne = gGeoManager->GetAlignableEntryByUID(uid);
192     hm = pne->GetGlobalOrig();
193     tr = hm->GetTranslation();
194
195     //Printf("symname %s", symname);
196     //Printf("x,y,z from survey: %f %f %f", sx, sy, sz);
197     //Printf("x,y,z from ideal : %f %f %f", tr[0], tr[1], tr[2]);
198     
199    new(alobj[imod]) AliAlignObjParams(symname, uid, sx-tr[0], sy-tr[1], sz-tr[2], 0., 0., 0., kTRUE);
200   }//module loop
201
202   delete entry;
203 }
204
205 //_________________________________________________________________________
206 void AliITSSurveyToAlignSSD::StoreAlignObj(){
207   // Stores the TClonesArray of AliAlignObj in
208   // $ALICE_ROOT/ITS/Align/Data/SSDfromSurvey.root
209   // In a later version these objects will be merged 
210   // with the SDD ones and will be put in the OCDB
211   const char* filename = "$ALICE_ROOT/ITS/Align/Data/SSDfromSurvey.root";
212   TFile *f = TFile::Open(filename,"RECREATE");
213   if(!f){
214     Error(filename,"cannot open file for output\n");
215     return;
216   }
217   cout<<"Saving alignment objects to the file "<<filename<<endl;
218   f->cd();
219   f->WriteObject(fSSDAlignObj,"ITSAlignObjs","kSingleKey");
220   f->Close();
221 }