]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliSurveyObj.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Additional getters (Haavard)
[u/mrichter/AliRoot.git] / STEER / AliSurveyObj.cxx
CommitLineData
324b1730 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// //
18// class AliSurveyObj //
19// Retrieve and Convert survey data into ROOT Objects //
20// //
21/////////////////////////////////////////////////////////////////////
22
23#include "AliSurveyObj.h"
24
25//ROOT includes
26#include "TROOT.h"
27#include "Riostream.h"
28#include "TObjArray.h"
29#include "TGrid.h"
30#include "TFile.h"
31#include "TObjString.h"
32
33//AliROOT includes
34#include "AliLog.h"
35
36//System includes
37//#include <time.h> //for sleep(3); // not used anymore!?
38
39ClassImp(AliSurveyObj)
40
41//_____________________________________________________________________________
42AliSurveyObj::AliSurveyObj():
43 TObject(),
44 fTitle(""),
45 fDate(""),
46 fDetector(""),
47 fURL("http://edms.cern.ch/"),
48 fReportNr(-1),
49 fVersion(-1),
50 fObs(""),
51 fCoordSys(""),
52 fUnits(""),
53 fNrColumns(-1),
54 fColNames(""),
55 fIsValid(kFALSE),
56 fDataPoints(new TObjArray(1))
57{
58 // constructor
59 fDataPoints->SetOwner(kTRUE);
60}
61
62
63//_____________________________________________________________________________
64AliSurveyObj::~AliSurveyObj() {
65 //destructor
66
67}
68
69
70//_____________________________________________________________________________
71AliSurveyObj::AliSurveyObj(const AliSurveyObj& surveyObj):
72 TObject(),
73 fTitle(surveyObj.fTitle),
74 fDate(surveyObj.fDate),
75 fDetector(surveyObj.fDetector),
76 fURL(surveyObj.fURL),
77 fReportNr(surveyObj.fReportNr),
78 fVersion(surveyObj.fVersion),
79 fObs(surveyObj.fObs),
80 fCoordSys(surveyObj.fCoordSys),
81 fUnits(surveyObj.fUnits),
82 fNrColumns(surveyObj.fNrColumns),
83 fColNames(surveyObj.fColNames),
84 fIsValid(surveyObj.fIsValid),
85 fDataPoints(new TObjArray(1))
86{
87 // copy constructor
88 TObject *curr = surveyObj.fDataPoints->First();
89 while (curr != 0) {
90 fDataPoints->Add(curr);
91 curr = surveyObj.fDataPoints->After(curr);
92 }
93}
94
95//_____________________________________________________________________________
96AliSurveyObj& AliSurveyObj::operator=(const AliSurveyObj& surveyObj)
97{
98 // assignment operator
99 if (this != &surveyObj) {
100 fTitle = surveyObj.fTitle;
101 fDate = surveyObj.fDate;
102 fDetector = surveyObj.fDetector;
103 fURL = surveyObj.fURL;
104 fReportNr = surveyObj.fReportNr;
105 fVersion = surveyObj.fVersion;
106 fObs = surveyObj.fObs;
107 fCoordSys = surveyObj.fCoordSys;
108 fUnits = surveyObj.fUnits;
109 fNrColumns = surveyObj.fNrColumns;
110 fColNames = surveyObj.fColNames;
111 fIsValid = surveyObj.fIsValid;
112
113 TObject *curr = surveyObj.fDataPoints->First();
114 while (curr != 0) {
115 fDataPoints->Add(curr);
116 curr = surveyObj.fDataPoints->After(curr);
117 }
118 }
119
120 return *this;
121}
122
123
124//_____________________________________________________________________________
125void AliSurveyObj::AddPoint(AliSurveyPoint* point) {
126 fDataPoints->Add(point);
127 return;
128}
129
130
131//_____________________________________________________________________________
132Bool_t AliSurveyObj::Connect(const char *gridUrl, const char *user) {
133
134 // if the same Grid is alreay active, skip connection
135 if ( !gGrid || gridUrl != gGrid->GridUrl() ||
136 (( user != "" ) && ( user != gGrid->GetUser() )) ) {
137 // connection to the Grid
138 AliInfo("\nConnecting to the Grid...");
139 if(gGrid){
140 AliInfo(Form("gGrid = %x; GridUrl = %s; gGrid->GridUrl() = %s",
141 gGrid, gridUrl, gGrid->GridUrl()));
142 AliInfo(Form("User = %s; gGrid->GetUser() = %s",
143 user, gGrid->GetUser()));
144 }
145 TGrid::Connect(gridUrl,user);
146 }
147
148 if(!gGrid) {
149 AliError("Connection failed!");
150 return kFALSE;
151 }
152 return kTRUE;
153}
154
155
156/*
157//_____________________________________________________________________________
158AliCDBGrid::~AliCDBGrid()
159{
160// destructor
161 delete gGrid; gGrid=0;
162
163}
164*/
165
166
167
168//_____________________________________________________________________________
169Bool_t AliSurveyObj::OpenFile(TString openString) {
170 TString storage = "alien://alice.cern.ch";
171
172 Printf("TFile::Open string: \n -> \"%s\"\n", openString.Data());
173
174 if (openString.BeginsWith("alien://")) Connect(storage.Data(), "rsilva");
175
176 TFile *file = TFile::Open(openString.Data(), "READ");
177 if ( !file ) {
178 AliError(Form("Error opening file \"%s\"", openString.Data()));
179 return kFALSE;
180 }
181
182 Int_t size = file->GetSize();
183
184 char *buf = new Char_t[size];
185 memset(buf, '\0', size);
186
187 --size;
188
189 file->Seek(0);
190 if ( file->ReadBuffer(buf, size) ) {
191 AliError("Error reading file contents to buffer!");
192 return kFALSE;
193 }
194
195 // Printf("%d bytes read!\n", size);
196 Printf("%d bytes read!\n", file->GetBytesRead());
197
198 // Printf("->\n%s\n<- ", buf);
199
200 // Printf("%d AQUI!\n", size);
201
202 ParseBuffer(buf);
203
204 file->Close();
205 delete[] buf;
206 return kTRUE;
207}
208
209
210
211//_____________________________________________________________________________
212Bool_t AliSurveyObj::Get(TString detector, Int_t reportNumber,
213 Int_t reportVersion) {
214 TString baseFolder = "/alice/cern.ch/user/r/rsilva/";
215
216 TString fullOpenString = "alien://" + baseFolder + detector + '/';
217 fullOpenString += Form("%d_v%d.txt?filetype=raw", reportNumber, reportVersion);
218 // !Still need to check it's a valid path before actually using it
219
220 return OpenFile(fullOpenString);
221}
222
223
224//_____________________________________________________________________________
225Bool_t AliSurveyObj::GetFromLocalFile(const Char_t* filename) {
226 TString fullOpenString = "file://" + TString(filename) + "?filetype=raw";
227
228 return OpenFile(fullOpenString);
229}
230
231
232//_____________________________________________________________________________
233TString &AliSurveyObj::Sanitize(TString str) {
234 str.Remove(TString::kTrailing, '\r');
235 str.Remove(TString::kTrailing, '\n');
236 return str.Remove(TString::kBoth, ' ');
237}
238
239
240//_____________________________________________________________________________
241Bool_t AliSurveyObj::ParseBuffer(const Char_t* buf) {
242 TString buffer = TString(buf);
243 TObjArray *lines = buffer.Tokenize('\n');
244 TObjArray *dataLine = NULL; // Used to Tokenize each point read
245 TObjArray *colLine = NULL; // Used to Tokenize the column names
246
247 const Int_t kFieldCheck = 10;
248 Bool_t check[kFieldCheck];
249 TString tmp_name = "";
250 Float_t tmp_x = 0.0, tmp_y = 0.0, tmp_z = 0.0;
251 Float_t tmp_precX = 0.0, tmp_precY = 0.0, tmp_precZ = 0.0;
252 Char_t tmp_type = '\0';
253 Bool_t tmp_targ = kTRUE;
254 AliSurveyPoint *dp = 0;
255
256 for (Int_t i = 0; i < kFieldCheck; ++i) check[i] = kFALSE;
257
258 Int_t nrLines = lines->GetEntries();
259 Printf("Lines in file: %d\n", nrLines);
260
261 TString currLine = "", nextLine = "";
262 for (Int_t i = 0; i < nrLines; ++i) {
263 currLine = ((TObjString *)(lines->At(i)))->GetString().Data();
264 nextLine = ((i + 1) < nrLines ? ((TObjString *)(lines->At(i + 1)))->GetString().Data() : "");
265 currLine = Sanitize(currLine);
266 nextLine = Sanitize(nextLine);
267 Printf("\n%d: \"\"%s\"\"\"\n", i + 1, currLine.Data());
268
269 if (0 == currLine.Length()) {
270 Printf("Info: Empty line skipped\n\n");
271 } else if (currLine.BeginsWith(">") && !nextLine.BeginsWith(">")) {
272 currLine.Remove(TString::kLeading, '>');
273 currLine.Remove(TString::kTrailing, ':');
274 currLine.Remove(TString::kBoth, ' ');
275 nextLine.Remove(TString::kBoth, ' ');
276 // Printf(" -> Field: \"%s\"\n", currLine.Data());
277
278 if (currLine.BeginsWith("Title", TString::kIgnoreCase)) {
279 // Report Title
280 fTitle = nextLine;
281 ++i;
282 } else if (currLine.BeginsWith("Date", TString::kIgnoreCase)) {
283 // Report(measurement) Date
284 fDate = nextLine;
285 ++i;
286 } else if (currLine.BeginsWith("Subdetector", TString::kIgnoreCase)) {
287 // Subdetector or structure
288 fDetector = nextLine;
289 ++i;
290 } else if (currLine.BeginsWith("Report URL", TString::kIgnoreCase)) {
291 // Report URL in EDMS
292 if (nextLine.BeginsWith("http://edms.cern.ch/document/", TString::kIgnoreCase) ||
293 nextLine.BeginsWith("https://edms.cern.ch/document/", TString::kIgnoreCase)) {
294 fURL = nextLine;
295 nextLine.Remove(TString::kTrailing, '/');
296 nextLine = nextLine(nextLine.Last('/') + 1, nextLine.Length() - nextLine.Last('/') + 1);
297 // Printf(nextLine);
298 if (!nextLine.IsDigit()) {
299 lines->Delete();
300 return kFALSE;
301 }
302 fReportNr = nextLine.Atoi();
303 ++i;
304 } else { // URL incorrectly formated
305 AliError("Survey text file sintax error! (incorrectly formated Report URL)");
306 return kFALSE;
307 }
308 } else if (currLine.BeginsWith("Version", TString::kIgnoreCase)) {
309 if (!nextLine.IsDigit()) {
310 lines->Delete();
311 return kFALSE;
312 }
313 fVersion = nextLine.Atoi();
314 ++i;
315 } else if (currLine.BeginsWith("General Observations", TString::kIgnoreCase)) {
316 fObs = "";
317 while (('>' != nextLine[0]) && nextLine.Length() > 0) {
318 fObs += (0 == fObs.Length()) ? nextLine : " / " + nextLine;
319 ++i;
320 nextLine = ((i + 1) < nrLines ? ((TObjString *)(lines->At(i + 1)))->GetString().Data() : "");
321 nextLine = Sanitize(nextLine);
322 }
323 // Printf("->%s<-\n", fObs.Data());
324 } else if (currLine.BeginsWith("Coordinate System", TString::kIgnoreCase)) {
325 fCoordSys = nextLine;
326 ++i;
327 } else if (currLine.BeginsWith("Units", TString::kIgnoreCase)) {
328 fUnits = nextLine;
329 ++i;
330 } else if (currLine.BeginsWith("Nr Columns", TString::kIgnoreCase)) {
331 if (!nextLine.IsDigit()) {
332 lines->Delete();
333 return kFALSE;
334 }
335 fNrColumns = nextLine.Atoi();
336 ++i;
337 } else if (currLine.BeginsWith("Column Names", TString::kIgnoreCase)) {
338 fColNames = nextLine;
339 colLine = nextLine.Tokenize(',');
340 if (colLine->GetEntries() != fNrColumns) {
341 colLine->Delete();
342 lines->Delete();
343 return kFALSE;
344 }
345 ++i;
346 } else if (currLine.BeginsWith("Data", TString::kIgnoreCase)) {
347 while (('>' != nextLine[0]) && nextLine.Length() > 0) {
348 Printf("Data LINE: \"%s\"\n", nextLine.Data());
349
350 dataLine = nextLine.Tokenize(' ');
351 if (dataLine->GetEntries() != fNrColumns) {
352 dataLine->Delete();
353 lines->Delete();
354 return kFALSE;
355 }
356
357 for (Int_t t = 0; t < kFieldCheck; ++t) check[t] = 0;
358
359 for (Int_t j = 0; j < dataLine->GetEntries(); ++j) {
360 TString cn = ((TObjString *)(colLine->At(j)))->GetString();
361 TString value = ((TObjString *)(dataLine->At(j)))->GetString();
362 if (cn.BeginsWith("Point Name", TString::kIgnoreCase)) {
363 tmp_name = value;
364 check[0] = kTRUE;
365 } else if (cn.BeginsWith("X", TString::kIgnoreCase)) {
366 tmp_x = value.Atof();
367 check[1] = kTRUE;
368 } else if (cn.BeginsWith("Y", TString::kIgnoreCase)) {
369 tmp_y = value.Atof();
370 check[2] = kTRUE;
371 } else if (cn.BeginsWith("Z", TString::kIgnoreCase)) {
372 tmp_z = value.Atof();
373 check[3] = kTRUE;
374 } else if (cn.BeginsWith("Precision", TString::kIgnoreCase)) {
375 TString tmpCN = cn(0, cn.First('('));
376 Int_t precLength = TString("Precision").Length();
377 //Printf(" ====== %d ======= %d ====== \n", precLength, tmpCN.Length());
378 //Printf(" ====== %s ======= \n", tmpCN.Data());
379 if (precLength == tmpCN.Length()) {
380 tmp_precX = tmp_precY = tmp_precZ = value.Atof();
381 check[6] = kTRUE;
382 } else {
383 TString axis = cn(precLength, tmpCN.Length() - precLength);
384 if (axis.Contains('X', TString::kIgnoreCase)) {
385 tmp_precX = value.Atof();
386 check[7] = kTRUE;
387 } else if (axis.Contains('Y', TString::kIgnoreCase)) {
388 tmp_precY = value.Atof();
389 check[8] = kTRUE;
390 } else if (axis.Contains('Z', TString::kIgnoreCase)) {
391 tmp_precZ = value.Atof();
392 check[9] = kTRUE;
393 } else {
394 AliError("Survey text file sintax error! (Precision column name invalid)");
395 dataLine->Delete();
396 lines->Delete();
397 return kFALSE;
398 }
399 }
400 } else if (cn.BeginsWith("Point Type", TString::kIgnoreCase)) {
401 tmp_type = value.Data()[0];
402 check[4] = kTRUE;
403 } else if (cn.BeginsWith("Target Used", TString::kIgnoreCase)) {
404 tmp_targ = (value.Data()[0] == 'Y') ? kTRUE : kFALSE;
405 check[5] = kTRUE;
406 }
407
408 Printf("--> %s\n", ((TObjString *)(dataLine->At(j)))->GetString().Data());
409 }
410 Bool_t res = kTRUE, precInd = kTRUE;
411
412 // Target
413 if (kFALSE == check[5]) {
414 tmp_targ = kTRUE;
415 check[5] = kTRUE;
416 }
417
418 // Individual axis precisions
419 for (Int_t t = 7; t < 10; ++t) precInd &= check[t];
420 if ((kFALSE == check[6]) && (kTRUE == precInd)) check[6] = kTRUE;
421
422 for (Int_t t = 0; t < kFieldCheck - 3; ++t) {
423 Printf("RES(%d): %d\n", t, check[t]);
424 res &= check[t];
425 }
426 if (kTRUE == res) {
427 dp = new AliSurveyPoint(tmp_name, tmp_x, tmp_y, tmp_z, tmp_precX, tmp_precY, tmp_precZ, tmp_type, tmp_targ);
428 dp->PrintPoint();
429 AddPoint(dp);
430 } else {
431 AliError("Parsing error processing data line!");
432 dataLine->Delete();
433 lines->Delete();
434 return kFALSE;
435 }
436
437 dataLine->Delete();
438 dataLine = NULL;
439 ++i;
440 nextLine = ((i + 1) < nrLines ? ((TObjString *)(lines->At(i + 1)))->GetString().Data() : "");
441 nextLine = Sanitize(nextLine);
442 }
443 }
444 } else { // 2 lines in a row start with ">"
445 AliError("Survey text file sintax error! (2 consecutive lines start with \'>\')");
446 lines->Delete();
447 return kFALSE;
448 }
449 }
450 lines->Delete();
451 fIsValid = kTRUE;
452 return kTRUE;
453}
454