]>
Commit | Line | Data |
---|---|---|
a654e257 | 1 | void AliBarrelRec_TPCparam(Int_t firstEvent=0,Int_t lastEvent=0) { |
056891c0 | 2 | // |
a654e257 | 3 | // Macro to create AliESDs.root using parametrized TPC tracking |
4 | // and AliITStrackerV2 | |
056891c0 | 5 | // |
a654e257 | 6 | // Input files: |
7 | // - galice.root | |
8 | // - Kinematics.root | |
9 | // - TrackReferences.root | |
10 | // - ITS.RecPoints.root (use AliRecontruction class) | |
11 | // - ITS.Vertex.root (use $ALICE_ROOT/ITS/AliITSVertexerZTest.C) | |
e0cf7c8a | 12 | // |
a654e257 | 13 | // A. Dainese - LNL |
e0cf7c8a | 14 | // |
056891c0 | 15 | |
056891c0 | 16 | |
a654e257 | 17 | /**** Initialization of the NewIO *******/ |
18 | ||
19 | if (gAlice) { | |
20 | delete gAlice->GetRunLoader(); | |
21 | delete gAlice; | |
22 | gAlice=0; | |
23 | } | |
24 | ||
25 | AliRunLoader *rl = AliRunLoader::Open("galice.root"); | |
26 | if (rl == 0x0) { | |
27 | cerr<<"Can not open session"<<endl; | |
28 | return; | |
29 | } | |
30 | Int_t retval = rl->LoadgAlice(); | |
31 | if (retval) { | |
32 | cerr<<"LoadgAlice returned error"<<endl; | |
33 | delete rl; | |
34 | return; | |
35 | } | |
36 | retval = rl->LoadHeader(); | |
37 | if (retval) { | |
38 | cerr<<"LoadHeader returned error"<<endl; | |
39 | delete rl; | |
40 | return; | |
41 | } | |
42 | gAlice=rl->GetAliRun(); | |
43 | ||
44 | ||
45 | TDatabasePDG *DataBase = TDatabasePDG::Instance(); | |
46 | ||
47 | // Get field from galice.root | |
48 | AliMagF *fiel = (AliMagF*)gAlice->Field(); | |
49 | Double_t fieval=TMath::Abs((Double_t)fiel->SolenoidField()/10.); | |
50 | // Set the conversion constant between curvature and Pt | |
51 | AliTracker::SetFieldMap(fiel,kTRUE); | |
52 | ||
53 | /**** The TPC corner ********************/ | |
54 | ||
55 | Int_t collcode = 1; // pp collisions | |
56 | AliTPCtrackerParam tpcTrackerPar(collcode,fieval); | |
57 | tpcTrackerPar.Init(); | |
58 | ||
59 | //**** Switch on the PID class (mandatory!) | |
60 | AliPID pid; | |
61 | ||
62 | /**** The ITS corner ********************/ | |
63 | ||
64 | AliITSLoader* itsl = (AliITSLoader*)rl->GetLoader("ITSLoader"); | |
65 | if (itsl == 0x0) { | |
66 | cerr<<"Cannot get the ITS loader"<<endl; | |
67 | return; | |
68 | } | |
69 | itsl->LoadRecPoints("read"); | |
70 | ||
71 | AliITS *dITS = (AliITS*)gAlice->GetDetector("ITS"); | |
72 | if (!dITS) { | |
73 | cerr<<"Cannot find the ITS detector !"<<endl; | |
74 | return; | |
75 | } | |
76 | AliITSgeom *geom = dITS->GetITSgeom(); | |
77 | ||
78 | //An instance of the ITS tracker | |
79 | AliITStrackerV2 itsTracker(geom); | |
80 | Int_t ITSclusters[6] = {1,1,1,1,1,1}; | |
81 | itsTracker.SetLayersNotToSkip(ITSclusters); | |
82 | ||
83 | /***** The TREE is born *****/ | |
84 | ||
85 | TTree *esdTree=new TTree("esdTree","Tree with ESD objects"); | |
86 | AliESD *event=0; | |
87 | esdTree->Branch("ESD","AliESD",&event); | |
88 | ||
89 | if(firstEvent>rl->GetNumberOfEvents()) firstEvent=rl->GetNumberOfEvents()-1; | |
90 | if(lastEvent>rl->GetNumberOfEvents()) lastEvent=rl->GetNumberOfEvents()-1; | |
91 | cout<<" Number of events: "<<1+lastEvent-firstEvent<<endl; | |
92 | ||
93 | TFile *ppZ = TFile::Open("ITS.Vertex.root"); // z vertices from SPD | |
94 | AliESDVertex *myvertex = new AliESDVertex(); | |
95 | Char_t zver[100]; | |
96 | Double_t vtx[3]; | |
97 | Double_t cvtx[6]; | |
98 | ||
99 | ||
100 | //<----------------------------------The Loop over events begins | |
101 | TStopwatch timer; | |
102 | Int_t trc; | |
103 | for(Int_t i=firstEvent; i<=lastEvent; i++) { | |
104 | ||
105 | cerr<<" Processing event number : "<<i<<endl; | |
106 | AliESD *event = new AliESD(); | |
107 | event->SetRunNumber(gAlice->GetRunNumber()); | |
108 | event->SetEventNumber(i); | |
109 | event->SetMagneticField(gAlice->Field()->SolenoidField()); | |
110 | rl->GetEvent(i); | |
111 | ||
112 | //***** Primary vertex | |
113 | sprintf(zver,"Event%d/Vertex",i); | |
114 | myvertex = (AliESDVertex*)ppZ->Get(zver); | |
115 | if(!myvertex) { | |
116 | esdTree->Fill(); delete event; | |
117 | continue; | |
118 | } | |
119 | event->SetVertex(myvertex); | |
120 | myvertex->GetXYZ(vtx); | |
121 | myvertex->GetCovMatrix(cvtx); | |
122 | ||
123 | //***** TPC tracking | |
124 | if ( (trc=tpcTrackerPar.BuildTPCtracks(event)) ) { | |
125 | printf("exiting TPC tracker with code %d in event %d\n",trc,i); | |
126 | esdTree->Fill(); delete event; | |
127 | continue; | |
128 | } | |
129 | ||
130 | //***** ITS tracking | |
131 | itsTracker.SetVertex(vtx,cvtx); | |
132 | TTree *itsTree=itsl->TreeR(); | |
133 | if (!itsTree) { | |
134 | cerr<<"Can't get the ITS cluster tree !\n"; | |
135 | esdTree->Fill(); delete event; | |
136 | return; | |
137 | } | |
138 | itsTracker.UnloadClusters(); | |
139 | itsTracker.LoadClusters(itsTree); | |
140 | if ( (trc=itsTracker.Clusters2Tracks(event)) ) { | |
141 | printf("exiting ITS tracker with code %d in event %d\n",trc,i); | |
142 | esdTree->Fill(); delete event; | |
143 | continue; | |
144 | } | |
145 | ||
146 | ||
147 | esdTree->Fill(); | |
148 | delete event; | |
149 | ||
150 | }//<-----------------------------------The Loop over events ends here | |
151 | timer.Stop(); timer.Print(); | |
152 | ||
153 | // The AliESDs.root is born | |
154 | TFile *ef = TFile::Open("AliESDs.root","RECREATE"); | |
155 | if (!ef || !ef->IsOpen()) {cerr<<"Can't open AliESDs.root !\n"; return;} | |
156 | ||
157 | esdTree->Write(); //Write the TREE and close everything | |
158 | delete esdTree; | |
159 | ef->Close(); | |
160 | ||
161 | delete rl; | |
056891c0 | 162 | |
e0cf7c8a | 163 | return; |
056891c0 | 164 | } |