FMD/AliFMD.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 //  Forward Multiplicity Detector based on Silicon plates                    //
  18 //  This class contains the base procedures for the Forward Multiplicity     //
  19 //  detector                                                                 //
  20 //  Detector consists of 6 Si volumes covered pseudorapidity interval         //
  21 //  from 1.6 to 6.0.                                                         //
  22 //                                                                           //
  23 //Begin_Html
  24 /*
  25 <img src="gif/AliFMDClass.gif">
  26 </pre>
  27 <br clear=left>
  28 <font size=+2 color=red>
  29 <p>The responsible person for this module is
  30 <a href="mailto:Alla.Maevskaia@cern.ch">Alla Maevskaia</a>.
  31 </font>
  32 <pre>
  33 */
  34 //End_Html
  35 //                                                                           //
  36 //                                                                           //
  37 ///////////////////////////////////////////////////////////////////////////////
  38
  39 #define DEBUG
  40 #include <TMath.h>
  41 #include <TGeometry.h>
  42 #include <TTUBE.h>
  43 #include <TTree.h>
  44 #include <TNode.h>
  45 #include <TFile.h>
  46
  47 #include <TClonesArray.h>
  48 #include <TLorentzVector.h>
  49 #include "AliFMDv1.h"
  50 #include "AliRun.h"
  51 #include "AliDetector.h"
  52 #include <Riostream.h>
  53 #include "AliMagF.h"
  54 #include "AliFMDhit.h"
  55 #include "AliFMDdigit.h"
  56 #include "AliFMDReconstruction.h"
  57 #include "AliFMDReconstParticles.h"
  58 #include <stdlib.h>
  59
  60
  61 ClassImp (AliFMD)
  62   //_____________________________________________________________________________
  63 AliFMD::AliFMD ():AliDetector ()
  64 {
  65   //
  66   // Default constructor for class AliFMD
  67   //
  68   fIshunt = 0;
  69   fHits     = 0;
  70   fDigits   = 0;
  71   fSDigits  = 0;
  72   fReconParticles=0;
  73 }
  74
  75 //_____________________________________________________________________________
  76 AliFMD::AliFMD (const char *name, const char *title):
  77 AliDetector (name, title)
  78 {
  79   //
  80   // Standard constructor for Forward Multiplicity Detector
  81   //
  82
  83   //
  84   // Initialise Hit array
  85   fHits = new TClonesArray ("AliFMDhit", 1000);
  86   // Digits for each Si disk
  87   fDigits = new TClonesArray ("AliFMDdigit", 1000);
  88   fSDigits = new TClonesArray ("AliFMDdigit", 1000);
  89   fReconParticles=new TClonesArray("AliFMDReconstParticles",1000);
  90   gAlice->AddHitList (fHits);
  91
  92   fIshunt = 0;
  93   fIdSens1 = 0;
  94   fIdSens2 = 0;
  95   fIdSens3 = 0;
  96   fIdSens4 = 0;
  97   fIdSens5 = 0;
  98   //  fMerger = 0;
  99   SetMarkerColor (kRed);
 100 }
 101
 102 //-----------------------------------------------------------------------------
 103 AliFMD::~AliFMD ()
 104 {
 105   if (fHits)
 106     {
 107       fHits->Delete ();
 108       delete fHits;
 109       fHits = 0;
 110     }
 111   if (fDigits)
 112     {
 113       fDigits->Delete ();
 114       delete fDigits;
 115       fDigits = 0;
 116     }
 117   if (fSDigits)
 118     {
 119       fSDigits->Delete ();
 120       delete fSDigits;
 121       fSDigits = 0;
 122     }
 123   if (fReconParticles)
 124     {
 125       fReconParticles->Delete ();
 126       delete fReconParticles;
 127       fReconParticles = 0;
 128     }
 129
 130 }
 131
 132 //_____________________________________________________________________________
 133 void AliFMD::AddHit (Int_t track, Int_t * vol, Float_t * hits)
 134 {
 135   //
 136   // Add a hit to the list
 137   //
 138   TClonesArray & lhits = *fHits;
 139   new (lhits[fNhits++]) AliFMDhit (fIshunt, track, vol, hits);
 140 }
 141
 142 //_____________________________________________________________________________
 143 void AliFMD::AddDigit (Int_t * digits)
 144 {
 145   // add a real digit - as coming from data
 146
 147
 148   TClonesArray & ldigits = *fDigits;
 149   new (ldigits[fNdigits++]) AliFMDdigit (digits);
 150
 151 }
 152 //_____________________________________________________________________________
 153 void AliFMD::AddSDigit (Int_t * digits)
 154 {
 155   // add a real digit - as coming from data
 156
 157   TClonesArray & ldigits = *fSDigits;
 158   new (ldigits[fNdigits++]) AliFMDdigit (digits);
 159
 160 }
 161 //_____________________________________________________________________________
 162 void AliFMD::BuildGeometry ()
 163 {
 164   //
 165   // Build simple ROOT TNode geometry for event display
 166   //
 167   TNode *node, *top;
 168   const int kColorFMD = 5;
 169   //
 170   top = gAlice->GetGeometry ()->GetNode ("alice");
 171
 172   // FMD define the different volumes
 173   new TRotMatrix ("rot901", "rot901", 90, 0, 90, 90, 180, 0);
 174
 175   new TTUBE ("S_FMD0", "FMD  volume 0", "void", 4.2, 17.2, 1.5);
 176   top->cd ();
 177   node = new TNode ("FMD0", "FMD0", "S_FMD0", 0, 0, 62.8, "");
 178   node->SetLineColor (kColorFMD);
 179   fNodes->Add (node);
 180
 181   new TTUBE ("S_FMD1", "FMD  volume 1", "void", 15.4, 28.4, 1.5);
 182   top->cd ();
 183   node = new TNode ("FMD1", "FMD1", "S_FMD1", 0, 0, 75.2, "");
 184   node->SetLineColor (kColorFMD);
 185   fNodes->Add (node);
 186
 187   new TTUBE ("S_FMD2", "FMD  volume 2", "void", 4.2, 17.2, 1.5);
 188   top->cd ();
 189   node = new TNode ("FMD2", "FMD2", "S_FMD2", 0, 0, -83.2, "");
 190   node->SetLineColor (kColorFMD);
 191   fNodes->Add (node);
 192
 193   new TTUBE ("S_FMD3", "FMD  volume 3", "void", 15.4, 28.4, 1.5);
 194   top->cd ();
 195   node = new TNode ("FMD3", "FMD3", "S_FMD3", 0, 0, -75.2, "");
 196   node->SetLineColor (kColorFMD);
 197   fNodes->Add (node);
 198
 199   new TTUBE ("S_FMD4", "FMD  volume 4", "void", 4.2, 17.2, 1.5);
 200   top->cd ();
 201   //  node = new TNode("FMD4","FMD4","S_FMD4",0,0,-270,"");
 202   node = new TNode ("FMD4", "FMD4", "S_FMD4", 0, 0, -340, "");
 203   node->SetLineColor (kColorFMD);
 204   fNodes->Add (node);
 205 }
 206
 207 //_____________________________________________________________________________
 208 Int_t AliFMD::DistanceToPrimitive (Int_t px, Int_t py)
 209 {
 210   //
 211   // Calculate the distance from the mouse to the FMD on the screen
 212   // Dummy routine
 213   //
 214   return 9999;
 215 }
 216
 217 //___________________________________________
 218 void AliFMD::ResetHits ()
 219 {
 220   // Reset number of clusters and the cluster array for this detector
 221   AliDetector::ResetHits ();
 222 }
 223
 224 //____________________________________________
 225 void AliFMD::ResetDigits ()
 226 {
 227   //
 228   // Reset number of digits and the digits array for this detector
 229   AliDetector::ResetDigits ();
 230   //
 231 }
 232
 233 //-------------------------------------------------------------------------
 234 void  AliFMD::Init ()
 235 {
 236   //
 237   // Initialis the FMD after it has been built
 238   Int_t i;
 239   //
 240   if (fDebug)
 241     {
 242       printf ("\n%s: ", ClassName ());
 243       for (i = 0; i < 35; i++)
 244         printf ("*");
 245       printf (" FMD_INIT ");
 246       for (i = 0; i < 35; i++)
 247         printf ("*");
 248       printf ("\n%s: ", ClassName ());
 249       //
 250       // Here the FMD initialisation code (if any!)
 251       for (i = 0; i < 80; i++)
 252         printf ("*");
 253       printf ("\n");
 254     }
 255   //
 256   //
 257     fIdSens1 = gMC->VolId ("GRN1");     //Si sensetive volume
 258     fIdSens2 = gMC->VolId ("GRN2");     //Si sensetive volume
 259     fIdSens3 = gMC->VolId ("GRN3");     //Si sensetive volume
 260     fIdSens4 = gMC->VolId ("GRN4");     //Si sensetive volume
 261
 262 }
 263 //---------------------------------------------------------------------
 264 void AliFMD::MakeBranch (Option_t * option, const char *file)
 265 {
 266   // Create Tree branches for the FMD.
 267   char branchname[10];
 268   const Int_t kBufferSize = 16000;
 269   sprintf (branchname, "%s", GetName ());
 270
 271   AliDetector::MakeBranch (option, file);
 272   const char *cD = strstr(option,"D");
 273   const char *cR = strstr(option,"R");
 274   const char *cS = strstr(option,"S");
 275
 276   if (cS){
 277
 278     MakeBranchInTree(gAlice->TreeS(),
 279                      branchname,&fSDigits,
 280                      kBufferSize, file);
 281   }
 282   if (cD){
 283
 284     MakeBranchInTree(gAlice->TreeD(),
 285                      branchname,&fDigits,
 286                      kBufferSize, file);
 287     cout<<" tree "<<gAlice->TreeD()<<" "<<branchname<<" "<<&fDigits<<endl;
 288   }
 289   if (cR){
 290     MakeBranchInTree(gAlice->TreeR(),
 291                      branchname,&fReconParticles,
 292                      kBufferSize, file);
 293   }
 294
 295 }
 296
 297 //_____________________________________________________________________________
 298 void AliFMD::SetTreeAddress ()
 299 {
 300   // Set branch address for the Hits and Digits Tree.
 301   char branchname[30];
 302   AliDetector::SetTreeAddress ();
 303
 304   TBranch *branch;
 305   TTree *treeD = gAlice->TreeD ();
 306
 307
 308   if (treeD)
 309     {
 310       if (fDigits)
 311         {
 312           branch = treeD->GetBranch (branchname);
 313           if (branch)
 314             branch->SetAddress (&fDigits);
 315         }
 316
 317     }
 318   if (fSDigits)
 319     //  fSDigits->Clear ();
 320
 321   if (gAlice->TreeS () && fSDigits)
 322     {
 323       branch = gAlice->TreeS ()->GetBranch ("FMD");
 324       if (branch)
 325         branch->SetAddress (&fSDigits);
 326     }
 327
 328   if (gAlice->TreeR() && fReconParticles)
 329     {
 330       branch = gAlice->TreeR()->GetBranch("FMD");
 331       if (branch) branch->SetAddress(&fReconParticles) ;
 332     }
 333 }
 334
 335 //---------------------------------------------------------------------
 336
 337 void AliFMD::SetRingsSi1(Int_t ringsSi1)
 338 {
 339   //  fRingsSi1=ringsSi1;
 340   fRingsSi1=256;
 341 }
 342 void AliFMD::SetSectorsSi1(Int_t sectorsSi1)
 343 {
 344   fSectorsSi1=20;
 345 }
 346 void AliFMD::SetRingsSi2(Int_t ringsSi2)
 347 {
 348   fRingsSi2=128;
 349 }
 350 void AliFMD::SetSectorsSi2(Int_t sectorsSi2)
 351 {
 352   fSectorsSi2=40;
 353 }
 354
 355 //---------------------------------------------------------------------
 356 /*
 357 void AliFMD::SDigits2Digits()
 358 {
 359   cout<<"AliFMD::SDigits2Digits"<<endl;
 360     if (!fMerger) {
 361       fMerger = new AliFMDMerger();
 362     }
 363
 364     fMerger ->SetRingsSi1(fRingsSi1);
 365     fMerger->SetRingsSi2(fRingsSi2);
 366     fMerger ->SetSectorsSi1(fSectorsSi1);
 367     fMerger ->SetSectorsSi2(fSectorsSi2);
 368
 369     fMerger->Init();
 370     cout<<"AliFMD::SDigits2Digits Init"<<endl;
 371     fMerger->Digitise();
 372     cout<<"AliFMD::SDigits2Digits Digitise() "<<endl;
 373
 374
 375     }
 376
 377     //---------------------------------------------------------------------
 378 void   AliFMD::SetMerger(AliFMDMerger* merger)
 379 {
 380 // Set pointer to merger
 381     fMerger = merger;
 382 }
 383
 384 AliFMDMerger*  AliFMD::Merger()
 385 {
 386 // Return pointer to merger
 387     return fMerger;
 388 }
 389 */
 390 //---------------------------------------------------------------------
 391
 392
 393
 394 void
 395 AliFMD::Eta2Radius (Float_t eta, Float_t zDisk, Float_t * radius)
 396 {
 397   Float_t expEta = TMath::Exp (-eta);
 398   Float_t theta = TMath::ATan (expEta);
 399   theta = 2. * theta;
 400   Float_t rad = zDisk * (TMath::Tan (theta));
 401   *radius = rad;
 402
 403   if (fDebug)
 404     printf ("%s: eta %f radius %f\n", ClassName (), eta, rad);
 405 }
 406
 407 //---------------------------------------------------------------------
 408
 409 void AliFMD::Hits2SDigits ()
 410 {
 411
 412   //#ifdef DEBUG
 413   cout<<"ALiFMD::Hits2SDigits> start...\n";
 414   //#endif
 415
 416   char * fileSDigits = "FMD.SDigits.root";
 417   char * fileHeader = 0;
 418   AliFMDSDigitizer * sd = new AliFMDSDigitizer(fileHeader,fileSDigits) ;
 419   sd->SetRingsSi1(fRingsSi1);
 420   sd->SetRingsSi2(fRingsSi2);
 421   sd->SetSectorsSi1(fSectorsSi1);
 422   sd->SetSectorsSi2(fSectorsSi2);
 423   //  sd->SetEventNumber(fEvNrSig);
 424   sd->Exec("") ;
 425
 426   delete sd ;
 427
 428 }
 429 //-----------------------------------------------------------------------
 430
 431 void AliFMD::Digits2Reco()
 432 {
 433   char * fileReconParticles=0;
 434   char * fileHeader=0;
 435   AliFMDReconstruction * reconstruction =
 436     new AliFMDReconstruction(fileHeader,fileReconParticles) ;
 437   //  fReconParticles=new TClonesArray("AliFMDReconstParticles",1000);
 438   reconstruction->Exec("");
 439   delete  reconstruction;
 440 }
 441 //-----------------------------------------------------------------------
 442
 443 void AliFMD::MakeBranchInTreeD(TTree *treeD, const char *file)
 444 {
 445     //
 446     // Create TreeD branches for the MUON.
 447     //
 448
 449     const Int_t kBufferSize = 4000;
 450     char branchname[20];
 451
 452
 453     sprintf(branchname,"%s",GetName());
 454     if(treeD){
 455     MakeBranchInTree(treeD,
 456                      branchname,&fDigits,
 457                      kBufferSize, file);
 458     }
 459 }
 460