Include Riostream.h
[u/mrichter/AliRoot.git] / MUON / MUONOfflineShift.C
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 /// \ingroup macros
19 /// \file MUONOfflineShift.C
20 /// \brief Macro to be used to check raw data during MUON offline shifts.
21 ///  
22 /// You NEED an access to the Grid to run this macro.
23 ///
24 /// Basic usage is : 
25 ///
26 /// MUONOfflineShift("path_to_raw_file","basename of output file"); > log
27 ///
28 /// (the redirection to an output log file is recommended as the output from
29 /// this macro might be quite long...)
30 ///
31 /// This will read the raw data and process it several times, varying what's done :
32 /// only decoding, decoding + zero-suppression, etc... (TBE)
33 ///
34 /// Two outputs files will be created : 
35 ///
36 /// - basename.root, containing AliMUONVTrackerData objects that can then 
37 ///   be displayed using the mchview program (using mchview --use basename.root)
38 ///
39 /// - basename.log, containing (for the moment) the occupancy numbers of the various detection elements
40 ///
41 /// Unless you really know what you're doing, please use this macro together with 
42 /// the grid OCDB (i.e. don't set it to a local OCDB). There are now a lot of things
43 /// that are grabbed from the OCDB that are run dependent...
44 ///
45 /// \author Laurent Aphecetche
46
47 #if !defined(__CINT__) || defined(__MAKECINT__)
48
49 #include "AliCDBManager.h"
50 #include "AliCodeTimer.h"
51 #include "AliMUONCDB.h"
52 #include "AliMUONPainterDataRegistry.h"
53 #include "AliMUONRecoParam.h"
54 #include "AliMUONTrackerDataMaker.h"
55 #include "AliMUONVTrackerData.h"
56 #include "AliMpCDB.h"
57 #include "AliMpConstants.h"
58 #include "AliMpDEIterator.h"
59 #include "AliRawReader.h"
60 #include <Riostream.h>
61 #include <TFile.h>
62 #include <TGrid.h>
63 #include <TStopwatch.h>
64 #include <TString.h>
65 #include <TSystem.h>
66
67 #endif
68
69 //______________________________________________________________________________
70 Int_t DataMakerReading(const char* input,
71                        TStopwatch& timer,
72                        const char* cdbPath="",
73                        const char* calibMode="",
74                        Bool_t histogram=kFALSE,
75                        Double_t xmin = 0.0,
76                        Double_t xmax = 4096.0)
77 {
78   /// Run over the data and calibrate it if so required (if cdbPath != "")
79   /// calibMode can be :
80   /// - NOGAIN           : only zero-suppression will be done
81   /// - GAINCONSTANTCAPA : zero-suppression + gain, but with a single capa value for all channels
82   /// - GAIN             : zero-suppression + gain w/ individual capacitance per channel.
83   
84   TString fileName(gSystem->ExpandPathName(input));
85   
86   AliRawReader* rawReader = AliRawReader::Create(fileName.Data());
87   
88   if (!rawReader) return 0;
89   
90   AliMUONVTrackerDataMaker* dm(0x0);
91   
92   AliMUONRecoParam* recoParam = AliMUONCDB::LoadRecoParam();
93   if (!recoParam) return 0;
94   
95   if ( strlen(cdbPath) > 0 ) 
96   {
97     dm = new AliMUONTrackerDataMaker(recoParam,rawReader,cdbPath,calibMode,histogram,xmin,xmax);
98   }
99   else  
100   {
101     dm = new AliMUONTrackerDataMaker(rawReader,histogram);
102   }
103   
104   AliMUONPainterDataRegistry::Instance()->Register(dm);
105
106   timer.Start(kTRUE);
107   Int_t n(0);
108   
109   dm->SetRunning(kTRUE);
110   
111   while (dm->NextEvent())
112   {
113     ++n;
114   }
115   
116   timer.Stop();
117   
118   return n;
119 }
120
121 //______________________________________________________________________________
122 void Print(const char* method, TStopwatch& timer, Int_t n)
123 {
124   /// Get the timing for a given method
125   
126   cout << Form("%20s %10d events. Total CPU time %7.2f seconds.",
127                method,n,timer.CpuTime());
128   
129   Double_t cpu = timer.CpuTime()/n;
130   Double_t real = timer.RealTime()/n;
131   
132   cout << Form(" ms real/event = %7.2f ms CPU/event = %7.2f",real*1E3,cpu*1E3)
133     << endl;
134 }
135
136 //______________________________________________________________________________
137 void Occupancy(ostream& outfile)
138 {
139   /// Write occupancy numbers to output text file
140   
141   outfile << "-----------------------------------------------------" << endl;
142   outfile << "Occupancy numbers" << endl;
143   outfile << "-----------------------------------------------------" << endl;
144   
145   const Int_t occIndex = 2;
146   
147   AliMUONPainterDataRegistry* reg = AliMUONPainterDataRegistry::Instance();
148
149   Int_t nofDataSources = reg->NumberOfDataSources();
150
151   outfile << Form("%11s|"," ");
152   
153   for ( Int_t ids = 0; ids < nofDataSources; ++ ids ) 
154   {
155     AliMUONVTrackerData* data = reg->DataSource(ids);
156     outfile << Form(" %13s |",data->GetName());
157   }
158
159   outfile << endl;
160
161   for ( Int_t chamberId = 0; chamberId < AliMpConstants::NofTrackingChambers(); ++chamberId ) 
162   {
163     Bool_t nonZero(kFALSE);
164     for ( Int_t ids = 0; ids < nofDataSources && nonZero == kFALSE; ++ ids ) 
165     {
166       if ( reg->DataSource(ids)->Chamber(chamberId,occIndex) ) nonZero = kTRUE;
167     }
168     
169     if ( !nonZero ) continue;
170     
171     outfile << Form("Chamber %2d |",chamberId);
172     for ( Int_t ids = 0; ids < nofDataSources; ++ ids ) 
173     {
174       AliMUONVTrackerData* data = reg->DataSource(ids);
175       outfile << Form("   %7.2f %%   |",100.0*data->Chamber(chamberId,occIndex));
176     }
177     outfile << endl;
178     
179     AliMpDEIterator it;
180     it.First(chamberId);
181     while (!it.IsDone())
182     {
183       Int_t detElemId = it.CurrentDEId();
184       Bool_t nonZero(kFALSE);
185       for ( Int_t ids = 0; ids < nofDataSources && nonZero == kFALSE; ++ ids ) 
186       {
187         AliMUONVTrackerData* data = reg->DataSource(ids);
188         if ( data->DetectionElement(detElemId,occIndex) > 0 ) 
189         {
190           nonZero = kTRUE;
191         }
192       }
193       
194       if ( nonZero ) 
195       {
196         outfile << Form("   DE %04d |",detElemId);
197         for ( Int_t ids = 0; ids < nofDataSources; ++ ids ) 
198         {
199           AliMUONVTrackerData* data = reg->DataSource(ids);
200           outfile << Form("   %7.2f %%   |",100.0*data->DetectionElement(detElemId,occIndex));
201         }
202         outfile << endl;
203       }
204       it.Next();
205     }
206   }
207 }
208
209 //______________________________________________________________________________
210 void MUONOfflineShift(const char* input="alien:///alice/data/2009/LHC09a/000067495/raw/09000067495031.10.root", 
211                       const char* outputBase="67495031.10",
212                       const char* ocdbPath="alien://folder=/alice/data/2009/OCDB")
213 {
214   /// Entry point of the macro. 
215
216   AliCodeTimer::Instance()->Reset();
217   
218   TGrid::Connect("alien://");
219   
220   AliRawReader* rawReader = AliRawReader::Create(input);
221   
222   rawReader->NextEvent();
223   
224   Int_t runNumber = rawReader->GetRunNumber();
225     
226   delete rawReader;
227   
228   AliCDBManager::Instance()->SetDefaultStorage(ocdbPath);
229   AliCDBManager::Instance()->SetRun(runNumber);
230
231   if (!AliMUONCDB::LoadMapping()) return;
232   
233   TStopwatch timer1;
234   TStopwatch timer2;
235   TStopwatch timer3;
236   TStopwatch timer4;
237   TStopwatch timer5;
238   
239   Int_t n1 = DataMakerReading(input,timer1,"","",kTRUE,0,0);
240
241   Int_t n2 = DataMakerReading(input,timer2,ocdbPath,"NOGAIN");
242
243   Int_t n3 = DataMakerReading(input,timer3,ocdbPath,"GAINCONSTANTCAPA");
244
245   Int_t n4 = DataMakerReading(input,timer4,ocdbPath,"GAIN");
246
247   Int_t n5 = DataMakerReading(input,timer5,ocdbPath,"INJECTIONGAIN");
248
249   Print("DataMakerReading(HRAW)",timer1,n1);  
250   Print("DataMakerReading(HCALZ)",timer2,n2);
251   Print("DataMakerReading(HCALG)",timer3,n3);
252   Print("DataMakerReading(HCALC)",timer4,n4);
253   Print("DataMakerReading(HCALE)",timer5,n5);
254   
255   AliMUONPainterDataRegistry* reg = AliMUONPainterDataRegistry::Instance();
256   
257   TFile f(gSystem->ExpandPathName(Form("%s.root",outputBase)),"RECREATE");
258   ofstream out(gSystem->ExpandPathName(Form("%s.log",outputBase)));
259
260   Occupancy(out);
261
262   for ( Int_t i = 0; i < reg->NumberOfDataSources(); ++i )
263   {
264     AliMUONVTrackerData* data = reg->DataSource(i);
265     data->Write();
266   }
267   
268   f.Close();
269   
270   AliCodeTimer::Instance()->Print();
271
272 }