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