Reconstruction of raw data (T.Kuhr)
[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
33
34 ClassImp(AliTPCReconstructor)
35
36 Double_t AliTPCReconstructor::fgCtgRange = 1.05;
37
38 //_____________________________________________________________________________
39 void AliTPCReconstructor::Reconstruct(AliRunLoader* runLoader) const
40 {
41 // reconstruct clusters
42
43   AliLoader* loader = runLoader->GetLoader("TPCLoader");
44   if (!loader) {
45     Error("Reconstruct", "TPC loader not found");
46     return;
47   }
48   loader->LoadRecPoints("recreate");
49   loader->LoadDigits("read");
50
51   AliTPCParam* param = GetTPCParam(runLoader);
52   if (!param) return;
53   AliTPCclustererMI clusterer(param);
54   Int_t nEvents = runLoader->GetNumberOfEvents();
55
56   for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
57     runLoader->GetEvent(iEvent);
58
59     TTree* treeClusters = loader->TreeR();
60     if (!treeClusters) {
61       loader->MakeTree("R");
62       treeClusters = loader->TreeR();
63     }
64     TTree* treeDigits = loader->TreeD();
65     if (!treeDigits) {
66       Error("Reconstruct", "Can't get digits tree !");
67       return;
68     }
69
70     clusterer.SetInput(treeDigits);
71     clusterer.SetOutput(treeClusters);
72     clusterer.Digits2Clusters();
73          
74     loader->WriteRecPoints("OVERWRITE");
75   }
76
77   loader->UnloadRecPoints();
78   loader->UnloadDigits();
79 }
80
81 //_____________________________________________________________________________
82 void AliTPCReconstructor::Reconstruct(AliRunLoader* runLoader,
83                                       AliRawReader* rawReader) const
84 {
85 // reconstruct clusters from raw data
86
87   AliLoader* loader = runLoader->GetLoader("TPCLoader");
88   if (!loader) {
89     Error("Reconstruct", "TPC loader not found");
90     return;
91   }
92   loader->LoadRecPoints("recreate");
93
94   AliTPCParam* param = GetTPCParam(runLoader);
95   if (!param) return;
96   AliTPCclustererMI clusterer(param);
97
98   Int_t iEvent = 0;
99   while (rawReader->NextEvent()) {
100     runLoader->GetEvent(iEvent++);
101
102     TTree* treeClusters = loader->TreeR();
103     if (!treeClusters) {
104       loader->MakeTree("R");
105       treeClusters = loader->TreeR();
106     }
107
108     clusterer.SetOutput(treeClusters);
109     clusterer.Digits2Clusters(rawReader);
110          
111     loader->WriteRecPoints("OVERWRITE");
112   }
113
114   loader->UnloadRecPoints();
115 }
116
117 //_____________________________________________________________________________
118 AliTracker* AliTPCReconstructor::CreateTracker(AliRunLoader* runLoader) const
119 {
120 // create a TPC tracker
121
122   AliTPCParam* param = GetTPCParam(runLoader);
123   if (!param) return NULL;
124   return new AliTPCtrackerMI(param);
125 }
126
127 //_____________________________________________________________________________
128 void AliTPCReconstructor::FillESD(AliRunLoader* /*runLoader*/, 
129                                   AliESD* esd) const
130 {
131 // make PID
132
133   Double_t parTPC[] = {47., 0.10, 10.};
134   AliTPCpidESD tpcPID(parTPC);
135   tpcPID.MakePID(esd);
136 }
137
138
139 //_____________________________________________________________________________
140 AliTPCParam* AliTPCReconstructor::GetTPCParam(AliRunLoader* runLoader) const
141 {
142 // get the TPC parameters
143
144   TDirectory* saveDir = gDirectory;
145   runLoader->CdGAFile();
146
147   AliTPCParam* param = (AliTPCParam*) gDirectory->Get("75x40_100x60_150x60");
148   if (!param) Error("GetTPCParam", "no TPC parameters found");
149
150   saveDir->cd();
151   return param;
152 }