Transition to NewIO
[u/mrichter/AliRoot.git] / TRD / AliTRDmcTrack.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 //                                                                           //
18 //  TRD MC track                                                             //
19 //  Used for efficiency estimates and matching of reconstructed tracks       //
20 //  to MC particles                                                          //                    
21 //                                                                           //
22 ///////////////////////////////////////////////////////////////////////////////
23
24 #include "AliTRDmcTrack.h"
25 #include "AliTRDgeometry.h"
26
27
28 ClassImp(AliTRDmcTrack)
29
30 //_____________________________________________________________________________
31 AliTRDmcTrack::AliTRDmcTrack() 
32
33
34   //
35   // Default constructor 
36   //
37
38   fLab = -1; 
39   fPrimary = kFALSE; 
40   fMass = 0; 
41   fCharge = 0;
42   fN = 0; 
43   
44   for(Int_t ltb=0; ltb<kMAX_TB; ltb++) {
45     for(Int_t plane=0; plane < 6; plane++) {
46       fIndex[ltb][plane][0] = -1;
47       fIndex[ltb][plane][1] = -1;
48     }
49   }
50
51   for(Int_t i=0; i<6; i++) {
52     for(Int_t j=0; j<3; j++) { 
53       Pin[i][j]=0.; 
54       Pout[i][j] = 0.;
55       XYZin[i][j]=0.; 
56       XYZout[i][j] = 0.;
57     }
58   }
59
60 }
61
62 //_____________________________________________________________________________
63 AliTRDmcTrack::AliTRDmcTrack(Int_t label, Int_t seedLabel, Bool_t primary, 
64                              Float_t mass, Int_t charge, Int_t pdg) 
65
66
67   //
68   // Main constructor 
69   //
70
71   fLab = label; 
72   fSeedLab = seedLabel; 
73   fPrimary = primary; 
74   fMass = mass; 
75   fCharge = charge;
76   fPDG = pdg;
77   fN = 0; 
78   
79   for(Int_t ltb=0; ltb<kMAX_TB; ltb++) {
80     for(Int_t plane=0; plane < 6; plane++) {
81       fIndex[ltb][plane][0] = -1;
82       fIndex[ltb][plane][1] = -1;
83     }
84   }
85   
86   for(Int_t i=0; i<6; i++) {
87     for(Int_t j=0; j<3; j++) { 
88       Pin[i][j]=0.; 
89       Pout[i][j] = 0.;
90       XYZin[i][j]=0.; 
91       XYZout[i][j] = 0.;
92     }
93   }
94
95 }
96
97 //_____________________________________________________________________________
98 void AliTRDmcTrack::GetPxPyPzXYZ(Double_t& px, Double_t& py, Double_t& pz, 
99                                  Double_t&  x, Double_t&  y, Double_t&  z, 
100                                  Int_t opt) const 
101 {
102   //
103   // Returns track momentum components and coordinates at the entrance 
104   // (opt >= 0), or exit (opt < 0) of TRD. 
105   //
106
107   Int_t i;
108
109   if(opt >= 0) {
110     for(i = 0; i < AliTRDgeometry::Nplan(); i++) {
111       if(  Pin[i][0] * Pin[i][0]
112          + Pin[i][1] * Pin[i][1]
113          + Pin[i][2] * Pin[i][2] > 0.0005) break;
114     }
115     px = Pin[i][0];    py = Pin[i][1];   pz = Pin[i][2];
116      x = XYZin[i][0];   y = XYZin[i][1];  z = XYZin[i][2];
117   }
118   else {
119     for(i = AliTRDgeometry::Nplan() - 1; i >= 0; i--) {
120       if(  Pout[i][0] * Pout[i][0]
121          + Pout[i][1] * Pout[i][1]
122          + Pout[i][2] * Pout[i][2] > 0.0005) break;
123     }
124     px = Pout[i][0];    py = Pout[i][1];   pz = Pout[i][2];
125      x = XYZout[i][0];   y = XYZout[i][1];  z = XYZout[i][2];
126   }
127   return;
128 }
129
130 //_____________________________________________________________________________
131 void AliTRDmcTrack::GetPlanePxPyPz(Double_t& px, Double_t& py, Double_t& pz, 
132                                    Int_t plane, Int_t opt) const 
133 {
134   //
135   // Returns momentum components at the entrance (opt >= 0), or
136   // exit (opt < 0) of TRD plane <plane>. 
137   //
138
139   if(opt >= 0) {
140     px = Pin[plane][0]; py = Pin[plane][1]; pz = Pin[plane][2];
141   }
142   else {
143     px = Pout[plane][0]; py = Pout[plane][1]; pz = Pout[plane][2];
144   }
145   return;
146 }