compilation error fixed
[u/mrichter/AliRoot.git] / HLT / ITS / AliHLTITSClusterHistoComponent.cxx
CommitLineData
c0b68f8e 1// $Id$
2//**************************************************************************
3//* This file is property of and copyright by the ALICE HLT Project *
4//* ALICE Experiment at CERN, All rights reserved. *
5//* *
6//* Primary Authors: Gaute Ovrebekk <ovrebekk@ift.uib.no> *
7//* for The ALICE HLT Project. *
8//* *
9//* Permission to use, copy, modify and distribute this software and its *
10//* documentation strictly for non-commercial purposes is hereby granted *
11//* without fee, provided that the above copyright notice appears in all *
12//* copies and that both the copyright notice and this permission notice *
13//* appear in the supporting documentation. The authors make no claims *
14//* about the suitability of this software for any purpose. It is *
15//* provided "as is" without express or implied warranty. *
16//**************************************************************************
17
18/** @file AliHLTITSClusterHistoComponent.cxx
19 @author Gaute Ovrebekk
f44c9362 20 @brief Component for plotting charge in clusters
c0b68f8e 21*/
22
23#if __GNUC__>= 3
24using namespace std;
25#endif
26
27#include "AliHLTITSClusterHistoComponent.h"
28#include "AliHLTITSClusterDataFormat.h"
29#include "AliCDBEntry.h"
30#include "AliCDBManager.h"
f44c9362 31#include "AliGeomManager.h"
c0b68f8e 32#include "AliITSRecPoint.h"
91edcbad 33#include "TMath.h"
c0b68f8e 34#include <TFile.h>
35#include <TString.h>
36#include "TObjString.h"
37#include "TObjArray.h"
38
c0b68f8e 39/** ROOT macro for the implementation of ROOT specific class methods */
40ClassImp(AliHLTITSClusterHistoComponent)
41
42AliHLTITSClusterHistoComponent::AliHLTITSClusterHistoComponent()
f44c9362 43 :
44 fXY(NULL),
91edcbad 45 fPhieta(NULL),
c0b68f8e 46 fCharge(NULL),
47 fPlotCharge(kFALSE),
59918cb1 48 fPlotXYPhiEta(kTRUE),
49 fOutputSize(5000)
c0b68f8e 50{
51 // see header file for class documentation
52 // or
53 // refer to README to build package
54 // or
55 // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
c0b68f8e 56}
57
f44c9362 58AliHLTITSClusterHistoComponent::~AliHLTITSClusterHistoComponent(){
c0b68f8e 59 // see header file for class documentation
60}
61
62// Public functions to implement AliHLTComponent's interface.
63// These functions are required for the registration process
64
f44c9362 65const char* AliHLTITSClusterHistoComponent::GetComponentID(){
c0b68f8e 66 // see header file for class documentation
c0b68f8e 67 return "ITSClusterHisto";
68}
69
f44c9362 70void AliHLTITSClusterHistoComponent::GetInputDataTypes(AliHLTComponentDataTypeList& list){
c0b68f8e 71 // see header file for class documentation
72 list.clear();
8ce2efb0 73 list.push_back( kAliHLTDataTypeClusters|kAliHLTDataOriginITSSPD );
74 list.push_back( kAliHLTDataTypeClusters|kAliHLTDataOriginITSSDD );
75 list.push_back( kAliHLTDataTypeClusters|kAliHLTDataOriginITSSSD );
c0b68f8e 76}
77
f44c9362 78AliHLTComponentDataType AliHLTITSClusterHistoComponent::GetOutputDataType(){
c0b68f8e 79 // see header file for class documentation
80 return kAliHLTDataTypeHistogram;
c0b68f8e 81}
82
f44c9362 83void AliHLTITSClusterHistoComponent::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier ){
c0b68f8e 84 // see header file for class documentation
85 // XXX TODO: Find more realistic values.
59918cb1 86 constBase = fOutputSize;
f44c9362 87 inputMultiplier = 2;
c0b68f8e 88}
89
f44c9362 90AliHLTComponent* AliHLTITSClusterHistoComponent::Spawn(){
c0b68f8e 91 // see header file for class documentation
92 return new AliHLTITSClusterHistoComponent;
93}
94
f44c9362 95int AliHLTITSClusterHistoComponent::DoInit( int argc, const char** argv ){
96 // see header file for class documentation
97
98 if(AliGeomManager::GetGeometry()==NULL){
99 AliGeomManager::LoadGeometry();
100 }
101
102 fPlotCharge = kFALSE;
103 fPlotXYPhiEta = kTRUE;
c0b68f8e 104
f44c9362 105 if(fPlotCharge) fCharge = new TH1F("fCharge","Total Charge of clusters",2000,0,2000);
0b26a583 106 if(fPlotXYPhiEta){
f44c9362 107 fXY = new TH2F("fXY","Global XY of ITS clusters",1600,-80,80,1600,-80,80);
108 Char_t name[50];
109 Char_t title[50];
110 fPhieta = new TH2F*[6];
111 for (Int_t iLay=0;iLay<6;iLay++){
112 sprintf(name,"Phi_vs_Eta_ITS_Layer%d",iLay+1);
113 sprintf(title,"Phi vs Eta - ITS Layer %d",iLay+1);
114 fPhieta[iLay]=new TH2F(name,title,60,-1.5,1.5,60,0.,2*TMath::Pi());
115 fPhieta[iLay]->GetXaxis()->SetTitle("Pseudorapidity");
116 fPhieta[iLay]->GetYaxis()->SetTitle("#varphi [rad]");
117 }
0b26a583 118 }
c0b68f8e 119
120 int iResult=0;
121 TString configuration="";
122 TString argument="";
f44c9362 123
124 for(int i=0; i<argc && iResult>=0; i++){
c0b68f8e 125 argument=argv[i];
126 if (!configuration.IsNull()) configuration+=" ";
127 configuration+=argument;
128 }
129
f44c9362 130 if(!configuration.IsNull()){
c0b68f8e 131 iResult=Configure(configuration.Data());
132 }
133
134 return iResult;
135}
136
f44c9362 137int AliHLTITSClusterHistoComponent::DoDeinit(){
c0b68f8e 138 // see header file for class documentation
139 if(fCharge!=NULL) delete fCharge;
0b26a583 140 if(fXY!=NULL) {
f44c9362 141 delete fXY;
142 for(Int_t i=0;i<6;i++) delete fPhieta[5-i];
0b26a583 143 }
c0b68f8e 144 return 0;
145}
146
f44c9362 147int AliHLTITSClusterHistoComponent::DoEvent(const AliHLTComponentEventData& /*evtData*/, AliHLTComponentTriggerData& /*trigData*/){
148 // see header file for class documentation
149
0b26a583 150 static Int_t event = 0;
151 event++;
c0b68f8e 152 int TotalSpacePoint = 0;
153
154 const AliHLTComponentBlockData* iter = NULL;
155
59918cb1 156 AliHLTUInt32_t eventType=gkAliEventTypeUnknown;
157 if(!IsDataEvent(&eventType) && eventType!=gkAliEventTypeEndOfRun) {
158 // publish the histograms at the end of run
159 return 0;
160 }
f44c9362 161
59918cb1 162 for(iter = GetFirstInputBlock(kAliHLTDataTypeClusters); eventType!=gkAliEventTypeEndOfRun && iter != NULL; iter = GetNextInputBlock()){
c0b68f8e 163
f44c9362 164 if(iter->fDataType!=(kAliHLTAnyDataType|kAliHLTDataOriginITSSPD) &&
165 iter->fDataType!=(kAliHLTAnyDataType|kAliHLTDataOriginITSSDD) &&
166 iter->fDataType!=(kAliHLTAnyDataType|kAliHLTDataOriginITSSSD))
167 continue;
8ce2efb0 168
f44c9362 169 const AliHLTITSClusterData* clusterData = (const AliHLTITSClusterData*) iter->fPtr;
170 Int_t nSpacepoint = (Int_t) clusterData->fSpacePointCnt;
171 TotalSpacePoint += nSpacepoint;
172 AliHLTITSSpacePointData *clusters = (AliHLTITSSpacePointData*) clusterData->fSpacePoints;
c0b68f8e 173
f44c9362 174 for(int i=0;i<nSpacepoint;i++){
175
176 Int_t lab[4] = {0,0,0,0};
177 Float_t hit[6] = {0,0,0,0,0,0};
178 Int_t info[3] = {0,0,0};
179
180 lab[0] = clusters[i].fTracks[0];
181 lab[1] = clusters[i].fTracks[1];
182 lab[2] = clusters[i].fTracks[2];
183 lab[3] = clusters[i].fIndex;
184 hit[0] = clusters[i].fY;
185 hit[1] = clusters[i].fZ;
186 hit[2] = clusters[i].fSigmaY2;
187 hit[3] = clusters[i].fSigmaZ2;
188 hit[4] = clusters[i].fQ;
189 hit[5] = clusters[i].fSigmaYZ;
190 info[0] = clusters[i].fNy;
191 info[1] = clusters[i].fNz;
192 info[2] = clusters[i].fLayer;
193
194 Float_t xyz[3];
195 AliITSRecPoint recpoint(lab,hit,info);
196 recpoint.GetGlobalXYZ(xyz);
197 Int_t layer = recpoint.GetLayer();
198
199 if(fPlotXYPhiEta){
200 fXY->Fill(xyz[0],xyz[1]);
201 Float_t rad=TMath::Sqrt(xyz[0]*xyz[0]+xyz[1]*xyz[1]);
202 Float_t theta=TMath::ATan2(rad,xyz[2]);
203 Float_t eta=-1*TMath::Log(TMath::Tan(theta/2.0));
204 Float_t phi=TMath::ATan2(xyz[1],xyz[0]);
205 if(phi<0.0){phi=2 * TMath::Pi() - TMath::Abs(phi);}
206 fPhieta[layer]->Fill(eta,phi);
207 }
208 if(fPlotCharge) fCharge->Fill(recpoint.GetQ());
c0b68f8e 209 }
210 }
211
212 if(fPlotCharge){
f44c9362 213 AliHLTUInt32_t fSpecification = 0x0;
59918cb1 214 if (PushBack( (TObject*) fCharge,kAliHLTDataTypeHistogram|kAliHLTDataOriginITS,fSpecification)==-ENOSPC) {
215 fOutputSize+=GetLastObjectSize();
216 }
c0b68f8e 217 }
0b26a583 218 if(fPlotXYPhiEta){
f44c9362 219 AliHLTUInt32_t fSpecification = 0x0;
59918cb1 220 if (PushBack( (TObject*) fXY,kAliHLTDataTypeHistogram|kAliHLTDataOriginITS,fSpecification)==-ENOSPC) {
221 fOutputSize+=GetLastObjectSize();
222 }
223 for(Int_t ii=0;ii<6;ii++)
224 if (PushBack( (TObject*) fPhieta[ii],kAliHLTDataTypeHistogram|kAliHLTDataOriginITS,fSpecification)==-ENOSPC) {
225 fOutputSize+=GetLastObjectSize();
226 }
c0b68f8e 227 }
228
f44c9362 229 HLTDebug("ITSClusterHisto found %d Total Spacepoints", TotalSpacePoint);
c0b68f8e 230
231 return 0;
232}
233
f44c9362 234int AliHLTITSClusterHistoComponent::Configure(const char* arguments){
235 // see header file for class documentation
236
c0b68f8e 237 int iResult=0;
238
239 if (!arguments) return iResult;
240
241 TString allArgs=arguments;
242 TString argument;
243
244 TObjArray* pTokens=allArgs.Tokenize(" ");
245
246 if (pTokens) {
247 for (int i=0; i<pTokens->GetEntries() && iResult>=0; i++) {
248 argument=((TObjString*)pTokens->At(i))->GetString();
249 if (argument.IsNull()) continue;
250
251 if (argument.CompareTo("-plot-all")==0) {
252 HLTInfo("Ploting all historgams");
0b26a583 253 fPlotXYPhiEta = kTRUE;
c0b68f8e 254 fPlotCharge = kTRUE;
255 continue;
256 }
257
258 else if (argument.CompareTo("-plot-xy")==0) {
259 HLTInfo("Ploting Global XY");
0b26a583 260 fPlotXYPhiEta = kTRUE;
c0b68f8e 261 continue;
262 }
263
c0b68f8e 264 else if (argument.CompareTo("-plot-charge")==0) {
265 HLTInfo("Ploting charge of clusters");
266 fPlotCharge = kTRUE;
267 continue;
268 }
269
270 else {
271 HLTError("unknown argument %s", argument.Data());
272 iResult=-EINVAL;
273 break;
274 }
275 }
276 delete pTokens;
277 }
278
91edcbad 279 if(!fCharge && fPlotCharge){fCharge = new TH1F("fCharge","Total Charge of clusters",2000,0,2000);}
0b26a583 280 if(!fXY && fPlotXYPhiEta){
281 fXY = new TH2F("fXY","Global XY of ITS clusters",1600,-80,80,1600,-80,80);
282 Char_t name[50];
283 Char_t title[50];
284 fPhieta = new TH2F*[6];
285 for (Int_t iLay=0;iLay<6;iLay++) {
286 sprintf(name,"Phi_vs_Eta_ITS_Layer%d",iLay+1);
287 sprintf(title,"Phi vs Eta - ITS Layer %d",iLay+1);
288 fPhieta[iLay]=new TH2F(name,title,30,-1.5,1.5,200,0.,2*TMath::Pi());
289 fPhieta[iLay]->GetXaxis()->SetTitle("Pseudorapidity");
290 fPhieta[iLay]->GetYaxis()->SetTitle("#varphi [rad]");
291 }
292 }
c0b68f8e 293
294 return iResult;
295}
296
f44c9362 297int AliHLTITSClusterHistoComponent::Reconfigure(const char* cdbEntry, const char* chainId){
c0b68f8e 298 // see header file for class documentation
299 int iResult=0;
300
301 const char* path="HLT/ConfigITS/HistoComponent";
302 const char* defaultNotify="";
303 if (cdbEntry) {
304 path=cdbEntry;
305 defaultNotify=" (default)";
306 }
307 if (path) {
308 HLTInfo("reconfigure from entry %s%s, chain id %s", path, defaultNotify,(chainId!=NULL && chainId[0]!=0)?chainId:"<none>");
309 AliCDBEntry *pEntry = AliCDBManager::Instance()->Get(path);
310 if (pEntry) {
311 TObjString* pString=dynamic_cast<TObjString*>(pEntry->GetObject());
312 if (pString) {
313 HLTInfo("received configuration object string: \'%s\'", pString->GetString().Data());
314 iResult=Configure(pString->GetString().Data());
315 } else {
316 HLTError("configuration object \"%s\" has wrong type, required TObjString", path);
317 }
318 } else {
319 HLTError("can not fetch object \"%s\" from CDB", path);
320 }
321 }
322
323 return iResult;
324}