Vertexing with tracks included (Andrea)
[u/mrichter/AliRoot.git] / TPC / AliTPCReconstructor.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 // class for TPC reconstruction                                              //
21 //                                                                           //
22 ///////////////////////////////////////////////////////////////////////////////
23
24
25 #include "AliTPCReconstructor.h"
26 #include "AliRunLoader.h"
27 #include "AliRun.h"
28 #include "AliRawReader.h"
29 #include "AliTPCclustererMI.h"
30 #include "AliTPCtrackerMI.h"
31 #include "AliTPCpidESD.h"
32 #include "AliTPCParam.h"
33 #include "AliTPCParamSR.h"
34 #include "AliTPCcalibDB.h"
35
36 ClassImp(AliTPCReconstructor)
37
38 Double_t AliTPCReconstructor::fgCtgRange = 1.05;
39 Double_t AliTPCReconstructor::fgMaxSnpTracker   = 0.95;   // max tangent in tracker - correspond to 3    
40 Double_t AliTPCReconstructor::fgMaxSnpTrack     = 0.999;  // tangent    
41 Int_t    AliTPCReconstructor::fgStreamLevel     = 0;      // stream (debug) level
42 //_____________________________________________________________________________
43 void AliTPCReconstructor::Reconstruct(AliRunLoader* runLoader) const
44 {
45 // reconstruct clusters
46
47   AliLoader* loader = runLoader->GetLoader("TPCLoader");
48   if (!loader) {
49     Error("Reconstruct", "TPC loader not found");
50     return;
51   }
52   loader->LoadRecPoints("recreate");
53   loader->LoadDigits("read");
54
55   AliTPCParam* param = GetTPCParam(runLoader);
56   if (!param) return;
57   AliTPCclustererMI clusterer(param);
58   Int_t nEvents = runLoader->GetNumberOfEvents();
59
60   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
61     runLoader->GetEvent(iEvent);
62
63     TTree* treeClusters = loader->TreeR();
64     if (!treeClusters) {
65       loader->MakeTree("R");
66       treeClusters = loader->TreeR();
67     }
68     TTree* treeDigits = loader->TreeD();
69     if (!treeDigits) {
70       Error("Reconstruct", "Can't get digits tree !");
71       return;
72     }
73
74     clusterer.SetInput(treeDigits);
75     clusterer.SetOutput(treeClusters);
76     clusterer.Digits2Clusters();
77          
78     loader->WriteRecPoints("OVERWRITE");
79   }
80
81   loader->UnloadRecPoints();
82   loader->UnloadDigits();
83 }
84
85 //_____________________________________________________________________________
86 void AliTPCReconstructor::Reconstruct(AliRunLoader* runLoader,
87                                       AliRawReader* rawReader) const
88 {
89 // reconstruct clusters from raw data
90
91   AliLoader* loader = runLoader->GetLoader("TPCLoader");
92   if (!loader) {
93     Error("Reconstruct", "TPC loader not found");
94     return;
95   }
96   loader->LoadRecPoints("recreate");
97
98   AliTPCParam* param = GetTPCParam(runLoader);
99   if (!param) {
100     AliWarning("Loading default TPC parameters !");
101     param = new AliTPCParamSR;
102   }
103   AliTPCclustererMI clusterer(param);
104
105   TString option = GetOption();
106   if (option.Contains("PedestalSubtraction"))
107     clusterer.SetPedSubtraction(kTRUE);
108   if (option.Contains("OldRCUFormat"))
109     clusterer.SetOldRCUFormat(kTRUE);
110  
111   Int_t iEvent = 0;
112   while (rawReader->NextEvent()) {  
113     runLoader->GetEvent(iEvent++);
114
115     TTree* treeClusters = loader->TreeR();
116     if (!treeClusters) {
117       loader->MakeTree("R");
118       treeClusters = loader->TreeR();
119     }
120
121     clusterer.SetOutput(treeClusters);
122     clusterer.Digits2Clusters(rawReader);
123          
124     loader->WriteRecPoints("OVERWRITE");
125   }
126
127   loader->UnloadRecPoints();
128 }
129
130 //_____________________________________________________________________________
131 AliTracker* AliTPCReconstructor::CreateTracker(AliRunLoader* runLoader) const
132 {
133 // create a TPC tracker
134
135   AliTPCParam* param = GetTPCParam(runLoader);
136   if (!param) {
137     AliWarning("Loading default TPC parameters !");
138     param = new AliTPCParamSR;
139   }
140   param->ReadGeoMatrices();
141   return new AliTPCtrackerMI(param);
142 }
143
144 //_____________________________________________________________________________
145 void AliTPCReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
146                                   AliESD* esd) const
147 {
148 // make PID
149
150   Double_t parTPC[] = {47., 0.10, 10.};
151   AliTPCpidESD tpcPID(parTPC);
152   tpcPID.MakePID(esd);
153 }
154
155
156 //_____________________________________________________________________________
157 AliTPCParam* AliTPCReconstructor::GetTPCParam(AliRunLoader* /*runLoader*/) const
158 {
159 // get the TPC parameters
160
161 //  TDirectory* saveDir = gDirectory;
162 //runLoader->CdGAFile();
163
164   AliTPCParam* param =  AliTPCcalibDB::Instance()->GetParameters();
165
166  //  AliTPCParam* param = (AliTPCParam*) gDirectory->Get("75x40_100x60_150x60");
167 //   if (!param) Error("GetTPCParam", "no TPC parameters found");
168
169 //  saveDir->cd();
170   return param;
171 }