MC-dependent part of AliRun extracted in AliMC (F.Carminati)
[u/mrichter/AliRoot.git] / 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