ATLAS Offline Software
Loading...
Searching...
No Matches
CovarianceFillerTool.cxx
Go to the documentation of this file.
1/*
2 Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
3*/
4
5// $Id: CovarianceFillerTool.cxx 281348 2010-02-24 23:15:11Z zaidan $
12
13
14#include "CovarianceFillerTool.h"
15//#include "TrkEventPrimitives/ErrorMatrix.h"
16#include "TrkEventPrimitives/ParamDefs.h"
17#include "AthenaKernel/errorcheck.h"
18#include "EventPrimitives/EventPrimitivesHelpers.h"
19
20#include <sstream>
21
22namespace D3PD {
23
24
31CovarianceFillerTool::CovarianceFillerTool
32 (const std::string& type,
33 const std::string& name,
34 const IInterface* parent)
35 : BlockFillerTool<Types<AmgSymMatrix(5),AmgSymMatrix(3),Amg::MatrixX>> (type, name, parent)
36{
37 declareProperty ("Error", m_writeError=true);
38 declareProperty ("DiagCovariance", m_writeDiagCovariance=false);
39 declareProperty ("OffDiagCovariance", m_writeOffDiagCovariance=false);
40 declareProperty ("IsPosition", m_isPosition=false);
41 declareProperty ("IsTrackPerigee", m_isTrackPerigee=false);
42 declareProperty ("NParameters", m_nParameters=0);
43}
44
45
49StatusCode CovarianceFillerTool::book()
50{
51
52 if( m_isPosition && m_isTrackPerigee ) {
53 REPORT_MESSAGE (MSG::WARNING) << "CovarianceFillerTool is configured as a position and a track perigee error matrix filler"
54 << " ===> Filling a generic covariance matrix";
55 m_isPosition = false;
56 m_isTrackPerigee = false;
57 }
58
59 if( m_isPosition ){
60 m_nParameters = 3;
61 }else if( m_isTrackPerigee ){
62 m_nParameters = 5;
63 }else{
64 //REPORT_MESSAGE (MSG::DEBUG) << "CovarianceFillerTool is filling a generic covariance matrix";
65 }
66
67 if(m_nParameters<=0){
68 REPORT_MESSAGE (MSG::WARNING) << "CovarianceFillerTool is configured to fill a generic covariance matrix of size <= 0";
69 return StatusCode::SUCCESS;
70 }
71
72 if( m_writeError )
73 m_err.resize (m_nParameters);
74
75 int ncov = 0;
76 if( m_writeDiagCovariance)
77 ncov = m_nParameters;
78 if( m_writeOffDiagCovariance)
79 ncov += m_nParameters * (m_nParameters - 1) / 2;
80 m_cov.resize (ncov);
81
82 if( m_isPosition ){
83 if( m_writeError ){
84 CHECK( addVariable ("err_x", m_err[0]) );
85 CHECK( addVariable ("err_y", m_err[1]) );
86 CHECK( addVariable ("err_z", m_err[2]) );
87 }
88 size_t icov = 0;
89 if( m_writeDiagCovariance ){
90 CHECK( addVariable ("cov_x", m_cov[icov++]) );
91 CHECK( addVariable ("cov_y", m_cov[icov++]) );
92 CHECK( addVariable ("cov_z", m_cov[icov++]) );
93 }
94 if( m_writeOffDiagCovariance ){
95 CHECK( addVariable ("cov_xy", m_cov[icov++]) );
96 CHECK( addVariable ("cov_xz", m_cov[icov++]) );
97 CHECK( addVariable ("cov_yz", m_cov[icov++]) );
98 }
99 }else if( m_isTrackPerigee ){
100 if( m_writeError ) {
101 CHECK( addVariable ("err_d0", m_err[0]) );
102 CHECK( addVariable ("err_z0", m_err[1]) );
103 CHECK( addVariable ("err_phi", m_err[2]) );
104 CHECK( addVariable ("err_theta", m_err[3]) );
105 CHECK( addVariable ("err_qoverp",m_err[4]) );
106 }
107
108 size_t icov = 0;
109 if( m_writeDiagCovariance ){
110 CHECK( addVariable ("cov_d0", m_cov[icov++]) );
111 CHECK( addVariable ("cov_z0", m_cov[icov++]) );
112 CHECK( addVariable ("cov_phi", m_cov[icov++]) );
113 CHECK( addVariable ("cov_theta", m_cov[icov++]) );
114 CHECK( addVariable ("cov_qoverp", m_cov[icov++]) );
115 }
116 if( m_writeOffDiagCovariance ){
117 CHECK( addVariable ("cov_d0_z0", m_cov[icov++]) );
118 CHECK( addVariable ("cov_d0_phi", m_cov[icov++]) );
119 CHECK( addVariable ("cov_d0_theta", m_cov[icov++]) );
120 CHECK( addVariable ("cov_d0_qoverp", m_cov[icov++]) );
121 CHECK( addVariable ("cov_z0_phi", m_cov[icov++]) );
122 CHECK( addVariable ("cov_z0_theta", m_cov[icov++]) );
123 CHECK( addVariable ("cov_z0_qoverp", m_cov[icov++]) );
124 CHECK( addVariable ("cov_phi_theta", m_cov[icov++]) );
125 CHECK( addVariable ("cov_phi_qoverp", m_cov[icov++]) );
126 CHECK( addVariable ("cov_theta_qoverp",m_cov[icov++]) );
127 }
128 }else{
129 std::ostringstream str;
130 if( m_writeError ) {
131 for(int i=0; i<m_nParameters; i++){
132 str.str("");
133 str << "err_par" << i;
134 CHECK( addVariable (str.str().c_str(), m_err[i]) );
135 }
136 }
137 size_t icov = 0;
138 if( m_writeDiagCovariance) {
139 for(int i=0; i<m_nParameters; i++){
140 str.str("");
141 str << "cov_par" << i << "_par" << i;
142 CHECK( addVariable (str.str().c_str(), m_cov[icov++]) );
143 }
144 }
145 if( m_writeOffDiagCovariance) {
146 for(int i=0; i<m_nParameters; i++){
147 for(int j=i+1; j<m_nParameters; j++){
148 str.str("");
149 str << "cov_par" << i << "_par" << j;
150 CHECK( addVariable (str.str().c_str(), m_cov[icov++]) );
151 }
152 }
153 }
154 assert (icov == m_cov.size());
155 }
156
157 return StatusCode::SUCCESS;
158}
159
160
169 StatusCode CovarianceFillerTool::fill (const AmgSymMatrix(5) &m)
170{
171
172 if(m_nParameters<=0){
173 return StatusCode::SUCCESS;
174 }
175
176 if( m_isTrackPerigee ) {
177 if (m.rows() > 0) {
178 if( m_writeError ) {
179 *m_err[0] = Amg::error(m,Trk::d0);
180 *m_err[1] = Amg::error(m,Trk::z0);
181 *m_err[2] = Amg::error(m,Trk::phi0);
182 *m_err[3] = Amg::error(m,Trk::theta);
183 *m_err[4] = Amg::error(m,Trk::qOverP);
184 }
185 size_t icov = 0;
186 if( m_writeDiagCovariance ) {
187 *m_cov[icov++] = m(Trk::d0,Trk::d0);
188 *m_cov[icov++] = m(Trk::z0,Trk::z0);
189 *m_cov[icov++] = m(Trk::phi0,Trk::phi0);
190 *m_cov[icov++] = m(Trk::theta,Trk::theta);
191 *m_cov[icov++] = m(Trk::qOverP,Trk::qOverP);
192 }
193 if( m_writeOffDiagCovariance ) {
194 *m_cov[icov++] = m(Trk::d0,Trk::z0);
195 *m_cov[icov++] = m(Trk::d0,Trk::phi0);
196 *m_cov[icov++] = m(Trk::d0,Trk::theta);
197 *m_cov[icov++] = m(Trk::d0,Trk::qOverP);
198 *m_cov[icov++] = m(Trk::z0,Trk::phi0);
199 *m_cov[icov++] = m(Trk::z0,Trk::theta);
200 *m_cov[icov++] = m(Trk::z0,Trk::qOverP);
201 *m_cov[icov++] = m(Trk::phi0,Trk::theta);
202 *m_cov[icov++] = m(Trk::phi0,Trk::qOverP);
203 *m_cov[icov++] = m(Trk::theta,Trk::qOverP);
204 }
205 assert (icov == m_cov.size());
206 }
207 }
208 else{
209 if( m_writeError ) {
210 for(int i=0; i<m_nParameters; i++){
211 *m_err[i] = Amg::error(m,(Trk::ParamDefs)i);
212 }
213 }
214
215 size_t icov = 0;
216 if( m_writeDiagCovariance ) {
217 for(int i=0; i<m_nParameters; i++)
218 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)i);
219 }
220 if( m_writeOffDiagCovariance ) {
221 for(int i=0; i<m_nParameters; i++) {
222 for(int j=i+1; j<m_nParameters; j++)
223 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)j);
224 }
225 }
226 assert (icov == m_cov.size());
227 }
228
229 return StatusCode::SUCCESS;
230}
231
232StatusCode CovarianceFillerTool::fill (const AmgSymMatrix(3) &m)
233{
234
235 if(m_nParameters<=0){
236 return StatusCode::SUCCESS;
237 }
238
239 if( m_isPosition ){
240 if( m_writeError ) {
241 *m_err[0] = sqrt(m(Trk::x,Trk::x));
242 *m_err[1] = sqrt(m(Trk::y,Trk::y));
243 *m_err[2] = sqrt(m(Trk::z,Trk::z));
244 }
245
246 size_t icov = 0;
247 if( m_writeDiagCovariance ) {
248 *m_cov[icov++] = m(Trk::x,Trk::x);
249 *m_cov[icov++] = m(Trk::y,Trk::y);
250 *m_cov[icov++] = m(Trk::z,Trk::z);
251 }
252 if( m_writeOffDiagCovariance ) {
253 *m_cov[icov++] = m(Trk::x,Trk::y);
254 *m_cov[icov++] = m(Trk::x,Trk::z);
255 *m_cov[icov++] = m(Trk::y,Trk::z);
256 }
257 assert (icov == m_cov.size());
258 }else{
259 if( m_writeError ) {
260 for(int i=0; i<m_nParameters; i++){
261 *m_err[i] = sqrt(m((Trk::ParamDefs)i,(Trk::ParamDefs)i));
262 }
263 }
264
265 size_t icov = 0;
266 if( m_writeDiagCovariance ) {
267 for(int i=0; i<m_nParameters; i++)
268 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)i);
269 }
270 if( m_writeOffDiagCovariance ) {
271 for(int i=0; i<m_nParameters; i++) {
272 for(int j=i+1; j<m_nParameters; j++)
273 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)j);
274 }
275 }
276 assert (icov == m_cov.size());
277 }
278
279 return StatusCode::SUCCESS;
280}
281
282 StatusCode CovarianceFillerTool::fill (const Amg::MatrixX &m)
283{
284
285 if(m_nParameters<=0){
286 return StatusCode::SUCCESS;
287 }
288
289 if( m_isTrackPerigee ) {
290 if (m.rows() > 0) {
291 if( m_writeError ) {
292 *m_err[0] = Amg::error(m,Trk::d0);
293 *m_err[1] = Amg::error(m,Trk::z0);
294 *m_err[2] = Amg::error(m,Trk::phi0);
295 *m_err[3] = Amg::error(m,Trk::theta);
296 *m_err[4] = Amg::error(m,Trk::qOverP);
297 }
298 size_t icov = 0;
299 if( m_writeDiagCovariance ) {
300 *m_cov[icov++] = m(Trk::d0,Trk::d0);
301 *m_cov[icov++] = m(Trk::z0,Trk::z0);
302 *m_cov[icov++] = m(Trk::phi0,Trk::phi0);
303 *m_cov[icov++] = m(Trk::theta,Trk::theta);
304 *m_cov[icov++] = m(Trk::qOverP,Trk::qOverP);
305 }
306 if( m_writeOffDiagCovariance ) {
307 *m_cov[icov++] = m(Trk::d0,Trk::z0);
308 *m_cov[icov++] = m(Trk::d0,Trk::phi0);
309 *m_cov[icov++] = m(Trk::d0,Trk::theta);
310 *m_cov[icov++] = m(Trk::d0,Trk::qOverP);
311 *m_cov[icov++] = m(Trk::z0,Trk::phi0);
312 *m_cov[icov++] = m(Trk::z0,Trk::theta);
313 *m_cov[icov++] = m(Trk::z0,Trk::qOverP);
314 *m_cov[icov++] = m(Trk::phi0,Trk::theta);
315 *m_cov[icov++] = m(Trk::phi0,Trk::qOverP);
316 *m_cov[icov++] = m(Trk::theta,Trk::qOverP);
317 }
318 assert (icov == m_cov.size());
319 }
320 } else if( m_isPosition ){
321 if( m_writeError ) {
322 *m_err[0] = Amg::error(m,Trk::x);
323 *m_err[1] = Amg::error(m,Trk::y);
324 *m_err[2] = Amg::error(m,Trk::z);
325 }
326
327 size_t icov = 0;
328 if( m_writeDiagCovariance ) {
329 *m_cov[icov++] = m(Trk::x,Trk::x);
330 *m_cov[icov++] = m(Trk::y,Trk::y);
331 *m_cov[icov++] = m(Trk::z,Trk::z);
332 }
333 if( m_writeOffDiagCovariance ) {
334 *m_cov[icov++] = m(Trk::x,Trk::y);
335 *m_cov[icov++] = m(Trk::x,Trk::z);
336 *m_cov[icov++] = m(Trk::y,Trk::z);
337 }
338 assert (icov == m_cov.size());
339 }else{
340 if( m_writeError ) {
341 for(int i=0; i<m_nParameters; i++){
342 *m_err[i] = Amg::error(m,(Trk::ParamDefs)i);
343 }
344 }
345
346 size_t icov = 0;
347 if( m_writeDiagCovariance ) {
348 for(int i=0; i<m_nParameters; i++)
349 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)i);
350 }
351 if( m_writeOffDiagCovariance ) {
352 for(int i=0; i<m_nParameters; i++) {
353 for(int j=i+1; j<m_nParameters; j++)
354 *m_cov[icov++] = m((Trk::ParamDefs)i,(Trk::ParamDefs)j);
355 }
356 }
357 assert (icov == m_cov.size());
358 }
359
360 return StatusCode::SUCCESS;
361}
362
363} // namespace D3PD
errorcheck.h
Helpers for checking error return status codes and reporting errors.
REPORT_MESSAGE
#define REPORT_MESSAGE(LVL)
Report a message.
Definition Control/AthenaKernel/AthenaKernel/errorcheck.h:365
CHECK
#define CHECK(...)
Evaluate an expression and check for errors.
Definition Control/AthenaKernel/AthenaKernel/errorcheck.h:422
CovarianceFillerTool.h
Block filler tool for error and covariance matrix.
AmgSymMatrix
#define AmgSymMatrix(dim)
Definition EventPrimitives.h:50
D3PD::BlockFillerTool< Types< AmgSymMatrix(5), AmgSymMatrix(3), Amg::MatrixX > >::addVariable
virtual StatusCode addVariable(const std::string &name, const std::type_info &ti, void *&ptr, const std::string &docstring="", const void *defval=0)
Definition AddVariable.cxx:85
D3PD::BlockFillerTool
Type-safe wrapper for block filler tools.
Definition BlockFillerTool.h:65
D3PD::CovarianceFillerTool::book
virtual StatusCode book()
Book variables for this block.
Definition CovarianceFillerTool.cxx:49
D3PD::CovarianceFillerTool::m_cov
std::vector< float * > m_cov
Variables: covariance matrix.
Definition CovarianceFillerTool.h:76
D3PD::CovarianceFillerTool::m_err
std::vector< float * > m_err
Variables: uncertainty.
Definition CovarianceFillerTool.h:73
D3PD::CovarianceFillerTool::fill
virtual StatusCode fill(const AmgSymMatrix(5) &p)
Fill one block — type-safe version.
Definition CovarianceFillerTool.cxx:169
D3PD::CovarianceFillerTool::CovarianceFillerTool
CovarianceFillerTool(const std::string &type, const std::string &name, const IInterface *parent)
Standard Gaudi tool constructor.
Definition CovarianceFillerTool.cxx:32
Amg
Definition of ATLAS Math & Geometry primitives (Amg)
Definition AmgStringHelpers.h:19
Amg::MatrixX
Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic > MatrixX
Dynamic Matrix - dynamic allocation.
Definition EventPrimitives.h:27
Amg::error
double error(const Amg::MatrixX &mat, int index)
return diagonal error of the matrix caller should ensure the matrix is symmetric and the index is in ...
Definition EventPrimitivesHelpers.h:40
D3PD
Block filler tool for noisy FEB information.
Definition CaloCellDetailsFillerTool.cxx:29
D3PD::AmgSymMatrix
const AmgSymMatrix(5) *PerigeeCovarianceAssociationTool
Return the target object.
Definition PerigeeCovarianceAssociationTool.cxx:43
D3PD::Types
std::tuple< WrapType< TYPES >... > Types
A simple tuple of multiple types.
Definition PhysicsAnalysis/D3PDMaker/D3PDMakerUtils/D3PDMakerUtils/Types.h:61
Trk::ParamDefs
ParamDefs
This file defines the parameter enums in the Trk namespace.
Definition ParamDefs.h:32
Trk::x
@ x
Definition ParamDefs.h:55
Trk::z
@ z
global position (cartesian)
Definition ParamDefs.h:57
Trk::phi0
@ phi0
Definition ParamDefs.h:65
Trk::theta
@ theta
Definition ParamDefs.h:66
Trk::qOverP
@ qOverP
perigee
Definition ParamDefs.h:67
Trk::y
@ y
Definition ParamDefs.h:56
Trk::d0
@ d0
Definition ParamDefs.h:63
Trk::z0
@ z0
Definition ParamDefs.h:64
type
Generated on for ATLAS Offline Software by doxygen 1.14.0