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