- added the ITS clusters/track in the track properties
[u/mrichter/AliRoot.git] / HLT / global / AliHLTGlobalHistoComponent.cxx
1 // $Id$
2
3 //**************************************************************************
4 //* This file is property of and copyright by the ALICE HLT Project        * 
5 //* ALICE Experiment at CERN, All rights reserved.                         *
6 //*                                                                        *
7 //* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no>        *
8 //*                  for The ALICE HLT Project.                            *
9 //*                                                                        *
10 //* Permission to use, copy, modify and distribute this software and its   *
11 //* documentation strictly for non-commercial purposes is hereby granted   *
12 //* without fee, provided that the above copyright notice appears in all   *
13 //* copies and that both the copyright notice and this permission notice   *
14 //* appear in the supporting documentation. The authors make no claims     *
15 //* about the suitability of this software for any purpose. It is          *
16 //* provided "as is" without express or implied warranty.                  *
17 //**************************************************************************
18
19 /// @file   AliHLTGlobalHistoComponent.cxx
20 /// @author Matthias Richter
21 /// @date   2010-09-16
22 /// @brief  A histogramming component for global ESD properties based
23 ///         on the AliHLTTTreeProcessor
24
25 #include "AliHLTGlobalHistoComponent.h"
26 #include "AliESDEvent.h"
27 #include "AliESDv0.h"
28 #include "AliKFParticle.h"
29 #include "AliKFVertex.h"
30
31 #include "TTree.h"
32 #include "TString.h"
33 #include <cassert>
34
35 /** ROOT macro for the implementation of ROOT specific class methods */
36 ClassImp(AliHLTGlobalHistoComponent)
37
38 AliHLTGlobalHistoComponent::AliHLTGlobalHistoComponent()
39   : AliHLTTTreeProcessor()
40   , fEvent(0)
41   , fNofTracks(0)
42   , fNofV0s(0)
43   , fNofContributors(0)
44   , fVertexX(-99)
45   , fVertexY(-99)
46   , fVertexZ(-99)
47   , fVertexStatus(kFALSE)
48   , fMaxTrackCount(20000)
49   , fMaxV0Count(1000)
50   , fFillV0(kFALSE)
51   , fTrackVariables()
52   , fTrackVariablesInt()
53   , fV0Variables()
54 {
55   // see header file for class documentation
56   // or
57   // refer to README to build package
58   // or
59   // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
60 }
61
62 AliHLTGlobalHistoComponent::~AliHLTGlobalHistoComponent(){
63   // see header file for class documentation
64   fTrackVariables.Reset();
65   fTrackVariablesInt.Reset();
66   fV0Variables.Reset();
67 }
68
69 void AliHLTGlobalHistoComponent::GetInputDataTypes(AliHLTComponentDataTypeList& list){
70   // see header file for class documentation
71   list.push_back(kAliHLTAllDataTypes);
72 }
73
74 AliHLTComponentDataType AliHLTGlobalHistoComponent::GetOutputDataType(){
75   // see header file for class documentation
76   return kAliHLTDataTypeHistogram|kAliHLTDataOriginOut;
77 }
78
79 TTree* AliHLTGlobalHistoComponent::CreateTree(int /*argc*/, const char** /*argv*/){
80 // create the tree and branches
81  
82   int iResult=0;
83   TTree* pTree = new TTree("ESDproperties", "HLT ESD properties");
84   if (!pTree) return NULL;
85
86   const char* trackVariableNames = {
87     // Note the black at the end of each name!
88     "Track_pt "
89     "Track_phi "
90     "Track_eta "
91     "Track_p "
92     "Track_theta "
93     "Track_TPCclus "
94     "Track_ITSclus "
95     "Track_status "
96     "Track_charge "
97     "Track_DCAr "
98     "Track_DCAz "
99     "Track_dEdx "
100   };
101   
102   const char* trackIntVariableNames = {
103     // Note the black at the end of each name!
104     "Track_status "
105   };
106
107   const char* V0VariableNames = {
108     // Note the black at the end of each name!
109     "V0_AP "
110     "V0_pt "  
111     "clust1 "
112     "clust2 "
113     "dev1 "
114     "dev2 "
115     "devPrim "
116     "length "
117     "sigmaLength "
118     "r "
119   };
120      
121   if((iResult=fTrackVariables.Init(fMaxTrackCount, trackVariableNames))<0){
122     HLTError("failed to initialize internal structure for track properties (float)");
123   }
124   if((iResult=fTrackVariablesInt.Init(fMaxTrackCount, trackIntVariableNames))<0){
125     HLTError("failed to initialize internal structure for track properties (int)");
126   }
127   if((iResult=fV0Variables.Init(fMaxV0Count, V0VariableNames))<0){
128       HLTError("failed to initialize internal structure for V0 properties (float)");
129   }
130   
131   if(iResult>=0){     
132      pTree->Branch("event",        &fEvent,           "event/I");
133      pTree->Branch("trackcount",   &fNofTracks,       "trackcount/I");
134      pTree->Branch("vertexX",      &fVertexX,         "vertexX/F");
135      pTree->Branch("vertexY",      &fVertexY,         "vertexY/F");
136      pTree->Branch("vertexZ",      &fVertexZ,         "vertexZ/F");
137      pTree->Branch("nContributors",&fNofContributors, "nContributors/I");
138      pTree->Branch("vertexStatus", &fVertexStatus,    "vertexStatus/I");
139      if(fFillV0==kTRUE) pTree->Branch("V0", &fNofV0s, "V0/I");
140
141      int i=0;
142      // FIXME: this is a bit ugly since type 'f' and 'i' are specified
143      // explicitely. Would be better to use a function like
144      // AliHLTGlobalHistoVariables::GetType but could not get this working
145     
146      for(i=0; i<fTrackVariables.Variables(); i++){
147          TString specifier=fTrackVariables.GetKey(i);
148          float* pArray=fTrackVariables.GetArray(specifier);
149          specifier+="[trackcount]/f";
150          pTree->Branch(fTrackVariables.GetKey(i), pArray, specifier.Data());
151      }
152      for(i=0; i<fTrackVariablesInt.Variables(); i++){
153          TString specifier=fTrackVariablesInt.GetKey(i);
154          int* pArray=fTrackVariablesInt.GetArray(specifier);
155          specifier+="[trackcount]/i";
156          pTree->Branch(fTrackVariablesInt.GetKey(i), pArray, specifier.Data());
157      }    
158      if(fFillV0==kTRUE){
159         for(i=0; i<fV0Variables.Variables(); i++){
160             TString specifier=fV0Variables.GetKey(i);
161             float* pArray=fV0Variables.GetArray(specifier);
162             specifier+="[V0]/f";
163             pTree->Branch(fV0Variables.GetKey(i), pArray, specifier.Data());
164         }
165      }
166   } else {
167     delete pTree;
168     pTree = NULL;
169   }
170   
171   return pTree;
172 }
173
174 void AliHLTGlobalHistoComponent::FillHistogramDefinitions(){
175   /// default histogram definitions
176 }
177
178 int AliHLTGlobalHistoComponent::FillTree(TTree* pTree, const AliHLTComponentEventData& /*evtData*/, AliHLTComponentTriggerData& /*trigData*/ ){
179
180   /// fill the tree from the ESD
181   int iResult=0;
182   if (!IsDataEvent()) return 0;
183
184   ResetVariables();
185
186   // fetch ESD from input stream
187   const TObject *obj = GetFirstInputObject(kAliHLTAllDataTypes, "AliESDEvent");
188   AliESDEvent *esd = dynamic_cast<AliESDEvent*>(const_cast<TObject*>(obj));
189   esd->GetStdContent();
190
191   // fill track variables
192   fNofTracks       = esd->GetNumberOfTracks();
193   fVertexX         = esd->GetPrimaryVertexTracks()->GetX();
194   fVertexY         = esd->GetPrimaryVertexTracks()->GetY();
195   fVertexZ         = esd->GetPrimaryVertexTracks()->GetZ();
196   fNofContributors = esd->GetPrimaryVertexTracks()->GetNContributors();
197   fVertexStatus    = esd->GetPrimaryVertexTracks()->GetStatus();
198   fNofV0s          = esd->GetNumberOfV0s();
199   
200   if(fFillV0==kTRUE && (fNofV0s > fMaxV0Count)){
201      HLTWarning("Found V0s are %d, while component argument is %d, the respective TTree branch is not filled properly. Need to reconfigure.\n", fNofV0s, fMaxV0Count);
202   }
203   
204   for(int i=0; i<fNofTracks; i++){    
205       AliESDtrack *esdTrack = esd->GetTrack(i);
206       if (!esdTrack) continue;
207       
208       Float_t DCAr, DCAz = -99;
209       esdTrack->GetImpactParametersTPC(DCAr, DCAz);
210       
211       fTrackVariables.Fill("Track_pt"        , esdTrack->Pt()                      );
212       fTrackVariables.Fill("Track_phi"       , esdTrack->Phi()*TMath::RadToDeg()   );
213       fTrackVariables.Fill("Track_eta"       , esdTrack->Eta()                     );
214       fTrackVariables.Fill("Track_p"         , esdTrack->P()                       );
215       fTrackVariables.Fill("Track_theta"     , esdTrack->Theta()*TMath::RadToDeg() );
216       fTrackVariables.Fill("Track_TPCclus"   , esdTrack->GetTPCNcls()              );
217       fTrackVariables.Fill("Track_ITSclus"   , esdTrack->GetNcls(0)                );
218       fTrackVariables.Fill("Track_status"    , esdTrack->GetStatus()               );
219       fTrackVariables.Fill("Track_charge"    , esdTrack->Charge()                  );
220       fTrackVariables.Fill("Track_DCAr"      , DCAr                                );
221       fTrackVariables.Fill("Track_DCAz"      , DCAz                                );   
222       fTrackVariables.Fill("Track_dEdx"      , esdTrack->GetTPCsignal()            );   
223       fTrackVariablesInt.Fill("Track_status" , esdTrack->GetStatus()               );      
224   }
225   
226   if(fFillV0==kTRUE){
227      for(int i=0; i<fNofV0s; i++){     
228          AliESDv0 *esdV0 = esd->GetV0(i);
229          if(!esdV0) continue;
230             
231         //fV0Variables.Fill("V0_AP", ap);
232         //fV0Variables.Fill("V0_pt", pt); 
233         //fV0Variables.Fill("clust1", t1->GetTPCNcls()); 
234         //fV0Variables.Fill("clust2", t2->GetTPCNcls()); 
235         //fV0Variables.Fill("dev1", dev1); 
236         //fV0Variables.Fill("dev2", dev2); 
237         //fV0Variables.Fill("devPrim", devPrim); 
238         //fV0Variables.Fill("length", length); 
239         //fV0Variables.Fill("sigmaLength", sigmaLength); 
240         //fV0Variables.Fill("r", r); 
241           
242      } // end of loop over V0s
243   }
244   
245   if(iResult<0){
246     // fill an empty event
247     ResetVariables();
248   }
249   
250   fEvent++;
251   pTree->Fill();
252   return iResult;
253 }
254
255 int AliHLTGlobalHistoComponent::ResetVariables(){
256 /// reset all filling variables
257   fNofTracks=0;
258   fNofV0s=0;
259   fTrackVariables.ResetCount();
260   fTrackVariablesInt.ResetCount();
261   fV0Variables.ResetCount();
262   return 0;
263 }
264
265 AliHLTComponentDataType AliHLTGlobalHistoComponent::GetOriginDataType() const{
266 // get the origin of the output data
267   return kAliHLTDataTypeHistogram|kAliHLTDataOriginHLT;
268 }
269
270 int AliHLTGlobalHistoComponent::ScanConfigurationArgument(int argc, const char** argv){
271 /// inherited from AliHLTComponent, scan argument
272
273   if (argv==NULL || argc<1) return 0;
274
275   int i=0;
276   TString argument = argv[i];
277
278   // -max-track-count
279   if(argument.CompareTo("-max-track-count")==0){    
280      if (++i>=argc) return -EPROTO; // missing parameter
281      argument = argv[i];
282      fMaxTrackCount = argument.Atoi();    
283      HLTInfo("got %s with parameter %s", argument.Data(), argv[i]);
284      return ++i; 
285   }
286  
287   // -max-V0-count 
288   if(argument.CompareTo("-max-V0-count")==0){    
289      if (++i>=argc) return -EPROTO; // missing parameter
290      argument = argv[i];
291      fMaxV0Count = argument.Atoi();    
292      HLTInfo("got %s with parameter %s", argument.Data(), argv[i]);
293      return ++i;
294   }
295   
296   // -fill-V0
297   if(argument.CompareTo("-fill-V0")==0){
298      fFillV0 = kTRUE;
299      HLTInfo("got %s", argument.Data());
300      return ++i;
301   }
302   // no recognized argument, forward to base class
303   return AliHLTTTreeProcessor::ScanConfigurationArgument(argc, argv);
304 }
305
306 int AliHLTGlobalHistoComponent::Reconfigure(const char* cdbEntry, const char* /*chainId*/){  
307 // see header file for class documentation
308
309   TString cdbPath;
310   if (cdbEntry) {
311     cdbPath=cdbEntry;
312   } else {
313     cdbPath="HLT/ConfigHLT/";
314     cdbPath+=GetComponentID();
315   }
316
317   return ConfigureFromCDBTObjString(cdbPath.Data());
318 }
319
320