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