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 5 Si volumes covered pseudorapidity interval         //
  24 //  from 1.7 to 5.1.                                                         //
  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 #include "AliMC.h"
  67
  68 ClassImp (AliFMD)
  69   //_____________________________________________________________________________
  70 AliFMD::AliFMD ():AliDetector ()
  71 {
  72   //
  73   // Default constructor for class AliFMD
  74   //
  75   fIshunt = 0;
  76   fHits     = 0;
  77   fDigits   = 0;
  78   fReconParticles=0;
  79 }
  80
  81 //_____________________________________________________________________________
  82 AliFMD::AliFMD (const char *name, const char *title):
  83 AliDetector (name, title)
  84 {
  85   //
  86   // Standard constructor for Forward Multiplicity Detector
  87   //
  88
  89   //
  90   // Initialise Hit array
  91   fHits = new TClonesArray ("AliFMDhit", 1000);
  92   // Digits for each Si disk
  93   fDigits = new TClonesArray ("AliFMDdigit", 1000);
  94   fReconParticles=new TClonesArray("AliFMDReconstParticles",1000);
  95   gAlice->GetMCApp()->AddHitList (fHits);
  96
  97   fIshunt = 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 (fReconParticles)
 118     {
 119       fReconParticles->Delete ();
 120       delete fReconParticles;
 121       fReconParticles = 0;
 122     }
 123
 124 }
 125
 126 //_____________________________________________________________________________
 127 void AliFMD::AddHit (Int_t track, Int_t * vol, Float_t * hits)
 128 {
 129   //
 130   // Add a hit to the list
 131   //
 132   TClonesArray & lhits = *fHits;
 133   new (lhits[fNhits++]) AliFMDhit (fIshunt, track, vol, hits);
 134 }
 135
 136 //_____________________________________________________________________________
 137 void AliFMD::AddDigit (Int_t * digits)
 138 {
 139   // add a real digit - as coming from data
 140
 141   if (fDigits == 0x0) fDigits = new TClonesArray ("AliFMDdigit", 1000);
 142   TClonesArray & ldigits = *fDigits;
 143   new (ldigits[fNdigits++]) AliFMDdigit (digits);
 144 }
 145
 146 //_____________________________________________________________________________
 147 void AliFMD::BuildGeometry ()
 148 {
 149   //
 150   // Build simple ROOT TNode geometry for event display
 151   //
 152   TNode *node, *top;
 153   const int kColorFMD = 5;
 154   //
 155   top = gAlice->GetGeometry ()->GetNode ("alice");
 156
 157   // FMD define the different volumes
 158   new TRotMatrix ("rot901", "rot901", 90, 0, 90, 90, 180, 0);
 159
 160   new TTUBE ("S_FMD0", "FMD  volume 0", "void", 4.2, 17.2, 1.5);
 161   top->cd ();
 162   node = new TNode ("FMD0", "FMD0", "S_FMD0", 0, 0, -62.8, "");
 163   node->SetLineColor (kColorFMD);
 164   fNodes->Add (node);
 165
 166   new TTUBE ("S_FMD1", "FMD  volume 1", "void", 15.4, 28.4, 1.5);
 167   top->cd ();
 168   node = new TNode ("FMD1", "FMD1", "S_FMD1", 0, 0, -75.2, "");
 169   node->SetLineColor (kColorFMD);
 170   fNodes->Add (node);
 171
 172   new TTUBE ("S_FMD2", "FMD  volume 2", "void", 4.2, 17.2, 1.5);
 173   top->cd ();
 174   node = new TNode ("FMD2", "FMD2", "S_FMD2", 0, 0, 83.2, "");
 175   node->SetLineColor (kColorFMD);
 176   fNodes->Add (node);
 177
 178   new TTUBE ("S_FMD3", "FMD  volume 3", "void", 15.4, 28.4, 1.5);
 179   top->cd ();
 180   node = new TNode ("FMD3", "FMD3", "S_FMD3", 0, 0, 75.2, "");
 181   node->SetLineColor (kColorFMD);
 182   fNodes->Add (node);
 183
 184   new TTUBE ("S_FMD4", "FMD  volume 4", "void", 4.2, 17.2, 1.5);
 185   top->cd ();
 186   node = new TNode ("FMD4", "FMD4", "S_FMD4", 0, 0, 340, "");
 187   node->SetLineColor (kColorFMD);
 188   fNodes->Add (node);
 189 }
 190
 191 //_____________________________________________________________________________
 192 Int_t AliFMD::DistanceToPrimitive (Int_t /*px*/, Int_t /*py*/)
 193 {
 194   //
 195   // Calculate the distance from the mouse to the FMD on the screen
 196   // Dummy routine
 197   //
 198   return 9999;
 199 }
 200
 201 //___________________________________________
 202 void AliFMD::ResetHits ()
 203 {
 204   // Reset number of clusters and the cluster array for this detector
 205   AliDetector::ResetHits ();
 206 }
 207
 208 //____________________________________________
 209 void AliFMD::ResetDigits ()
 210 {
 211   //
 212   // Reset number of digits and the digits array for this detector
 213   AliDetector::ResetDigits ();
 214   //
 215 }
 216
 217 //-------------------------------------------------------------------------
 218 void  AliFMD::Init ()
 219 {
 220   //
 221   // Initialis the FMD after it has been built
 222   Int_t i;
 223   //
 224   if (fDebug)
 225     {
 226       printf ("\n%s: ", ClassName ());
 227       for (i = 0; i < 35; i++)
 228         printf ("*");
 229       printf (" FMD_INIT ");
 230       for (i = 0; i < 35; i++)
 231         printf ("*");
 232       printf ("\n%s: ", ClassName ());
 233       //
 234       // Here the FMD initialisation code (if any!)
 235       for (i = 0; i < 80; i++)
 236         printf ("*");
 237       printf ("\n");
 238     }
 239   //
 240   //
 241
 242 }
 243 //---------------------------------------------------------------------
 244 void AliFMD::MakeBranch (Option_t * option)
 245 {
 246   // Create Tree branches for the FMD.
 247   char branchname[10];
 248   const Int_t kBufferSize = 16000;
 249   sprintf (branchname, "%s", GetName ());
 250
 251   const char *cH = strstr(option,"H");
 252   const char *cD = strstr(option,"D");
 253   const char *cR = strstr(option,"R");
 254
 255   if (cH && (fHits == 0x0)) fHits = new TClonesArray ("AliFMDhit", 1000);
 256
 257   AliDetector::MakeBranch (option);
 258
 259   if (cD){
 260     if (fDigits == 0x0) fDigits = new TClonesArray ("AliFMDdigit", 1000);
 261     MakeBranchInTree(fLoader->TreeD(), branchname,&fDigits, kBufferSize, 0);
 262   }
 263
 264   if (cR){
 265     if (fReconParticles == 0x0)
 266       fReconParticles=new TClonesArray("AliFMDReconstParticles",1000);
 267     MakeBranchInTree(fLoader->TreeR(), branchname,&fReconParticles, kBufferSize, 0);
 268   }
 269
 270 }
 271
 272 //_____________________________________________________________________________
 273 void AliFMD::SetTreeAddress ()
 274 {
 275   // Set branch address for the Hits and Digits Tree.
 276   char branchname[30];
 277
 278   if (fLoader->TreeH() && (fHits == 0x0))
 279     fHits = new TClonesArray ("AliFMDhit", 1000);
 280
 281   AliDetector::SetTreeAddress ();
 282
 283   TBranch *branch;
 284   TTree *treeD = fLoader->TreeD();
 285
 286   if (treeD)
 287     {
 288       if (fDigits == 0x0) fDigits = new TClonesArray ("AliFMDdigit", 1000);
 289       branch = treeD->GetBranch (branchname);
 290       if (branch)
 291        branch->SetAddress (&fDigits);
 292     }
 293
 294   if (fLoader->TreeR() && fReconParticles)
 295     {
 296       if (fReconParticles == 0x0)
 297         fReconParticles=new TClonesArray("AliFMDReconstParticles",1000);
 298       branch = fLoader->TreeR()->GetBranch("FMD");
 299       if (branch) branch->SetAddress(&fReconParticles) ;
 300     }
 301 }
 302
 303
 304
 305 void
 306 AliFMD::Eta2Radius (Float_t eta, Float_t zDisk, Float_t * radius)
 307 {
 308   Float_t expEta = TMath::Exp (-eta);
 309   Float_t theta = TMath::ATan (expEta);
 310   theta = 2. * theta;
 311   Float_t rad = zDisk * (TMath::Tan (theta));
 312   *radius = rad;
 313
 314   if (fDebug)
 315     printf ("%s: eta %f radius %f\n", ClassName (), eta, rad);
 316 }
 317
 318 //---------------------------------------------------------------------
 319
 320
 321 void AliFMD::Digits2Reco()
 322 {
 323   AliFMDReconstruction * reconstruction =  new AliFMDReconstruction(fLoader->GetRunLoader());
 324   cout<<" AliFMD::Digits2Reco >> "<<reconstruction<<endl;
 325   reconstruction->Exec();
 326   delete  reconstruction;
 327 }
 328 //-----------------------------------------------------------------------
 329
 330 void AliFMD::MakeBranchInTreeD(TTree *treeD, const char *file)
 331 {
 332     //
 333     // Create TreeD branches for the FMD
 334     //
 335     const Int_t kBufferSize = 4000;
 336     char branchname[20];
 337     sprintf(branchname,"%s",GetName());
 338     if(treeD)
 339      {
 340        MakeBranchInTree(treeD,  branchname,&fDigits, kBufferSize, file);
 341      }
 342 }
 343