code according Convertions
[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 //                                                                            //
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
40 #define DEBUG
41 #include <TMath.h>
42 #include <TGeometry.h>
43 #include <TTUBE.h>
44 #include <TTree.h>
45 #include <TNode.h>
46 #include <TFile.h>
47
48 #include <TClonesArray.h>
49 #include <TLorentzVector.h>
50 #include "AliFMDv1.h"
51 #include "AliRun.h"
52 #include "AliDetector.h"
53 #include <TBranch.h>
54 #include <Riostream.h>
55 #include "AliMagF.h"
56 #include "AliFMDhit.h"
57 #include "AliFMDdigit.h"
58 #include "AliFMDReconstruction.h"
59 #include "AliFMDReconstParticles.h"
60 #include <stdlib.h>
61
62 ClassImp (AliFMD)
63   //_____________________________________________________________________________
64 AliFMD::AliFMD ():AliDetector ()
65 {
66   //
67   // Default constructor for class AliFMD
68   //
69   fIshunt = 0;
70   fHits     = 0;
71   fDigits   = 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   fReconParticles=new TClonesArray("AliFMDReconstParticles",1000); 
89   gAlice->AddHitList (fHits);
90
91   fIshunt = 0;
92   fIdSens1 = 0;
93   fIdSens2 = 0;
94   fIdSens3 = 0;
95   fIdSens4 = 0;
96   fIdSens5 = 0;
97   //  fMerger = 0;
98   SetMarkerColor (kRed);
99 }
100
101 //-----------------------------------------------------------------------------
102 AliFMD::~AliFMD ()
103 {
104   if (fHits)
105     {
106       fHits->Delete ();
107       delete fHits;
108       fHits = 0;
109     }
110   if (fDigits)
111     {
112       fDigits->Delete ();
113       delete fDigits;
114       fDigits = 0;
115     }
116   if (fReconParticles)
117     {
118       fReconParticles->Delete ();
119       delete fReconParticles;
120       fReconParticles = 0;
121     }
122
123 }
124
125 //_____________________________________________________________________________
126 void AliFMD::AddHit (Int_t track, Int_t * vol, Float_t * hits)
127 {
128   //
129   // Add a hit to the list
130   //
131   TClonesArray & lhits = *fHits;
132   new (lhits[fNhits++]) AliFMDhit (fIshunt, track, vol, hits);
133 }
134
135 //_____________________________________________________________________________
136 void AliFMD::AddDigit (Int_t * digits)
137 {
138   // add a real digit - as coming from data
139
140
141   TClonesArray & ldigits = *fDigits;
142   new (ldigits[fNdigits++]) AliFMDdigit (digits);
143
144 }
145 //_____________________________________________________________________________
146 void AliFMD::BuildGeometry ()
147 {
148   //
149   // Build simple ROOT TNode geometry for event display
150   //
151   TNode *node, *top;
152   const int kColorFMD = 5;
153   //
154   top = gAlice->GetGeometry ()->GetNode ("alice");
155
156   // FMD define the different volumes
157   new TRotMatrix ("rot901", "rot901", 90, 0, 90, 90, 180, 0);
158
159   new TTUBE ("S_FMD0", "FMD  volume 0", "void", 4.2, 17.2, 1.5);
160   top->cd ();
161   node = new TNode ("FMD0", "FMD0", "S_FMD0", 0, 0, 62.8, "");
162   node->SetLineColor (kColorFMD);
163   fNodes->Add (node);
164
165   new TTUBE ("S_FMD1", "FMD  volume 1", "void", 15.4, 28.4, 1.5);
166   top->cd ();
167   node = new TNode ("FMD1", "FMD1", "S_FMD1", 0, 0, 75.2, "");
168   node->SetLineColor (kColorFMD);
169   fNodes->Add (node);
170
171   new TTUBE ("S_FMD2", "FMD  volume 2", "void", 4.2, 17.2, 1.5);
172   top->cd ();
173   node = new TNode ("FMD2", "FMD2", "S_FMD2", 0, 0, -83.2, "");
174   node->SetLineColor (kColorFMD);
175   fNodes->Add (node);
176
177   new TTUBE ("S_FMD3", "FMD  volume 3", "void", 15.4, 28.4, 1.5);
178   top->cd ();
179   node = new TNode ("FMD3", "FMD3", "S_FMD3", 0, 0, -75.2, "");
180   node->SetLineColor (kColorFMD);
181   fNodes->Add (node);
182
183   new TTUBE ("S_FMD4", "FMD  volume 4", "void", 4.2, 17.2, 1.5);
184   top->cd ();
185   //  node = new TNode("FMD4","FMD4","S_FMD4",0,0,-270,"");
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
219 void  AliFMD::Init ()
220 {
221   //
222   // Initialis the FMD after it has been built
223   Int_t i;
224   //
225   if (fDebug)
226     {
227       printf ("\n%s: ", ClassName ());
228       for (i = 0; i < 35; i++)
229         printf ("*");
230       printf (" FMD_INIT ");
231       for (i = 0; i < 35; i++)
232         printf ("*");
233       printf ("\n%s: ", ClassName ());
234       //
235       // Here the FMD initialisation code (if any!)
236       for (i = 0; i < 80; i++)
237         printf ("*");
238       printf ("\n");
239     }
240   //
241   //
242     fIdSens1 = gMC->VolId ("GRN1");     //Si sensetive volume
243     fIdSens2 = gMC->VolId ("GRN2");     //Si sensetive volume
244     fIdSens3 = gMC->VolId ("GRN3");     //Si sensetive volume
245     fIdSens4 = gMC->VolId ("GRN4");     //Si sensetive volume
246     fIdSens5 = gMC->VolId ("GRN5");     //Si sensetive volume
247
248 }
249
250 //---------------------------------------------------------------------
251 void AliFMD::MakeBranch (Option_t * option, const char *file)
252 {
253   // Create Tree branches for the FMD.
254   char branchname[10];
255   const Int_t kBufferSize = 16000;
256   sprintf (branchname, "%s", GetName ());
257   
258   AliDetector::MakeBranch (option, file);
259   const char *cD = strstr(option,"D");
260   const char *cR = strstr(option,"R");
261   
262   if (cD){
263
264     MakeBranchInTree(gAlice->TreeD(), 
265                      branchname,&fDigits, 
266                      kBufferSize, file);
267     cout<<" tree "<<gAlice->TreeD()<<" "<<branchname<<" "<<&fDigits<<endl;
268   }
269   if (cR){
270     MakeBranchInTree(gAlice->TreeR(), 
271                      branchname,&fReconParticles,
272                      kBufferSize, file);
273   }
274   
275 }
276
277 //_____________________________________________________________________________
278 void AliFMD::SetTreeAddress ()
279 {
280   // Set branch address for the Hits and Digits Tree.
281   char branchname[30];
282   AliDetector::SetTreeAddress ();
283
284   TBranch *branch;
285   TTree *treeD = gAlice->TreeD ();
286
287
288   if (treeD)
289     {
290       if (fDigits)
291         {
292           branch = treeD->GetBranch (branchname);
293           if (branch)
294             branch->SetAddress (&fDigits);
295         }
296
297     }
298
299  
300   if (gAlice->TreeR() && fReconParticles) 
301     {
302       branch = gAlice->TreeR()->GetBranch("FMD"); 
303       if (branch) branch->SetAddress(&fReconParticles) ;
304     }   
305 }
306
307 //---------------------------------------------------------------------
308
309 void AliFMD::SetRingsSi1(Int_t ringsSi1)
310 {
311   //  fRingsSi1=ringsSi1;
312   fRingsSi1=768;
313 }
314 void AliFMD::SetSectorsSi1(Int_t sectorsSi1)
315 {
316   fSectorsSi1=20;
317 }
318 void AliFMD::SetRingsSi2(Int_t ringsSi2)
319 {
320   fRingsSi2=384;
321 }
322 void AliFMD::SetSectorsSi2(Int_t sectorsSi2)
323 {
324   fSectorsSi2=40;
325 }
326
327 //---------------------------------------------------------------------
328
329
330
331 void
332 AliFMD::Eta2Radius (Float_t eta, Float_t zDisk, Float_t * radius)
333 {
334   Float_t expEta = TMath::Exp (-eta);
335   Float_t theta = TMath::ATan (expEta);
336   theta = 2. * theta;
337   Float_t rad = zDisk * (TMath::Tan (theta));
338   *radius = rad;
339
340   if (fDebug)
341     printf ("%s: eta %f radius %f\n", ClassName (), eta, rad);
342 }
343
344 //-----------------------------------------------------------------------
345
346 void AliFMD::Digits2Reco()
347 {
348   char * fileReconParticles=0;
349   char * fileHeader=0;
350   AliFMDReconstruction * reconstruction =
351     new AliFMDReconstruction(fileHeader,fileReconParticles) ;
352   reconstruction->Exec("");
353   delete  reconstruction;
354 }
355 //-----------------------------------------------------------------------
356
357 void AliFMD::MakeBranchInTreeD(TTree *treeD, const char *file)
358 {
359     //
360     // Create TreeD branches for the MUON.
361     //
362
363     const Int_t kBufferSize = 4000;
364     char branchname[20];
365     
366
367     sprintf(branchname,"%s",GetName()); 
368     if(treeD){
369     MakeBranchInTree(treeD, 
370                      branchname,&fDigits, 
371                      kBufferSize, file);
372     }
373 }
374