fed4be7bec6a569b8b6c22dcb27b3753e527752a
[u/mrichter/AliRoot.git] / STEER / AliESDFMD.cxx
1 /**************************************************************************
2  * Copyright(c) 2004, 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 //  ESD information from the FMD 
21 //  Contains information on:
22 //      Charged particle multiplicty per strip (rough estimate)
23 //      Psuedo-rapdity per strip
24 //  Latest changes by Christian Holm Christensen
25 //
26 #include "AliESDFMD.h"          // ALIFMDESD_H
27 #include "AliLog.h"             // ALILOG_H
28 #include "Riostream.h"          // ROOT_Riostream
29
30 //____________________________________________________________________
31 ClassImp(AliESDFMD)
32 #if 0
33   ; // This is here to keep Emacs for indenting the next line
34 #endif
35
36
37 //____________________________________________________________________
38 AliESDFMD::AliESDFMD()
39   : fMultiplicity(AliFMDFloatMap::kMaxDetectors, 
40                   AliFMDFloatMap::kMaxRings, 
41                   AliFMDFloatMap::kMaxSectors, 
42                   AliFMDFloatMap::kMaxStrips), 
43     fEta(AliFMDFloatMap::kMaxDetectors, 
44          AliFMDFloatMap::kMaxRings, 
45          1,
46          AliFMDFloatMap::kMaxStrips), 
47     fAngleCorrected(kFALSE)
48 {
49   // Default CTOR
50 }
51   
52 //____________________________________________________________________
53 AliESDFMD::AliESDFMD(const AliESDFMD& other)
54   : TObject(other), 
55     fMultiplicity(other.fMultiplicity),
56     fEta(other.fEta)
57 {
58   // Default CTOR
59 }
60
61 //____________________________________________________________________
62 AliESDFMD& 
63 AliESDFMD::operator=(const AliESDFMD& other)
64 {
65   // Default CTOR
66   if(this!=&other){
67     TObject::operator=(other);
68     fMultiplicity = other.fMultiplicity;
69     fEta          = other.fEta;
70   }
71   return *this;
72 }
73
74 //____________________________________________________________________
75 void
76 AliESDFMD::CheckNeedUShort(TFile* file) 
77 {
78   fMultiplicity.CheckNeedUShort(file);
79   fEta.CheckNeedUShort(file);
80 }
81
82 //____________________________________________________________________
83 void
84 AliESDFMD::Clear(Option_t* )
85 {
86   fMultiplicity.Reset(kInvalidMult);
87   fEta.Reset(kInvalidEta);
88 }
89
90
91 //____________________________________________________________________
92 Float_t
93 AliESDFMD::Multiplicity(UShort_t detector, Char_t ring, UShort_t sector, 
94                         UShort_t strip) const
95 {
96   // Return rough estimate of charged particle multiplicity in the
97   // strip FMD<detector><ring>[<sector>,<strip>]. 
98   // 
99   // Note, that this should at most be interpreted as the sum
100   // multiplicity of secondaries and primaries. 
101   return fMultiplicity(detector, ring, sector, strip);
102 }
103
104 //____________________________________________________________________
105 Float_t
106 AliESDFMD::Eta(UShort_t detector, Char_t ring, UShort_t /* sector */, 
107                UShort_t strip) const
108 {
109   // Return pseudo-rapidity of the strip
110   // FMD<detector><ring>[<sector>,<strip>].  (actually, the sector
111   // argument is ignored, as it is assumed that the primary vertex is
112   // a (x,y) = (0,0), and that the modules are aligned with a
113   // precision better than 2 degrees in the azimuthal angle). 
114   // 
115   return fEta(detector, ring, 0, strip);
116 }
117
118 //____________________________________________________________________
119 void
120 AliESDFMD::SetMultiplicity(UShort_t detector, Char_t ring, UShort_t sector, 
121                            UShort_t strip, Float_t mult)
122 {
123   // Return rough estimate of charged particle multiplicity in the
124   // strip FMD<detector><ring>[<sector>,<strip>]. 
125   // 
126   // Note, that this should at most be interpreted as the sum
127   // multiplicity of secondaries and primaries. 
128   fMultiplicity(detector, ring, sector, strip) = mult;
129 }
130
131 //____________________________________________________________________
132 void
133 AliESDFMD::SetEta(UShort_t detector, Char_t ring, UShort_t /* sector */, 
134                   UShort_t strip, Float_t eta)
135 {
136   // Set pseudo-rapidity of the strip
137   // FMD<detector><ring>[<sector>,<strip>].  (actually, the sector
138   // argument is ignored, as it is assumed that the primary vertex is
139   // a (x,y) = (0,0), and that the modules are aligned with a
140   // precision better than 2 degrees in the azimuthal angle). 
141   // 
142   fEta(detector, ring, 0, strip) = eta;
143 }
144
145 //____________________________________________________________________
146 void
147 AliESDFMD::Print(Option_t* /* option*/) const
148 {
149   // Print all information to standard output. 
150   std::cout << "AliESDFMD:" << std::endl;
151   for (UShort_t det = 1; det <= fMultiplicity.MaxDetectors(); det++) {
152     for (UShort_t ir = 0; ir < fMultiplicity.MaxRings(); ir++) {
153       Char_t ring = (ir == 0 ? 'I' : 'O');
154       std::cout << "FMD" << det << ring << ":" << std::endl;
155       for  (UShort_t sec = 0; sec < fMultiplicity.MaxSectors(); sec++) {
156         std::cout << " Sector # " << sec << ":" << std::flush;
157         for (UShort_t str = 0; str < fMultiplicity.MaxStrips(); str++) {
158           if (str % 6 == 0) std::cout << "\n  " << std::flush;
159           Float_t m = fMultiplicity(det, ring, sec, str);
160           Float_t e = fEta(det, ring, 0, str);
161           if (m == kInvalidMult && e == kInvalidEta) break;
162           if (m == kInvalidMult) std::cout << " ---- ";
163           else                   std::cout << Form("%6.3f", m);
164           std::cout << "/";
165           if (e == kInvalidEta)  std::cout << " ---- ";
166           else                   std::cout << Form("%-6.3f", e);
167           std::cout << " " << std::flush;
168         }
169         std::cout << std::endl;
170       }
171     }
172   }
173 }
174
175
176 //____________________________________________________________________
177 //
178 // EOF
179 //