]> git.uio.no Git - u/mrichter/AliRoot.git/blame - ITS/AliITS.cxx
Make the code compatible with the new AliRun
[u/mrichter/AliRoot.git] / ITS / AliITS.cxx
CommitLineData
4c039060 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/*
17$Log$
340cd0d8 18Revision 1.34 2001/02/11 15:51:39 mariana
19Set protection in MakeBranch
20
c34b4885 21Revision 1.33 2001/02/10 22:26:39 mariana
22Move the initialization of the containers for raw clusters in MakeTreeC()
23
caf37aec 24Revision 1.32 2001/02/08 23:55:31 nilsen
25Removed fMajor/MinorVersion variables in favor of variables in derived classes.
26Set arrays char *det[3] = {"SPD","SDD","SSD"} as const.
27
04874c1a 28Revision 1.31 2001/02/02 23:57:28 nilsen
29Added include file that are no londer included in AliITSgeom.h
30
d953664a 31Revision 1.30 2001/01/30 09:23:13 hristov
32Streamers removed (R.Brun)
33
a8a6107b 34Revision 1.29 2001/01/26 20:01:09 hristov
35Major upgrade of AliRoot code
36
2ab0c725 37Revision 1.28 2000/12/18 14:02:00 barbera
38new version of the ITS tracking to take into account the new TPC track parametrization
39
cc6e75dd 40Revision 1.27 2000/12/08 13:49:27 barbera
41Hidden declaration in a for loop removed to be compliant with HP-UX compiler
42
c5b22044 43Revision 1.26 2000/11/27 13:12:13 barbera
44New version containing the files for tracking
45
55b5a0b3 46Revision 1.25 2000/11/12 22:38:05 barbera
47Added header file for the SPD Bari model
48
dfe2a93f 49Revision 1.24 2000/10/09 22:18:12 barbera
50Bug fixes from MAriana to le AliITStest.C run correctly
51
65d4384f 52Revision 1.23 2000/10/05 20:47:42 nilsen
53fixed dependencies of include files. Tryed but failed to get a root automaticly
54generates streamer function to work. Modified SetDefaults.
55
143d1056 56Revision 1.9.2.15 2000/10/04 16:56:40 nilsen
57Needed to include stdlib.h
58
59=======
60Revision 1.22 2000/10/04 19:45:52 barbera
61Corrected by F. Carminati for v3.04
62
114da3a4 63Revision 1.21 2000/10/02 21:28:08 fca
64Removal of useless dependecies via forward declarations
65
94de3818 66Revision 1.20 2000/10/02 16:31:39 barbera
67General code clean-up
68
d5da1ecf 69Revision 1.9.2.14 2000/10/02 15:43:51 barbera
70General code clean-up (e.g., printf -> cout)
71
72Revision 1.19 2000/09/22 12:13:25 nilsen
73Patches and updates for fixes to this and other routines.
74
9c74c52b 75Revision 1.18 2000/07/12 05:32:20 fca
76Correcting several syntax problem with static members
77
b669392e 78Revision 1.17 2000/07/10 16:07:18 fca
79Release version of ITS code
80
3bd79107 81Revision 1.9.2.3 2000/02/02 13:42:09 barbera
82fixed AliITS.cxx for new AliRun structure. Added ITS hits list to list of hits which will have their track numbers updated
83
84Revision 1.9.2.2 2000/01/23 03:03:13 nilsen
85//fixed FillModule. Removed fi(fabs(xl)<dx....
86
87Revision 1.9.2.1 2000/01/12 19:03:32 nilsen
88This is the version of the files after the merging done in December 1999.
89See the ReadMe110100.txt file for details
c9a71be1 90
1cedd08a 91Revision 1.9 1999/11/14 14:33:25 fca
92Correct problems with distructors and pointers, thanks to I.Hrivnacova
93
6c854497 94Revision 1.8 1999/09/29 09:24:19 fca
95Introduction of the Copyright and cvs Log
96
4c039060 97*/
98
fe4da5cc 99///////////////////////////////////////////////////////////////////////////////
58005f18 100//
101// An overview of the basic philosophy of the ITS code development
102// and analysis is show in the figure below.
fe4da5cc 103//Begin_Html
104/*
a92b2b7d 105<img src="picts/ITS/ITS_Analysis_schema.gif">
fe4da5cc 106</pre>
107<br clear=left>
108<font size=+2 color=red>
58005f18 109<p>Roberto Barbera is in charge of the ITS Offline code (1999).
fe4da5cc 110<a href="mailto:roberto.barbera@ct.infn.it">Roberto Barbera</a>.
111</font>
112<pre>
113*/
114//End_Html
58005f18 115//
116// AliITS. Inner Traking System base class.
117// This class contains the base procedures for the Inner Tracking System
118//
119//Begin_Html
120/*
a92b2b7d 121<img src="picts/ITS/AliITS_Class_Diagram.gif">
58005f18 122</pre>
123<br clear=left>
124<font size=+2 color=red>
125<p>This show the class diagram of the different elements that are part of
126the AliITS class.
127</font>
128<pre>
129*/
130//End_Html
131//
132// Version: 0
133// Written by Rene Brun, Federico Carminati, and Roberto Barbera
134//
135// Version: 1
136// Modified and documented by Bjorn S. Nilsen
137// July 11 1999
138//
3bd79107 139// Version: 2
140// Modified and documented by A. Bologna
141// October 18 1999
142//
58005f18 143// AliITS is the general base class for the ITS. Also see AliDetector for
144// futher information.
145//
fe4da5cc 146///////////////////////////////////////////////////////////////////////////////
d953664a 147#include <iostream.h>
148#include <iomanip.h>
149#include <fstream.h>
143d1056 150#include <stdlib.h>
fe4da5cc 151#include <TMath.h>
152#include <TRandom.h>
9c74c52b 153#include <TBranch.h>
fe4da5cc 154#include <TVector.h>
caf37aec 155#include <TClonesArray.h>
3bd79107 156#include <TROOT.h>
157#include <TObjectTable.h>
9c74c52b 158#include <TFile.h>
159#include <TTree.h>
d5da1ecf 160#include <TString.h>
3bd79107 161
162
bb8ac1e0 163
fe4da5cc 164#include "AliRun.h"
3bd79107 165#include "AliITS.h"
166#include "AliITSMap.h"
e8189707 167#include "AliITSDetType.h"
3bd79107 168#include "AliITSClusterFinder.h"
caf37aec 169//#include "AliITSsimulation.h"
170#include "AliITSsimulationSPD.h"
171#include "AliITSsimulationSDD.h"
172#include "AliITSsimulationSSD.h"
143d1056 173#include "AliITSresponse.h"
3bd79107 174#include "AliITSsegmentationSPD.h"
175#include "AliITSresponseSPD.h"
dfe2a93f 176#include "AliITSresponseSPDbari.h"
3bd79107 177#include "AliITSsegmentationSDD.h"
178#include "AliITSresponseSDD.h"
179#include "AliITSsegmentationSSD.h"
180#include "AliITSresponseSSD.h"
9c74c52b 181#include "AliITShit.h"
182#include "AliITSgeom.h"
183#include "AliITSdigit.h"
184#include "AliITSmodule.h"
185#include "AliITSRecPoint.h"
186#include "AliITSRawCluster.h"
94de3818 187#include "AliMC.h"
114da3a4 188#include "stdlib.h"
e8189707 189
55b5a0b3 190#include "AliITStrack.h"
191#include "AliITSiotrack.h"
192#include "AliITStracking.h"
193#include "../TPC/AliTPC.h"
194#include "../TPC/AliTPCParam.h"
195
c9a71be1 196
3bd79107 197ClassImp(AliITS)
198
fe4da5cc 199//_____________________________________________________________________________
3bd79107 200AliITS::AliITS() : AliDetector() {
fe4da5cc 201 //
202 // Default initialiser for ITS
58005f18 203 // The default constructor of the AliITS class. In addition to
204 // creating the AliITS class it zeros the variables fIshunt (a member
205 // of AliDetector class), fEuclidOut, and fIdN, and zeros the pointers
3bd79107 206 // fITSpoints, fIdSens, and fIdName. The AliDetector default constructor
207 // is also called.
fe4da5cc 208 //
3bd79107 209
9c74c52b 210
58005f18 211 fIshunt = 0;
212 fEuclidOut = 0;
e8189707 213
a8a6107b 214 fNDetTypes = kNTYPES;
58005f18 215 fIdN = 0;
216 fIdName = 0;
217 fIdSens = 0;
6c854497 218 fITSmodules = 0;
3bd79107 219 //
220 fDetTypes = 0;
3bd79107 221 //
222 fDtype = 0;
223 fNdtype = 0;
224 fCtype = 0;
225 fNctype = 0;
226 fRecPoints = 0;
227 fNRecPoints = 0;
e8189707 228 fTreeC = 0;
3bd79107 229 //
230 fITSgeom=0;
fe4da5cc 231}
3bd79107 232
fe4da5cc 233//_____________________________________________________________________________
58005f18 234AliITS::AliITS(const char *name, const char *title):AliDetector(name,title){
fe4da5cc 235 //
236 // Default initialiser for ITS
58005f18 237 // The constructor of the AliITS class. In addition to creating the
238 // AliITS class, it allocates memory for the TClonesArrays fHits and
239 // fDigits, and for the TObjArray fITSpoints. It also zeros the variables
240 // fIshunt (a member of AliDetector class), fEuclidOut, and fIdN, and zeros
241 // the pointers fIdSens and fIdName. To help in displaying hits via the ROOT
242 // macro display.C AliITS also sets the marker color to red. The variables
243 // passes with this constructor, const char *name and *title, are used by
244 // the constructor of AliDetector class. See AliDetector class for a
245 // description of these parameters and its constructor functions.
fe4da5cc 246 //
58005f18 247
9c74c52b 248
58005f18 249 fHits = new TClonesArray("AliITShit", 1560);
1cedd08a 250 gAlice->AddHitList(fHits);
3bd79107 251
a8a6107b 252 fNDetTypes = kNTYPES;
e8189707 253
a8a6107b 254 fNdtype = new Int_t[kNTYPES];
255 fDtype = new TObjArray(kNTYPES);
e8189707 256
a8a6107b 257 fNctype = new Int_t[kNTYPES];
258 fCtype = new TObjArray(kNTYPES);
3bd79107 259
3bd79107 260
261 fRecPoints = 0;
262 fNRecPoints = 0;
263
e8189707 264 fTreeC = 0;
3bd79107 265
266 fITSmodules = 0;
58005f18 267
268 fIshunt = 0;
269 fEuclidOut = 0;
270 fIdN = 0;
271 fIdName = 0;
272 fIdSens = 0;
3bd79107 273
a8a6107b 274 fDetTypes = new TObjArray(kNTYPES);
3bd79107 275
276 Int_t i;
a8a6107b 277 for(i=0;i<kNTYPES;i++) {
3bd79107 278 (*fDetTypes)[i]=new AliITSDetType();
e8189707 279 fNdtype[i]=0;
280 fNctype[i]=0;
3bd79107 281 }
282 //
58005f18 283
fe4da5cc 284 SetMarkerColor(kRed);
58005f18 285
3bd79107 286 fITSgeom=0;
287}
288//___________________________________________________________________________
289AliITS::AliITS(AliITS &source){
e8189707 290 // copy constructor
3bd79107 291 if(this==&source) return;
d5da1ecf 292 Error("AliITS::Copy constructor",
293 "You are not allowed to make a copy of the AliITS");
3bd79107 294 exit(1);
fe4da5cc 295}
3bd79107 296//____________________________________________________________________________
297AliITS& AliITS::operator=(AliITS &source){
e8189707 298 // assignment operator
3bd79107 299 if(this==&source) return *this;
d5da1ecf 300 Error("AliITS::operator=",
301 "You are not allowed to make a copy of the AliITS");
3bd79107 302 exit(1);
e8189707 303 return *this; //fake return
3bd79107 304}
305//____________________________________________________________________________
306void AliITS::ClearModules(){
307 //clear the modules TObjArray
3bd79107 308
9c74c52b 309 if(fITSmodules) fITSmodules->Delete();
fe4da5cc 310
3bd79107 311}
fe4da5cc 312//_____________________________________________________________________________
58005f18 313AliITS::~AliITS(){
fe4da5cc 314 //
315 // Default distructor for ITS
58005f18 316 // The default destructor of the AliITS class. In addition to deleting
317 // the AliITS class it deletes the memory pointed to by the fHits, fDigits,
318 // fIdSens, fIdName, and fITSpoints.
fe4da5cc 319 //
3bd79107 320
321
fe4da5cc 322 delete fHits;
323 delete fDigits;
3bd79107 324 delete fRecPoints;
143d1056 325// delete fIdName; // TObjArray of TObjStrings
326 if(fIdName!=0) delete[] fIdName; // Array of TStrings
58005f18 327 if(fIdSens!=0) delete[] fIdSens;
3bd79107 328 if(fITSmodules!=0) {
329 this->ClearModules();
330 delete fITSmodules;
331 }// end if fITSmodules!=0
fe4da5cc 332
fe4da5cc 333 //
3bd79107 334 if(fDtype) {
9c74c52b 335 fDtype->Delete();
336 delete fDtype;
3bd79107 337 }
9c74c52b 338 delete [] fNdtype;
339 if (fCtype) {
340 fCtype->Delete();
341 delete fCtype;
3bd79107 342 }
9c74c52b 343 delete [] fNctype;
fe4da5cc 344 //
3bd79107 345
346 if (fDetTypes) {
347 fDetTypes->Delete();
348 delete fDetTypes;
349 }
350
351 if (fTreeC) delete fTreeC;
e8189707 352
9c74c52b 353 if (fITSgeom) delete fITSgeom;
354
3bd79107 355}
356
357//___________________________________________
358AliITSDetType* AliITS::DetType(Int_t id)
359{
360 //return pointer to id detector type
361 return ((AliITSDetType*) (*fDetTypes)[id]);
362
363}
364//___________________________________________
e8189707 365void AliITS::SetClasses(Int_t id, const char *digit, const char *cluster)
3bd79107 366{
367 //set the digit and cluster classes to be used for the id detector type
368 ((AliITSDetType*) (*fDetTypes)[id])->ClassNames(digit,cluster);
369
370}
371//___________________________________________
372void AliITS::SetResponseModel(Int_t id, AliITSresponse *response)
373{
374 //set the response model for the id detector type
375
376 ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response);
377
fe4da5cc 378}
379
3bd79107 380//___________________________________________
381void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg)
382{
383 //set the segmentation model for the id detector type
384
385 ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg);
386
387}
388
389//___________________________________________
390void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim)
391{
392 //set the simulation model for the id detector type
393
394 ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim);
395
396}
397//___________________________________________
398void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst)
399{
400 //set the cluster finder model for the id detector type
401
402 ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst);
58005f18 403
58005f18 404}
405
fe4da5cc 406//_____________________________________________________________________________
58005f18 407void AliITS::AddHit(Int_t track, Int_t *vol, Float_t *hits){
fe4da5cc 408 //
409 // Add an ITS hit
58005f18 410 // The function to add information to the AliITShit class. See the
411 // AliITShit class for a full description. This function allocates the
412 // necessary new space for the hit information and passes the variable
413 // track, and the pointers *vol and *hits to the AliITShit constructor
414 // function.
fe4da5cc 415 //
416 TClonesArray &lhits = *fHits;
417 new(lhits[fNhits++]) AliITShit(fIshunt,track,vol,hits);
418}
3bd79107 419//_____________________________________________________________________________
420void AliITS::AddRealDigit(Int_t id, Int_t *digits)
421{
422 // add a real digit - as coming from data
423
424 TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
425 new(ldigits[fNdtype[id]++]) AliITSdigit(digits);
426
427}
428//_____________________________________________________________________________
e8189707 429void AliITS::AddSimDigit(Int_t id, AliITSdigit *d)
3bd79107 430{
431
432 // add a simulated digit
433
3bd79107 434 TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
435
436 switch(id)
437 {
438 case 0:
439 new(ldigits[fNdtype[id]++]) AliITSdigitSPD(*((AliITSdigitSPD*)d));
440 break;
441 case 1:
442 new(ldigits[fNdtype[id]++]) AliITSdigitSDD(*((AliITSdigitSDD*)d));
443 break;
444 case 2:
445 new(ldigits[fNdtype[id]++]) AliITSdigitSSD(*((AliITSdigitSSD*)d));
446 break;
447 }
448
449}
450
451//_____________________________________________________________________________
e8189707 452void AliITS::AddSimDigit(Int_t id,Float_t phys,Int_t *digits,Int_t *tracks,Int_t *hits,Float_t *charges){
3bd79107 453
454 // add a simulated digit to the list
455
456 TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
457 switch(id)
458 {
459 case 0:
e8189707 460 new(ldigits[fNdtype[id]++]) AliITSdigitSPD(digits,tracks,hits);
3bd79107 461 break;
462 case 1:
e8189707 463 new(ldigits[fNdtype[id]++]) AliITSdigitSDD(phys,digits,tracks,hits,charges);
3bd79107 464 break;
465 case 2:
e8189707 466 new(ldigits[fNdtype[id]++]) AliITSdigitSSD(digits,tracks,hits);
3bd79107 467 break;
468 }
469
470}
471
472//_____________________________________________________________________________
473void AliITS::AddCluster(Int_t id, AliITSRawCluster *c)
474{
475
476 // add a cluster to the list
477
3bd79107 478 TClonesArray &lcl = *((TClonesArray*)(*fCtype)[id]);
479
480 switch(id)
481 {
482 case 0:
483 new(lcl[fNctype[id]++]) AliITSRawClusterSPD(*((AliITSRawClusterSPD*)c));
484 break;
485 case 1:
486 new(lcl[fNctype[id]++]) AliITSRawClusterSDD(*((AliITSRawClusterSDD*)c));
487 break;
488 case 2:
489 new(lcl[fNctype[id]++]) AliITSRawClusterSSD(*((AliITSRawClusterSSD*)c));
490 break;
491 }
492
493}
494
495
496//_____________________________________________________________________________
497void AliITS::AddRecPoint(const AliITSRecPoint &r)
498{
499 //
500 // Add a reconstructed space point to the list
501 //
502 TClonesArray &lrecp = *fRecPoints;
503 new(lrecp[fNRecPoints++]) AliITSRecPoint(r);
504}
505
506
507//____________________________________________
508void AliITS::ResetDigits()
509{
510 //
9c74c52b 511 // Reset number of digits and the digits array for the ITS detector
3bd79107 512 //
513
514 if (!fDtype) return;
515
516 Int_t i;
a8a6107b 517 for (i=0;i<kNTYPES;i++ ) {
3bd79107 518 if ((*fDtype)[i]) ((TClonesArray*)(*fDtype)[i])->Clear();
519 if (fNdtype) fNdtype[i]=0;
520 }
521}
522
523//____________________________________________
524void AliITS::ResetDigits(Int_t i)
525{
526 //
527 // Reset number of digits and the digits array for this branch
528 //
e8189707 529 if ((*fDtype)[i]) ((TClonesArray*)(*fDtype)[i])->Clear();
530 if (fNdtype) fNdtype[i]=0;
3bd79107 531}
e8189707 532
533
3bd79107 534//____________________________________________
535void AliITS::ResetClusters()
536{
537 //
538 // Reset number of clusters and the clusters array for ITS
539 //
540
541 Int_t i;
a8a6107b 542 for (i=0;i<kNTYPES;i++ ) {
3bd79107 543 if ((*fCtype)[i]) ((TClonesArray*)(*fCtype)[i])->Clear();
544 if (fNctype) fNctype[i]=0;
545 }
546
547}
e8189707 548
3bd79107 549//____________________________________________
550void AliITS::ResetClusters(Int_t i)
551{
552 //
553 // Reset number of clusters and the clusters array for this branch
554 //
555 if ((*fCtype)[i]) ((TClonesArray*)(*fCtype)[i])->Clear();
556 if (fNctype) fNctype[i]=0;
557
558}
559
560
561//____________________________________________
562void AliITS::ResetRecPoints()
563{
564 //
565 // Reset number of rec points and the rec points array
566 //
3bd79107 567 if (fRecPoints) fRecPoints->Clear();
9c74c52b 568 fNRecPoints = 0;
3bd79107 569
570}
571
fe4da5cc 572//_____________________________________________________________________________
58005f18 573Int_t AliITS::DistancetoPrimitive(Int_t , Int_t ){
fe4da5cc 574 //
575 // Distance from mouse to ITS on the screen. Dummy routine
58005f18 576 // A dummy routine used by the ROOT macro display.C to allow for the
577 // use of the mouse (pointing device) in the macro. In general this should
578 // never be called. If it is it returns the number 9999 for any value of
579 // x and y.
fe4da5cc 580 //
581 return 9999;
582}
583
584//_____________________________________________________________________________
58005f18 585void AliITS::Init(){
fe4da5cc 586 //
587 // Initialise ITS after it has been built
58005f18 588 // This routine initializes the AliITS class. It is intended to be called
589 // from the Init function in AliITSv?. Besides displaying a banner
3bd79107 590 // indicating that it has been called it initializes the array fIdSens
591 // and sets the default segmentation, response, digit and raw cluster classes
58005f18 592 // Therefore it should be called after a call to CreateGeometry.
fe4da5cc 593 //
594 Int_t i;
143d1056 595
143d1056 596//
597 SetDefaults();
143d1056 598// Array of TStrings
58005f18 599 for(i=0;i<fIdN;i++) fIdSens[i] = gMC->VolId(fIdName[i]);
143d1056 600//
fe4da5cc 601}
602
3bd79107 603//_____________________________________________________________________________
604void AliITS::SetDefaults()
605{
606 // sets the default segmentation, response, digit and raw cluster classes
607
65d4384f 608 printf("SetDefaults\n");
3bd79107 609
610 AliITSDetType *iDetType;
611
65d4384f 612
3bd79107 613 //SPD
614
65d4384f 615 iDetType=DetType(0);
143d1056 616 if (!iDetType->GetSegmentationModel()) {
65d4384f 617 AliITSsegmentationSPD *seg0=new AliITSsegmentationSPD(fITSgeom);
618 SetSegmentationModel(0,seg0);
619 }
620 if (!iDetType->GetResponseModel()) {
621 SetResponseModel(0,new AliITSresponseSPD());
622 }
3bd79107 623 // set digit and raw cluster classes to be used
65d4384f 624
625 const char *kData0=(iDetType->GetResponseModel())->DataType();
e8189707 626 if (strstr(kData0,"real")) {
3bd79107 627 iDetType->ClassNames("AliITSdigit","AliITSRawClusterSPD");
628 } else iDetType->ClassNames("AliITSdigitSPD","AliITSRawClusterSPD");
629
65d4384f 630 // SDD //
631 iDetType=DetType(1);
632 if (!iDetType->GetResponseModel()) {
633 SetResponseModel(1,new AliITSresponseSDD());
634 }
635 AliITSresponse *resp1=iDetType->GetResponseModel();
143d1056 636 if (!iDetType->GetSegmentationModel()) {
65d4384f 637 AliITSsegmentationSDD *seg1=new AliITSsegmentationSDD(fITSgeom,resp1);
143d1056 638 SetSegmentationModel(1,seg1);
143d1056 639 }
65d4384f 640 const char *kData1=(iDetType->GetResponseModel())->DataType();
641 const char *kopt=iDetType->GetResponseModel()->ZeroSuppOption();
e8189707 642 if ((!strstr(kopt,"2D")) && (!strstr(kopt,"1D")) || strstr(kData1,"real") ) {
3bd79107 643 iDetType->ClassNames("AliITSdigit","AliITSRawClusterSDD");
644 } else iDetType->ClassNames("AliITSdigitSDD","AliITSRawClusterSDD");
645
646 // SSD
65d4384f 647 iDetType=DetType(2);
143d1056 648 if (!iDetType->GetSegmentationModel()) {
65d4384f 649 AliITSsegmentationSSD *seg2=new AliITSsegmentationSSD(fITSgeom);
143d1056 650 SetSegmentationModel(2,seg2);
65d4384f 651 }
652 if (!iDetType->GetResponseModel()) {
653 SetResponseModel(2,new AliITSresponseSSD());
654 }
655 const char *kData2=(iDetType->GetResponseModel())->DataType();
e8189707 656 if (strstr(kData2,"real")) {
3bd79107 657 iDetType->ClassNames("AliITSdigit","AliITSRawClusterSSD");
658 } else iDetType->ClassNames("AliITSdigitSSD","AliITSRawClusterSSD");
659
a8a6107b 660 if (kNTYPES>3) {
3bd79107 661 Warning("SetDefaults","Only the three basic detector types are initialised!");
65d4384f 662 }
3bd79107 663
664}
65d4384f 665
666
667
e8189707 668//_____________________________________________________________________________
669void AliITS::SetDefaultSimulation()
670{
671 // to be written
672
673}
674//_____________________________________________________________________________
675void AliITS::SetDefaultClusterFinders()
676{
677 // to be written
678
679}
3bd79107 680//_____________________________________________________________________________
681
682void AliITS::MakeTreeC(Option_t *option)
683{
684 // create a separate tree to store the clusters
685
d5da1ecf 686 cout << "AliITS::MakeTreeC" << endl;
9c74c52b 687
3bd79107 688 char *optC = strstr(option,"C");
689 if (optC && !fTreeC) fTreeC = new TTree("TC","Clusters in ITS");
9c74c52b 690 else return;
3bd79107 691
692 Int_t buffersize = 4000;
693 char branchname[30];
694
04874c1a 695 const char *det[3] = {"SPD","SDD","SSD"};
3bd79107 696
caf37aec 697 char digclass[40];
698 char clclass[40];
699
3bd79107 700 // one branch for Clusters per type of detector
701 Int_t i;
a8a6107b 702 for (i=0; i<kNTYPES ;i++) {
caf37aec 703 AliITSDetType *iDetType=DetType(i);
704 iDetType->GetClassNames(digclass,clclass);
caf37aec 705 // clusters
706 (*fCtype)[i] = new TClonesArray(clclass,10000);
a8a6107b 707 if (kNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
3bd79107 708 else sprintf(branchname,"%sClusters%d",GetName(),i+1);
709 if (fCtype && fTreeC) {
710 TreeC()->Branch(branchname,&((*fCtype)[i]), buffersize);
d5da1ecf 711 cout << "Making Branch " << branchname;
712 cout << " for Clusters of detector type " << i+1 << endl;
3bd79107 713 }
714 }
715
716}
717
718//_____________________________________________________________________________
719void AliITS::GetTreeC(Int_t event)
720{
721
d5da1ecf 722 cout << "AliITS::GetTreeC" << endl;
9c74c52b 723
3bd79107 724 // get the clusters tree for this event and set the branch address
725 char treeName[20];
726 char branchname[30];
727
04874c1a 728 const char *det[3] = {"SPD","SDD","SSD"};
3bd79107 729
730 ResetClusters();
731 if (fTreeC) {
732 delete fTreeC;
733 }
734
735 sprintf(treeName,"TreeC%d",event);
736 fTreeC = (TTree*)gDirectory->Get(treeName);
737
3bd79107 738 TBranch *branch;
739 if (fTreeC) {
740 Int_t i;
a8a6107b 741 for (i=0; i<kNTYPES; i++) {
742 if (kNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
e8189707 743 else sprintf(branchname,"%sClusters%d",GetName(),i+1);
744 if (fCtype) {
3bd79107 745 branch = fTreeC->GetBranch(branchname);
746 if (branch) branch->SetAddress(&((*fCtype)[i]));
747 }
748 }
749 } else {
d5da1ecf 750 Error("AliITS::GetTreeC",
751 "cannot find Clusters Tree for event:%d\n",event);
3bd79107 752 }
753
754}
fe4da5cc 755//_____________________________________________________________________________
2ab0c725 756void AliITS::MakeBranch(Option_t* option, char *file)
757{
fe4da5cc 758 //
3bd79107 759 // Creates Tree branches for the ITS.
fe4da5cc 760 //
2ab0c725 761 //
fe4da5cc 762 Int_t buffersize = 4000;
3bd79107 763 char branchname[30];
fe4da5cc 764 sprintf(branchname,"%s",GetName());
765
2ab0c725 766 AliDetector::MakeBranch(option,file);
fe4da5cc 767
2ab0c725 768 char *cD = strstr(option,"D");
769 char *cR = strstr(option,"R");
fe4da5cc 770
2ab0c725 771 if (cD) {
772 //
773 // one branch for digits per type of detector
774 //
04874c1a 775 const char *det[3] = {"SPD","SDD","SSD"};
3bd79107 776
e8189707 777 char digclass[40];
778 char clclass[40];
3bd79107 779
780 Int_t i;
a8a6107b 781 for (i=0; i<kNTYPES ;i++) {
3bd79107 782 AliITSDetType *iDetType=DetType(i);
e8189707 783 iDetType->GetClassNames(digclass,clclass);
3bd79107 784 // digits
c34b4885 785 if(!((*fDtype)[i])) (*fDtype)[i] = new TClonesArray(digclass,10000);
786 else ResetDigits(i);
3bd79107 787 }
788
a8a6107b 789 for (i=0; i<kNTYPES ;i++) {
790 if (kNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
2ab0c725 791 else sprintf(branchname,"%sDigits%d",GetName(),i+1);
792 if (fDtype && gAlice->TreeD()) {
793 gAlice->MakeBranchInTree(gAlice->TreeD(),
c34b4885 794 branchname, &((*fDtype)[i]), buffersize, file);
2ab0c725 795 cout << "Making Branch " << branchname;
796 cout << " for digits of type "<< i+1 << endl;
3bd79107 797 }
2ab0c725 798 }
3bd79107 799 }
800
2ab0c725 801 if (cR) {
802 //
3bd79107 803 // only one branch for rec points for all detector types
2ab0c725 804 //
805 sprintf(branchname,"%sRecPoints",GetName());
806
340cd0d8 807 if(!fRecPoints) fRecPoints=new TClonesArray("AliITSRecPoint",10000);
3bd79107 808
2ab0c725 809 if (fRecPoints && gAlice->TreeR()) {
810 gAlice->MakeBranchInTree(gAlice->TreeR(),
811 branchname, &fRecPoints, buffersize, file) ;
812 cout << "Making Branch " << branchname;
813 cout << " for reconstructed space points" << endl;
814 }
3bd79107 815 }
3bd79107 816}
817
818//___________________________________________
819void AliITS::SetTreeAddress()
820{
821
822 // Set branch address for the Trees.
823
824 char branchname[30];
825 AliDetector::SetTreeAddress();
826
04874c1a 827 const char *det[3] = {"SPD","SDD","SSD"};
3bd79107 828
829 TBranch *branch;
830 TTree *treeD = gAlice->TreeD();
831 TTree *treeR = gAlice->TreeR();
832
833 Int_t i;
834 if (treeD) {
a8a6107b 835 for (i=0; i<kNTYPES; i++) {
836 if (kNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
3bd79107 837 else sprintf(branchname,"%sDigits%d",GetName(),i+1);
838 if (fDtype) {
839 branch = treeD->GetBranch(branchname);
840 if (branch) branch->SetAddress(&((*fDtype)[i]));
841 }
842 }
843 }
844
845
846 if (treeR) {
847 sprintf(branchname,"%sRecPoints",GetName());
3bd79107 848 branch = treeR->GetBranch(branchname);
849 if (branch) branch->SetAddress(&fRecPoints);
3bd79107 850 }
851
852
853}
854
855//____________________________________________________________________________
856void AliITS::InitModules(Int_t size,Int_t &nmodules){
857
858 //initialize the modules array
859
e8189707 860 if(fITSmodules){
9c74c52b 861 fITSmodules->Delete();
e8189707 862 delete fITSmodules;
863 }
864
3bd79107 865 Int_t nl,indexMAX,index;
3bd79107 866
867 if(size<=0){ // default to using data stored in AliITSgeom
868 if(fITSgeom==0) {
d5da1ecf 869 Error("AliITS::InitModules",
870 "in AliITS::InitModule fITSgeom not defined\n");
3bd79107 871 return;
872 } // end if fITSgeom==0
873 nl = fITSgeom->GetNlayers();
874 indexMAX = fITSgeom->GetModuleIndex(nl,fITSgeom->GetNladders(nl),
875 fITSgeom->GetNdetectors(nl))+1;
876 nmodules = indexMAX;
877 fITSmodules = new TObjArray(indexMAX);
3bd79107 878 for(index=0;index<indexMAX;index++){
3bd79107 879 fITSmodules->AddAt( new AliITSmodule(index),index);
3bd79107 880 } // end for index
881 }else{
882 fITSmodules = new TObjArray(size);
9c74c52b 883 for(index=0;index<size;index++) {
884 fITSmodules->AddAt( new AliITSmodule(index),index);
885 }
886
3bd79107 887 nmodules = size;
888 } // end i size<=0
889}
890
891//____________________________________________________________________________
e8189707 892void AliITS::FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,Option_t *option,Text_t *filename){
3bd79107 893
894 // fill the modules with the sorted by module hits; add hits from background
895 // if option=Add
896
9c74c52b 897
3bd79107 898 static TTree *trH1; //Tree with background hits
899 static TClonesArray *fHits2; //List of hits for one track only
900
901 static Bool_t first=kTRUE;
902 static TFile *file;
e8189707 903 char *addBgr = strstr(option,"Add");
3bd79107 904
905
e8189707 906 if (addBgr ) {
3bd79107 907 if(first) {
908 cout<<"filename "<<filename<<endl;
909 file=new TFile(filename);
910 cout<<"I have opened "<<filename<<" file "<<endl;
911 fHits2 = new TClonesArray("AliITShit",1000 );
912 }
913 first=kFALSE;
914 file->cd();
915 file->ls();
916 // Get Hits Tree header from file
917 if(fHits2) fHits2->Clear();
918 if(trH1) delete trH1;
919 trH1=0;
920
921 char treeName[20];
922 sprintf(treeName,"TreeH%d",bgrev);
923 trH1 = (TTree*)gDirectory->Get(treeName);
924 //printf("TrH1 %p of treename %s for event %d \n",trH1,treeName,bgrev);
925
926 if (!trH1) {
d5da1ecf 927 Error("AliITS::FillModules",
928 "cannot find Hits Tree for event:%d\n",bgrev);
3bd79107 929 }
930 // Set branch addresses
931 TBranch *branch;
932 char branchname[20];
933 sprintf(branchname,"%s",GetName());
934 if (trH1 && fHits2) {
935 branch = trH1->GetBranch(branchname);
936 if (branch) branch->SetAddress(&fHits2);
937 }
938
939 // test
940 //Int_t ntracks1 =(Int_t)TrH1->GetEntries();
941 //printf("background - ntracks1 - %d\n",ntracks1);
942 }
943
2ab0c725 944 //Int_t npart = gAlice->GetEvent(evnt);
945 //if(npart<=0) return;
3bd79107 946 TClonesArray *itsHits = this->Hits();
947 Int_t lay,lad,det,index;
948 AliITShit *itsHit=0;
949 AliITSmodule *mod=0;
950
951 TTree *iTH = gAlice->TreeH();
952 Int_t ntracks =(Int_t) iTH->GetEntries();
953
954 Int_t t,h;
955 for(t=0; t<ntracks; t++){
956 gAlice->ResetHits();
957 iTH->GetEvent(t);
958 Int_t nhits = itsHits->GetEntriesFast();
9c74c52b 959 //printf("nhits %d\n",nhits);
3bd79107 960 if (!nhits) continue;
3bd79107 961 for(h=0; h<nhits; h++){
962 itsHit = (AliITShit *)itsHits->UncheckedAt(h);
963 itsHit->GetDetectorID(lay,lad,det);
9c74c52b 964 // temporarily index=det-1 !!!
965 if(fITSgeom) index = fITSgeom->GetModuleIndex(lay,lad,det);
966 else index=det-1;
967 //
3bd79107 968 mod = this->GetModule(index);
9c74c52b 969 mod->AddHit(itsHit,t,h);
3bd79107 970 } // end loop over hits
971 } // end loop over tracks
972
973 // open the file with background
974
e8189707 975 if (addBgr ) {
976 Int_t track,i;
3bd79107 977 ntracks =(Int_t)trH1->GetEntries();
978 //printf("background - ntracks1 %d\n",ntracks);
979 //printf("background - Start loop over tracks \n");
980 // Loop over tracks
981
e8189707 982 for (track=0; track<ntracks; track++) {
3bd79107 983
984 if (fHits2) fHits2->Clear();
985 trH1->GetEvent(track);
986 // Loop over hits
987 for(i=0;i<fHits2->GetEntriesFast();++i) {
988
989 itsHit=(AliITShit*) (*fHits2)[i];
990 itsHit->GetDetectorID(lay,lad,det);
9c74c52b 991 // temporarily index=det-1 !!!
992 if(fITSgeom) index = fITSgeom->GetModuleIndex(lay,lad,det);
993 else index=det-1;
994 //
3bd79107 995 mod = this->GetModule(index);
9c74c52b 996 mod->AddHit(itsHit,track,i);
3bd79107 997 } // end loop over hits
998 } // end loop over tracks
999
1000 TTree *fAli=gAlice->TreeK();
1001 TFile *fileAli=0;
1002
1003 if (fAli) fileAli =fAli->GetCurrentFile();
9c74c52b 1004 fileAli->cd();
3bd79107 1005
1006 } // end if add
1007
3bd79107 1008 //gObjectTable->Print();
1009
1010}
1011
2ab0c725 1012//____________________________________________________________________________
1013
1014void AliITS::SDigits2Digits()
1015{
1016
1017 AliITSgeom *geom = GetITSgeom();
1018
1019 // SPD
1020 AliITSDetType *iDetType;
1021 iDetType=DetType(0);
1022 AliITSsegmentationSPD *seg0=(AliITSsegmentationSPD*)iDetType->GetSegmentationModel();
1023 AliITSresponseSPD *res0 = (AliITSresponseSPD*)iDetType->GetResponseModel();
1024 AliITSsimulationSPD *sim0=new AliITSsimulationSPD(seg0,res0);
1025 SetSimulationModel(0,sim0);
1026 // test
1027 // printf("SPD dimensions %f %f \n",seg0->Dx(),seg0->Dz());
1028 // printf("SPD npixels %d %d \n",seg0->Npz(),seg0->Npx());
1029 // printf("SPD pitches %d %d \n",seg0->Dpz(0),seg0->Dpx(0));
1030 // end test
1031 //
1032 // SDD
1033 //Set response functions
1034 Float_t baseline = 10.;
1035 Float_t noise = 1.75;
1036
1037 // SDD compression param: 2 fDecrease, 2fTmin, 2fTmax or disable, 2 fTolerance
1038
1039 iDetType=DetType(1);
1040 AliITSresponseSDD *res1 = (AliITSresponseSDD*)iDetType->GetResponseModel();
1041 if (!res1) {
1042 res1=new AliITSresponseSDD();
1043 SetResponseModel(1,res1);
1044 }
1045 res1->SetMagicValue(900.);
1046 Float_t maxadc = res1->MaxAdc();
1047 Float_t topValue = res1->MagicValue();
1048 Float_t norm = maxadc/topValue;
1049
1050 Float_t fCutAmp = baseline + 2.*noise;
1051 fCutAmp *= norm;
1052 Int_t cp[8]={0,0,(int)fCutAmp,(int)fCutAmp,0,0,0,0}; //1D
1053
1054 //res1->SetZeroSupp("2D");
1055 res1->SetZeroSupp("1D");
1056 res1->SetNoiseParam(noise,baseline);
1057 res1->SetDo10to8(kTRUE);
1058 res1->SetCompressParam(cp);
1059 res1->SetMinVal(4);
1060 res1->SetDiffCoeff(3.6,40.);
1061 //res1->SetMagicValue(96.95);
1062 AliITSsegmentationSDD *seg1=(AliITSsegmentationSDD*)iDetType->GetSegmentationModel();
1063 if (!seg1) {
1064 seg1 = new AliITSsegmentationSDD(geom,res1);
1065 SetSegmentationModel(1,seg1);
1066 }
1067 AliITSsimulationSDD *sim1=new AliITSsimulationSDD(seg1,res1);
1068 sim1->SetDoFFT(1);
1069 sim1->SetCheckNoise(kFALSE);
1070 SetSimulationModel(1,sim1);
1071
1072 // SSD
1073 iDetType=DetType(2);
1074 AliITSsegmentationSSD *seg2=(AliITSsegmentationSSD*)iDetType->GetSegmentationModel();
1075 AliITSresponseSSD *res2 = (AliITSresponseSSD*)iDetType->GetResponseModel();
1076 res2->SetSigmaSpread(3.,2.);
1077 AliITSsimulationSSD *sim2=new AliITSsimulationSSD(seg2,res2);
1078 SetSimulationModel(2,sim2);
1079
1080 cerr<<"Digitizing ITS...\n";
1081
1082 TStopwatch timer;
1083 timer.Start();
1084 HitsToDigits(0,0,-1," ","All"," ");
1085 timer.Stop(); timer.Print();
1086
1087 delete sim0;
1088 delete sim1;
1089 delete sim2;
1090}
1091
3bd79107 1092
1093//____________________________________________________________________________
e8189707 1094void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, Option_t *option, Option_t *opt,Text_t *filename)
3bd79107 1095{
1096 // keep galice.root for signal and name differently the file for
1097 // background when add! otherwise the track info for signal will be lost !
1098
9c74c52b 1099 // the condition below will disappear when the geom class will be
1100 // initialised for all versions - for the moment it is only for v5 !
1101 // 7 is the SDD beam test version
1102 Int_t ver = this->IsVersion();
1103 if(ver!=5 && ver!=7) return;
1104
3bd79107 1105 char *all = strstr(opt,"All");
1106 char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
3bd79107 1107
1108 Int_t nmodules;
1109 InitModules(size,nmodules);
e8189707 1110 FillModules(evNumber,bgrev,nmodules,option,filename);
3bd79107 1111
9c74c52b 1112 //TBranch *branch;
3bd79107 1113 AliITSsimulation* sim;
9c74c52b 1114 //TObjArray *branches=gAlice->TreeD()->GetListOfBranches();
3bd79107 1115 AliITSgeom *geom = GetITSgeom();
1116
1117 Int_t id,module;
9c74c52b 1118 Int_t first,last;
a8a6107b 1119 for (id=0;id<kNTYPES;id++) {
3bd79107 1120 if (!all && !det[id]) continue;
9c74c52b 1121 //branch = (TBranch*)branches->UncheckedAt(id);
3bd79107 1122 AliITSDetType *iDetType=DetType(id);
1123 sim = (AliITSsimulation*)iDetType->GetSimulationModel();
1124 if (!sim) {
1125 Error("HitsToDigits","The simulation class was not instantiated!");
1126 exit(1);
e8189707 1127 // or SetDefaultSimulation();
3bd79107 1128 }
9c74c52b 1129 if(geom) {
1130 first = geom->GetStartDet(id);
1131 last = geom->GetLastDet(id);
1132 } else first=last=0;
d5da1ecf 1133 cout << "det type " << id << " first, last "<< first << last << endl;
3bd79107 1134 for(module=first;module<=last;module++) {
1135 AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module);
1136 sim->DigitiseModule(mod,module,evNumber);
1137 // fills all branches - wasted disk space
1138 gAlice->TreeD()->Fill();
1139 ResetDigits();
1140 // try and fill only the branch
1141 //branch->Fill();
1142 //ResetDigits(id);
1143 } // loop over modules
1144 } // loop over detector types
1145
3bd79107 1146 ClearModules();
1147
e8189707 1148 Int_t nentries=(Int_t)gAlice->TreeD()->GetEntries();
d5da1ecf 1149 cout << "nentries in TreeD" << nentries << endl;
3bd79107 1150
1151 char hname[30];
1152 sprintf(hname,"TreeD%d",evNumber);
2ab0c725 1153 gAlice->TreeD()->Write(hname,TObject::kOverwrite);
3bd79107 1154 // reset tree
1155 gAlice->TreeD()->Reset();
1156
1157}
1158
1159
1160//____________________________________________________________________________
1161void AliITS::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt)
1162{
e8189707 1163 // cluster finding and reconstruction of space points
3bd79107 1164
9c74c52b 1165 // the condition below will disappear when the geom class will be
1166 // initialised for all versions - for the moment it is only for v5 !
1167 // 7 is the SDD beam test version
1168 Int_t ver = this->IsVersion();
340cd0d8 1169 if(ver!=5) return;
9c74c52b 1170
3bd79107 1171 char *all = strstr(opt,"All");
1172 char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
1173
1174 static Bool_t first=kTRUE;
caf37aec 1175 if (!TreeC() && first) {
3bd79107 1176 MakeTreeC("C");
1177 first=kFALSE;
1178 }
caf37aec 1179
1180 TTree *treeC=TreeC();
3bd79107 1181
9c74c52b 1182
1183 //TBranch *branch;
3bd79107 1184 AliITSClusterFinder* rec;
1185
9c74c52b 1186 //TObjArray *branches=gAlice->TreeR()->GetListOfBranches();
3bd79107 1187 AliITSgeom *geom = GetITSgeom();
1188
1189 Int_t id,module;
a8a6107b 1190 for (id=0;id<kNTYPES;id++) {
3bd79107 1191 if (!all && !det[id]) continue;
9c74c52b 1192 //branch = (TBranch*)branches->UncheckedAt(id);
3bd79107 1193 AliITSDetType *iDetType=DetType(id);
1194 rec = (AliITSClusterFinder*)iDetType->GetReconstructionModel();
1195 if (!rec) {
1196 Error("DigitsToRecPoints","The cluster finder class was not instantiated!");
1197 exit(1);
e8189707 1198 // or SetDefaultClusterFinders();
3bd79107 1199 }
1200 TClonesArray *itsDigits = this->DigitsAddress(id);
1201
9c74c52b 1202 Int_t first,last;
1203 if(geom) {
1204 first = geom->GetStartDet(id);
1205 last = geom->GetLastDet(id);
1206 } else first=last=0;
1207 //printf("first last %d %d\n",first,last);
3bd79107 1208 for(module=first;module<=last;module++) {
3bd79107 1209 this->ResetDigits();
1210 if (all) gAlice->TreeD()->GetEvent(lastentry+module);
1211 else gAlice->TreeD()->GetEvent(lastentry+(module-first));
1212 Int_t ndigits = itsDigits->GetEntriesFast();
1213 if (ndigits) rec->FindRawClusters();
1214 gAlice->TreeR()->Fill();
1215 ResetRecPoints();
caf37aec 1216 treeC->Fill();
3bd79107 1217 ResetClusters();
1218 // try and fill only the branch
1219 //branch->Fill();
1220 //ResetRecPoints(id);
1221 } // loop over modules
1222 } // loop over detector types
1223
1224
e8189707 1225 Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries();
caf37aec 1226 Int_t ncentries=(Int_t)treeC->GetEntries();
d5da1ecf 1227 cout << " nentries ncentries " << nentries << ncentries << endl;
3bd79107 1228
1229 char hname[30];
1230 sprintf(hname,"TreeR%d",evNumber);
2ab0c725 1231 gAlice->TreeR()->Write(hname,TObject::kOverwrite);
3bd79107 1232 // reset tree
1233 gAlice->TreeR()->Reset();
1234
1235 sprintf(hname,"TreeC%d",evNumber);
caf37aec 1236 treeC->Write(hname,TObject::kOverwrite);
1237 treeC->Reset();
3bd79107 1238}
1239
1240
1241//____________________________________________________________________________
e8189707 1242void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
3bd79107 1243Option_t *option,Option_t *opt,Text_t *filename)
1244{
1245 // keep galice.root for signal and name differently the file for
1246 // background when add! otherwise the track info for signal will be lost !
1247
9c74c52b 1248
1249 // the condition below will disappear when the geom class will be
1250 // initialised for all versions - for the moment it is only for v5 !
1251 Int_t ver = this->IsVersion();
1252 if(ver!=5) return;
1253
3bd79107 1254 char *all = strstr(opt,"All");
1255 char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
1256
1257 Int_t nmodules;
e8189707 1258 InitModules(size,nmodules);
1259 FillModules(evNumber,bgrev,nmodules,option,filename);
3bd79107 1260
1261
e8189707 1262 AliITSsimulation* sim;
3bd79107 1263 AliITSgeom *geom = GetITSgeom();
1264
e8189707 1265 TRandom *random=new TRandom[9];
1266 random[0].SetSeed(111);
1267 random[1].SetSeed(222);
1268 random[2].SetSeed(333);
1269 random[3].SetSeed(444);
1270 random[4].SetSeed(555);
1271 random[5].SetSeed(666);
1272 random[6].SetSeed(777);
1273 random[7].SetSeed(888);
1274 random[8].SetSeed(999);
1275
1276
3bd79107 1277 Int_t id,module;
a8a6107b 1278 for (id=0;id<kNTYPES;id++) {
3bd79107 1279 if (!all && !det[id]) continue;
e8189707 1280 AliITSDetType *iDetType=DetType(id);
1281 sim = (AliITSsimulation*)iDetType->GetSimulationModel();
1282 if (!sim) {
1283 Error("HitsToFastPoints","The simulation class was not instantiated!");
1284 exit(1);
1285 // or SetDefaultSimulation();
1286 }
3bd79107 1287 Int_t first = geom->GetStartDet(id);
1288 Int_t last = geom->GetLastDet(id);
1289 for(module=first;module<=last;module++) {
1290 AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module);
e8189707 1291 sim->CreateFastRecPoints(mod,module,random);
3bd79107 1292 gAlice->TreeR()->Fill();
1293 ResetRecPoints();
1294 } // loop over modules
1295 } // loop over detector types
1296
1297
1298 ClearModules();
1299
1300 //Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries();
1301
1302 char hname[30];
1303 sprintf(hname,"TreeR%d",evNumber);
2ab0c725 1304 gAlice->TreeR()->Write(hname,TObject::kOverwrite);
3bd79107 1305 // reset tree
1306 gAlice->TreeR()->Reset();
1307
e8189707 1308 delete [] random;
1309
fe4da5cc 1310}
1311
55b5a0b3 1312//________________________________________________________________
1313
1314AliITStrack AliITS::Tracking(AliITStrack &track, AliITStrack *reference,TObjArray *fastpoints, Int_t
1315**vettid, Bool_t flagvert ) {
1316
1317//Origin A. Badala' and G.S. Pappalardo: e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
1318
cc6e75dd 1319
55b5a0b3 1320 TList *list= new TList();
1321
1322 AliITStrack tr(track);
cc6e75dd 1323
55b5a0b3 1324 list->AddLast(&tr);
1325
1326 Double_t Pt=(tr).GetPt();
1327 //cout << "\n Pt = " << Pt <<"\n";
1328
55b5a0b3 1329 AliITStracking obj(list, reference, this, fastpoints,TMath::Abs(Pt),vettid, flagvert);
1330 list->Delete();
1331 delete list;
1332
1333 Int_t itot=-1;
1334 TVector VecTotLabref(18);
cc6e75dd 1335 Int_t lay, k;
1336 for(lay=5; lay>=0; lay--) {
55b5a0b3 1337 TVector VecLabref(3);
1338 VecLabref=(*reference).GetLabTrack(lay);
cc6e75dd 1339 for(k=0; k<3; k++) {itot++; VecTotLabref(itot)=VecLabref(k);}
55b5a0b3 1340 }
1341 Long_t labref;
1342 Int_t freq;
1343 (*reference).Search(VecTotLabref, labref, freq);
1344
1345 if(freq < 5) labref=-labref;
1346 (*reference).SetLabel(labref);
1347
1348 return *reference;
1349
1350}
1351
1352
1353
1354//________________________________________________________________
1355
1356
1357
1358void AliITS::DoTracking(Int_t evNumber, Int_t min_t, Int_t max_t, TFile *file, Bool_t flagvert) {
1359
1360// ex macro for tracking ITS
1361
1362 printf("begin DoTracking - file %p\n",file);
1363
1364 const char *pname="75x40_100x60";
1365
1366 struct GoodTrack {
1367 Int_t lab,code;
1368 Float_t px,py,pz,x,y,z,pxg,pyg,pzg,ptg;
1369 Bool_t flag;
1370 };
1371
1372
1373 gAlice->GetEvent(0);
1374
1375 AliTPC *TPC=(AliTPC*)gAlice->GetDetector("TPC");
1376 AliTPCParam *digp = (AliTPCParam*)file->Get(pname);
1377 if (digp!=0) TPC->SetParam(digp);
1378
1379 GoodTrack gt[7000];
1380 Int_t ngood=0;
1381 ifstream in("good_tracks");
1382
1383 cerr<<"Reading good tracks...\n";
1384 while (in>>gt[ngood].lab>>gt[ngood].code
1385 >>gt[ngood].px >>gt[ngood].py>>gt[ngood].pz
1386 >>gt[ngood].x >>gt[ngood].y >>gt[ngood].z
1387 >>gt[ngood].pxg >>gt[ngood].pyg >>gt[ngood].pzg
1388 >>gt[ngood].ptg >>gt[ngood].flag) {
1389 ngood++;
1390 cerr<<ngood<<'\r';
1391 if (ngood==7000) {
1392 cerr<<"Too many good tracks !\n";
1393 break;
1394 }
1395 }
1396 if (!in.eof()) cerr<<"Read error (good_tracks) !\n";
1397
1398
1399// Load tracks
1400 TFile *tf=TFile::Open("tpctracks.root");
1401 if (!tf->IsOpen()) {cerr<<"Can't open tpctracks.root !\n"; return ;}
1402 TObjArray tracks(200000);
1403 TTree *tracktree=(TTree*)tf->Get("TreeT");
1404 TBranch *tbranch=tracktree->GetBranch("tracks");
1405 Int_t nentr=(Int_t)tracktree->GetEntries();
cc6e75dd 1406 Int_t kk;
1407 for (kk=0; kk<nentr; kk++) {
55b5a0b3 1408 AliTPCtrack *iotrack=new AliTPCtrack;
1409 tbranch->SetAddress(&iotrack);
cc6e75dd 1410 tracktree->GetEvent(kk);
55b5a0b3 1411 tracks.AddLast(iotrack);
1412 }
1413 tf->Close();
1414
1415
1416 Int_t nt = tracks.GetEntriesFast();
1417 cerr<<"Number of found tracks "<<nt<<endl;
1418
1419 TVector DataOut(9);
1420 Int_t kkk=0;
1421
1422 Double_t ptg=0.,pxg=0.,pyg=0.,pzg=0.;
1423
1424 ////////////////////////////// good tracks definition in TPC ////////////////////////////////
1425
1426 ofstream out1 ("AliITSTrag.out");
cc6e75dd 1427 Int_t i;
1428 for (i=0; i<ngood; i++) out1 << gt[i].ptg << "\n";
55b5a0b3 1429 out1.close();
1430
1431
1432 TVector vec(5);
1433 TTree *TR=gAlice->TreeR();
1434 Int_t nent=(Int_t)TR->GetEntries();
1435 TClonesArray *recPoints = RecPoints();
1436 Int_t numbpoints;
1437 Int_t totalpoints=0;
1438 Int_t *np = new Int_t[nent];
1439 Int_t **vettid = new Int_t* [nent];
cc6e75dd 1440 Int_t mod;
1441 for (mod=0; mod<nent; mod++) {
55b5a0b3 1442 vettid[mod]=0;
1443 this->ResetRecPoints();
340cd0d8 1444 //gAlice->TreeR()->GetEvent(mod+1); //first entry in TreeR is empty
1445 gAlice->TreeR()->GetEvent(mod); //first entry in TreeR is empty
55b5a0b3 1446 numbpoints = recPoints->GetEntries();
1447 totalpoints+=numbpoints;
1448 np[mod] = numbpoints;
1449 //cout<<" mod = "<<mod<<" numbpoints = "<<numbpoints<<"\n"; getchar();
1450 vettid[mod] = new Int_t[numbpoints];
cc6e75dd 1451 Int_t ii;
1452 for (ii=0;ii<numbpoints; ii++) *(vettid[mod]+ii)=0;
55b5a0b3 1453 }
1454
1455 AliTPCtrack *track;
1456
1457
1458 if(min_t < 0) {min_t = 0; max_t = nt-1;}
1459
1460/*
1461 ///////////////////////////////// Definition of vertex end its error ////////////////////////////
1462 ////////////////////////// In the future it will be given by a method ///////////////////////////
1463 Double_t Vx=0.;
1464 Double_t Vy=0.;
1465 Double_t Vz=0.;
1466
1467 Float_t sigmavx=0.0050; // 50 microns
1468 Float_t sigmavy=0.0050; // 50 microns
1469 Float_t sigmavz=0.010; // 100 microns
1470
1471 //Vx+=gRandom->Gaus(0,sigmavx); Vy+=gRandom->Gaus(0,sigmavy); Vz+=gRandom->Gaus(0,sigmavz);
1472 TVector vertex(3), ervertex(3)
1473 vertex(0)=Vx; vertex(1)=Vy; vertex(2)=Vz;
1474 ervertex(0)=sigmavx; ervertex(1)=sigmavy; ervertex(2)=sigmavz;
1475 /////////////////////////////////////////////////////////////////////////////////////////////////
1476*/
1477
1478 //TDirectory *savedir=gDirectory;
1479
1480 TTree tracktree1("TreeT","Tree with ITS tracks");
1481 AliITSiotrack *iotrack=0;
1482 tracktree1.Branch("ITStracks","AliITSiotrack",&iotrack,32000,0);
1483
1484 ofstream out ("AliITSTra.out");
cc6e75dd 1485
1486 Int_t j;
1487 for (j=min_t; j<=max_t; j++) {
55b5a0b3 1488 track=(AliTPCtrack*)tracks.UncheckedAt(j);
1489 Int_t flaglab=0;
1490 if (!track) continue;
1491 ////// elimination of not good tracks ////////////
1492 Int_t ilab=TMath::Abs(track->GetLabel());
cc6e75dd 1493 Int_t iii;
1494 for (iii=0;iii<ngood;iii++) {
1495 //cout<<" ilab, gt[iii].lab = "<<ilab<<" "<<gt[iii].lab<<"\n"; getchar();
1496 if (ilab==gt[iii].lab) {
55b5a0b3 1497 flaglab=1;
cc6e75dd 1498 ptg=gt[iii].ptg;
1499 pxg=gt[iii].pxg;
1500 pyg=gt[iii].pyg;
1501 pzg=gt[iii].pzg;
55b5a0b3 1502 break;
1503 }
1504 }
1505 //cout<<" j flaglab = " <<j<<" "<<flaglab<<"\n"; getchar();
1506 if (!flaglab) continue;
1507 //cout<<" j = " <<j<<"\n"; getchar();
1508 /*
1509 ////// old propagation to the end of TPC //////////////
1510 Double_t xk=76.;
1511 track->PropagateTo(xk);
1512 xk-=0.11;
1513 track->PropagateTo(xk,42.7,2.27); //C
1514 xk-=2.6;
1515 track->PropagateTo(xk,36.2,1.98e-3); //C02
1516 xk-=0.051;
1517 track->PropagateTo(xk,42.7,2.27); //C
1518 ///////////////////////////////////////////////////
1519 */
1520
1521 ////// new propagation to the end of TPC //////////////
1522 Double_t xk=77.415;
1523 track->PropagateTo(xk, 28.94, 1.204e-3); //Ne
1524 xk -=0.01;
1525 track->PropagateTo(xk, 44.77, 1.71); //Tedlar
1526 xk -=0.04;
1527 track->PropagateTo(xk, 44.86, 1.45); //Kevlar
1528 xk -=2.0;
1529 track->PropagateTo(xk, 41.28, 0.029); //Nomex
1530 xk-=16;
1531 track->PropagateTo(xk,36.2,1.98e-3); //C02
1532 xk -=0.01;
1533 track->PropagateTo(xk, 24.01, 2.7); //Al
1534 xk -=0.01;
1535 track->PropagateTo(xk, 44.77, 1.71); //Tedlar
1536 xk -=0.04;
1537 track->PropagateTo(xk, 44.86, 1.45); //Kevlar
1538 xk -=0.5;
1539 track->PropagateTo(xk, 41.28, 0.029); //Nomex
1540
1541 ///////////////////////////////////////////////////////////////
1542
1543 ///////////////////////////////////////////////////////////////
1544 AliITStrack trackITS(*track);
1545 AliITStrack result(*track);
1546 AliITStrack primarytrack(*track);
1547
1548///////////////////////////////////////////////////////////////////////////////////////////////
1549 TVector Vgeant(3);
1550 Vgeant=result.GetVertex();
1551
1552 // Definition of Dv and Zv for vertex constraint
cc6e75dd 1553 Double_t sigmaDv=0.0050; Double_t sigmaZv=0.010;
1554 //Double_t sigmaDv=0.0015; Double_t sigmaZv=0.0015;
55b5a0b3 1555 Double_t uniform= gRandom->Uniform();
1556 Double_t signdv;
1557 if(uniform<=0.5) signdv=-1.;
1558 else
1559 signdv=1.;
1560
1561 Double_t Vr=TMath::Sqrt(Vgeant(0)*Vgeant(0)+ Vgeant(1)*Vgeant(1));
1562 Double_t Dv=gRandom->Gaus(signdv*Vr,(Float_t)sigmaDv);
1563 Double_t Zv=gRandom->Gaus(Vgeant(2),(Float_t)sigmaZv);
1564
1565 //cout<<" Dv e Zv = "<<Dv<<" "<<Zv<<"\n";
1566 trackITS.SetDv(Dv); trackITS.SetZv(Zv);
1567 trackITS.SetsigmaDv(sigmaDv); trackITS.SetsigmaZv(sigmaZv);
1568 result.SetDv(Dv); result.SetZv(Zv);
1569 result.SetsigmaDv(sigmaDv); result.SetsigmaZv(sigmaZv);
1570 primarytrack.SetDv(Dv); primarytrack.SetZv(Zv);
1571 primarytrack.SetsigmaDv(sigmaDv); primarytrack.SetsigmaZv(sigmaZv);
1572
1573/////////////////////////////////////////////////////////////////////////////////////////////////
1574
1575 primarytrack.PrimaryTrack();
1576 TVector d2=primarytrack.Getd2();
1577 TVector tgl2=primarytrack.Gettgl2();
1578 TVector dtgl=primarytrack.Getdtgl();
1579 trackITS.Setd2(d2); trackITS.Settgl2(tgl2); trackITS.Setdtgl(dtgl);
1580 result.Setd2(d2); result.Settgl2(tgl2); result.Setdtgl(dtgl);
1581 /*
1582 trackITS.SetVertex(vertex); trackITS.SetErrorVertex(ervertex);
1583 result.SetVertex(vertex); result.SetErrorVertex(ervertex);
1584 */
1585 Tracking(trackITS,&result,recPoints,vettid, flagvert);
1586
cc6e75dd 1587 cout<<" progressive track number = "<<j<<"\r";
1588 // cout<<j<<"\r";
1589 //cout<<" progressive track number = "<<j<<"\n";
55b5a0b3 1590 Long_t labITS=result.GetLabel();
cc6e75dd 1591 //cout << " ITS track label = " << labITS << "\n";
55b5a0b3 1592 int lab=track->GetLabel();
cc6e75dd 1593 // cout << " TPC track label = " << lab <<"\n";
55b5a0b3 1594 //result.GetClusters(); getchar(); //to print the cluster matrix
1595
1596//propagation to vertex
1597
1598 Double_t rbeam=3.;
cc6e75dd 1599
1600 result.Propagation(rbeam);
1601
55b5a0b3 1602 TMatrix *cov;
1603 cov=&result.GetCMatrix();
1604 Double_t pt=TMath::Abs(result.GetPt());
1605 Double_t Dr=result.GetD();
1606 Double_t Z=result.GetZ();
1607 Double_t tgl=result.GetTgl();
1608 Double_t C=result.GetC();
1609 Double_t Cy=C/2.;
1610 Double_t Dz=Z-(tgl/Cy)*TMath::ASin(result.arga(rbeam));
1611 Dz-=Vgeant(2);
cc6e75dd 1612
55b5a0b3 1613 // cout<<" Dr e dz alla fine = "<<Dr<<" "<<Dz<<"\n"; getchar();
1614 Double_t phi=result.Getphi();
1615 Double_t phivertex = phi - TMath::ASin(result.argA(rbeam));
1616 Double_t duepi=2.*TMath::Pi();
1617 if(phivertex>duepi) phivertex-=duepi;
1618 if(phivertex<0.) phivertex+=duepi;
cc6e75dd 1619 Double_t Dtot=TMath::Sqrt(Dr*Dr+Dz*Dz);
1620
55b5a0b3 1621//////////////////////////////////////////////////////////////////////////////////////////
1622 Int_t NumofCluster, idmodule,idpoint;
1623 NumofCluster=result.GetNumClust();
1624 if(NumofCluster >=5) {
1625
1626
1627 AliITSiotrack outtrack;
1628
1629 iotrack=&outtrack;
1630
1631 iotrack->SetStatePhi(phi);
1632 iotrack->SetStateZ(Z);
1633 iotrack->SetStateD(Dr);
1634 iotrack->SetStateTgl(tgl);
1635 iotrack->SetStateC(C);
1636 Double_t radius=result.Getrtrack();
1637 iotrack->SetRadius(radius);
1638 Int_t charge;
1639 if(C>0.) charge=-1; else charge=1;
1640 iotrack->SetCharge(charge);
1641
1642
1643
1644 iotrack->SetCovMatrix(cov);
1645
1646 Double_t px=pt*TMath::Cos(phi);
1647 Double_t py=pt*TMath::Sin(phi);
1648 Double_t pz=pt*tgl;
1649
1650 Double_t xtrack=Dr*TMath::Sin(phi);
1651 Double_t ytrack=Dr*TMath::Cos(phi);
1652 Double_t ztrack=Dz+Vgeant(2);
1653
1654
1655 iotrack->SetPx(px);
1656 iotrack->SetPy(py);
1657 iotrack->SetPz(pz);
1658 iotrack->SetX(xtrack);
1659 iotrack->SetY(ytrack);
1660 iotrack->SetZ(ztrack);
1661 iotrack->SetLabel(labITS);
1662
c5b22044 1663 Int_t il;
1664 for(il=0;il<6; il++){
55b5a0b3 1665 iotrack->SetIdPoint(il,result.GetIdPoint(il));
1666 iotrack->SetIdModule(il,result.GetIdModule(il));
1667 }
1668 tracktree1.Fill();
1669
1670 //cout<<" labITS = "<<labITS<<"\n";
1671 //cout<<" phi z Dr tgl C = "<<phi<<" "<<Z<<" "<<Dr<<" "<<tgl<<" "<<C<<"\n"; getchar();
1672
1673 DataOut(kkk) = ptg; kkk++; DataOut(kkk)=labITS; kkk++; DataOut(kkk)=lab; kkk++;
1674
c5b22044 1675 for (il=0;il<6;il++) {
55b5a0b3 1676 idpoint=result.GetIdPoint(il);
1677 idmodule=result.GetIdModule(il);
1678 *(vettid[idmodule]+idpoint)=1;
1679 iotrack->SetIdPoint(il,idpoint);
1680 iotrack->SetIdModule(il,idmodule);
1681 }
cc6e75dd 1682
1683 // cout<<" +++++++++++++ pt e ptg = "<<pt<<" "<<ptg<<" ++++++++++\n";
55b5a0b3 1684 Double_t difpt= (pt-ptg)/ptg*100.;
1685 DataOut(kkk)=difpt; kkk++;
1686 Double_t lambdag=TMath::ATan(pzg/ptg);
1687 Double_t lam=TMath::ATan(tgl);
1688 Double_t diflam = (lam - lambdag)*1000.;
1689 DataOut(kkk) = diflam; kkk++;
1690 Double_t phig=TMath::ATan(pyg/pxg);
1691 Double_t phi=phivertex;
1692 Double_t difphi = (phi - phig)*1000.;
1693 DataOut(kkk)=difphi; kkk++;
1694 DataOut(kkk)=Dtot*1.e4; kkk++;
1695 DataOut(kkk)=Dr*1.e4; kkk++;
1696 DataOut(kkk)=Dz*1.e4; kkk++;
cc6e75dd 1697 Int_t r;
1698 for (r=0; r<9; r++) { out<<DataOut(r)<<" ";}
55b5a0b3 1699 out<<"\n";
1700 kkk=0;
1701
1702
1703 } // end if on NumofCluster
1704 //gObjectTable->Print(); // stampa memoria
1705 } // end for (int j=min_t; j<=max_t; j++)
1706
1707 out.close();
1708
cc6e75dd 1709
55b5a0b3 1710 static Bool_t first=kTRUE;
1711 static TFile *tfile;
1712
1713 if(first) {
1714 tfile=new TFile("itstracks.root","RECREATE");
1715 //cout<<"I have opened itstracks.root file "<<endl;
1716 }
1717 first=kFALSE;
1718 tfile->cd();
1719 tfile->ls();
1720
1721 char hname[30];
1722 sprintf(hname,"TreeT%d",evNumber);
1723
1724 tracktree1.Write(hname);
1725
1726
1727
1728 TTree *fAli=gAlice->TreeK();
1729 TFile *fileAli=0;
1730
1731 if (fAli) fileAli =fAli->GetCurrentFile();
1732 fileAli->cd();
1733
1734 ////////////////////////////////////////////////////////////////////////////////////////////////
1735
1736 printf("delete vectors\n");
1737 if(np) delete [] np;
1738 if(vettid) delete [] vettid;
1739
1740}