fcb1c2767248abbb52fc1c2185c4b78a936654f0
[u/mrichter/AliRoot.git] / ITS / AliITS.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 /*
17 $Log$
18 Revision 1.15  2000/06/28 14:41:12  fca
19 Corrections to the custom Streamer
20
21 Revision 1.14  2000/06/15 09:27:52  barbera
22 Problems with the HP compiler fixed
23
24 Revision 1.13  2000/06/13 15:32:44  nilsen
25 fix compilation error on HP and DEC unix.
26
27 Revision 1.12  2000/06/12 23:43:16  nilsen
28 New ITS code replacing the old structure and simulations code.
29
30 Revision 1.9.2.8  2000/06/12 18:05:59  barbera
31 fixed posible compilation errors on HP unix
32
33 Revision 1.9.2.7  2000/06/11 20:20:18  barbera
34 New AliITS base clase  for the new structure.
35
36 Revision 1.9.2.3  2000/02/02 13:42:09  barbera
37 fixed AliITS.cxx for new AliRun structure. Added ITS hits list to list of hits which will have their track numbers updated
38
39 Revision 1.9.2.2  2000/01/23 03:03:13  nilsen
40 //fixed FillModule. Removed fi(fabs(xl)<dx....
41
42 Revision 1.9.2.1  2000/01/12 19:03:32  nilsen
43 This is the version of the files after the merging done in December 1999.
44 See the ReadMe110100.txt file for details
45
46 Revision 1.9  1999/11/14 14:33:25  fca
47 Correct problems with distructors and pointers, thanks to I.Hrivnacova
48
49 Revision 1.8  1999/09/29 09:24:19  fca
50 Introduction of the Copyright and cvs Log
51
52 */
53
54 ///////////////////////////////////////////////////////////////////////////////
55 //
56 //      An overview of the basic philosophy of the ITS code development
57 // and analysis is show in the figure below.
58 //Begin_Html
59 /*
60 <img src="picts/ITS/ITS_Analysis_schema.gif">
61 </pre>
62 <br clear=left>
63 <font size=+2 color=red>
64 <p>Roberto Barbera is in charge of the ITS Offline code (1999).
65 <a href="mailto:roberto.barbera@ct.infn.it">Roberto Barbera</a>.
66 </font>
67 <pre>
68 */
69 //End_Html
70 //
71 //  AliITS. Inner Traking System base class.
72 //  This class contains the base procedures for the Inner Tracking System
73 //
74 //Begin_Html
75 /*
76 <img src="picts/ITS/AliITS_Class_Diagram.gif">
77 </pre>
78 <br clear=left>
79 <font size=+2 color=red>
80 <p>This show the class diagram of the different elements that are part of
81 the AliITS class.
82 </font>
83 <pre>
84 */
85 //End_Html
86 //
87 // Version: 0
88 // Written by Rene Brun, Federico Carminati, and Roberto Barbera
89 //
90 // Version: 1
91 // Modified and documented by Bjorn S. Nilsen
92 // July 11 1999
93 //
94 // Version: 2
95 // Modified and documented by A. Bologna
96 // October 18 1999
97 //
98 // AliITS is the general base class for the ITS. Also see AliDetector for
99 // futher information.
100 //
101 ///////////////////////////////////////////////////////////////////////////////
102  
103 #include <TMath.h>
104 #include <TRandom.h>
105 #include <TVector.h>
106 #include <TObjArray.h>
107 #include <TClonesArray.h>
108 #include <TROOT.h>
109 #include <TObjectTable.h>
110
111
112
113 #include "AliRun.h"
114 #include "AliITS.h"
115 #include "AliITSMap.h"
116 #include "AliITSClusterFinder.h"
117 #include "AliITSsimulation.h"
118 #include "AliITSsimulationFastPoints.h"
119 #include "AliITSsegmentationSPD.h"
120 #include "AliITSresponseSPD.h"
121 #include "AliITSsegmentationSDD.h"
122 #include "AliITSresponseSDD.h"
123 #include "AliITSsegmentationSSD.h"
124 #include "AliITSresponseSSD.h"
125 //#include "AliITStrack.h"
126
127
128 ClassImp(AliITS)
129  
130 //_____________________________________________________________________________
131 AliITS::AliITS() : AliDetector() {
132   //
133   // Default initialiser for ITS
134   //     The default constructor of the AliITS class. In addition to
135   // creating the AliITS class it zeros the variables fIshunt (a member
136   // of AliDetector class), fEuclidOut, and fIdN, and zeros the pointers
137   // fITSpoints, fIdSens, and fIdName. The AliDetector default constructor
138   // is also called.
139   //
140
141   fIshunt     = 0;
142   fEuclidOut  = 0;
143   fIdN        = 0;
144   fIdName     = 0;
145   fIdSens     = 0;
146   fITSmodules = 0;
147   //
148   fDetTypes   = 0;
149   SetNDetTypes();
150   //
151   fDtype  = 0;
152   fNdtype = 0;
153   fCtype  = 0;
154   fNctype = 0;
155   fRecPoints = 0;
156   fNRecPoints = 0;
157   //
158   fITSgeom=0;
159 }
160
161 //_____________________________________________________________________________
162 AliITS::AliITS(const char *name, const char *title):AliDetector(name,title){
163   //
164   // Default initialiser for ITS
165   //     The constructor of the AliITS class. In addition to creating the
166   // AliITS class, it allocates memory for the TClonesArrays fHits and
167   // fDigits, and for the TObjArray fITSpoints. It also zeros the variables
168   // fIshunt (a member of AliDetector class), fEuclidOut, and fIdN, and zeros
169   // the pointers fIdSens and fIdName. To help in displaying hits via the ROOT
170   // macro display.C AliITS also sets the marker color to red. The variables
171   // passes with this constructor, const char *name and *title, are used by
172   // the constructor of AliDetector class. See AliDetector class for a
173   // description of these parameters and its constructor functions.
174   //
175
176   fHits       = new TClonesArray("AliITShit", 1560);
177   gAlice->AddHitList(fHits);
178
179   SetNDetTypes();
180
181   fNdtype = 0;
182   fDtype = 0;
183   fCtype  = 0;
184   fNctype = 0;
185
186   fRecPoints = 0;
187   fNRecPoints = 0;
188
189
190   fITSmodules = 0; 
191
192   fIshunt     = 0;
193   fEuclidOut  = 0;
194   fIdN        = 0;
195   fIdName     = 0;
196   fIdSens     = 0;
197  
198   fDetTypes = new TObjArray(fNDetTypes);  
199
200   Int_t i;
201   for(i=0;i<fNDetTypes;i++) {
202     (*fDetTypes)[i]=new AliITSDetType(); 
203    }
204   //
205
206    fNdtype = new Int_t[fNDetTypes];
207    fDtype = new TObjArray(fNDetTypes);
208
209    fNctype = new Int_t[fNDetTypes];
210    fCtype = new TObjArray(fNDetTypes);
211
212   SetMarkerColor(kRed);
213
214   fITSgeom=0;
215 }
216 //___________________________________________________________________________
217 AliITS::AliITS(AliITS &source){
218   if(this==&source) return;
219   printf("Error: You are not allowed to make a copy of the AliITS\n");
220   exit(1);
221 }
222 //____________________________________________________________________________
223 AliITS& AliITS::operator=(AliITS &source){
224   if(this==&source) return *this;
225   printf("Error: You are not allowed to make a copy of the AliITS\n");
226   exit(1);
227 }
228 //____________________________________________________________________________
229 void AliITS::ClearModules(){
230   //clear the modules TObjArray
231   Int_t i;
232
233   if(fITSmodules!=0) {
234         Int_t indSPD = fITSgeom->GetModuleIndex(2,fITSgeom->GetNladders(2),
235                                                 fITSgeom->GetNdetectors(2));
236         Int_t indSDD = fITSgeom->GetModuleIndex(4,fITSgeom->GetNladders(4),
237                                                 fITSgeom->GetNdetectors(4));
238       for(i=0;i<fITSmodules->GetEntriesFast();i++){
239             if(i<indSPD)
240                 delete (AliITSmodule *) fITSmodules->At(i);
241             else if(i<indSDD)
242                 delete (AliITSmodule *) fITSmodules->At(i);
243             else
244                 delete (AliITSmodule *) fITSmodules->At(i);
245       } // end for i
246   }// end if fITSmodules!=0
247
248 }
249 //_____________________________________________________________________________
250 AliITS::~AliITS(){
251   //
252   // Default distructor for ITS
253   //     The default destructor of the AliITS class. In addition to deleting
254   // the AliITS class it deletes the memory pointed to by the fHits, fDigits,
255   // fIdSens, fIdName, and fITSpoints.
256   //
257
258
259   delete fHits;
260   delete fDigits;
261   delete fRecPoints;
262   if(fIdName!=0) delete[] fIdName;
263   if(fIdSens!=0) delete[] fIdSens;
264   if(fITSmodules!=0) {
265       this->ClearModules();
266       delete fITSmodules;
267   }// end if fITSmodules!=0
268
269   //
270   Int_t i;
271   if(fDtype) {
272     for(i=0;i<fNDetTypes;i++) {
273       delete (*fDtype)[i];
274       fNdtype[i]=0;
275     }
276   }
277
278   for(i=0;i<fNDetTypes;i++) {
279       delete (*fCtype)[i];
280       fNctype[i]=0;
281   }
282
283   //
284
285   if (fDetTypes) {
286     fDetTypes->Delete();
287     delete fDetTypes;
288   }
289
290   if (fTreeC) delete fTreeC;
291   
292 }
293
294 //___________________________________________
295 AliITSDetType* AliITS::DetType(Int_t id)
296 {
297   //return pointer to id detector type
298     return ((AliITSDetType*) (*fDetTypes)[id]);
299
300 }
301 //___________________________________________
302 void AliITS::SetClasses(Int_t id, char* digit, char* cluster)
303 {
304   //set the digit and cluster classes to be used for the id detector type
305     ((AliITSDetType*) (*fDetTypes)[id])->ClassNames(digit,cluster);
306
307 }
308 //___________________________________________
309 void AliITS::SetResponseModel(Int_t id, AliITSresponse *response)
310 {
311   //set the response model for the id detector type
312
313     ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response);
314
315 }
316
317 //___________________________________________
318 void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg)
319 {
320   //set the segmentation model for the id detector type
321
322     ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg);
323
324 }
325
326 //___________________________________________
327 void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim)
328 {
329   //set the simulation model for the id detector type
330
331    ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim);
332
333 }
334 //___________________________________________
335 void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst)
336 {
337   //set the cluster finder model for the id detector type
338
339    ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst);
340
341 }
342
343 //_____________________________________________________________________________
344 void AliITS::AddHit(Int_t track, Int_t *vol, Float_t *hits){
345   //
346   // Add an ITS hit
347   //     The function to add information to the AliITShit class. See the
348   // AliITShit class for a full description. This function allocates the
349   // necessary new space for the hit information and passes the variable
350   // track, and the pointers *vol and *hits to the AliITShit constructor
351   // function.
352   //
353   TClonesArray &lhits = *fHits;
354   new(lhits[fNhits++]) AliITShit(fIshunt,track,vol,hits);
355 }
356 //_____________________________________________________________________________
357 void AliITS::AddRealDigit(Int_t id, Int_t *digits) 
358 {
359   // add a real digit - as coming from data
360
361   TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
362   new(ldigits[fNdtype[id]++]) AliITSdigit(digits);
363
364 }
365 //_____________________________________________________________________________
366 void AliITS::AddDigit(Int_t id, AliITSdigit *d) 
367 {
368
369   // add a simulated digit
370
371   // should have ctors of type AliITSdigitSDD(const AliITSdigitSDD &)
372
373   TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
374
375   switch(id)
376   {
377   case 0:
378      new(ldigits[fNdtype[id]++]) AliITSdigitSPD(*((AliITSdigitSPD*)d));
379      break;
380   case 1:
381      new(ldigits[fNdtype[id]++]) AliITSdigitSDD(*((AliITSdigitSDD*)d));
382      break;
383   case 2:
384      new(ldigits[fNdtype[id]++]) AliITSdigitSSD(*((AliITSdigitSSD*)d));
385      break;
386   }
387
388 }
389
390 //_____________________________________________________________________________
391 void AliITS::AddDigit(Int_t id,Float_t phys,Int_t *digits,Int_t *tracks,Float_t *charges){
392
393   // add a simulated digit to the list
394
395   TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]);
396   switch(id)
397   {
398   case 0:
399      new(ldigits[fNdtype[id]++]) AliITSdigitSPD(digits,tracks);
400      break;
401   case 1:
402      new(ldigits[fNdtype[id]++]) AliITSdigitSDD(phys,digits,tracks,charges);
403      break;
404   case 2:
405      new(ldigits[fNdtype[id]++]) AliITSdigitSSD(digits,tracks);
406      break;
407   }
408  
409 }
410
411 //_____________________________________________________________________________
412 void AliITS::AddCluster(Int_t id, AliITSRawCluster *c) 
413 {
414
415   // add a cluster to the list
416
417   // should have ctors of type AliITSRawClusterSDD(const AliITSRawClusterSDD &)
418
419   TClonesArray &lcl = *((TClonesArray*)(*fCtype)[id]);
420
421   switch(id)
422   {
423   case 0:
424      new(lcl[fNctype[id]++]) AliITSRawClusterSPD(*((AliITSRawClusterSPD*)c));
425      break;
426   case 1:
427      new(lcl[fNctype[id]++]) AliITSRawClusterSDD(*((AliITSRawClusterSDD*)c));
428      break;
429   case 2:
430      new(lcl[fNctype[id]++]) AliITSRawClusterSSD(*((AliITSRawClusterSSD*)c));
431      break;
432   }
433
434 }
435
436
437 //_____________________________________________________________________________
438 void AliITS::AddRecPoint(const AliITSRecPoint &r)
439 {
440   //
441   // Add a reconstructed space point to the list
442   //
443   TClonesArray &lrecp = *fRecPoints;
444   new(lrecp[fNRecPoints++]) AliITSRecPoint(r);
445 }
446  
447
448 //____________________________________________
449 void AliITS::ResetDigits()
450 {
451     //
452     // Reset number of digits and the digits array for thE ITS detector
453     //
454
455     if (!fDtype) return;
456
457     Int_t i;
458     for(i=0;i<fNDetTypes;i++ ) {
459         if ((*fDtype)[i])    ((TClonesArray*)(*fDtype)[i])->Clear();
460         if (fNdtype)  fNdtype[i]=0;
461     }
462 }
463
464 //____________________________________________
465 void AliITS::ResetDigits(Int_t i)
466 {
467     //
468     // Reset number of digits and the digits array for this branch
469     //
470         if ((*fDtype)[i])    ((TClonesArray*)(*fDtype)[i])->Clear();
471         if (fNdtype)  fNdtype[i]=0;
472
473 }
474 //____________________________________________
475 void AliITS::ResetClusters()
476 {
477     //
478     // Reset number of clusters and the clusters array for ITS
479     //
480
481     Int_t i;
482     for(i=0;i<fNDetTypes;i++ ) {
483         if ((*fCtype)[i])    ((TClonesArray*)(*fCtype)[i])->Clear();
484         if (fNctype)  fNctype[i]=0;
485     }
486
487 }
488 //____________________________________________
489 void AliITS::ResetClusters(Int_t i)
490 {
491     //
492     // Reset number of clusters and the clusters array for this branch
493     //
494         if ((*fCtype)[i])    ((TClonesArray*)(*fCtype)[i])->Clear();
495         if (fNctype)  fNctype[i]=0;
496
497 }
498
499
500 //____________________________________________
501 void AliITS::ResetRecPoints()
502 {
503     //
504     // Reset number of rec points and the rec points array 
505     //
506     fNRecPoints = 0;
507     if (fRecPoints) fRecPoints->Clear();
508
509 }
510
511 //_____________________________________________________________________________
512 Int_t AliITS::DistancetoPrimitive(Int_t , Int_t ){
513   //
514   // Distance from mouse to ITS on the screen. Dummy routine
515   //     A dummy routine used by the ROOT macro display.C to allow for the
516   // use of the mouse (pointing device) in the macro. In general this should
517   // never be called. If it is it returns the number 9999 for any value of
518   // x and y.
519   //
520   return 9999;
521 }
522
523 //_____________________________________________________________________________
524 void AliITS::Init(){
525   //
526   // Initialise ITS after it has been built
527   //     This routine initializes the AliITS class. It is intended to be called
528   // from the Init function in AliITSv?. Besides displaying a banner
529   // indicating that it has been called it initializes the array fIdSens
530   // and sets the default segmentation, response, digit and raw cluster classes
531   // Therefore it should be called after a call to CreateGeometry.
532   //
533
534
535   SetDefaults();
536
537   Int_t i;
538
539   printf("\n");
540   for(i=0;i<35;i++) printf("*");
541   printf(" ITS_INIT ");
542   for(i=0;i<35;i++) printf("*");
543   printf("\n");
544   //
545   //
546   for(i=0;i<fIdN;i++) fIdSens[i] = gMC->VolId(fIdName[i]);
547   //
548   for(i=0;i<80;i++) printf("*");
549   printf("\n");
550 }
551
552 //_____________________________________________________________________________
553 void AliITS::SetDefaults()
554 {
555   // sets the default segmentation, response, digit and raw cluster classes
556
557
558   AliITSDetType *iDetType;
559
560   //SPD 
561
562   AliITSsegmentationSPD *seg0=new AliITSsegmentationSPD(fITSgeom);
563   AliITSresponseSPD *resp0=new AliITSresponseSPD();
564   iDetType=DetType(0); 
565   if (!iDetType->GetSegmentationModel()) SetSegmentationModel(0,seg0); 
566   if (!iDetType->GetResponseModel()) SetResponseModel(0,resp0); 
567   // set digit and raw cluster classes to be used
568   const char *kData=resp0->DataType();
569   if (strstr(kData,"real")) {
570       iDetType->ClassNames("AliITSdigit","AliITSRawClusterSPD");
571   } else iDetType->ClassNames("AliITSdigitSPD","AliITSRawClusterSPD");
572
573   // SDD                                          //
574   AliITSresponseSDD *resp1=new AliITSresponseSDD();
575   AliITSsegmentationSDD *seg1=new AliITSsegmentationSDD(fITSgeom,resp1);
576   iDetType=DetType(1); 
577   if (!iDetType->GetSegmentationModel()) SetSegmentationModel(1,seg1); 
578   if (!iDetType->GetResponseModel()) SetResponseModel(1,resp1); 
579   kData=resp1->DataType();
580   Option_t *opt=resp1->ZeroSuppOption();
581   if ((!strstr(opt,"2D")) && (!strstr(opt,"1D")) || strstr(kData,"real") ) {
582       iDetType->ClassNames("AliITSdigit","AliITSRawClusterSDD");
583   } else iDetType->ClassNames("AliITSdigitSDD","AliITSRawClusterSDD");
584
585   // SSD
586   AliITSsegmentationSSD *seg2=new AliITSsegmentationSSD(fITSgeom);
587   AliITSresponseSSD *resp2=new AliITSresponseSSD();
588   iDetType=DetType(2); 
589   if (!iDetType->GetSegmentationModel()) SetSegmentationModel(2,seg2); 
590   if (!iDetType->GetResponseModel()) SetResponseModel(2,resp2); 
591   kData=resp2->DataType();
592   if (strstr(kData,"real")) {
593       iDetType->ClassNames("AliITSdigit","AliITSRawClusterSSD");
594   } else iDetType->ClassNames("AliITSdigitSSD","AliITSRawClusterSSD");
595
596   if (fNDetTypes>3) {
597     Warning("SetDefaults","Only the three basic detector types are initialised!");
598   } 
599
600 }
601
602
603 //_____________________________________________________________________________
604
605 void AliITS::MakeTreeC(Option_t *option)
606 {
607   // create a separate tree to store the clusters
608
609      char *optC = strstr(option,"C");
610      if (optC && !fTreeC) fTreeC = new TTree("TC","Clusters in ITS");
611
612      Int_t buffersize = 4000;
613      char branchname[30];
614
615      char *det[3] = {"SPD","SDD","SSD"};
616
617      // one branch for Clusters per type of detector
618      Int_t i;
619      for(i=0; i<fNDetTypes ;i++) {
620         if (fNDetTypes==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
621         else  sprintf(branchname,"%sClusters%d",GetName(),i+1);
622         if (fCtype   && fTreeC) {
623            TreeC()->Branch(branchname,&((*fCtype)[i]), buffersize);
624            printf("Making Branch %s for Clusters of detector type %d\n",branchname,i+1);
625         }       
626      }
627
628 }
629
630 //_____________________________________________________________________________
631 void AliITS::GetTreeC(Int_t event)
632 {
633
634   // get the clusters tree for this event and set the branch address
635     char treeName[20];
636     char branchname[30];
637
638     char *det[3] = {"SPD","SDD","SSD"};
639
640     ResetClusters();
641     if (fTreeC) {
642           delete fTreeC;
643     }
644
645     sprintf(treeName,"TreeC%d",event);
646     fTreeC = (TTree*)gDirectory->Get(treeName);
647
648
649     TBranch *branch;
650     if (fTreeC) {
651         Int_t i;
652         for(i=0; i<fNDetTypes; i++) {
653             if (fNDetTypes==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
654             else  sprintf(branchname,"%sClusters%d",GetName(),i+1);
655             if (fCtype) {
656                 branch = fTreeC->GetBranch(branchname);
657                 if (branch) branch->SetAddress(&((*fCtype)[i]));
658             }
659         }
660     } else {
661         printf("ERROR: cannot find Clusters Tree for event:%d\n",event);
662     }
663
664 }
665 //_____________________________________________________________________________
666 void AliITS::MakeBranch(Option_t* option){
667   //
668   // Creates Tree branches for the ITS.
669   //
670
671
672   Int_t buffersize = 4000;
673   char branchname[30];
674   sprintf(branchname,"%s",GetName());
675
676   AliDetector::MakeBranch(option);
677
678
679 // one branch for digits per type of detector
680   
681    char *det[3] = {"SPD","SDD","SSD"};
682
683    char *kDigclass;
684    char *kClclass;
685
686    Int_t i;
687    for(i=0; i<fNDetTypes ;i++) {
688        AliITSDetType *iDetType=DetType(i); 
689        iDetType->GetClassNames(kDigclass,kClclass);
690        //printf("i, digclass, recclass %d %s %s\n",i,kDigclass,kClclass); 
691        // digits
692        (*fDtype)[i] = new TClonesArray(kDigclass,100); 
693        fNdtype[i]=0;
694        // clusters
695        (*fCtype)[i] = new TClonesArray(kClclass,100); 
696        fNctype[i]=0;
697    }
698
699
700   for(i=0; i<fNDetTypes ;i++) {
701       if (fNDetTypes==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
702       else  sprintf(branchname,"%sDigits%d",GetName(),i+1);
703       
704       if (fDtype   && gAlice->TreeD()) {
705           gAlice->TreeD()->Branch(branchname,&((*fDtype)[i]), buffersize);
706           printf("Making Branch %s for digits of type %d\n",branchname,i+1);
707       } 
708   }
709
710   // only one branch for rec points for all detector types
711   sprintf(branchname,"%sRecPoints",GetName());
712
713   fRecPoints=new TClonesArray("AliITSRecPoint",1000);
714
715   if (fRecPoints && gAlice->TreeR()) {
716     gAlice->TreeR()->Branch(branchname,&fRecPoints, buffersize);
717     printf("Making Branch %s for reconstructed space points\n",branchname);
718   }     
719
720
721 }
722
723 //___________________________________________
724 void AliITS::SetTreeAddress()
725 {
726
727   // Set branch address for the Trees.
728
729   char branchname[30];
730   AliDetector::SetTreeAddress();
731
732   char *det[3] = {"SPD","SDD","SSD"};
733
734   TBranch *branch;
735   TTree *treeD = gAlice->TreeD();
736   TTree *treeR = gAlice->TreeR();
737
738   Int_t i;
739   if (treeD) {
740       for(i=0; i<fNDetTypes; i++) {
741           if (fNDetTypes==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
742           else  sprintf(branchname,"%sDigits%d",GetName(),i+1);
743           if (fDtype) {
744               branch = treeD->GetBranch(branchname);
745               if (branch) branch->SetAddress(&((*fDtype)[i]));
746           }
747       }
748   }
749
750  
751   if (treeR) {
752     sprintf(branchname,"%sRecPoints",GetName());
753     if (fRecPoints) {
754       branch = treeR->GetBranch(branchname);
755       if (branch) branch->SetAddress(&fRecPoints);
756     }
757   }
758   
759
760 }
761
762 //____________________________________________________________________________
763 void AliITS::InitModules(Int_t size,Int_t &nmodules){
764
765   //initialize the modules array
766
767     Int_t nl,indexMAX,index;
768     Int_t indSPD,indSDD;
769
770     if(size<=0){ // default to using data stored in AliITSgeom
771         if(fITSgeom==0) {
772             printf("Error in AliITS::InitModule fITSgeom not defined\n");
773             return;
774         } // end if fITSgeom==0
775         nl = fITSgeom->GetNlayers();
776         indexMAX = fITSgeom->GetModuleIndex(nl,fITSgeom->GetNladders(nl),
777                                             fITSgeom->GetNdetectors(nl))+1;
778         nmodules = indexMAX;
779         fITSmodules = new TObjArray(indexMAX);
780         indSPD = fITSgeom->GetModuleIndex(2,fITSgeom->GetNladders(2),
781                                             fITSgeom->GetNdetectors(2));
782         indSDD = fITSgeom->GetModuleIndex(4,fITSgeom->GetNladders(4),
783                                             fITSgeom->GetNdetectors(4));
784         for(index=0;index<indexMAX;index++){
785             if(index<=indSPD)
786                 fITSmodules->AddAt( new AliITSmodule(index),index);
787             else if(index<=indSDD)
788                 fITSmodules->AddAt( new AliITSmodule(index),index);
789             else
790                 fITSmodules->AddAt( new AliITSmodule(index),index);
791         } // end for index
792     }else{
793         fITSmodules = new TObjArray(size);
794         nmodules = size;
795     } // end i size<=0
796 }
797
798 //____________________________________________________________________________
799 void AliITS::FillModules(Int_t evnt,Int_t bgrev,Int_t lastev,Int_t nmodules,Option_t *option,Text_t *filename){
800
801   // fill the modules with the sorted by module hits; add hits from background
802   // if option=Add
803
804     static TTree *trH1;                 //Tree with background hits
805     static TClonesArray *fHits2;        //List of hits for one track only
806
807     static Bool_t first=kTRUE;
808     static TFile *file;
809     char *add = strstr(option,"Add");
810
811
812     if (add ) {
813         if(first) {
814             cout<<"filename "<<filename<<endl;
815             file=new TFile(filename);
816             cout<<"I have opened "<<filename<<" file "<<endl;
817             fHits2     = new TClonesArray("AliITShit",1000  );
818         }           
819         first=kFALSE;
820         file->cd();
821         file->ls();
822         // Get Hits Tree header from file
823         if(fHits2) fHits2->Clear();
824         if(trH1) delete trH1;
825         trH1=0;
826         
827         char treeName[20];
828         sprintf(treeName,"TreeH%d",bgrev);
829         trH1 = (TTree*)gDirectory->Get(treeName);
830         //printf("TrH1 %p of treename %s for event %d \n",trH1,treeName,bgrev);
831         
832         if (!trH1) {
833             printf("ERROR: cannot find Hits Tree for event:%d\n",bgrev);
834         }
835         // Set branch addresses
836         TBranch *branch;
837         char branchname[20];
838         sprintf(branchname,"%s",GetName());
839         if (trH1 && fHits2) {
840             branch = trH1->GetBranch(branchname);
841             if (branch) branch->SetAddress(&fHits2);
842         }
843
844         // test
845         //Int_t ntracks1 =(Int_t)TrH1->GetEntries();
846         //printf("background - ntracks1 - %d\n",ntracks1);
847    }
848
849     // store temporarily coordinates of signal particles - see where delete
850     static Int_t *signal;
851     Int_t i;
852     if(!signal) signal=new Int_t[nmodules];
853     memset(signal,0,sizeof(int)*nmodules);
854 //    Float_t xhit[nmodules][4];
855     Float_t **xhit = new Float_t*[nmodules];
856     for(i=0;i<nmodules;i++) xhit[i] = new Float_t[4];
857 //    Float_t yhit[nmodules][4];
858     Float_t **yhit = new Float_t*[nmodules];
859     for(i=0;i<nmodules;i++) yhit[i] = new Float_t[4];
860
861     Int_t npart = gAlice->GetEvent(evnt);
862     if(npart<=0) return;
863     TClonesArray *itsHits = this->Hits();
864     Int_t lay,lad,det,index;
865     AliITShit *itsHit=0;
866     AliITSmodule *mod=0;
867
868     TTree *iTH = gAlice->TreeH();
869     Int_t ntracks =(Int_t) iTH->GetEntries();
870
871     Int_t t,h;
872     for(t=0; t<ntracks; t++){
873         gAlice->ResetHits();
874         iTH->GetEvent(t);
875         Int_t nhits = itsHits->GetEntriesFast();
876         if (!nhits) continue;
877         // cout << nhits << " hits in track " << t << endl;
878         for(h=0; h<nhits; h++){
879             itsHit = (AliITShit *)itsHits->UncheckedAt(h);
880             itsHit->GetDetectorID(lay,lad,det);
881             index = fITSgeom->GetModuleIndex(lay,lad,det);
882             mod = this->GetModule(index);
883             if (add) {
884                 xhit[index][signal[index]]=itsHit->fX;
885                 yhit[index][signal[index]]=itsHit->fY;
886                 signal[index]++;
887                 if (signal[index] >4) 
888                       printf("index,nsignal %d %d\n",index,signal[index]);
889             }
890             if(lay == 1 || lay == 2)
891                 mod->AddHit((AliITShit *) itsHit,t,h);
892             else if(lay == 3 || lay == 4)
893                     mod->AddHit((AliITShit *) itsHit,t,h);
894             else if(lay == 5 || lay ==6)
895                 mod->AddHit((AliITShit *)itsHit,t,h);
896             else
897                 mod->AddHit(itsHit,t,h);
898
899         } // end loop over hits 
900     } // end loop over tracks
901
902     // open the file with background
903     
904     if (add ) {
905           Int_t track,i,isig;
906           ntracks =(Int_t)trH1->GetEntries();
907             //printf("background - ntracks1 %d\n",ntracks);
908             //printf("background - Start loop over tracks \n");     
909             //   Loop over tracks
910
911             for(track=0; track<ntracks; track++) {
912
913                 if (fHits2)       fHits2->Clear();
914                 trH1->GetEvent(track);
915                 //   Loop over hits
916                 for(i=0;i<fHits2->GetEntriesFast();++i) {
917
918                     itsHit=(AliITShit*) (*fHits2)[i];
919                     itsHit->GetDetectorID(lay,lad,det);
920                     index = fITSgeom->GetModuleIndex(lay,lad,det);
921                     mod = this->GetModule(index);
922
923                     Float_t xbgr=itsHit->fX;
924                     Float_t ybgr=itsHit->fY;
925                     Float_t ebgr=itsHit->GetIonization();
926                     Bool_t cond=kFALSE;
927                     
928                     for(isig =0; isig < signal[index]; isig++) {
929                         Float_t dist= 
930                             (xbgr-xhit[index][isig])*(xbgr-xhit[index][isig])
931                             +(ybgr-yhit[index][isig])*(ybgr-yhit[index][isig]);
932                         if (dist<0.2&& ebgr!=0) cond=kTRUE; // check this number for ITS!
933                     }
934                     if (!cond) continue;
935                     if(lay == 1 || lay == 2)
936                        mod->AddHit((AliITShit *) itsHit,track,i);
937                     else if(lay == 3 || lay == 4)
938                             mod->AddHit((AliITShit *) itsHit,track,i);
939                     else if(lay == 5 || lay ==6)
940                             mod->AddHit((AliITShit *)itsHit,track,i);
941                     else
942                        mod->AddHit(itsHit,track,i);
943
944
945                }  // end loop over hits
946             } // end loop over tracks
947
948             TTree *fAli=gAlice->TreeK();
949             TFile *fileAli=0;
950             
951             if (fAli) fileAli =fAli->GetCurrentFile();
952             //printf("fAli, file %p %p\n",fAli,file);
953             file->cd();
954
955     } // end if add
956
957
958     for(i=0;i<nmodules;i++) delete[] xhit[i];
959     delete[] xhit;
960     for(i=0;i<nmodules;i++) delete[] yhit[i];
961     delete[] yhit;
962     //if (evnt==lastev) {delete [] signal; delete signal;}
963
964     //gObjectTable->Print();
965
966 }
967
968
969 //____________________________________________________________________________
970 void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
971 Option_t *option,Option_t *opt,Text_t *filename)
972 {
973     // keep galice.root for signal and name differently the file for 
974     // background when add! otherwise the track info for signal will be lost !
975   
976    char *all = strstr(opt,"All");
977    char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
978    //printf("Det 1 2 3 %s %s %s \n",det[0],det[1],det[2]);
979
980    Int_t nmodules;
981    InitModules(size,nmodules); 
982    FillModules(evNumber,bgrev,lastev,nmodules,option,filename);
983    //printf("nmodules %d\n",nmodules);
984
985    TBranch *branch;
986    AliITSsimulation* sim;
987
988    TObjArray *branches=gAlice->TreeD()->GetListOfBranches();
989    AliITSgeom *geom = GetITSgeom();
990
991    Int_t id,module;
992    for(id=0;id<3;id++) {
993         //printf("id %d All %s  det[id] %s \n",id,all,det[id]);
994         if (!all && !det[id]) continue;
995         branch = (TBranch*)branches->UncheckedAt(id);
996         AliITSDetType *iDetType=DetType(id); 
997         sim = (AliITSsimulation*)iDetType->GetSimulationModel();
998         if (!sim) {
999            Error("HitsToDigits","The simulation class was not instantiated!");
1000            exit(1);
1001            // or SetDefaultSimulation(id,iDetType*);
1002         }
1003         Int_t first = geom->GetStartDet(id);
1004         Int_t last = geom->GetLastDet(id);
1005         //printf("det type %d first, last %d %d \n",id,first,last);
1006         for(module=first;module<=last;module++) {
1007             AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module);
1008             sim->DigitiseModule(mod,module,evNumber);
1009             // fills all branches - wasted disk space
1010             gAlice->TreeD()->Fill(); 
1011             ResetDigits();
1012             // try and fill only the branch 
1013             //branch->Fill();
1014             //ResetDigits(id);
1015         } // loop over modules
1016    } // loop over detector types
1017
1018
1019    ClearModules();
1020
1021    //Int_t nentries=(Int_t)gAlice->TreeD()->GetEntries();
1022
1023    char hname[30];
1024    sprintf(hname,"TreeD%d",evNumber);
1025    gAlice->TreeD()->Write(hname);
1026    // reset tree
1027    gAlice->TreeD()->Reset();
1028
1029 }
1030
1031
1032 //____________________________________________________________________________
1033 void AliITS::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt)
1034 {
1035   
1036    char *all = strstr(opt,"All");
1037    char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
1038
1039    static Bool_t first=kTRUE;
1040    if (first) {
1041        MakeTreeC("C");
1042        first=kFALSE;
1043    }
1044    TTree *iTC=TreeC();
1045  
1046    TBranch *branch;
1047    AliITSClusterFinder* rec;
1048
1049    TObjArray *branches=gAlice->TreeR()->GetListOfBranches();
1050    AliITSgeom *geom = GetITSgeom();
1051
1052    Int_t id,module;
1053    for(id=0;id<fNDetTypes;id++) {
1054         if (!all && !det[id]) continue;
1055         branch = (TBranch*)branches->UncheckedAt(id);
1056         AliITSDetType *iDetType=DetType(id); 
1057         rec = (AliITSClusterFinder*)iDetType->GetReconstructionModel();
1058         if (!rec) {
1059            Error("DigitsToRecPoints","The cluster finder class was not instantiated!");
1060            exit(1);
1061            // or SetDefaultClusterFinder(id,iDetType*);
1062         }
1063         TClonesArray *itsDigits  = this->DigitsAddress(id);
1064
1065         Int_t first = geom->GetStartDet(id);
1066         Int_t last = geom->GetLastDet(id);
1067         //printf("det type %d first, last %d %d \n",id,first,last);
1068         for(module=first;module<=last;module++) {
1069               //printf("AliITS: module=%d\n",module);
1070               this->ResetDigits();
1071               if (all) gAlice->TreeD()->GetEvent(lastentry+module);
1072               else gAlice->TreeD()->GetEvent(lastentry+(module-first));
1073               Int_t ndigits = itsDigits->GetEntriesFast();
1074               if (ndigits) rec->FindRawClusters();
1075               gAlice->TreeR()->Fill(); 
1076               ResetRecPoints();
1077               iTC->Fill();
1078               ResetClusters();
1079               // try and fill only the branch 
1080               //branch->Fill();
1081               //ResetRecPoints(id);
1082         } // loop over modules
1083    } // loop over detector types
1084
1085
1086    //Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries();
1087
1088    //Int_t ncentries=(Int_t)TC->GetEntries();
1089
1090    char hname[30];
1091    sprintf(hname,"TreeR%d",evNumber);
1092    gAlice->TreeR()->Write(hname);
1093    // reset tree
1094    gAlice->TreeR()->Reset();
1095
1096    sprintf(hname,"TreeC%d",evNumber);
1097    iTC->Write(hname);
1098    iTC->Reset();
1099 }
1100
1101
1102 //____________________________________________________________________________
1103 void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
1104 Option_t *option,Option_t *opt,Text_t *filename)
1105 {
1106     // keep galice.root for signal and name differently the file for 
1107     // background when add! otherwise the track info for signal will be lost !
1108   
1109    char *all = strstr(opt,"All");
1110    char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")};
1111
1112    Int_t nmodules;
1113    InitModules(size,nmodules); 
1114    FillModules(evNumber,bgrev,lastev,nmodules,option,filename);
1115
1116    static AliITSsimulationFastPoints* sim=0;
1117    if (!sim) sim = new AliITSsimulationFastPoints(); 
1118
1119
1120    AliITSgeom *geom = GetITSgeom();
1121
1122    Int_t id,module;
1123    for(id=0;id<3;id++) {
1124         if (!all && !det[id]) continue;
1125         Int_t first = geom->GetStartDet(id);
1126         Int_t last = geom->GetLastDet(id);
1127         for(module=first;module<=last;module++) {
1128             AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module);
1129             sim->CreateFastRecPoints(mod);
1130             gAlice->TreeR()->Fill(); 
1131             ResetRecPoints();
1132         } // loop over modules
1133    } // loop over detector types
1134
1135
1136    ClearModules();
1137
1138    //Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries();
1139
1140    char hname[30];
1141    sprintf(hname,"TreeR%d",evNumber);
1142    gAlice->TreeR()->Write(hname);
1143    // reset tree
1144    gAlice->TreeR()->Reset();
1145
1146 }
1147
1148 //______________________________________________________________________________
1149 void AliITS::Streamer(TBuffer &R__b)
1150 {
1151    // Stream an object of class AliITS.
1152
1153   Int_t i, j, l;
1154
1155    if (R__b.IsReading()) {
1156       Version_t R__v = R__b.ReadVersion();
1157       if (R__v) {
1158           AliDetector::Streamer(R__b);
1159           R__b >> fITSgeom;
1160           R__b >> fITSmodules;
1161           R__b >> fEuclidOut;
1162           R__b >> fIdN;
1163           delete []fIdSens; 
1164           fIdSens = new Int_t[fIdN];
1165           R__b.ReadFastArray(fIdSens,fIdN);
1166           if(fIdName!=0) delete[] fIdName;
1167           fIdName = new char*[fIdN];
1168           for(i=0;i<fIdN;i++){
1169               R__b >> l;
1170               fIdName[i] = new char[l+1]; // add room for null character.
1171               for(j=0;j<l;j++) R__b >> fIdName[i][j];
1172               fIdName[i][l] = '\0'; // Null terminate this string.
1173           } // end for i
1174           //R__b.ReadArray(fIdName);
1175           R__b >> fMajorVersion;
1176           R__b >> fMinorVersion;
1177           R__b >> fDetTypes;
1178           R__b >> fNDetTypes;
1179           R__b >> fDtype;
1180           delete []fNdtype; 
1181           fNdtype = new Int_t[fNDetTypes];
1182           R__b.ReadFastArray(fNdtype,fNDetTypes);
1183           R__b >> fCtype;
1184           delete []fNctype; 
1185           fNctype = new Int_t[fNDetTypes];
1186           R__b.ReadFastArray(fNctype,fNDetTypes);
1187           R__b >> fRecPoints;
1188           R__b >> fNRecPoints;
1189           R__b >> fTracks;
1190           R__b >> fTreeC;
1191       } // end if R__v
1192    } else { // writing
1193       R__b.WriteVersion(AliITS::IsA());
1194       AliDetector::Streamer(R__b);
1195       R__b << fITSgeom;
1196       R__b << fITSmodules;
1197       R__b << fEuclidOut;
1198       R__b << fIdN;
1199       R__b.WriteFastArray(fIdSens,fIdN);
1200       for(i=0;i<fIdN;i++){
1201           l = strlen(fIdName[i]);
1202           R__b << l;
1203           for(j=0;j<l;j++) R__b << fIdName[i][j];
1204       } // end for i
1205       //R__b.WriteArray(fIdName, __COUNTER__);
1206       R__b << fMajorVersion;
1207       R__b << fMinorVersion;
1208       R__b << fDetTypes;
1209       R__b << fNDetTypes;
1210       R__b << fDtype;
1211       R__b.WriteFastArray(fNdtype,fNDetTypes);
1212       R__b << fCtype;
1213       R__b.WriteFastArray(fNctype,fNDetTypes);
1214       R__b << fRecPoints;
1215       // fRecPoints->Streamer(R__b);
1216       R__b << fNRecPoints;
1217       R__b << fTracks;
1218       //      fTracks->Streamer(R__b);
1219       R__b << fTreeC;
1220    } // end if
1221 }
1222
1223 ClassImp(AliITSRecPoint)
1224 ClassImp(AliITSsegmentation)
1225 ClassImp(AliITSresponse)        
1226 //ClassImp(AliITStrack)
1227