]>
Commit | Line | Data |
---|---|---|
1 | /**************************************************************************** | |
2 | * Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch * | |
3 | ****************************************************************************/ | |
4 | ||
5 | #if !defined(__CINT__) || defined(__MAKECINT__) | |
6 | #include <Riostream.h> | |
7 | #include "AliTPCParam.h" | |
8 | #include "AliTPCtracker.h" | |
9 | #include "AliTPCtrackerMI.h" | |
10 | #include "AliRun.h" | |
11 | #include "AliRunLoader.h" | |
12 | #include "AliTPCLoader.h" | |
13 | #include "AliESD.h" | |
14 | #include "TFile.h" | |
15 | #include "TStopwatch.h" | |
16 | #endif | |
17 | ||
18 | extern AliRun *gAlice; | |
19 | ||
20 | ||
21 | Int_t AliTPCFindTracksMI(Int_t N=-1) { | |
22 | ||
23 | cerr<<"Looking for tracks...\n"; | |
24 | ||
25 | if (gAlice) | |
26 | { | |
27 | delete AliRunLoader::Instance(); | |
28 | delete gAlice; | |
29 | gAlice = 0x0; | |
30 | } | |
31 | ||
32 | AliRunLoader *rl = AliRunLoader::Open("galice.root"); | |
33 | if (rl == 0x0) | |
34 | { | |
35 | cerr<<"Can not open session"<<endl; | |
36 | return 1; | |
37 | } | |
38 | AliTPCLoader *tpcl = (AliTPCLoader*)rl->GetLoader("TPCLoader"); | |
39 | if (tpcl == 0x0) | |
40 | { | |
41 | cerr<<"Can not get TPC Loader"<<endl; | |
42 | return 1; | |
43 | } | |
44 | ||
45 | if (rl->LoadgAlice()) | |
46 | { | |
47 | cerr<<"Error occured while l"<<endl; | |
48 | return 1; | |
49 | } | |
50 | AliKalmanTrack::SetFieldMap(rl->GetAliRun()->Field()); | |
51 | ||
52 | rl->CdGAFile(); | |
53 | ||
54 | AliTPCParam *param=(AliTPCParam *)gDirectory->Get("75x40_100x60_150x60"); | |
55 | if (!param) | |
56 | { | |
57 | param=(AliTPCParam *)gDirectory->Get("75x40_100x60"); | |
58 | if (!param) | |
59 | { | |
60 | cerr<<"TPC parameters have not been found !\n"; | |
61 | return 1; | |
62 | } | |
63 | else | |
64 | { | |
65 | cout<<"TPC 75x40_100x60 geometry found"<<endl; | |
66 | } | |
67 | } | |
68 | else | |
69 | { | |
70 | cout<<"TPC 75x40_100x60_150x60 geometry found"<<endl; | |
71 | } | |
72 | ||
73 | ||
74 | tpcl->LoadTracks("recreate"); | |
75 | ||
76 | Int_t eventn; | |
77 | if (N<=0) | |
78 | { | |
79 | eventn = rl->GetNumberOfEvents(); | |
80 | rl->UnloadHeader(); | |
81 | } | |
82 | else | |
83 | eventn = N; | |
84 | ||
85 | TStopwatch timer; | |
86 | Int_t rc=0; | |
87 | for (Int_t i=0;i<eventn;i++) | |
88 | { | |
89 | rl->GetEvent(i); | |
90 | TTree * input = tpcl->TreeR(); | |
91 | if (input == 0x0) | |
92 | { | |
93 | tpcl->LoadRecPoints("read"); | |
94 | input = tpcl->TreeR(); | |
95 | if (input == 0x0) | |
96 | { | |
97 | cerr << "Problems with input tree (TreeR) for event " << i <<endl; | |
98 | continue; | |
99 | } | |
100 | } | |
101 | TTree * output = tpcl->TreeT(); | |
102 | if (output == 0x0) | |
103 | { | |
104 | tpcl->MakeTree("T"); | |
105 | output = tpcl->TreeT(); | |
106 | if (output == 0x0) | |
107 | { | |
108 | cerr << "Problems with output tree (TreeT) for event " << i <<endl; | |
109 | continue; | |
110 | } | |
111 | } | |
112 | ||
113 | printf("Processing event %d\n",i); | |
114 | AliTPCtrackerMI *tracker = new AliTPCtrackerMI(param); | |
115 | tracker->SetIO(); | |
116 | tracker->LoadClusters(); | |
117 | rc=tracker->Clusters2Tracks(); | |
118 | tracker->WriteTracks(output); | |
119 | tracker->UnloadClusters(); | |
120 | tpcl->WriteTracks("OVERWRITE"); | |
121 | //output->GetDirectory()->cd(); | |
122 | //output->Write(); | |
123 | delete tracker; | |
124 | } | |
125 | timer.Stop(); timer.Print(); | |
126 | rl->UnloadgAlice(); | |
127 | ||
128 | delete param; | |
129 | delete rl; | |
130 | return rc; | |
131 | } |