Modifications by Alberto
[u/mrichter/AliRoot.git] / TRD / AliTRDPreprocessor.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 ////////////////////////////////////////////////////////////////////////////
19 //                                                                        //
20 // This class is a first implementation for the TRD.                      //
21 // It takes data from HLT and computes the parameters                     //
22 // and stores both reference data and online calibration                  //
23 // parameters in the CDB                                                  //
24 //                                                                        //
25 // Author:                                                                //
26 //   R. Bailhache (R.Bailhache@gsi.de)                                    //
27 //                                                                        //
28 ////////////////////////////////////////////////////////////////////////////
29
30 #include "AliTRDPreprocessor.h"
31
32 #include <TTimeStamp.h>
33 #include <TFile.h>
34 #include <TProfile2D.h>
35 #include <TH2I.h>
36 #include <TStopwatch.h>
37 #include <TObjString.h>
38 #include <TString.h>
39 #include <TList.h>
40 #include <TCollection.h>
41
42 #include "AliCDBMetaData.h"
43 #include "AliDCSValue.h"
44 #include "AliLog.h"
45
46 #include "AliTRDCalibra.h"
47 #include "Cal/AliTRDCalDet.h"
48
49
50 ClassImp(AliTRDPreprocessor)
51
52 //______________________________________________________________________________________________
53 AliTRDPreprocessor::AliTRDPreprocessor(AliShuttleInterface *shuttle)
54                    :AliPreprocessor("TRD", shuttle)
55 {
56   //
57   // Constructor
58   //
59
60 }
61
62 //______________________________________________________________________________________________
63 AliTRDPreprocessor::~AliTRDPreprocessor()
64 {
65   //
66   // Destructor
67   //
68
69 }
70
71 //______________________________________________________________________________________________
72 void AliTRDPreprocessor::Initialize(Int_t run, UInt_t startTime, UInt_t endTime)
73 {
74   //
75   // Initialization routine for the TRD preprocessor
76   //
77
78   AliPreprocessor::Initialize(run,startTime,endTime);
79
80 }
81
82 //______________________________________________________________________________________________
83 UInt_t AliTRDPreprocessor::Process(TMap* /*dcsAliasMap*/)
84 {
85   //
86   // Process the calibraion data for the HLT part
87   //
88
89   // How long does it take for the HLT part?
90   TStopwatch timer;
91   timer.Start();
92
93   // Metadata for the reference data
94   AliCDBMetaData metaData;
95   metaData.SetBeamPeriod(1);
96   metaData.SetResponsible("Raphaelle Bailhache");
97   metaData.SetComment("This preprocessor fills reference data.");
98
99
100   // Take the file from the HLT file exchange server
101   TList *filesources = GetFileSources(kHLT,"GAINDRIFTPRF");
102   if (!filesources) {
103     AliError(Form("No sources found for GAINDRIFTPRF for run %d !",fRun));
104     return 0;
105   }
106   if (filesources->GetSize() != 1) {
107     AliError(Form("More than one source found for GAINDRIFTPRF for run %d!",fRun));
108     return 0;
109   }
110
111   // Call a AliTRDCalibra instance for fit
112   AliTRDCalibra *calibra = AliTRDCalibra::Instance();
113
114   // Init some things
115   AliTRDCalDet *objgaindet          = 0x0; // Object for det average gain factor
116   AliTRDCalDet *objdriftvelocitydet = 0x0; // Object for det average drift velocity
117   AliTRDCalDet *objtime0det         = 0x0; // Object for det average time0 
118   TObject      *objgainpad          = 0x0; // Object for pad (relative to the det) gain factor
119   TObject      *objdriftvelocitypad = 0x0; // Object for pad (relative to the det) drift velocity
120   TObject      *objtime0pad         = 0x0; // Object for pad (relative to the det) time0
121   TObject      *objPRFpad           = 0x0; // Object for pad prf width
122   TH2I         *histogain           = 0x0; // Histogram taken from HLT for gain factor
123   TProfile2D   *histodriftvelocity  = 0x0; // Profile taken from HLT for drift velocity and time0
124   TProfile2D   *histoprf            = 0x0; // Profile taken from HLT for prf
125
126   Int_t    numberfit[3]        = { 0,   0,   0   }; // Number of histos fitted for gain, drift velocity and prf
127   Int_t    numberEnt[3]        = { 0,   0,   0   }; // Number of histos with entries
128   Double_t statisticmean[3]    = { 0.0, 0.0, 0.0 }; // Mean values of the number of entries in these histos
129   Int_t    numbertotalgroup[3] = { 0,   0,   0   }; // Total number of groups
130
131
132   // Loop over the files taken from the HLT/
133   TIter iter(filesources);
134   TObjString *source;
135   while ((source = dynamic_cast<TObjString *> (iter.Next()))) {
136     
137     TString filename = GetFile(kHLT,"GAINDRIFTPRF",source->GetName());
138     if (!filename.Data()) {
139       AliError(Form("Error retrieving file from source %d failed!", source->GetName()));
140       delete filesources;
141       return 0;
142     }
143
144     // Take the histos
145     TFile *file = TFile::Open(filename);
146     histogain = (TH2I *) file->Get("CH2d");
147     histogain->SetDirectory(0);
148     if (!histogain) {
149       AliError("Error retrieving 2D histos for gain failed!");
150     }
151     histodriftvelocity = (TProfile2D *) file->Get("PH2d");
152     histodriftvelocity->SetDirectory(0);
153     if (!histodriftvelocity) {
154       AliError("Error retrieving 2D Profile for average pulse height failed!");
155     }
156     histoprf = (TProfile2D *) file->Get("PRF2d");
157     histoprf->SetDirectory(0);
158     if (!histoprf) {
159       AliError("Error retrieving 2D Profile for Pad Response Function failed!");
160     }
161     file->Close();
162
163     // Set the mode of calibration from the TObject, store the reference data and try to fit them
164     if (histogain) {
165       calibra->SetModeCalibrationFromTObject((TObject *) histogain,0);
166       Store("Data","Gain",(TObject *) histogain,&metaData,0,0);
167       AliInfo("Take the CH reference data. Now we will try to fit\n");
168       calibra->SetMinEntries(100); // If there is less than 100 entries in the histo: no fit
169       calibra->FitCHOnline(histogain);
170       numbertotalgroup[0] = 6*4*18*((Int_t) calibra->GetDetChamb0(0))
171                           + 6*  18*((Int_t) calibra->GetDetChamb2(0));
172       numberfit[0]        = calibra->GetNumberFit();
173       statisticmean[0]    = calibra->GetStatisticMean(); 
174       numberEnt[0]        = calibra->GetNumberEnt();
175       objgaindet          = calibra->CreateDetObjectTree(calibra->GetGain(),0);
176       objgainpad          = calibra->CreatePadObjectTree(calibra->GetGain(),0,objgaindet);
177     }
178     
179     if (histodriftvelocity) {
180       calibra->SetModeCalibrationFromTObject((TObject *) histodriftvelocity,1);
181       Store("Data","VdriftT0",(TObject *) histodriftvelocity,&metaData,0,0);
182       AliInfo("Take the PH reference data. Now we will try to fit\n");
183       calibra->SetMinEntries(100*20); // If there is less than 2000
184       calibra->FitPHOnline(histodriftvelocity);
185       numbertotalgroup[1] = 6*4*18*((Int_t) calibra->GetDetChamb0(1))
186                           + 6*  18*((Int_t) calibra->GetDetChamb2(1));
187       numberfit[1]        = calibra->GetNumberFit();
188       statisticmean[1]    = calibra->GetStatisticMean(); 
189       numberEnt[1]        = calibra->GetNumberEnt();
190       objdriftvelocitydet = calibra->CreateDetObjectTree(calibra->GetVdrift(),1);
191       objdriftvelocitypad = calibra->CreatePadObjectTree(calibra->GetVdrift(),1,objdriftvelocitydet);
192       objtime0det         = calibra->CreateDetObjectTree(calibra->GetT0(),3);
193       objtime0pad         = calibra->CreatePadObjectTree(calibra->GetT0(),3,objtime0det);
194     }
195     
196     if (histoprf) {
197       calibra->SetModeCalibrationFromTObject((TObject *) histoprf,2);
198       Store("Data","PRF",(TObject *) histoprf,&metaData,0,0);
199       AliInfo("Take the PRF reference data. Now we will try to fit\n");
200       calibra->SetMinEntries(100*20); // If there is less than 2000
201       calibra->SetRangeFitPRF(0.5);
202       calibra->FitPRFOnline(histoprf);
203       numbertotalgroup[2] = 6*4*18*((Int_t) calibra->GetDetChamb0(2))
204                           + 6*  18*((Int_t) calibra->GetDetChamb2(2));
205       numberfit[2]        = calibra->GetNumberFit();
206       statisticmean[2]    = calibra->GetStatisticMean(); 
207       numberEnt[2]        = calibra->GetNumberEnt();
208       objPRFpad           = calibra->CreatePadObjectTree(calibra->GetPRF());
209     }
210
211   }
212   
213   // Bilan of the fit statistic
214   AliInfo(Form("The mean number of entries required for a fit is: %d"
215               ,(Int_t) calibra->GetMinEntries()));
216   AliInfo(Form("FOR THE CH: There is a mean statistic of: %f, with %d fits for %d groups and %d histos with entries"
217               ,statisticmean[0],numberfit[0],numbertotalgroup[0],numberEnt[0]));
218   AliInfo(Form("FOR THE PH: There is a mean statistic of: %f, with %d fits for %d groups and %d histos with entries"
219               ,statisticmean[1],numberfit[1],numbertotalgroup[1],numberEnt[1]));
220   AliInfo(Form("FOR THE PRF: There is a mean statistic of: %f, with %d fits for %d groups and %d histos with entries"
221               ,statisticmean[2],numberfit[2],numbertotalgroup[2],numberEnt[2]));
222   
223
224   //
225   // Store the coefficients in the grid OCDB if enough statistics
226   //
227   
228   // Store the infos for the detector
229   AliCDBMetaData *md1= new AliCDBMetaData(); 
230   md1->SetObjectClassName("AliTRDCalDet");
231   md1->SetResponsible("Raphaelle Bailhache");
232   md1->SetBeamPeriod(1);
233   md1->SetAliRootVersion("01-10-06"); // root version
234   md1->SetComment("The dummy values in this calibration file are for testing only");
235   if ((numbertotalgroup[0] >                  0) && 
236       (numberfit[0]        >= 0.95*numberEnt[0])) {
237     Store("Calib","ChamberGainFactor",(TObject *) objgaindet         ,md1,0,kTRUE);
238   }
239   if ((numbertotalgroup[1] >                  0) && 
240       (numberfit[1]        >= 0.95*numberEnt[1])) {
241     Store("Calib","ChamberVdrift"    ,(TObject *) objdriftvelocitydet,md1,0,kTRUE);
242     Store("Calib","ChamberT0"        ,(TObject *) objtime0det        ,md1,0,kTRUE);
243   }
244   
245   // Store the infos for the pads
246   AliCDBMetaData *md2= new AliCDBMetaData(); 
247   md2->SetObjectClassName("AliTRDCalPad");
248   md2->SetResponsible("Raphaelle Bailhache");
249   md2->SetBeamPeriod(1);
250   md2->SetAliRootVersion("01-10-06"); //root version
251   md2->SetComment("The dummy values in this calibration file are for testing only");
252   if ((numbertotalgroup[0] >                  0) && 
253       (numberfit[0]        >= 0.95*numberEnt[0])) {
254     Store("Calib","LocalGainFactor"  ,(TObject *) objgainpad         ,md2,0,kTRUE);
255   }
256   if ((numbertotalgroup[1] >                  0) && 
257       (numberfit[1]        >= 0.95*numberEnt[1])) {
258     Store("Calib","LocalVdrift"      ,(TObject *) objdriftvelocitypad,md2,0,kTRUE);
259     Store("Calib","LocalT0"          ,(TObject *) objtime0pad        ,md2,0,kTRUE);
260   }
261   if ((numbertotalgroup[2] >                  0) && 
262       (numberfit[2]        >= 0.95*numberEnt[2])) {
263     Store("Calib","PRFWidth"         ,(TObject *) objPRFpad          ,md2,0,kTRUE);
264   }
265   
266   // End
267   delete filesources;
268   timer.Stop();
269   timer.Print();
270   return 1;  
271
272 }