Fixed Coverity issues
[u/mrichter/AliRoot.git] / FMD / AliFMDv1.cxx
CommitLineData
37c4363a 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 **************************************************************************/
37c4363a 15/* $Id$ */
c2fc1258 16/** @file AliFMDv1.cxx
17 @author Christian Holm Christensen <cholm@nbi.dk>
18 @date Mon Mar 27 12:48:51 2006
19 @brief Concrete implementation of FMD detector driver - detailed
20 version
02a27b50 21 @ingroup FMD_sim
c2fc1258 22*/
37c4363a 23//____________________________________________________________________
24//
25// Forward Multiplicity Detector based on Silicon wafers. This class
26// contains the base procedures for the Forward Multiplicity detector
1a1fdef7 27// Detector consists of 3 sub-detectors FMD1, FMD2, and FMD3, each of
28// which has 1 or 2 rings of silicon sensors.
37c4363a 29// This class contains the detailed version of the FMD - that is, hits
30// are produced during simulation.
31//
37c4363a 32// See also the class AliFMD for a more detailed explanation of the
33// various componets.
1a1fdef7 34//
56b1929b 35#include <TVirtualMC.h> // ROOT_TVirtualMC
36#include <AliRun.h> // ALIRUN_H
37#include <AliMC.h> // ALIMC_H
f95a63c4 38// #include <AliLog.h> // ALILOG_H
39#include "AliFMDDebug.h" // Better debug macros
e802be3e 40#include "AliFMDv1.h" // ALIFMDV1_H
02a27b50 41// #include "AliFMDGeometryBuilder.h"
54e415a8 42#include "AliFMDGeometry.h"
43#include "AliFMDDetector.h"
44#include "AliFMDRing.h"
45#include <TParticlePDG.h>
46#include <TDatabasePDG.h>
47#include "AliFMDHit.h"
37c4363a 48
49//____________________________________________________________________
925e6570 50ClassImp(AliFMDv1)
1a1fdef7 51#if 0
52 ; // This is here to keep Emacs for indenting the next line
53#endif
37c4363a 54
55
54e415a8 56//____________________________________________________________________
57Bool_t
58AliFMDv1::VMC2FMD(TLorentzVector& v, UShort_t& detector,
59 Char_t& ring, UShort_t& sector, UShort_t& strip) const
60{
02a27b50 61 // Convert VMC coordinates to detector coordinates
65af05bc 62 TVirtualMC* mc = TVirtualMC::GetMC();
54e415a8 63 AliFMDGeometry* fmd = AliFMDGeometry::Instance();
64
65 // Get track position
66 mc->TrackPosition(v);
67 Int_t moduleno; mc->CurrentVolOffID(fmd->GetModuleOff(), moduleno);
68 Int_t iring; mc->CurrentVolOffID(fmd->GetRingOff(), iring);
69 ring = Char_t(iring);
70 Int_t det; mc->CurrentVolOffID(fmd->GetDetectorOff(), det);
71 detector = det;
72
73
74 // Get the ring geometry
75 //Int_t nsec = fmd->GetDetector(detector)->GetRing(ring)->GetNSectors();
76 Int_t nstr = fmd->GetDetector(detector)->GetRing(ring)->GetNStrips();
77 Double_t lowr = fmd->GetDetector(detector)->GetRing(ring)->GetMinR();
78 Double_t theta = fmd->GetDetector(detector)->GetRing(ring)->GetTheta();
79 Double_t pitch = fmd->GetDetector(detector)->GetRing(ring)->GetPitch();
80
81 // Figure out the strip number
82 Double_t r = TMath::Sqrt(v.X() * v.X() + v.Y() * v.Y());
83 Int_t str = Int_t((r - lowr) / pitch);
84 if (str < 0 || str >= nstr) return kFALSE;
85 strip = str;
86
87 // Figure out the sector number
88 Double_t phi = TMath::ATan2(v.Y(), v.X()) * 180. / TMath::Pi();
89 if (phi < 0) phi = 360. + phi;
90 Double_t t = phi - 2 * moduleno * theta;
91 sector = 2 * moduleno;
92 if (t < 0 || t > 2 * theta) return kFALSE;
93 else if (t > theta) sector += 1;
94
f95a63c4 95 AliFMDDebug(40, ("<1> Inside an active FMD volume FMD%d%c[%2d,%3d] %s",
54e415a8 96 detector, ring, sector, strip, mc->CurrentVolPath()));
97 return kTRUE;
98}
99
100//____________________________________________________________________
101Bool_t
102AliFMDv1::VMC2FMD(Int_t copy, TLorentzVector& v,
103 UShort_t& detector, Char_t& ring,
104 UShort_t& sector, UShort_t& strip) const
105{
02a27b50 106 // Convert VMC coordinates to detector coordinates
65af05bc 107 TVirtualMC* mc = TVirtualMC::GetMC();
54e415a8 108 AliFMDGeometry* fmd = AliFMDGeometry::Instance();
109
110 strip = copy - 1;
111 Int_t sectordiv; mc->CurrentVolOffID(fmd->GetSectorOff(), sectordiv);
112 if (fmd->GetModuleOff() >= 0) {
113 Int_t module; mc->CurrentVolOffID(fmd->GetModuleOff(), module);
114 sector = 2 * module + sectordiv;
115 }
116 else
117 sector = sectordiv;
f95a63c4 118 AliFMDDebug(30, ("Getting ring volume with offset %d -> %s",
bf000c32 119 fmd->GetRingOff(),
120 mc->CurrentVolOffName(fmd->GetRingOff())));
54e415a8 121 Int_t iring; mc->CurrentVolOffID(fmd->GetRingOff(), iring);
122 ring = Char_t(iring);
123 Int_t det; mc->CurrentVolOffID(fmd->GetDetectorOff(), det);
124 detector = det;
125
126 //Double_t rz = fmd->GetDetector(detector)->GetRingZ(ring);
088f8e79 127 AliFMDDetector* gdet = fmd->GetDetector(detector);
128 AliFMDRing* gring = gdet->GetRing(ring);
bf000c32 129 if (!gring) {
130 AliFatal(Form("Ring %c not found (volume was %s at offset %d in path %s)",
a828379a 131 ring,
bf000c32 132 mc->CurrentVolOffName(fmd->GetRingOff()),
a828379a 133 fmd->GetRingOff(),
bf000c32 134 mc->CurrentVolPath()));
135 }
136 Int_t n = gring->GetNSectors();
54e415a8 137#if 0
138 if (rz < 0) {
139 Int_t s = ((n - sector + n / 2) % n) + 1;
f95a63c4 140 AliFMDDebug(1, ("Recalculating sector to %d (=%d-%d+%d/2%%%d+1 z=%f)",
54e415a8 141 s, n, sector, n, n, rz));
142 sector = s;
143 }
144#endif
145 if (sector < 1 || sector > n) {
8f6ee336 146 AliWarning(Form("sector # %d out of range (0-%d)", sector-1, n-1));
54e415a8 147 return kFALSE;
148 }
149 sector--;
150 // Get track position
151 mc->TrackPosition(v);
65af05bc 152 AliFMDDebug(40, ("<2> Inside an active FMD volume FMD%d%c[%2d,%3d] %s",
54e415a8 153 detector, ring, sector, strip, mc->CurrentVolPath()));
154
155 return kTRUE;
156}
bf000c32 157
158
159//____________________________________________________________________
160Bool_t
161AliFMDv1::CheckHit(Int_t trackno, Int_t pdg, Float_t absQ,
162 const TLorentzVector& p, Float_t edep) const
163{
02a27b50 164 // Check that a hit is good
bf000c32 165 if (AliLog::GetDebugLevel("FMD", "AliFMD") < 5) return kFALSE;
166 TVirtualMC* mc = TVirtualMC::GetMC();
167 Double_t mass = mc->TrackMass();
168 Double_t poverm = (mass == 0 ? 0 : p.P() / mass);
169
170 // This `if' is to debug abnormal energy depositions. We trigger on
171 // p/m approx larger than or equal to a MIP, and a large edep - more
172 // than 1 keV - a MIP is 100 eV.
173 if (!(edep > absQ * absQ && poverm > 1)) return kFALSE;
174
175 TArrayI procs;
176 mc->StepProcesses(procs);
177 TString processes;
178 for (Int_t ip = 0; ip < procs.fN; ip++) {
179 if (ip != 0) processes.Append(",");
180 processes.Append(TMCProcessName[procs.fArray[ip]]);
181 }
182 TDatabasePDG* pdgDB = TDatabasePDG::Instance();
183 TParticlePDG* particleType = pdgDB->GetParticle(pdg);
184 TString pname(particleType ? particleType->GetName() : "???");
185 TString what;
186 if (mc->IsTrackEntering()) what.Append("entering ");
187 if (mc->IsTrackExiting()) what.Append("exiting ");
188 if (mc->IsTrackInside()) what.Append("inside ");
189 if (mc->IsTrackDisappeared()) what.Append("disappeared ");
190 if (mc->IsTrackStop()) what.Append("stopped ");
191 if (mc->IsNewTrack()) what.Append("new ");
192 if (mc->IsTrackAlive()) what.Append("alive ");
193 if (mc->IsTrackOut()) what.Append("out ");
194
195 Int_t mother = gAlice->GetMCApp()->GetPrimary(trackno);
f95a63c4 196 AliFMDDebug(15, ("Track # %5d deposits a lot of energy\n"
bf000c32 197 " Volume: %s\n"
198 " Momentum: (%7.4f,%7.4f,%7.4f)\n"
199 " PDG: %d (%s)\n"
200 " Edep: %-14.7f keV (mother %d)\n"
201 " p/m: %-7.4f/%-7.4f = %-14.7f\n"
202 " Processes: %s\n"
203 " What: %s\n",
204 trackno, mc->CurrentVolPath(), p.X(), p.Y(), p.Z(),
205 pdg, pname.Data(), edep, mother, p.P(), mass,
206 poverm, processes.Data(), what.Data()));
207 return kTRUE;
208}
209
54e415a8 210
37c4363a 211//____________________________________________________________________
212void
213AliFMDv1::StepManager()
214{
54e415a8 215 // Member function that is executed each time a hit is made in the
216 // FMD. None-charged particles are ignored. Dead tracks are
217 // ignored.
218 //
219 // The procedure is as follows:
220 //
221 // - IF NOT track is alive THEN RETURN ENDIF
222 // - IF NOT particle is charged THEN RETURN ENDIF
223 // - IF NOT volume name is "STRI" or "STRO" THEN RETURN ENDIF
224 // - Get strip number (volume copy # minus 1)
225 // - Get phi division number (mother volume copy #)
226 // - Get module number (grand-mother volume copy #)
227 // - section # = 2 * module # + phi division # - 1
228 // - Get ring Id from volume name
229 // - Get detector # from grand-grand-grand-mother volume name
230 // - Get pointer to sub-detector object.
231 // - Get track position
232 // - IF track is entering volume AND track is inside real shape THEN
233 // - Reset energy deposited
234 // - Get track momentum
235 // - Get particle ID #
236 /// - ENDIF
237 // - IF track is inside volume AND inside real shape THEN
238 /// - Update energy deposited
239 // - ENDIF
240 // - IF track is inside real shape AND (track is leaving volume,
241 // or it died, or it is stopped THEN
242 // - Create a hit
243 // - ENDIF
244 //
54e415a8 245 TVirtualMC* mc = TVirtualMC::GetMC();
246 if (!mc->IsTrackAlive()) return;
247 Double_t absQ = TMath::Abs(mc->TrackCharge());
248 if (absQ <= 0) return;
249
250 Int_t copy;
251 Int_t vol = mc->CurrentVolID(copy);
252 AliFMDGeometry* fmd = AliFMDGeometry::Instance();
253 if (!fmd->IsActive(vol)) {
f95a63c4 254 AliFMDDebug(50, ("Not an FMD volume %d '%s'",vol,mc->CurrentVolName()));
54e415a8 255 return;
256 }
257 TLorentzVector v;
258 UShort_t detector;
259 Char_t ring;
260 UShort_t sector;
261 UShort_t strip;
262
263 if (fmd->IsDetailed()) {
264 if (!VMC2FMD(copy, v, detector, ring, sector, strip)) return;
265 } else {
266 if (!VMC2FMD(v, detector, ring, sector, strip)) return;
267 }
268 TLorentzVector p;
269 mc->TrackMomentum(p);
270 Int_t trackno = gAlice->GetMCApp()->GetCurrentTrackNumber();
271 Int_t pdg = mc->TrackPid();
54e415a8 272 Double_t edep = mc->Edep() * 1000; // keV
bf000c32 273 Bool_t isBad = CheckHit(trackno, pdg, absQ, p, edep);
54e415a8 274
275 // Check that the track is actually within the active area
276 Bool_t entering = mc->IsTrackEntering();
277 Bool_t inside = mc->IsTrackInside();
278 Bool_t out = (mc->IsTrackExiting()|| mc->IsTrackDisappeared()||
279 mc->IsTrackStop());
280 // Reset the energy deposition for this track, and update some of
281 // our parameters.
282 if (entering) {
f95a63c4 283 AliFMDDebug(15, ("Track # %8d entering active FMD volume %s: "
54e415a8 284 "Edep=%f (%f,%f,%f)", trackno, mc->CurrentVolPath(),
285 edep, v.X(), v.Y(), v.Z()));
286 fCurrentP = p;
287 fCurrentV = v;
288 fCurrentDeltaE = edep;
289 fCurrentPdg = pdg; // mc->IdFromPDG(pdg);
290 }
291 // If the track is inside, then update the energy deposition
292 if (inside && fCurrentDeltaE >= 0) {
293 fCurrentDeltaE += edep;
f95a63c4 294 AliFMDDebug(15, ("Track # %8d inside active FMD volume %s: Edep=%f, "
54e415a8 295 "Accumulated Edep=%f (%f,%f,%f)", trackno,
296 mc->CurrentVolPath(), edep, fCurrentDeltaE,
297 v.X(), v.Y(), v.Z()));
298 }
299 // The track exits the volume, or it disappeared in the volume, or
300 // the track is stopped because it no longer fulfills the cuts
301 // defined, then we create a hit.
302 if (out) {
303 if (fCurrentDeltaE >= 0) {
304 fCurrentDeltaE += edep;
f95a63c4 305 AliFMDDebug(15, ("Track # %8d exiting active FMD volume %s: Edep=%g, "
54e415a8 306 "Accumulated Edep=%g (%f,%f,%f)", trackno,
307 mc->CurrentVolPath(), edep, fCurrentDeltaE,
308 v.X(), v.Y(), v.Z()));
088f8e79 309 TVector3 cur(v.Vect());
310 cur -= fCurrentV.Vect();
311 Double_t len = cur.Mag();
54e415a8 312 AliFMDHit* h =
313 AddHitByFields(trackno, detector, ring, sector, strip,
314 fCurrentV.X(), fCurrentV.Y(), fCurrentV.Z(),
315 fCurrentP.X(), fCurrentP.Y(), fCurrentP.Z(),
088f8e79 316 fCurrentDeltaE, fCurrentPdg, fCurrentV.T(),
317 len, mc->IsTrackDisappeared()||mc->IsTrackStop());
54e415a8 318 // Add a copy
319 if (isBad && fBad) {
320 new ((*fBad)[fBad->GetEntries()]) AliFMDHit(*h);
321 }
bf000c32 322 // Check the geometry that we can get back the coordinates.
323#ifdef CHECK_TRANS
324 Double_t x, y, z;
325 fmd->Detector2XYZ(detector, ring, sector, strip, x, y ,z);
f95a63c4 326 AliFMDDebug(1, ("Hit at (%f,%f,%f), geometry says (%f,%f,%f)",
bf000c32 327 fCurrentV.X(), fCurrentV.Y(), fCurrentV.Z(), x, y, z));
328#endif
54e415a8 329 }
330 fCurrentDeltaE = -1;
331 }
37c4363a 332}
333//___________________________________________________________________
334//
335// EOF
336//