ATLAS Offline Software
Classes | Public Member Functions | Private Member Functions | Static Private Member Functions | Private Attributes | Friends | List of all members
Trk::TrkVKalVrtFitter Class Reference

#include <TrkVKalVrtFitter.h>

Inheritance diagram for Trk::TrkVKalVrtFitter:
Collaboration diagram for Trk::TrkVKalVrtFitter:

Classes

class  CascadeState
 
class  State
 
struct  TrkMatControl
 

Public Member Functions

virtual StatusCode initialize () override final
 
virtual StatusCode finalize () override final
 
 TrkVKalVrtFitter (const std::string &t, const std::string &name, const IInterface *parent)
 
virtual ~TrkVKalVrtFitter ()
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &perigeeList, const Amg::Vector3D &startingPoint) const override final
 Interface for MeasuredPerigee with starting point. More...
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &perigeeList, const std::vector< const NeutralParameters * > &, const Amg::Vector3D &startingPoint) const override final
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &perigeeList, const xAOD::Vertex &constraint) const override final
 Interface for MeasuredPerigee with vertex constraint. More...
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &perigeeList, const std::vector< const NeutralParameters * > &, const xAOD::Vertex &constraint) const override final
 
virtual std::unique_ptr< xAOD::Vertex > fit (const EventContext &ctx, const std::vector< const xAOD::TrackParticle * > &vectorTrk, const Amg::Vector3D &startingPoint) const override final
 Interface for xAOD::TrackParticle with starting point Implements the new style (unique_ptr,EventContext) More...
 
virtual xAOD::Vertex * fit (const std::vector< const xAOD::TrackParticle * > &vectorTrk, const xAOD::Vertex &constraint) const override final
 Interface for xAOD::TrackParticle with vertex constraint. More...
 
virtual xAOD::Vertex * fit (const std::vector< const xAOD::TrackParticle * > &vectorTrk, const std::vector< const xAOD::NeutralParticle * > &vectorNeu, const Amg::Vector3D &startingPoint) const override final
 
virtual xAOD::Vertex * fit (const std::vector< const xAOD::TrackParticle * > &vectorTrk, const std::vector< const xAOD::NeutralParticle * > &vectorNeu, const xAOD::Vertex &constraint) const override final
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &) const override final
 
virtual xAOD::Vertex * fit (const std::vector< const TrackParameters * > &, const std::vector< const Trk::NeutralParameters * > &) const override final
 
xAOD::Vertex * fit (const std::vector< const xAOD::TrackParticle * > &vectorTrk, const Amg::Vector3D &constraint, IVKalState &istate) const
 
xAOD::Vertex * fit (const std::vector< const xAOD::TrackParticle * > &vectorTrk, const xAOD::Vertex &constraint, IVKalState &istate) const
 
VertexID startVertex (const std::vector< const xAOD::TrackParticle * > &list, std::span< const double > particleMass, IVKalState &istate, double massConstraint=0.) const override final
 Interface for cascade fit. More...
 
VertexID nextVertex (const std::vector< const xAOD::TrackParticle * > &list, std::span< const double > particleMass, IVKalState &istate, double massConstraint=0.) const override final
 
VertexID nextVertex (const std::vector< const xAOD::TrackParticle * > &list, std::span< const double > particleMass, const std::vector< VertexID > &precedingVertices, IVKalState &istate, double massConstraint=0.) const override final
 
VxCascadeInfo * fitCascade (IVKalState &istate, const Vertex *primVertex=0, bool FirstDecayAtPV=false) const override final
 
StatusCode addMassConstraint (VertexID Vertex, const std::vector< const xAOD::TrackParticle * > &tracksInConstraint, const std::vector< VertexID > &verticesInConstraint, IVKalState &istate, double massConstraint) const override final
 
virtual std::unique_ptr< IVKalState > makeState (const EventContext &ctx) const override final
 
virtual StatusCode VKalVrtFit (const std::vector< const xAOD::TrackParticle * > &, const std::vector< const xAOD::NeutralParticle * > &, Amg::Vector3D &Vertex, TLorentzVector &Momentum, long int &Charge, dvect &ErrorMatrix, dvect &Chi2PerTrk, std::vector< std::vector< double >> &TrkAtVrt, double &Chi2, IVKalState &istate, bool ifCovV0=false) const override final
 
virtual StatusCode VKalVrtFit (const std::vector< const Perigee * > &, Amg::Vector3D &Vertex, TLorentzVector &Momentum, long int &Charge, dvect &ErrorMatrix, dvect &Chi2PerTrk, std::vector< std::vector< double >> &TrkAtVrt, double &Chi2, IVKalState &istate, bool ifCovV0=false) const override final
 
virtual StatusCode VKalVrtFit (const std::vector< const TrackParameters * > &, const std::vector< const NeutralParameters * > &, Amg::Vector3D &Vertex, TLorentzVector &Momentum, long int &Charge, dvect &ErrorMatrix, dvect &Chi2PerTrk, std::vector< std::vector< double >> &TrkAtVrt, double &Chi2, IVKalState &istate, bool ifCovV0=false) const override final
 
virtual StatusCode VKalVrtCvtTool (const Amg::Vector3D &Vertex, const TLorentzVector &Momentum, const dvect &CovVrtMom, const long int &Charge, dvect &Perigee, dvect &CovPerigee, IVKalState &istate) const override final
 
virtual StatusCode VKalVrtFitFast (std::span< const xAOD::TrackParticle *const >, Amg::Vector3D &Vertex, double &minDZ, IVKalState &istate) const
 
virtual StatusCode VKalVrtFitFast (const std::span< const xAOD::TrackParticle *const >, Amg::Vector3D &Vertex, IVKalState &istate) const override final
 
virtual StatusCode VKalVrtFitFast (const std::vector< const TrackParameters * > &, Amg::Vector3D &Vertex, IVKalState &istate) const override final
 
virtual std::unique_ptr< Trk::Perigee > CreatePerigee (const std::vector< double > &VKPerigee, const std::vector< double > &VKCov, IVKalState &istate) const override final
 
virtual StatusCode VKalGetTrkWeights (dvect &Weights, const IVKalState &istate) const override final
 
virtual StatusCode VKalGetFullCov (long int, dvect &CovMtx, IVKalState &istate, bool=false) const override final
 
virtual StatusCode VKalGetMassError (double &Mass, double &MassError, const IVKalState &istate) const override final
 
virtual void setApproximateVertex (double X, double Y, double Z, IVKalState &istate) const override final
 
virtual void setMassForConstraint (double Mass, IVKalState &istate) const override final
 
virtual void setMassForConstraint (double Mass, std::span< const int >, IVKalState &istate) const override final
 
virtual void setRobustness (int, IVKalState &istate) const override final
 
virtual void setRobustScale (double, IVKalState &istate) const override final
 
virtual void setCnstType (int, IVKalState &istate) const override final
 
virtual void setVertexForConstraint (const xAOD::Vertex &, IVKalState &istate) const override final
 
virtual void setVertexForConstraint (double X, double Y, double Z, IVKalState &istate) const override final
 
virtual void setCovVrtForConstraint (double XX, double XY, double YY, double XZ, double YZ, double ZZ, IVKalState &istate) const override final
 
virtual void setMassInputParticles (const std::vector< double > &, IVKalState &istate) const override final
 
virtual double VKalGetImpact (const xAOD::TrackParticle *, const Amg::Vector3D &Vertex, const long int Charge, dvect &Impact, dvect &ImpactError, IVKalState &istate) const override final
 
virtual double VKalGetImpact (const Trk::Perigee *, const Amg::Vector3D &Vertex, const long int Charge, dvect &Impact, dvect &ImpactError, IVKalState &istate) const override final
 
virtual double VKalGetImpact (const xAOD::TrackParticle *, const Amg::Vector3D &Vertex, const long int Charge, dvect &Impact, dvect &ImpactError) const override final
 
virtual double VKalGetImpact (const Trk::Perigee *, const Amg::Vector3D &Vertex, const long int Charge, dvect &Impact, dvect &ImpactError) const override final
 

Private Member Functions

void initCnstList ()
 
void setAthenaPropagator (const Trk::IExtrapolator *)
 
void initState (const EventContext &ctx, State &state) const
 
void initState (State &state) const
 
bool convertAmg5SymMtx (const AmgSymMatrix(5) *, double[15]) const
 
void VKalTransform (double MAG, double A0V, double ZV, double PhiV, double ThetaV, double PInv, const double[15], long int &Charge, double[5], double[15]) const
 
xAOD::Vertex * makeXAODVertex (int, const Amg::Vector3D &, const dvect &, const dvect &, const std::vector< dvect > &, double, State &state) const
 
StatusCode CvtPerigee (const std::vector< const Perigee * > &list, int &ntrk, State &state) const
 
StatusCode CvtTrackParticle (std::span< const xAOD::TrackParticle *const > list, int &ntrk, State &state) const
 
StatusCode CvtNeutralParticle (const std::vector< const xAOD::NeutralParticle * > &list, int &ntrk, State &state) const
 
StatusCode CvtTrackParameters (const std::vector< const TrackParameters * > &InpTrk, int &ntrk, State &state) const
 
StatusCode CvtNeutralParameters (const std::vector< const NeutralParameters * > &InpTrk, int &ntrk, State &state) const
 
void VKalVrtConfigureFitterCore (int NTRK, State &state) const
 
void VKalToTrkTrack (double curBMAG, double vp1, double vp2, double vp3, double &tp1, double &tp2, double &tp3) const
 
int VKalVrtFit3 (int ntrk, Amg::Vector3D &Vertex, TLorentzVector &Momentum, long int &Charge, dvect &ErrorMatrix, dvect &Chi2PerTrk, std::vector< std::vector< double >> &TrkAtVrt, double &Chi2, State &state, bool ifCovV0) const
 
std::unique_ptr< Perigee > CreatePerigee (double Vx, double Vy, double Vz, const std::vector< double > &VKPerigee, const std::vector< double > &VKCov, State &state) const
 

Static Private Member Functions

static void makeSimpleCascade (std::vector< std::vector< int > > &, std::vector< std::vector< int > > &, CascadeState &cstate)
 
static void printSimpleCascade (std::vector< std::vector< int > > &, std::vector< std::vector< int > > &, const CascadeState &cstate)
 
static int findPositions (const std::vector< int > &, const std::vector< int > &, std::vector< int > &)
 
static int getSimpleVIndex (const VertexID &, const CascadeState &cstate)
 
static int indexInV (const VertexID &, const CascadeState &cstate)
 
static int getCascadeNDoF (const CascadeState &cstate)
 
static void FillMatrixP (AmgSymMatrix(5)&, std::vector< double > &)
 
static void FillMatrixP (int iTrk, AmgSymMatrix(5)&, std::vector< double > &)
 
static Amg::MatrixX * GiveFullMatrix (int NTrk, std::vector< double > &)
 
static const Perigee * GetPerigee (const TrackParameters *i_ntrk)
 
static int VKalGetNDOF (const State &state)
 

Private Attributes

Gaudi::Property< int > m_Robustness {this, "Robustness", 0}
 
Gaudi::Property< double > m_RobustScale {this, "RobustScale", 1.0}
 
Gaudi::Property< double > m_cascadeCnstPrecision {this, "CascadeCnstPrecision", 1.e-4}
 
Gaudi::Property< double > m_massForConstraint {this, "MassForConstraint", -1.0}
 
Gaudi::Property< int > m_IterationNumber {this, "IterationNumber", 0}
 
Gaudi::Property< double > m_IterationPrecision {this, "IterationPrecision", 0.0}
 
Gaudi::Property< double > m_IDsizeR {this, "IDsizeR", 1150.0}
 
Gaudi::Property< double > m_IDsizeZ {this, "IDsizeZ", 3000.0}
 
Gaudi::Property< double > m_MSsizeR {this, "MSsizeR", 8000.0}
 
Gaudi::Property< double > m_MSsizeZ {this, "MSsizeZ", 10000.0}
 
Gaudi::Property< std::vector< double > > m_c_VertexForConstraint {this, "VertexForConstraint", {0,0,0}}
 
Gaudi::Property< std::vector< double > > m_c_CovVrtForConstraint {this, "CovVrtForConstraint", {0,0,0,0,0,0}}
 
Gaudi::Property< std::vector< double > > m_c_MassInputParticles {this, "InputParticleMasses", {}, "List of masses of input particles (pions assumed if absent)"}
 
ToolHandle< IExtrapolator > m_extPropagator {this, "Extrapolator", "", "External propagator"}
 
SG::ReadCondHandleKey< AtlasFieldCacheCondObj > m_fieldCacheCondObjInputKey
 
Gaudi::Property< bool > m_firstMeasuredPoint {this, "FirstMeasuredPoint", false, "Use FirstMeasuredPoint strategy in fits"}
 
Gaudi::Property< bool > m_firstMeasuredPointLimit {this, "FirstMeasuredPointLimit", false, "Use FirstMeasuredPointLimit strategy"}
 
Gaudi::Property< bool > m_firstMeasuredRadiusLimit
 
Gaudi::Property< bool > m_makeExtendedVertex {this, "MakeExtendedVertex", false, "Return VxCandidate with full covariance matrix"}
 
Gaudi::Property< bool > m_useFixedField {this, "useFixedField", false, "Use fixed magnetic field instead of exact Atlas one"}
 
bool m_isAtlasField {false}
 
Gaudi::Property< bool > m_useAprioriVertex {this, "useAprioriVertexCnst", false, "Use a priori vertex constraint"}
 
Gaudi::Property< bool > m_useThetaCnst {this, "useThetaCnst", false, "Use angle dTheta=0 constraint"}
 
Gaudi::Property< bool > m_usePhiCnst {this, "usePhiCnst", false, "Use angle dPhi=0 constraint"}
 
Gaudi::Property< bool > m_usePointingCnst {this, "usePointingCnst", false, "Use pointing to other vertex constraint"}
 
Gaudi::Property< bool > m_useZPointingCnst {this, "useZPointingCnst", false, "Use ZPointing to other vertex constraint"}
 
Gaudi::Property< bool > m_usePassNear {this, "usePassNearCnst", false, "Use combined particle pass near other vertex constraint"}
 
Gaudi::Property< bool > m_usePassWithTrkErr {this, "usePassWithTrkErrCnst", false, "Use pass near with combined particle errors constraint"}
 
Gaudi::Property< bool > m_frozenVersionForBTagging {this, "FrozenVersionForBTagging", false, "Frozen version for BTagging"}
 
Gaudi::Property< bool > m_allowUltraDisplaced {this, "allowUltraDisplaced", false, "Allow ultra displaced vertices"}
 
double m_BMAG {1.997}
 
double m_CNVMAG {0.29979246}
 
VKalExtPropagator * m_fitPropagator {}
 
const IExtrapolator * m_InDetExtrapolator {}
 Pointer to Extrapolator AlgTool. More...
 

Friends

class VKalExtPropagator
 

Detailed Description

Definition at line 64 of file TrkVKalVrtFitter.h.

Constructor & Destructor Documentation

◆ TrkVKalVrtFitter()

Trk::TrkVKalVrtFitter::TrkVKalVrtFitter ( const std::string &  t,
const std::string &  name,
const IInterface *  parent 
)

Definition at line 30 of file TrkVKalVrtFitter.cxx.

32  :
33  base_class(type,name,parent)
34  {
35  declareInterface<IVertexFitter>(this);
36  declareInterface<ITrkVKalVrtFitter>(this);
37  declareInterface<IVertexCascadeFitter>(this);
38 
39 /*--------------------------------------------------------------------------*/
40 /* New propagator object is created. It's provided to VKalVrtCore. */
41 /* VKalVrtFitter must set up Core BEFORE any call required propagation!!! */
42 /* This object is created ONLY if IExtrapolator pointer is provideded. */
43 /* see VKalExtPropagator.cxx for details */
44 /*--------------------------------------------------------------------------*/
45  m_fitPropagator = nullptr; //Pointer to VKalVrtFitter propagator object to supply to VKalVrtCore (specific interface)
46  m_InDetExtrapolator = nullptr; //Direct pointer to Athena propagator
47 }

◆ ~TrkVKalVrtFitter()

Trk::TrkVKalVrtFitter::~TrkVKalVrtFitter ( )
virtual

Definition at line 51 of file TrkVKalVrtFitter.cxx.

51  {
52  //log << MSG::DEBUG << "TrkVKalVrtFitter destructor called" << endmsg;
53  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<<"TrkVKalVrtFitter destructor called" << endmsg;
54  if(m_fitPropagator) delete m_fitPropagator;
55 }

Member Function Documentation

◆ addMassConstraint()

StatusCode Trk::TrkVKalVrtFitter::addMassConstraint ( VertexID  Vertex,
const std::vector< const xAOD::TrackParticle * > &  tracksInConstraint,
const std::vector< VertexID > &  verticesInConstraint,
IVKalState &  istate,
double  massConstraint 
) const
finaloverride

Definition at line 728 of file TrkCascadeFitter.cxx.

733 {
734  assert(dynamic_cast<State*> (&istate)!=nullptr);
735  State& state = static_cast<State&> (istate);
736  CascadeState& cstate = *state.m_cascadeState;
737 
738  int ivc, it, itc;
739  //int NV=m_cstate.cascadeVList.size(); // cascade size
740 //----
741  if(Vertex < 0) return StatusCode::FAILURE;
742  //if(Vertex >= NV) return StatusCode::FAILURE; //Now this check is WRONG. Use indexInV(..) instead
743 //
744 //---- real tracks
745 //
746  int cnstNTRK=tracksInConstraint.size(); // number of real tracks in constraints
747  int indexV = indexInV(Vertex, cstate); // index of vertex in cascade structure
748  if(indexV<0) return StatusCode::FAILURE;
749  int NTRK = cstate.m_cascadeVList[indexV].trkInVrt.size(); // number of real tracks in chosen vertex
750  int totNTRK = cstate.m_partListForCascade.size(); // total number of real tracks
751  if( cnstNTRK > NTRK ) return StatusCode::FAILURE;
752 //-
753  PartialMassConstraint tmpMcnst;
754  tmpMcnst.Mass = massConstraint;
755  tmpMcnst.VRT = Vertex;
756 //
757  double totMass=0;
758  for(itc=0; itc<cnstNTRK; itc++) {
759  for(it=0; it<totNTRK; it++) if(tracksInConstraint[itc]==cstate.m_partListForCascade[it]) break;
760  if(it==totNTRK) return StatusCode::FAILURE; //track in constraint doesn't correspond to any track in vertex
761  tmpMcnst.trkInVrt.push_back(it);
762  totMass += cstate.m_partMassForCascade[it];
763  }
764  if(totMass > massConstraint)return StatusCode::FAILURE;
765 //
766 //---- pseudo tracks
767 //
768  int cnstNVP = pseudotracksInConstraint.size(); // number of pseudo-tracks in constraints
769  int NVP = cstate.m_cascadeVList[indexV].inPointingV.size(); // number of pseudo-tracks in chosen vertex
770  if( cnstNVP > NVP ) return StatusCode::FAILURE;
771 //-
772  for(ivc=0; ivc<cnstNVP; ivc++) {
773  int tmpV = indexInV(pseudotracksInConstraint[ivc], cstate); // index of vertex in cascade structure
774  if( tmpV< 0) return StatusCode::FAILURE; //pseudotrack in constraint doesn't correspond to any pseudotrack in vertex
775  tmpMcnst.pseudoInVrt.push_back( pseudotracksInConstraint[ivc] );
776  }
777 
778  cstate.m_partMassCnstForCascade.push_back(std::move(tmpMcnst));
779 
780  return StatusCode::SUCCESS;
781 }

◆ convertAmg5SymMtx()

bool Trk::TrkVKalVrtFitter::convertAmg5SymMtx ( const AmgSymMatrix(5) *  AmgMtx,
double  stdSymMtx[15] 
) const
private

Definition at line 26 of file VKalTransform.cxx.

27  {
28  if(!AmgMtx) return false;
29  //----- Check perigee covarince matrix for safety
30  double DET=AmgMtx->determinant();
31  if( DET!=DET ) {
32  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<<" NaN in Perigee covariance is detected! Stop fit."<<endmsg;
33  return false;
34  }
35  if( fabs(DET) < 1000.*std::numeric_limits<double>::min()) {
36  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<<"Zero Perigee covariance DET is detected! Stop fit."<<endmsg;
37  return false;
38  }
39 //std::cout.setf(std::ios::scientific); std::cout<<"VKMINNUMB="<<std::numeric_limits<double>::min()<<", "<<DET<<'\n';
40  //---------------------------------------------------------
41  stdSymMtx[ 0] =(*AmgMtx)(0,0);
42  stdSymMtx[ 1] =(*AmgMtx)(1,0);
43  stdSymMtx[ 2] =(*AmgMtx)(1,1);
44  stdSymMtx[ 3] =(*AmgMtx)(2,0);
45  stdSymMtx[ 4] =(*AmgMtx)(2,1);
46  stdSymMtx[ 5] =(*AmgMtx)(2,2);
47  stdSymMtx[ 6] =(*AmgMtx)(3,0);
48  stdSymMtx[ 7] =(*AmgMtx)(3,1);
49  stdSymMtx[ 8] =(*AmgMtx)(3,2);
50  stdSymMtx[ 9] =(*AmgMtx)(3,3);
51  stdSymMtx[10] =(*AmgMtx)(4,0);
52  stdSymMtx[11] =(*AmgMtx)(4,1);
53  stdSymMtx[12] =(*AmgMtx)(4,2);
54  stdSymMtx[13] =(*AmgMtx)(4,3);
55  stdSymMtx[14] =(*AmgMtx)(4,4);
56  return true;
57  }

◆ CreatePerigee() [1/2]

std::unique_ptr< Perigee > Trk::TrkVKalVrtFitter::CreatePerigee ( const std::vector< double > &  VKPerigee,
const std::vector< double > &  VKCov,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 159 of file CvtPerigee.cxx.

162  {
163  assert(dynamic_cast<const State*> (&istate)!=nullptr);
164  State& state = static_cast<State&> (istate);
165  return CreatePerigee(0., 0., 0., VKPerigee, VKCov, state);
166  }

◆ CreatePerigee() [2/2]

std::unique_ptr< Perigee > Trk::TrkVKalVrtFitter::CreatePerigee ( double  Vx,
double  Vy,
double  Vz,
const std::vector< double > &  VKPerigee,
const std::vector< double > &  VKCov,
State &  state 
) const
private

!!! Change of sign !!!!

!!! Change of sign of charge!!!!

Definition at line 172 of file CvtPerigee.cxx.

176  {
177 
178  // ------ Magnetic field access
179  double fx = 0., fy = 0., BMAG_CUR = 0.;
180  state.m_fitField.getMagFld(vX,vY,vZ,fx,fy,BMAG_CUR);
181  if(std::abs(BMAG_CUR) < 0.01) BMAG_CUR=0.01; //safety
182 
183  double TrkP3 = 0., TrkP4 = 0., TrkP5 = 0.;
184  VKalToTrkTrack(BMAG_CUR, VKPerigee[2], VKPerigee[3], VKPerigee[4],
185  TrkP3, TrkP4, TrkP5);
186  double TrkP1 = -VKPerigee[0];
187  double TrkP2 = VKPerigee[1];
188  TrkP5 = -TrkP5;
190  AmgSymMatrix(5) CovMtx;
191  double Deriv[5][5],CovMtxOld[5][5];
192  for(int i=0; i<5; i++){
193  for(int j=0; j<5; j++){
194  Deriv[i][j]=0.;
195  CovMtxOld[i][j]=0.;
196  }
197  }
198  Deriv[0][0] = -1.;
199  Deriv[1][1] = 1.;
200  Deriv[2][3] = 1.;
201  Deriv[3][2] = 1.;
202  Deriv[4][2] = (std::cos(VKPerigee[2])/(m_CNVMAG*BMAG_CUR)) * VKPerigee[4];
203  Deriv[4][4] = -(std::sin(VKPerigee[2])/(m_CNVMAG*BMAG_CUR));
204 
205  CovMtxOld[0][0] = VKCov[0];
206  CovMtxOld[0][1] = CovMtxOld[1][0] = VKCov[1];
207  CovMtxOld[1][1] = VKCov[2];
208  CovMtxOld[0][2] = CovMtxOld[2][0] = VKCov[3];
209  CovMtxOld[1][2] = CovMtxOld[2][1] = VKCov[4];
210  CovMtxOld[2][2] = VKCov[5];
211  CovMtxOld[0][3] = CovMtxOld[3][0] = VKCov[6];
212  CovMtxOld[1][3] = CovMtxOld[3][1] = VKCov[7];
213  CovMtxOld[2][3] = CovMtxOld[3][2] = VKCov[8];
214  CovMtxOld[3][3] = VKCov[9];
215  CovMtxOld[0][4] = CovMtxOld[4][0] = VKCov[10];
216  CovMtxOld[1][4] = CovMtxOld[4][1] = VKCov[11];
217  CovMtxOld[2][4] = CovMtxOld[4][2] = VKCov[12];
218  CovMtxOld[3][4] = CovMtxOld[4][3] = VKCov[13];
219  CovMtxOld[4][4] = VKCov[14];
220 
221  for(int i=0; i<5; i++){
222  for(int j=i; j<5; j++){
223  double tmp=0.;
224  for(int ik=4; ik>=0; ik--){
225  if(Deriv[i][ik]==0.)continue;
226  for(int jk=4; jk>=0; jk--){
227  if(Deriv[j][jk]==0.)continue;
228  tmp += Deriv[i][ik]*CovMtxOld[ik][jk]*Deriv[j][jk];
229  }
230  }
231  CovMtx(i,j) = CovMtx(j,i)=tmp;
232  }
233  }
234 
235  auto surface = PerigeeSurface(Amg::Vector3D(state.m_refFrameX+vX,
236  state.m_refFrameY+vY,
237  state.m_refFrameZ+vZ));
238 
239  return std::make_unique<Perigee>(TrkP1, TrkP2, TrkP3, TrkP4, TrkP5,
240  surface,
241  std::move(CovMtx));
242  }

◆ CvtNeutralParameters()

StatusCode Trk::TrkVKalVrtFitter::CvtNeutralParameters ( const std::vector< const NeutralParameters * > &  InpTrk,
int &  ntrk,
State &  state 
) const
private

Definition at line 179 of file CvtParametersBase.cxx.

182  {
183 
184  std::vector<const NeutralParameters*>::const_iterator i_pbase;
185  AmgVector(5) VectPerig;
186  Amg::Vector3D perGlobalPos,perGlobalVrt;
187  const NeutralPerigee* mPerN = nullptr;
188  double CovVertTrk[15];
189  double tmp_refFrameX = 0, tmp_refFrameY = 0, tmp_refFrameZ = 0;
190  double rxyMin = 1000000.;
191 
192  //
193  // ----- Set reference frame to (0.,0.,0.) == ATLAS frame
194  // ----- Magnetic field is taken in reference point
195  //
196  state.m_refFrameX = state.m_refFrameY = state.m_refFrameZ = 0.;
197  state.m_fitField.setAtlasMagRefFrame(0., 0., 0.);
198 
199  if( m_InDetExtrapolator == nullptr ){
200  if(msgLvl(MSG::WARNING))msg()<< "No InDet extrapolator given. Can't use TrackParameters!!!" << endmsg;
201  return StatusCode::FAILURE;
202  }
203 
204  //
205  // Cycle to determine common reference point for the fit
206  //
207  int counter = 0;
208  state.m_trkControl.clear();
209  state.m_trkControl.reserve(InpTrk.size());
210  for (i_pbase = InpTrk.begin(); i_pbase != InpTrk.end(); ++i_pbase) {
211  // Global position of hit
212  perGlobalPos = (*i_pbase)->position();
213  // Crazy user protection
214  if(std::abs(perGlobalPos.z()) > m_IDsizeZ) return StatusCode::FAILURE;
215  if(perGlobalPos.perp() > m_IDsizeR)return StatusCode::FAILURE;
216 
217  tmp_refFrameX += perGlobalPos.x() ;
218  tmp_refFrameY += perGlobalPos.y() ;
219  tmp_refFrameZ += perGlobalPos.z() ;
220 
221  // Here we create structure to control material effects
222  TrkMatControl tmpMat;
223  tmpMat.trkSavedLocalVertex.setZero();
224  // on track extrapolation
225  tmpMat.trkRefGlobPos = Amg::Vector3D(perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z());
226  // First measured point strategy
227  tmpMat.extrapolationType = 0;
228  //No reference point for neutral track for the moment !!!
229  tmpMat.TrkPnt = nullptr;
230  tmpMat.prtMass = ParticleConstants::chargedPionMassInMeV;
231  if(counter<(int)state.m_MassInputParticles.size()){
232  tmpMat.prtMass = state.m_MassInputParticles[counter];
233  }
234  tmpMat.TrkID = counter;
235  state.m_trkControl.push_back(tmpMat);
236  counter++;
237  if(perGlobalPos.perp()<rxyMin){
238  rxyMin = perGlobalPos.perp();
239  state.m_globalFirstHit=nullptr;
240  }
241  }
242 
243  if(counter == 0) return StatusCode::FAILURE;
244  // Reference frame for the fit based on hits positions
245  tmp_refFrameX /= counter;
246  tmp_refFrameY /= counter;
247  tmp_refFrameZ /= counter;
248  Amg::Vector3D refGVertex (tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ);
249 
250  // Rotation parameters in case of rotation use
251  double fx,fy,fz,BMAG_FIXED;
252  state.m_fitField.getMagFld(tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ,
253  fx, fy, fz);
254 
255  //
256  // Common reference frame is ready. Start extraction of parameters for fit.
257  // TracksParameters are extrapolated to common point and converted to Perigee
258  // This is needed for VKalVrtCore engine.
259  //
260 
261  for (i_pbase = InpTrk.begin(); i_pbase != InpTrk.end(); ++i_pbase) {
262  const Trk::NeutralParameters* neuparO = (*i_pbase);
263  if(neuparO == nullptr) return StatusCode::FAILURE;
264  const Trk::NeutralParameters* neuparN = m_fitPropagator->myExtrapNeutral(neuparO, &refGVertex);
265  mPerN = dynamic_cast<const Trk::NeutralPerigee*>(neuparN);
266  if(mPerN == nullptr) {
267  delete neuparN;
268  return StatusCode::FAILURE;
269  }
270 
271  VectPerig = mPerN->parameters();
272  // Global position of perigee point
273  perGlobalPos = mPerN->position();
274  // Global position of reference point
275  perGlobalVrt = mPerN->associatedSurface().center();
276  // VK no good covariance matrix!
277  if( !convertAmg5SymMtx(mPerN->covariance(), CovVertTrk) ) return StatusCode::FAILURE;
278  delete neuparN;
279 
280  state.m_refFrameX = state.m_refFrameY = state.m_refFrameZ = 0.;
281  // restore ATLAS frame for safety
282  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.);
283  // Magnetic field at perigee point
284  state.m_fitField.getMagFld(perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z(),
285  fx, fy, BMAG_FIXED);
286  if(std::abs(BMAG_FIXED) < 0.01) BMAG_FIXED = 0.01;
287 
288  VKalTransform(BMAG_FIXED,
289  (double)VectPerig[0], (double)VectPerig[1],
290  (double)VectPerig[2], (double)VectPerig[3],
291  (double)VectPerig[4], CovVertTrk,
292  state.m_ich[ntrk], &state.m_apar[ntrk][0],
293  &state.m_awgt[ntrk][0]);
294 
295  state.m_ich[ntrk]=0;
296  if(state.m_apar[ntrk][4]<0){
297  // Charge=0 is always equal to Charge=+1
298  state.m_apar[ntrk][4] = -state.m_apar[ntrk][4];
299  state.m_awgt[ntrk][10] = -state.m_awgt[ntrk][10];
300  state.m_awgt[ntrk][11] = -state.m_awgt[ntrk][11];
301  state.m_awgt[ntrk][12] = -state.m_awgt[ntrk][12];
302  state.m_awgt[ntrk][13] = -state.m_awgt[ntrk][13];
303  }
304  ntrk++;
305  if(ntrk>=NTrMaxVFit) return StatusCode::FAILURE;
306  }
307 
308  //-------------- Finally setting new reference frame common for ALL tracks
309  state.m_refFrameX = tmp_refFrameX;
310  state.m_refFrameY = tmp_refFrameY;
311  state.m_refFrameZ = tmp_refFrameZ;
312  state.m_fitField.setAtlasMagRefFrame(state.m_refFrameX, state.m_refFrameY, state.m_refFrameZ);
313 
314  return StatusCode::SUCCESS;
315  }

◆ CvtNeutralParticle()

StatusCode Trk::TrkVKalVrtFitter::CvtNeutralParticle ( const std::vector< const xAOD::NeutralParticle * > &  list,
int &  ntrk,
State &  state 
) const
private

Definition at line 124 of file CvtTrackParticle.cxx.

127  {
128  std::vector<const xAOD::NeutralParticle*>::const_iterator i_ntrk;
129  AmgVector(5) VectPerig; VectPerig.setZero();
130  const NeutralPerigee* mPer=nullptr;
131  double CovVertTrk[15]; std::fill(CovVertTrk,CovVertTrk+15,0.);
132  double tmp_refFrameX=0, tmp_refFrameY=0, tmp_refFrameZ=0;
133  double fx,fy,BMAG_FIXED;
134 //
135 // ----- Set reference frame to (0.,0.,0.) == ATLAS frame
136 // ----- Magnetic field is taken in reference point
137 //
138  state.m_refFrameX=state.m_refFrameY=state.m_refFrameZ=0.;
139  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.);
140 //
141 // Cycle to determine common reference point for the fit
142 //
143  int counter =0;
144  Amg::Vector3D perGlobalPos;
145  state.m_trkControl.clear(); state.m_trkControl.reserve(InpTrk.size());
146  for (i_ntrk = InpTrk.begin(); i_ntrk != InpTrk.end(); ++i_ntrk) {
147 //-- (Measured)Perigee in xAOD::NeutralParticle
148  mPer = &(*i_ntrk)->perigeeParameters();
149  if( mPer==nullptr ) continue; // No perigee!!!
150  perGlobalPos = mPer->position(); //Global position of perigee point
151  if(fabs(perGlobalPos.z()) > m_IDsizeZ)return StatusCode::FAILURE; // Crazy user protection
152  if( perGlobalPos.perp() > m_IDsizeR)return StatusCode::FAILURE;
153  tmp_refFrameX += perGlobalPos.x() ; // Reference system calculation
154  tmp_refFrameY += perGlobalPos.y() ; // Use hit position itself to get more precise
155  tmp_refFrameZ += perGlobalPos.z() ; // magnetic field
156  TrkMatControl tmpMat;
157  tmpMat.trkSavedLocalVertex.setZero();
158  tmpMat.trkRefGlobPos=Amg::Vector3D( perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z());
159  tmpMat.extrapolationType=2; // Perigee point strategy
160  tmpMat.TrkPnt=nullptr; //No reference point for neutral particle for the moment
161  tmpMat.prtMass = ParticleConstants::chargedPionMassInMeV;
162  if(counter<(int)state.m_MassInputParticles.size())tmpMat.prtMass = state.m_MassInputParticles[counter];
163  tmpMat.TrkID=counter; state.m_trkControl.push_back(tmpMat);
164  counter++;
165  }
166  if(counter == 0) return StatusCode::FAILURE;
167  tmp_refFrameX /= counter; // Reference frame for the fit
168  tmp_refFrameY /= counter;
169  tmp_refFrameZ /= counter;
170  Amg::Vector3D refGVertex (tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ);
171  PerigeeSurface surfGRefPoint( refGVertex ); // Reference perigee surface for current fit
172 //
173 //std::cout.setf( std::ios::scientific); std::cout.precision(5);
174 //std::cout<<" VK ref.frame="<<tmp_refFrameX<<", "<<tmp_refFrameY<<", "<<tmp_refFrameZ<<'\n';
175 //
176 // Common reference frame is ready. Start extraction of parameters for fit.
177 //
178 
179  state.m_refFrameX=state.m_refFrameY=state.m_refFrameZ=0.; //set ATLAS frame
180  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.); //set ATLAS frame
181  for (i_ntrk = InpTrk.begin(); i_ntrk != InpTrk.end(); ++i_ntrk) {
182 //
183 //-- (Measured)Perigee in TrackParticle
184 //
185  mPer = &(*i_ntrk)->perigeeParameters();
186  if( mPer==nullptr ) continue; // No perigee!!!
187  perGlobalPos = mPer->position(); //Global position of perigee point
188  if( !convertAmg5SymMtx(mPer->covariance(), CovVertTrk) ) return StatusCode::FAILURE; //VK no good covariance matrix!;
189  state.m_fitField.getMagFld( perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z(), // Magnetic field
190  fx, fy, BMAG_FIXED); // at track perigee point
191  if(fabs(BMAG_FIXED) < 0.01) BMAG_FIXED=0.01;
192 
193 //
194 //--- Move ref. frame to the track common point refGVertex
195 // Small beamline inclination doesn't change track covariance matrix
196 //
197  AmgSymMatrix(5) tmpCov = AmgSymMatrix(5)(*(mPer->covariance()));
198  const Perigee tmpPer(mPer->position(),mPer->momentum(),mPer->charge(),surfGRefPoint,std::move(tmpCov));
199  VectPerig = tmpPer.parameters();
200  //--- Transform to internal parametrisation
201  VKalTransform( BMAG_FIXED, (double)VectPerig[0], (double)VectPerig[1],
202  (double)VectPerig[2], (double)VectPerig[3], (double)VectPerig[4], CovVertTrk,
203  state.m_ich[ntrk],&state.m_apar[ntrk][0],&state.m_awgt[ntrk][0]);
204  state.m_ich[ntrk]=0;
205  if(state.m_apar[ntrk][4]<0){ state.m_apar[ntrk][4] = -state.m_apar[ntrk][4]; // Charge=0 is always equal to Charge=+1
206  state.m_awgt[ntrk][10] = -state.m_awgt[ntrk][10];
207  state.m_awgt[ntrk][11] = -state.m_awgt[ntrk][11];
208  state.m_awgt[ntrk][12] = -state.m_awgt[ntrk][12];
209  state.m_awgt[ntrk][13] = -state.m_awgt[ntrk][13]; }
210 
211  ntrk++;
212  if(ntrk>=NTrMaxVFit) {
213  return StatusCode::FAILURE;
214  }
215  }
216 //-------------- Finally setting new reference frame common for ALL tracks
217  state.m_refFrameX=tmp_refFrameX;
218  state.m_refFrameY=tmp_refFrameY;
219  state.m_refFrameZ=tmp_refFrameZ;
220  state.m_fitField.setAtlasMagRefFrame( state.m_refFrameX, state.m_refFrameY, state.m_refFrameZ);
221  return StatusCode::SUCCESS;
222  }

◆ CvtPerigee()

StatusCode Trk::TrkVKalVrtFitter::CvtPerigee ( const std::vector< const Perigee * > &  list,
int &  ntrk,
State &  state 
) const
private

Definition at line 25 of file CvtPerigee.cxx.

28  {
29 
30  double tmp_refFrameX = 0, tmp_refFrameY = 0, tmp_refFrameZ = 0;
31 
32  //
33  // ----- Set reference frame to (0.,0.,0.) == ATLAS frame
34  // ----- Magnetic field is taken in reference point
35  //
36  state.m_refFrameX = state.m_refFrameY = state.m_refFrameZ = 0.;
37  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.);
38 
39  //
40  // Cycle to determine common reference point for the fit
41  //
42  int counter =0;
43  state.m_trkControl.clear();
44  for (const auto& mPer : InpPerigee) {
45  if( mPer == nullptr ){ continue; }
46 
47  // Global position of perigee point
48  Amg::Vector3D perGlobalPos = mPer->position();
49  // Crazy user protection
50  if(!(state.m_allowUltraDisplaced) && std::abs(perGlobalPos.z()) > m_IDsizeZ) return StatusCode::FAILURE;
51  if(!(state.m_allowUltraDisplaced) && perGlobalPos.perp() > m_IDsizeR) return StatusCode::FAILURE;
52 
53  // Reference system calculation
54  // Use hit position itself to get more precise magnetic field
55  tmp_refFrameX += perGlobalPos.x() ;
56  tmp_refFrameY += perGlobalPos.y() ;
57  tmp_refFrameZ += perGlobalPos.z() ;
58 
59  TrkMatControl tmpMat;
60  tmpMat.trkRefGlobPos = Amg::Vector3D(perGlobalPos.x(),
61  perGlobalPos.y(),
62  perGlobalPos.z());
63  // Perigee point strategy
64  tmpMat.extrapolationType = 2;
65  tmpMat.TrkPnt = mPer;
66  tmpMat.prtMass = ParticleConstants::chargedPionMassInMeV;
67  if(counter < static_cast<int>(state.m_MassInputParticles.size())){
68  tmpMat.prtMass = state.m_MassInputParticles[counter];
69  }
70  tmpMat.trkSavedLocalVertex.setZero();
71  tmpMat.TrkID=counter;
72  state.m_trkControl.push_back(tmpMat);
73  counter++;
74  }
75 
76  if(counter == 0) return StatusCode::FAILURE;
77 
78  // Reference frame for the fit
79  tmp_refFrameX /= counter;
80  tmp_refFrameY /= counter;
81  tmp_refFrameZ /= counter;
82 
83  //
84  // Common reference frame is ready. Start extraction of parameters for fit.
85  //
86  double fx = 0., fy = 0., BMAG_FIXED = 0.;
87  for (const auto& mPer : InpPerigee) {
88  if(mPer == nullptr){ continue; }
89  AmgVector(5) VectPerig = mPer->parameters();
90  // Global position of perigee point
91  Amg::Vector3D perGlobalPos = mPer->position();
92  // Global position of reference point
93  Amg::Vector3D perGlobalVrt = mPer->associatedSurface().center();
94  // Restore ATLAS frame
95  state.m_refFrameX = state.m_refFrameY = state.m_refFrameZ = 0.;
96  state.m_fitField.setAtlasMagRefFrame(0., 0., 0.);
97  // Magnetic field at perigee point
98  state.m_fitField.getMagFld(perGlobalPos.x(),
99  perGlobalPos.y(),
100  perGlobalPos.z(),
101  fx, fy, BMAG_FIXED);
102  if(std::abs(BMAG_FIXED) < 0.01) BMAG_FIXED = 0.01;
103 
104  double CovVertTrk[15];
105  std::fill(CovVertTrk,CovVertTrk+15,0.);
106  // No good covariance matrix!
107  if(!convertAmg5SymMtx(mPer->covariance(), CovVertTrk)) return StatusCode::FAILURE;
108  VKalTransform(BMAG_FIXED,
109  static_cast<double>(VectPerig(0)),
110  static_cast<double>(VectPerig(1)),
111  static_cast<double>(VectPerig(2)),
112  static_cast<double>(VectPerig(3)),
113  static_cast<double>(VectPerig(4)),
114  CovVertTrk,
115  state.m_ich[ntrk], &state.m_apar[ntrk][0],
116  &state.m_awgt[ntrk][0]);
117 
118  // Check if propagation to common reference point is needed and make it
119  // initial track reference position
120  state.m_refFrameX=perGlobalVrt.x();
121  state.m_refFrameY=perGlobalVrt.y();
122  state.m_refFrameZ=perGlobalVrt.z();
123  state.m_fitField.setAtlasMagRefFrame(state.m_refFrameX,
124  state.m_refFrameY,
125  state.m_refFrameZ);
126  double dX = tmp_refFrameX-perGlobalVrt.x();
127  double dY = tmp_refFrameY-perGlobalVrt.y();
128  double dZ = tmp_refFrameZ-perGlobalVrt.z();
129  if(std::abs(dX)+std::abs(dY)+std::abs(dZ) != 0.) {
130  double pari[5], covi[15];
131  double vrtini[3] = {0.,0.,0.};
132  double vrtend[3] = {dX,dY,dZ};
133  for(int i=0; i<5; i++) pari[i] = state.m_apar[ntrk][i];
134  for(int i=0; i<15;i++) covi[i] = state.m_awgt[ntrk][i];
135  long int Charge = (long int) mPer->charge();
136  long int TrkID = ntrk;
137  Trk::vkalPropagator::Propagate(TrkID, Charge, pari, covi,
138  vrtini, vrtend, &state.m_apar[ntrk][0],
139  &state.m_awgt[ntrk][0],
140  &state.m_vkalFitControl);
141  }
142 
143  ntrk++;
144  if(ntrk>=NTrMaxVFit) return StatusCode::FAILURE;
145  }
146 
147  //-------------- Finally setting new reference frame common for ALL tracks
148  state.m_refFrameX = tmp_refFrameX;
149  state.m_refFrameY = tmp_refFrameY;
150  state.m_refFrameZ = tmp_refFrameZ;
151  state.m_fitField.setAtlasMagRefFrame(state.m_refFrameX,
152  state.m_refFrameY,
153  state.m_refFrameZ);
154 
155  return StatusCode::SUCCESS;
156  }

◆ CvtTrackParameters()

StatusCode Trk::TrkVKalVrtFitter::CvtTrackParameters ( const std::vector< const TrackParameters * > &  InpTrk,
int &  ntrk,
State &  state 
) const
private

Definition at line 24 of file CvtParametersBase.cxx.

27  {
28 
29  std::vector<const TrackParameters*>::const_iterator i_pbase;
30  AmgVector(5) VectPerig;
31  VectPerig.setZero();
32  Amg::Vector3D perGlobalPos,perGlobalVrt;
33  const Trk::Perigee* mPer=nullptr;
34 
35  double tmp_refFrameX = 0, tmp_refFrameY = 0, tmp_refFrameZ = 0;
36  double rxyMin = 1000000.;
37 
38  //
39  // ----- Set reference frame to (0.,0.,0.) == ATLAS frame
40  // ----- Magnetic field is taken in reference point
41  //
42  state.m_refFrameX=state.m_refFrameY=state.m_refFrameZ=0.;
43  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.);
44 
45  if( m_InDetExtrapolator == nullptr ){
46  if(msgLvl(MSG::WARNING))msg()<< "No InDet extrapolator given. Can't use TrackParameters!!!" << endmsg;
47  return StatusCode::FAILURE;
48  }
49 
50  //
51  // Cycle to determine common reference point for the fit
52  //
53  int counter =0;
54  state.m_trkControl.clear();
55  state.m_trkControl.reserve(InpTrk.size());
56  for (i_pbase = InpTrk.begin(); i_pbase != InpTrk.end(); ++i_pbase) {
57  // Global position of hit
58  perGlobalPos = (*i_pbase)->position();
59  // Crazy user protection
60  if(!(state.m_allowUltraDisplaced) && std::abs(perGlobalPos.z()) > m_IDsizeZ) return StatusCode::FAILURE;
61  if(!(state.m_allowUltraDisplaced) && perGlobalPos.perp() > m_IDsizeR) return StatusCode::FAILURE;
62  tmp_refFrameX += perGlobalPos.x();
63  tmp_refFrameY += perGlobalPos.y();
64  tmp_refFrameZ += perGlobalPos.z();
65 
66  // Here we create structure to control material effects
67  TrkMatControl tmpMat;
68  tmpMat.trkSavedLocalVertex.setZero();
69  tmpMat.trkRefGlobPos = Amg::Vector3D(perGlobalPos.x(),
70  perGlobalPos.y(),
71  perGlobalPos.z());
72  // First measured point strategy
73  tmpMat.extrapolationType = m_firstMeasuredPoint ? 0 : 1;
74  tmpMat.TrkPnt = (*i_pbase);
75  tmpMat.prtMass = ParticleConstants::chargedPionMassInMeV;
76  if(counter < (int)state.m_MassInputParticles.size()){
77  tmpMat.prtMass = state.m_MassInputParticles[counter];
78  }
79  tmpMat.TrkID = counter;
80  state.m_trkControl.push_back(tmpMat);
81  counter++;
82 
83  if(perGlobalPos.perp() < rxyMin){
84  rxyMin=perGlobalPos.perp();
85  state.m_globalFirstHit=(*i_pbase);
86  }
87  }
88 
89  if(counter == 0) return StatusCode::FAILURE;
90  // Reference frame for the fit based on hits positions
91  tmp_refFrameX /= counter;
92  tmp_refFrameY /= counter;
93  tmp_refFrameZ /= counter;
94  Amg::Vector3D refGVertex(tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ);
95 
96  //Rotation parameters in case of rotation use
97  double fx, fy, fz, BMAG_FIXED;
98  state.m_fitField.getMagFld(tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ,
99  fx, fy, fz);
100 
101  //
102  // Common reference frame is ready. Start extraction of parameters for fit.
103  // TracksParameters are extrapolated to common point and converted to Perigee
104  // This is needed for VKalVrtCore engine.
105  //
106 
107  double CovVertTrk[15];
108  std::fill(CovVertTrk, CovVertTrk+15, 0.);
109 
110  for (i_pbase = InpTrk.begin(); i_pbase != InpTrk.end(); ++i_pbase) {
111  long int TrkID=ntrk;
112  const TrackParameters* trkparO = (*i_pbase);
113 
114  if(trkparO){
115  const Trk::TrackParameters* trkparN =
116  m_fitPropagator->myExtrapWithMatUpdate(TrkID,
117  trkparO,
118  &refGVertex,
119  state);
120  if(trkparN == nullptr) return StatusCode::FAILURE;
121  mPer = dynamic_cast<const Trk::Perigee*>(trkparN);
122  if(mPer == nullptr) {
123  delete trkparN;
124  return StatusCode::FAILURE;
125  }
126 
127  VectPerig = mPer->parameters();
128  // Global position of perigee point
129  perGlobalPos = mPer->position();
130  // Global position of reference point
131  perGlobalVrt = mPer->associatedSurface().center();
132  // VK no good covariance matrix!
133  if( !convertAmg5SymMtx(mPer->covariance(), CovVertTrk) ) return StatusCode::FAILURE;
134  delete trkparN;
135  }
136 
137  state.m_refFrameX = state.m_refFrameY = state.m_refFrameZ = 0.;
138  // Restore ATLAS frame for safety
139  state.m_fitField.setAtlasMagRefFrame(0., 0., 0.);
140  // Magnetic field at perigee point
141  state.m_fitField.getMagFld(perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z(),
142  fx, fy, BMAG_FIXED);
143  if(std::abs(BMAG_FIXED) < 0.01) BMAG_FIXED = 0.01;
144 
145  VKalTransform(BMAG_FIXED,
146  (double)VectPerig[0], (double)VectPerig[1],
147  (double)VectPerig[2], (double)VectPerig[3],
148  (double)VectPerig[4], CovVertTrk,
149  state.m_ich[ntrk], &state.m_apar[ntrk][0],
150  &state.m_awgt[ntrk][0]);
151 
152  // Neutral track
153  if( trkparO==nullptr ) {
154  state.m_ich[ntrk]=0;
155  if(state.m_apar[ntrk][4]<0){
156  // Charge=0 is always equal to Charge=+1
157  state.m_apar[ntrk][4] = -state.m_apar[ntrk][4];
158  state.m_awgt[ntrk][10] = -state.m_awgt[ntrk][10];
159  state.m_awgt[ntrk][11] = -state.m_awgt[ntrk][11];
160  state.m_awgt[ntrk][12] = -state.m_awgt[ntrk][12];
161  state.m_awgt[ntrk][13] = -state.m_awgt[ntrk][13];
162  }
163  }
164  ntrk++;
165  if(ntrk>=NTrMaxVFit) return StatusCode::FAILURE;
166  }
167 
168  //-------------- Finally setting new reference frame common for ALL tracks
169  state.m_refFrameX = tmp_refFrameX;
170  state.m_refFrameY = tmp_refFrameY;
171  state.m_refFrameZ = tmp_refFrameZ;
172  state.m_fitField.setAtlasMagRefFrame(state.m_refFrameX, state.m_refFrameY, state.m_refFrameZ);
173 
174  return StatusCode::SUCCESS;
175  }

◆ CvtTrackParticle()

StatusCode Trk::TrkVKalVrtFitter::CvtTrackParticle ( std::span< const xAOD::TrackParticle *const >  list,
int &  ntrk,
State &  state 
) const
private

Definition at line 27 of file CvtTrackParticle.cxx.

30  {
31 
32  AmgVector(5) VectPerig; VectPerig.setZero();
33  const Trk::Perigee* mPer=nullptr;
34  double CovVertTrk[15]; std::fill(CovVertTrk,CovVertTrk+15,0.);
35  double tmp_refFrameX=0, tmp_refFrameY=0, tmp_refFrameZ=0;
36  double fx,fy,BMAG_FIXED;
37 //
38 // ----- Set reference frame to (0.,0.,0.) == ATLAS frame
39 // ----- Magnetic field is taken in reference point
40 //
41  state.m_refFrameX=state.m_refFrameY=state.m_refFrameZ=0.;
42  state.m_fitField.setAtlasMagRefFrame( 0., 0., 0.);
43 //
44 // Cycle to determine common reference point for the fit
45 //
46  int counter =0;
47  Amg::Vector3D perGlobalPos;
48  state.m_trkControl.clear(); state.m_trkControl.reserve(InpTrk.size());
49  for (auto i_ntrk = InpTrk.begin(); i_ntrk != InpTrk.end(); ++i_ntrk) {
50 //-- (Measured)Perigee in xAOD::TrackParticle
51  mPer = &(*i_ntrk)->perigeeParameters();
52  if( mPer==nullptr ) continue; // No perigee!!!
53  perGlobalPos = mPer->position(); //Global position of perigee point
54  if(!(state.m_allowUltraDisplaced) && std::abs(perGlobalPos.z()) > m_IDsizeZ)return StatusCode::FAILURE; // Crazy user protection
55  if(!(state.m_allowUltraDisplaced) && perGlobalPos.perp() > m_IDsizeR)return StatusCode::FAILURE;
56  tmp_refFrameX += perGlobalPos.x() ; // Reference system calculation
57  tmp_refFrameY += perGlobalPos.y() ; // Use hit position itself to get more precise
58  tmp_refFrameZ += perGlobalPos.z() ; // magnetic field
59  TrkMatControl tmpMat;
60  tmpMat.trkSavedLocalVertex.setZero();
61  tmpMat.trkRefGlobPos=Amg::Vector3D( perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z());
62  tmpMat.extrapolationType=2; // Perigee point strategy
63  tmpMat.TrkPnt=mPer;
64  tmpMat.prtMass = ParticleConstants::chargedPionMassInMeV;
65  if(counter<(int)state.m_MassInputParticles.size())tmpMat.prtMass = state.m_MassInputParticles[counter];
66  tmpMat.TrkID=counter; state.m_trkControl.push_back(tmpMat);
67  counter++;
68  }
69  if(counter == 0) return StatusCode::FAILURE;
70  tmp_refFrameX /= counter; // Reference frame for the fit
71  tmp_refFrameY /= counter;
72  tmp_refFrameZ /= counter;
73  Amg::Vector3D refGVertex (tmp_refFrameX, tmp_refFrameY, tmp_refFrameZ);
74 
75  PerigeeSurface surfGRefPoint( refGVertex ); // Reference perigee surface for current fit
76 //
77 //std::cout.setf( std::ios::scientific); std::cout.precision(9);
78 //std::cout<<" VK ref.frame="<<tmp_refFrameX<<", "<<tmp_refFrameY<<", "<<tmp_refFrameZ<<'\n';
79 //
80 // Common reference frame is ready. Start extraction of parameters for fit.
81 //
82 
83  for (auto i_ntrk = InpTrk.begin(); i_ntrk != InpTrk.end(); ++i_ntrk) {
84 //
85 //-- (Measured)Perigee in TrackParticle
86 //
87  mPer = &(*i_ntrk)->perigeeParameters();
88  if( mPer==nullptr ) continue; // No perigee!!!
89  perGlobalPos = mPer->position(); //Global position of perigee point
90  if( !convertAmg5SymMtx(mPer->covariance(), CovVertTrk) ) return StatusCode::FAILURE; //VK no good covariance matrix!;
91  state.m_fitField.getMagFld( perGlobalPos.x(), perGlobalPos.y(), perGlobalPos.z(), // Magnetic field
92  fx, fy, BMAG_FIXED); // at the track perigee point
93  if(fabs(BMAG_FIXED) < 0.01) BMAG_FIXED=0.01;
94 //
95 //--- Move ref. frame to the track common point refGVertex
96 // Small beamline inclination doesn't change track covariance matrix
97  AmgSymMatrix(5) tmpCov = AmgSymMatrix(5)(*(mPer->covariance()));
98  const Perigee tmpPer(mPer->position(),mPer->momentum(),mPer->charge(),surfGRefPoint,std::move(tmpCov));
99  VectPerig = tmpPer.parameters();
100 //--- Transform to internal parametrisation
101  VKalTransform( BMAG_FIXED, (double)VectPerig[0], (double)VectPerig[1],
102  (double)VectPerig[2], (double)VectPerig[3], (double)VectPerig[4], CovVertTrk,
103  state.m_ich[ntrk],&state.m_apar[ntrk][0],&state.m_awgt[ntrk][0]);
104 //
105  ntrk++;
106  if(ntrk>=NTrMaxVFit) {
107  return StatusCode::FAILURE;
108  }
109  }
110 //-------------- Finally setting new reference frame common for ALL tracks
111  state.m_refFrameX=tmp_refFrameX;
112  state.m_refFrameY=tmp_refFrameY;
113  state.m_refFrameZ=tmp_refFrameZ;
114  state.m_fitField.setAtlasMagRefFrame( state.m_refFrameX, state.m_refFrameY, state.m_refFrameZ);
115 
116  return StatusCode::SUCCESS;
117  }

◆ FillMatrixP() [1/2]

void Trk::TrkVKalVrtFitter::FillMatrixP ( AmgSymMatrix(5)&  CovMtx,
std::vector< double > &  Matrix 
)
staticprivate

Definition at line 614 of file TrkVKalVrtFitter.cxx.

615 {
616  CovMtx.setIdentity();
617  if( Matrix.size() < 21) return;
618  CovMtx(0,0) = 0;
619  CovMtx(1,1) = 0;
620  CovMtx(2,2)= Matrix[ 9];
621  CovMtx.fillSymmetric(2,3,Matrix[13]);
622  CovMtx(3,3)= Matrix[14];
623  CovMtx.fillSymmetric(2,4,Matrix[18]);
624  CovMtx.fillSymmetric(3,4,Matrix[19]);
625  CovMtx(4,4)= Matrix[20];
626 }

◆ FillMatrixP() [2/2]

void Trk::TrkVKalVrtFitter::FillMatrixP ( int  iTrk,
AmgSymMatrix(5)&  CovMtx,
std::vector< double > &  Matrix 
)
staticprivate

Definition at line 629 of file TrkVKalVrtFitter.cxx.

630 {
631  int iTmp=(iTrk+1)*3;
632  int NContent = Matrix.size();
633  CovMtx.setIdentity(); //Clean matrix for the beginning, then fill needed elements
634  CovMtx(0,0) = 0;
635  CovMtx(1,1) = 0;
636  int pnt = (iTmp+1)*iTmp/2 + iTmp; if( pnt > NContent ) return;
637  CovMtx(2,2) = Matrix[pnt];
638  pnt = (iTmp+1+1)*(iTmp+1)/2 + iTmp; if( pnt+1 > NContent ){ CovMtx.setIdentity(); return; }
639  CovMtx.fillSymmetric(2,3,Matrix[pnt]);
640  CovMtx(3,3) = Matrix[pnt+1];
641  pnt = (iTmp+2+1)*(iTmp+2)/2 + iTmp; if( pnt+2 > NContent ){ CovMtx.setIdentity(); return; }
642  CovMtx.fillSymmetric(2,4,Matrix[pnt]);
643  CovMtx.fillSymmetric(3,4,Matrix[pnt+1]);
644  CovMtx(4,4) = Matrix[pnt+2];
645 }

◆ finalize()

StatusCode Trk::TrkVKalVrtFitter::finalize ( )
finaloverridevirtual

Definition at line 65 of file TrkVKalVrtFitter.cxx.

66 {
67  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG) <<"TrkVKalVrtFitter finalize() successful" << endmsg;
68  return StatusCode::SUCCESS;
69 }

◆ findPositions()

int Trk::TrkVKalVrtFitter::findPositions ( const std::vector< int > &  inputList,
const std::vector< int > &  refList,
std::vector< int > &  index 
)
staticprivate

Definition at line 787 of file TrkCascadeFitter.cxx.

788 {
789  int R,I;
790  index.clear();
791  int nI=inputList.size(); if(nI==0) return 0; //all ok
792  int nR=refList.size(); if(nR==0) return 0; //all ok
793  //std::cout<<"inp="; for(I=0; I<nI; I++)std::cout<<inputList[I]; std::cout<<'\n';
794  //std::cout<<"ref="; for(R=0; R<nR; R++)std::cout<<refList[R]; std::cout<<'\n';
795  for(I=0; I<nI; I++){
796  for(R=0; R<nR; R++) if(inputList[I]==refList[R]){index.push_back(R); break;}
797  if(R==nR) return -1; //input element not found in reference list
798  }
799  return 0;
800 }

◆ fit() [1/12]

std::unique_ptr< xAOD::Vertex > Trk::TrkVKalVrtFitter::fit ( const EventContext &  ctx,
const std::vector< const xAOD::TrackParticle * > &  vectorTrk,
const Amg::Vector3D &  startingPoint 
) const
finaloverridevirtual

Interface for xAOD::TrackParticle with starting point Implements the new style (unique_ptr,EventContext)

Definition at line 365 of file TrkVKalVrtFitter.cxx.

368 {
369  State state;
370  initState(ctx, state);
371  return std::unique_ptr<xAOD::Vertex>(fit(xtpListC, startingPoint, state));
372 }

◆ fit() [2/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeListC) const
finaloverridevirtual

Definition at line 557 of file TrkVKalVrtFitter.cxx.

558 {
559  State state;
560  initState (state);
561  Amg::Vector3D VertexIni(0.,0.,0.);
562  StatusCode sc=VKalVrtFitFast(perigeeListC, VertexIni, state);
563  if( sc.isSuccess()) setApproximateVertex(VertexIni.x(),VertexIni.y(),VertexIni.z(),state);
564  std::vector<const NeutralParameters*> perigeeListN(0);
565  Amg::Vector3D Vertex;
566  TLorentzVector Momentum;
567  long int Charge;
568  std::vector<double> ErrorMatrix;
569  std::vector<double> Chi2PerTrk;
570  std::vector< std::vector<double> > TrkAtVrt;
571  double Chi2;
572  sc=VKalVrtFit( perigeeListC, perigeeListN,
573  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
574 
575  xAOD::Vertex * tmpVertex = nullptr;
576  if(sc.isSuccess()) {
577  tmpVertex = makeXAODVertex( 0, Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
578  }
579  return tmpVertex;
580 }

◆ fit() [3/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeListC,
const std::vector< const Trk::NeutralParameters * > &  perigeeListN 
) const
finaloverridevirtual

Definition at line 582 of file TrkVKalVrtFitter.cxx.

584 {
585  State state;
586  initState (state);
587  Amg::Vector3D VertexIni(0.,0.,0.);
588  StatusCode sc=VKalVrtFitFast(perigeeListC, VertexIni, state);
589  if( sc.isSuccess()) setApproximateVertex(VertexIni.x(),VertexIni.y(),VertexIni.z(),state);
590  Amg::Vector3D Vertex;
591  TLorentzVector Momentum;
592  long int Charge;
593  std::vector<double> ErrorMatrix;
594  std::vector<double> Chi2PerTrk;
595  std::vector< std::vector<double> > TrkAtVrt;
596  double Chi2;
597  sc=VKalVrtFit( perigeeListC, perigeeListN,
598  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
599 
600  xAOD::Vertex * tmpVertex = nullptr;
601  if(sc.isSuccess()) {
602  tmpVertex = makeXAODVertex( (int)perigeeListN.size(), Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
603  }
604  return tmpVertex;
605 }

◆ fit() [4/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeList,
const Amg::Vector3D &  startingPoint 
) const
finaloverridevirtual

Interface for MeasuredPerigee with starting point.

Definition at line 203 of file TrkVKalVrtFitter.cxx.

205 {
206  State state;
207  initState (state);
208  setApproximateVertex(startingPoint.x(),
209  startingPoint.y(),
210  startingPoint.z(),
211  state);
212  std::vector<const NeutralParameters*> perigeeListN(0);
213  Amg::Vector3D Vertex;
214  TLorentzVector Momentum;
215  long int Charge;
216  std::vector<double> ErrorMatrix;
217  std::vector<double> Chi2PerTrk;
218  std::vector< std::vector<double> > TrkAtVrt;
219  double Chi2;
220  StatusCode sc=VKalVrtFit( perigeeListC, perigeeListN,
221  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
222 
223  xAOD::Vertex * tmpVertex = nullptr;
224  if(sc.isSuccess()) {
225  tmpVertex = makeXAODVertex( 0, Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
226  }
227  return tmpVertex;
228 }

◆ fit() [5/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeList,
const std::vector< const NeutralParameters * > &  perigeeListN,
const Amg::Vector3D &  startingPoint 
) const
finaloverridevirtual

Definition at line 231 of file TrkVKalVrtFitter.cxx.

234 {
235  State state;
236  initState (state);
237  setApproximateVertex(startingPoint.x(),
238  startingPoint.y(),
239  startingPoint.z(),
240  state);
241  Amg::Vector3D Vertex;
242  TLorentzVector Momentum;
243  long int Charge;
244  std::vector<double> ErrorMatrix;
245  std::vector<double> Chi2PerTrk;
246  std::vector< std::vector<double> > TrkAtVrt;
247  double Chi2;
248  StatusCode sc=VKalVrtFit( perigeeListC,perigeeListN,
249  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
250 
251  xAOD::Vertex * tmpVertex = nullptr;
252  if(sc.isSuccess()) {
253  tmpVertex = makeXAODVertex( (int)perigeeListN.size(), Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
254  }
255  return tmpVertex;
256 }

◆ fit() [6/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeList,
const std::vector< const NeutralParameters * > &  perigeeListN,
const xAOD::Vertex &  constraint 
) const
finaloverridevirtual

Definition at line 313 of file TrkVKalVrtFitter.cxx.

316 {
317  State state;
318  initState (state);
319 
320  if(msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)<< "A priori vertex constraint is activated in VKalVrt fitter!" << endmsg;
321  Amg::Vector3D VertexIni(0.,0.,0.);
322  StatusCode sc=VKalVrtFitFast(perigeeListC, VertexIni, state);
323  if( sc.isSuccess()){
324  setApproximateVertex(VertexIni.x(),VertexIni.y(),VertexIni.z(),state);
325  }else{
326  setApproximateVertex(constraint.position().x(),
327  constraint.position().y(),
328  constraint.position().z(),
329  state);
330  }
331  setVertexForConstraint(constraint.position().x(),
332  constraint.position().y(),
333  constraint.position().z(),
334  state);
335  setCovVrtForConstraint(constraint.covariancePosition()(Trk::x,Trk::x),
336  constraint.covariancePosition()(Trk::y,Trk::x),
337  constraint.covariancePosition()(Trk::y,Trk::y),
338  constraint.covariancePosition()(Trk::z,Trk::x),
339  constraint.covariancePosition()(Trk::z,Trk::y),
340  constraint.covariancePosition()(Trk::z,Trk::z),
341  state);
342  state.m_useAprioriVertex=true;
343  Amg::Vector3D Vertex;
344  TLorentzVector Momentum;
345  long int Charge;
346  std::vector<double> ErrorMatrix;
347  std::vector<double> Chi2PerTrk;
348  std::vector< std::vector<double> > TrkAtVrt;
349  double Chi2;
350  sc=VKalVrtFit( perigeeListC, perigeeListN,
351  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
352 
353 
354  xAOD::Vertex * tmpVertex = nullptr;
355  if(sc.isSuccess()) {
356  tmpVertex = makeXAODVertex( (int)perigeeListN.size(), Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
357  }
358  return tmpVertex;
359 }

◆ fit() [7/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const TrackParameters * > &  perigeeListC,
const xAOD::Vertex &  constraint 
) const
finaloverridevirtual

Interface for MeasuredPerigee with vertex constraint.

the position of the constraint is ALWAYS the starting point

Definition at line 265 of file TrkVKalVrtFitter.cxx.

267 {
268  State state;
269  initState (state);
270  if(msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)<< "A priori vertex constraint is activated in VKalVrt fitter!" << endmsg;
271  Amg::Vector3D VertexIni(0.,0.,0.);
272  StatusCode sc=VKalVrtFitFast(perigeeListC, VertexIni, state);
273  if( sc.isSuccess()){
274  setApproximateVertex(VertexIni.x(),VertexIni.y(),VertexIni.z(),state);
275  }else{
276  setApproximateVertex(constraint.position().x(),
277  constraint.position().y(),
278  constraint.position().z(),
279  state);
280  }
281  setVertexForConstraint(constraint.position().x(),
282  constraint.position().y(),
283  constraint.position().z(),
284  state);
285  setCovVrtForConstraint(constraint.covariancePosition()(Trk::x,Trk::x),
286  constraint.covariancePosition()(Trk::y,Trk::x),
287  constraint.covariancePosition()(Trk::y,Trk::y),
288  constraint.covariancePosition()(Trk::z,Trk::x),
289  constraint.covariancePosition()(Trk::z,Trk::y),
290  constraint.covariancePosition()(Trk::z,Trk::z),
291  state);
292  state.m_useAprioriVertex=true;
293  std::vector<const NeutralParameters*> perigeeListN(0);
294  Amg::Vector3D Vertex;
295  TLorentzVector Momentum;
296  long int Charge;
297  std::vector<double> ErrorMatrix;
298  std::vector<double> Chi2PerTrk;
299  std::vector< std::vector<double> > TrkAtVrt;
300  double Chi2;
301  sc=VKalVrtFit( perigeeListC, perigeeListN,
302  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
303 
304 
305  xAOD::Vertex * tmpVertex = nullptr;
306  if(sc.isSuccess()) {
307  tmpVertex = makeXAODVertex( 0, Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
308  }
309  return tmpVertex;
310 }

◆ fit() [8/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const xAOD::TrackParticle * > &  vectorTrk,
const Amg::Vector3D &  constraint,
IVKalState &  istate 
) const

Definition at line 374 of file TrkVKalVrtFitter.cxx.

377 {
378  assert(dynamic_cast<State*> (&istate)!=nullptr);
379  State& state = static_cast<State&> (istate);
380 
381  xAOD::Vertex * tmpVertex = nullptr;
382  setApproximateVertex(startingPoint.x(),
383  startingPoint.y(),
384  startingPoint.z(),
385  state);
386  std::vector<const xAOD::NeutralParticle*> xtpListN(0);
387  Amg::Vector3D Vertex;
388  TLorentzVector Momentum;
389  long int Charge;
390  std::vector<double> ErrorMatrix;
391  std::vector<double> Chi2PerTrk;
392  std::vector< std::vector<double> > TrkAtVrt;
393  double Chi2;
394  StatusCode sc=VKalVrtFit( xtpListC, xtpListN,
395  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
396  if(sc.isSuccess()) {
397  tmpVertex = makeXAODVertex( 0, Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
398  dvect fittrkwgt;
399  sc=VKalGetTrkWeights(fittrkwgt, state); if(sc.isFailure())fittrkwgt.clear();
400  for(int ii=0; ii<state.m_FitStatus; ii++) {
401  ElementLink<xAOD::TrackParticleContainer> TEL; TEL.setElement( xtpListC[ii] );
402  if(!fittrkwgt.empty()) tmpVertex->addTrackAtVertex(TEL,fittrkwgt[ii]);
403  else tmpVertex->addTrackAtVertex(TEL,1.);
404  }
405  }
406 
407  return tmpVertex;
408 }

◆ fit() [9/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const xAOD::TrackParticle * > &  vectorTrk,
const std::vector< const xAOD::NeutralParticle * > &  vectorNeu,
const Amg::Vector3D &  startingPoint 
) const
finaloverridevirtual

Definition at line 410 of file TrkVKalVrtFitter.cxx.

413 {
414  State state;
415  initState (state);
416  xAOD::Vertex * tmpVertex = nullptr;
417  setApproximateVertex(startingPoint.x(),
418  startingPoint.y(),
419  startingPoint.z(),
420  state);
421  Amg::Vector3D Vertex;
422  TLorentzVector Momentum;
423  long int Charge;
424  std::vector<double> ErrorMatrix;
425  std::vector<double> Chi2PerTrk;
426  std::vector< std::vector<double> > TrkAtVrt;
427  double Chi2;
428  StatusCode sc=VKalVrtFit( xtpListC, xtpListN,
429  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
430  if(sc.isSuccess()) {
431  tmpVertex = makeXAODVertex( (int)xtpListN.size(), Vertex, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state );
432  dvect fittrkwgt;
433  sc=VKalGetTrkWeights(fittrkwgt, state); if(sc.isFailure())fittrkwgt.clear();
434  for(int ii=0; ii<state.m_FitStatus; ii++) {
435  if(ii<(int)xtpListC.size()) {
436  ElementLink<xAOD::TrackParticleContainer> TEL; TEL.setElement( xtpListC[ii] );
437  if(!fittrkwgt.empty()) tmpVertex->addTrackAtVertex(TEL,fittrkwgt[ii]);
438  else tmpVertex->addTrackAtVertex(TEL,1.);
439  }else{
440  ElementLink<xAOD::NeutralParticleContainer> TEL; TEL.setElement( xtpListN[ii] );
441  if(!fittrkwgt.empty()) tmpVertex->addNeutralAtVertex(TEL,fittrkwgt[ii]);
442  else tmpVertex->addNeutralAtVertex(TEL,1.);
443  }
444  }
445  }
446 
447  return tmpVertex;
448 }

◆ fit() [10/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const xAOD::TrackParticle * > &  vectorTrk,
const std::vector< const xAOD::NeutralParticle * > &  vectorNeu,
const xAOD::Vertex &  constraint 
) const
finaloverridevirtual

Definition at line 505 of file TrkVKalVrtFitter.cxx.

508 {
509  State state;
510  initState (state);
511 
512  if(msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)<< "A priori vertex constraint is activated in VKalVrt fitter!" << endmsg;
513  xAOD::Vertex * tmpVertex = nullptr;
514  setApproximateVertex(constraint.position().x(), constraint.position().y(),constraint.position().z(),state);
515  setVertexForConstraint(constraint.position().x(),
516  constraint.position().y(),
517  constraint.position().z(),
518  state);
519  setCovVrtForConstraint(constraint.covariancePosition()(Trk::x,Trk::x),
520  constraint.covariancePosition()(Trk::y,Trk::x),
521  constraint.covariancePosition()(Trk::y,Trk::y),
522  constraint.covariancePosition()(Trk::z,Trk::x),
523  constraint.covariancePosition()(Trk::z,Trk::y),
524  constraint.covariancePosition()(Trk::z,Trk::z),
525  state);
526  state.m_useAprioriVertex=true;
527  Amg::Vector3D Vertex;
528  TLorentzVector Momentum;
529  long int Charge;
530  std::vector<double> ErrorMatrix;
531  std::vector<double> Chi2PerTrk;
532  std::vector< std::vector<double> > TrkAtVrt;
533  double Chi2;
534  StatusCode sc=VKalVrtFit( xtpListC, xtpListN,
535  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
536  if(sc.isSuccess()){
537  tmpVertex = makeXAODVertex( (int)xtpListN.size(), Vertex, ErrorMatrix,Chi2PerTrk, TrkAtVrt, Chi2, state );
538  dvect fittrkwgt;
539  sc=VKalGetTrkWeights(fittrkwgt, state); if(sc.isFailure())fittrkwgt.clear();
540  for(int ii=0; ii<state.m_FitStatus; ii++) {
541  if(ii<(int)xtpListC.size()) {
542  ElementLink<xAOD::TrackParticleContainer> TEL; TEL.setElement( xtpListC[ii] );
543  if(!fittrkwgt.empty()) tmpVertex->addTrackAtVertex(TEL,fittrkwgt[ii]);
544  else tmpVertex->addTrackAtVertex(TEL,1.);
545  }else{
546  ElementLink<xAOD::NeutralParticleContainer> TEL; TEL.setElement( xtpListN[ii] );
547  if(!fittrkwgt.empty()) tmpVertex->addNeutralAtVertex(TEL,fittrkwgt[ii]);
548  else tmpVertex->addNeutralAtVertex(TEL,1.);
549  }
550  }
551  }
552 
553  return tmpVertex;
554 }

◆ fit() [11/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const xAOD::TrackParticle * > &  xtpListC,
const xAOD::Vertex &  constraint 
) const
finaloverridevirtual

Interface for xAOD::TrackParticle with vertex constraint.

the position of the constraint is ALWAYS the starting point

Definition at line 452 of file TrkVKalVrtFitter.cxx.

454 {
455  State state;
456  initState (state);
457  return fit (xtpListC, constraint, state);
458 }

◆ fit() [12/12]

xAOD::Vertex * Trk::TrkVKalVrtFitter::fit ( const std::vector< const xAOD::TrackParticle * > &  vectorTrk,
const xAOD::Vertex &  constraint,
IVKalState &  istate 
) const

Definition at line 459 of file TrkVKalVrtFitter.cxx.

462 {
463  assert(dynamic_cast<State*> (&istate)!=nullptr);
464  State& state = static_cast<State&> (istate);
465 
466  if(msgLvl(MSG::DEBUG)) msg(MSG::DEBUG)<< "A priori vertex constraint is activated in VKalVrt fitter!" << endmsg;
467  xAOD::Vertex * tmpVertex = nullptr;
468  setApproximateVertex(constraint.position().x(), constraint.position().y(),constraint.position().z(),state);
469  setVertexForConstraint(constraint.position().x(),
470  constraint.position().y(),
471  constraint.position().z(),
472  state);
473  setCovVrtForConstraint(constraint.covariancePosition()(Trk::x,Trk::x),
474  constraint.covariancePosition()(Trk::y,Trk::x),
475  constraint.covariancePosition()(Trk::y,Trk::y),
476  constraint.covariancePosition()(Trk::z,Trk::x),
477  constraint.covariancePosition()(Trk::z,Trk::y),
478  constraint.covariancePosition()(Trk::z,Trk::z),
479  state);
480  state.m_useAprioriVertex=true;
481  std::vector<const xAOD::NeutralParticle*> xtpListN(0);
482  Amg::Vector3D Vertex;
483  TLorentzVector Momentum;
484  long int Charge;
485  std::vector<double> ErrorMatrix;
486  std::vector<double> Chi2PerTrk;
487  std::vector< std::vector<double> > TrkAtVrt;
488  double Chi2;
489  StatusCode sc=VKalVrtFit( xtpListC, xtpListN,
490  Vertex, Momentum, Charge, ErrorMatrix, Chi2PerTrk, TrkAtVrt, Chi2, state, true );
491  if(sc.isSuccess()) {
492  tmpVertex = makeXAODVertex( 0, Vertex, ErrorMatrix,Chi2PerTrk, TrkAtVrt, Chi2, state );
493  dvect fittrkwgt;
494  sc=VKalGetTrkWeights(fittrkwgt, state); if(sc.isFailure())fittrkwgt.clear();
495  for(int ii=0; ii<state.m_FitStatus; ii++) {
496  ElementLink<xAOD::TrackParticleContainer> TEL; TEL.setElement( xtpListC[ii] );
497  if(!fittrkwgt.empty()) tmpVertex->addTrackAtVertex(TEL,fittrkwgt[ii]);
498  else tmpVertex->addTrackAtVertex(TEL,1.);
499  }
500  }
501 
502  return tmpVertex;
503 }

◆ fitCascade()

VxCascadeInfo * Trk::TrkVKalVrtFitter::fitCascade ( IVKalState &  istate,
const Vertex *  primVertex = 0,
bool  FirstDecayAtPV = false 
) const
finaloverride

Definition at line 274 of file TrkCascadeFitter.cxx.

276 {
277  assert(dynamic_cast<State*> (&istate)!=nullptr);
278  State& state = static_cast<State&> (istate);
279  CascadeState& cstate = *state.m_cascadeState;
280 
281  int iv,it,jt;
282  std::vector< Vect3DF > cVertices;
283  std::vector< std::vector<double> > covVertices;
284  std::vector< std::vector< VectMOM> > fittedParticles;
285  std::vector< std::vector<double> > fittedCovariance;
286  std::vector<double> fitFullCovariance;
287  std::vector<double> particleChi2;
288 //
289  int ntrk=0;
290  StatusCode sc;
291  std::vector<const TrackParameters*> baseInpTrk;
292  if(m_firstMeasuredPoint){ //First measured point strategy
293  std::vector<const xAOD::TrackParticle*>::const_iterator i_ntrk;
294  //for (i_ntrk = cstate.m_partListForCascade.begin(); i_ntrk < cstate.m_partListForCascade.end(); ++i_ntrk) baseInpTrk.push_back(GetFirstPoint(*i_ntrk));
295  unsigned int indexFMP;
296  for (i_ntrk = cstate.m_partListForCascade.begin(); i_ntrk < cstate.m_partListForCascade.end(); ++i_ntrk) {
297  if ((*i_ntrk)->indexOfParameterAtPosition(indexFMP, xAOD::FirstMeasurement)){
298  ATH_MSG_DEBUG("FirstMeasuredPoint on track is discovered. Use it.");
299  baseInpTrk.push_back(new CurvilinearParameters((*i_ntrk)->curvilinearParameters(indexFMP)));
300  }else{
301  ATH_MSG_DEBUG("No FirstMeasuredPoint on track in CascadeFitter. Stop fit");
302  { CLEANCASCADE(); return nullptr; }
303  }
304  }
305  sc=CvtTrackParameters(baseInpTrk,ntrk,state);
306  if(sc.isFailure()){ntrk=0; sc=CvtTrackParticle(cstate.m_partListForCascade,ntrk,state);}
307  }else{
308  sc=CvtTrackParticle(cstate.m_partListForCascade,ntrk,state);
309  }
310  if(sc.isFailure()){ CLEANCASCADE(); return nullptr; }
311 
312  VKalVrtConfigureFitterCore(ntrk, state);
313 
314  std::vector< std::vector<int> > vertexDefinition; // track indices for vertex;
315  std::vector< std::vector<int> > cascadeDefinition; // cascade structure
316  makeSimpleCascade(vertexDefinition, cascadeDefinition, cstate);
317 
318  double * partMass=new double[ntrk];
319  for(int i=0; i<ntrk; i++) partMass[i] = cstate.m_partMassForCascade[i];
320  int IERR = makeCascade(state.m_vkalFitControl, ntrk, state.m_ich, partMass, &state.m_apar[0][0], &state.m_awgt[0][0],
321  vertexDefinition,
322  cascadeDefinition,
323  m_cascadeCnstPrecision); delete[] partMass; if(IERR){ CLEANCASCADE(); return nullptr;}
324  CascadeEvent & refCascadeEvent=*(state.m_vkalFitControl.getCascadeEvent());
325 //
326 // Then set vertex mass constraints
327 //
328  std::vector<int> indexT,indexV,indexTT,indexVV,tmpInd; // track indices for vertex;
329  for (const PartialMassConstraint& c : cstate.m_partMassCnstForCascade) {
330  //int index=c.VRT; // vertex position in simple structure
331  int index=getSimpleVIndex(c.VRT, cstate); // vertex position in simple structure
332  IERR = findPositions(c.trkInVrt, vertexDefinition[index], indexT); if(IERR)break;
333  tmpInd.clear();
334  for (int idx : c.pseudoInVrt)
335  tmpInd.push_back( getSimpleVIndex(idx, cstate) );
336  IERR = findPositions( tmpInd, cascadeDefinition[index], indexV); if(IERR)break;
337  //IERR = findPositions(c.pseudoInVrt, cascadeDefinition[index], indexV); if(IERR)break; //VK 31.10.2011 ERROR!!!
338  IERR = setCascadeMassConstraint(refCascadeEvent,index, indexT, indexV, c.Mass);
339  if(IERR)break;
340  }
341  if(IERR){ CLEANCASCADE(); return nullptr;}
342  if(msgLvl(MSG::DEBUG)){
343  msg(MSG::DEBUG)<<"Standard cascade fit" << endmsg;
344  printSimpleCascade(vertexDefinition,cascadeDefinition, cstate);
345  }
346 //
347 // At last fit of cascade
348 // primVrt == 0 - no primary vertex
349 // primVrt is <Vertex*> - exact pointing to primary vertex
350 // primVrt is <RecVertex*> - summary track pass near primary vertex
351 //
352  if(primVrt){
353  double vertex[3] = {primVrt->position().x()-state.m_refFrameX, primVrt->position().y()-state.m_refFrameY,primVrt->position().z()-state.m_refFrameZ};
354  const RecVertex* primVrtRec=dynamic_cast< const RecVertex* > (primVrt);
355  if(primVrtRec){
356  double covari[6] = {primVrtRec->covariancePosition()(0,0),primVrtRec->covariancePosition()(0,1),
357  primVrtRec->covariancePosition()(1,1),primVrtRec->covariancePosition()(0,2),
358  primVrtRec->covariancePosition()(1,2),primVrtRec->covariancePosition()(2,2)};
359  if(FirstDecayAtPV) { IERR = processCascadePV(refCascadeEvent,vertex,covari);}
360  else { IERR = processCascade(refCascadeEvent,vertex,covari);}
361  }else{
362  IERR = processCascade(refCascadeEvent,vertex);
363  }
364  }else{
365  IERR = processCascade(refCascadeEvent);
366  }
367  if(IERR){ CLEANCASCADE(); return nullptr;}
368  getFittedCascade(refCascadeEvent, cVertices, covVertices, fittedParticles, fittedCovariance, particleChi2, fitFullCovariance );
369 
370 // for(int iv=0; iv<(int)cVertices.size(); iv++){ std::cout<<"iv="<<iv<<" masses=";
371 // for(int it=0; it<(int)fittedParticles[iv].size(); it++){
372 // double m=sqrt( fittedParticles[iv][it].E *fittedParticles[iv][it].E
373 // -fittedParticles[iv][it].Pz*fittedParticles[iv][it].Pz
374 // -fittedParticles[iv][it].Py*fittedParticles[iv][it].Py
375 // -fittedParticles[iv][it].Px*fittedParticles[iv][it].Px);
376 // std::cout<<m<<", "; } std::cout<<'\n'; }
377 //-----------------------------------------------------------------------------
378 //
379 // Check cascade correctness
380 //
381  int ip,ivFrom=0,ivTo;
382  double px,py,pz,Sign=10.;
383  for( ivTo=0; ivTo<(int)vertexDefinition.size(); ivTo++){ //Vertex to check
384  if(cascadeDefinition[ivTo].empty()) continue; //no pointing to it
385  for( ip=0; ip<(int)cascadeDefinition[ivTo].size(); ip++){
386  ivFrom=cascadeDefinition[ivTo][ip]; //pointing vertex
387  px=py=pz=0;
388  for(it=0; it<(int)fittedParticles[ivFrom].size(); it++){
389  px += fittedParticles[ivFrom][it].Px;
390  py += fittedParticles[ivFrom][it].Py;
391  pz += fittedParticles[ivFrom][it].Pz;
392  }
393  Sign= (cVertices[ivFrom].X-cVertices[ivTo].X)*px
394  +(cVertices[ivFrom].Y-cVertices[ivTo].Y)*py
395  +(cVertices[ivFrom].Z-cVertices[ivTo].Z)*pz;
396  if(Sign<0) break;
397  }
398  if(Sign<0) break;
399  }
400 //
401 //--------------- Wrong vertices in cascade precedence. Squeeze cascade and refit-----------
402 //
403  int NDOFsqueezed=0;
404  if(Sign<0.){
405  int index,itmp;
406  std::vector< std::vector<int> > new_vertexDefinition; // track indices for vertex;
407  std::vector< std::vector<int> > new_cascadeDefinition; // cascade structure
408  cstate.m_cascadeVList[ivFrom].mergedTO=cstate.m_cascadeVList[ivTo].vID;
409  cstate.m_cascadeVList[ivTo].mergedIN.push_back(ivFrom);
410  makeSimpleCascade(new_vertexDefinition, new_cascadeDefinition, cstate);
411  if(msgLvl(MSG::DEBUG)){
412  msg(MSG::DEBUG)<<"Compressed cascade fit" << endmsg;
413  printSimpleCascade(new_vertexDefinition,new_cascadeDefinition, cstate);
414  }
415 //-----------------------------------------------------------------------------------------
416  state.m_vkalFitControl.renewCascadeEvent(new CascadeEvent());
417  partMass=new double[ntrk];
418  for(int i=0; i<ntrk; i++) partMass[i] = cstate.m_partMassForCascade[i];
419  int IERR = makeCascade(state.m_vkalFitControl, ntrk, state.m_ich, partMass, &state.m_apar[0][0], &state.m_awgt[0][0],
420  new_vertexDefinition,
421  new_cascadeDefinition); delete[] partMass; if(IERR){ CLEANCASCADE(); return nullptr;}
422 //------Set up mass constraints
423  for (const PartialMassConstraint& c : cstate.m_partMassCnstForCascade) {
424  indexT.clear(); indexV.clear();
425  index=getSimpleVIndex( c.VRT, cstate);
426  IERR = findPositions(c.trkInVrt, new_vertexDefinition[index], indexT); if(IERR)break;
427  for (VertexID inV : c.pseudoInVrt) { //cycle over pseudotracks
428  int icv=indexInV(inV, cstate); if(icv<0) break;
429  if(cstate.m_cascadeVList[icv].mergedTO == c.VRT){
430  IERR = findPositions(cstate.m_cascadeVList[icv].trkInVrt, new_vertexDefinition[index], indexTT);
431  if(IERR)break;
432  indexT.insert (indexT.end(), indexTT.begin(), indexTT.end());
433  }else{
434  std::vector<int> tmpI(1); tmpI[0]=inV;
435  IERR = findPositions(tmpI, new_cascadeDefinition[index], indexVV);
436  if(IERR)break;
437  indexV.insert (indexV.end(), indexVV.begin(), indexVV.end());
438  }
439  } if(IERR)break;
440  //std::cout<<"trk2="; for(int I=0; I<(int)indexT.size(); I++)std::cout<<indexT[I]; std::cout<<'\n';
441  //std::cout<<"pse="; for(int I=0; I<(int)indexV.size(); I++)std::cout<<indexV[I]; std::cout<<'\n';
442  IERR = setCascadeMassConstraint(*(state.m_vkalFitControl.getCascadeEvent()), index , indexT, indexV, c.Mass); if(IERR)break;
443  }
444  ATH_MSG_DEBUG("Setting compressed mass constraints ierr="<<IERR);
445  if(IERR){ CLEANCASCADE(); return nullptr;}
446 //
447 //--------------------------- Refit
448 //
449  if(primVrt){
450  double vertex[3] = {primVrt->position().x()-state.m_refFrameX, primVrt->position().y()-state.m_refFrameY,primVrt->position().z()-state.m_refFrameZ};
451  const RecVertex* primVrtRec=dynamic_cast< const RecVertex* > (primVrt);
452  if(primVrtRec){
453  double covari[6] = {primVrtRec->covariancePosition()(0,0),primVrtRec->covariancePosition()(0,1),
454  primVrtRec->covariancePosition()(1,1),primVrtRec->covariancePosition()(0,2),
455  primVrtRec->covariancePosition()(1,2),primVrtRec->covariancePosition()(2,2)};
456  IERR = processCascade(*(state.m_vkalFitControl.getCascadeEvent()),vertex,covari);
457  }else{
458  IERR = processCascade(*(state.m_vkalFitControl.getCascadeEvent()),vertex);
459  }
460  }else{
461  IERR = processCascade(*(state.m_vkalFitControl.getCascadeEvent()));
462  }
463  if(IERR){ CLEANCASCADE(); return nullptr;}
464  NDOFsqueezed=getCascadeNDoF(cstate)+3-2; // Remove vertex (+3 ndf) and this vertex pointing (-2 ndf)
465 //
466 //-------------------- Get information according to old cascade structure
467 //
468  std::vector< Vect3DF > t_cVertices;
469  std::vector< std::vector<double> > t_covVertices;
470  std::vector< std::vector< VectMOM> > t_fittedParticles;
471  std::vector< std::vector<double> > t_fittedCovariance;
472  std::vector<double> t_fitFullCovariance;
473  getFittedCascade(*(state.m_vkalFitControl.getCascadeEvent()), t_cVertices, t_covVertices, t_fittedParticles,
474  t_fittedCovariance, particleChi2, t_fitFullCovariance);
475  cVertices.clear(); covVertices.clear();
476 //
477 //------------------------- Real tracks
478 //
479  if(msgLvl(MSG::DEBUG)){
480  msg(MSG::DEBUG)<<"Initial cascade momenta"<<endmsg;
481  for(int kv=0; kv<(int)fittedParticles.size(); kv++){
482  for(int kt=0; kt<(int)fittedParticles[kv].size(); kt++)
483  std::cout<<
484  " Px="<<fittedParticles[kv][kt].Px<<" Py="<<fittedParticles[kv][kt].Py<<";";
485  std::cout<<'\n';
486  }
487  msg(MSG::DEBUG)<<"Squized cascade momenta"<<endmsg;
488  for(int kv=0; kv<(int)t_fittedParticles.size(); kv++){
489  for(int kt=0; kt<(int)t_fittedParticles[kv].size(); kt++)
490  std::cout<<
491  " Px="<<t_fittedParticles[kv][kt].Px<<" Py="<<t_fittedParticles[kv][kt].Py<<";";
492  std::cout<<'\n';
493  }
494  }
495  for(iv=0; iv<(int)cstate.m_cascadeVList.size(); iv++){
496  index=getSimpleVIndex( cstate.m_cascadeVList[iv].vID, cstate ); //index of vertex in simplified structure
497  cVertices.push_back(t_cVertices[index]);
498  covVertices.push_back(t_covVertices[index]);
499  for(it=0; it<(int)cstate.m_cascadeVList[iv].trkInVrt.size(); it++){
500  int numTrk=cstate.m_cascadeVList[iv].trkInVrt[it]; //track itself
501  for(itmp=0; itmp<(int)new_vertexDefinition[index].size(); itmp++) if(numTrk==new_vertexDefinition[index][itmp])break;
502  fittedParticles[iv][it]=t_fittedParticles[index][itmp];
503 //Update only particle covariance. Cross particle covariance remains old.
504  fittedCovariance[iv][SymIndex(it,0,0)]=t_fittedCovariance[index][SymIndex(itmp,0,0)];
505  fittedCovariance[iv][SymIndex(it,1,0)]=t_fittedCovariance[index][SymIndex(itmp,1,0)];
506  fittedCovariance[iv][SymIndex(it,1,1)]=t_fittedCovariance[index][SymIndex(itmp,1,1)];
507  fittedCovariance[iv][SymIndex(it,2,0)]=t_fittedCovariance[index][SymIndex(itmp,2,0)];
508  fittedCovariance[iv][SymIndex(it,2,1)]=t_fittedCovariance[index][SymIndex(itmp,2,1)];
509  fittedCovariance[iv][SymIndex(it,2,2)]=t_fittedCovariance[index][SymIndex(itmp,2,2)];
510  }
511  fittedCovariance[iv][SymIndex(0,0,0)]=t_fittedCovariance[index][SymIndex(0,0,0)]; // Update also vertex
512  fittedCovariance[iv][SymIndex(0,1,0)]=t_fittedCovariance[index][SymIndex(0,1,0)]; // covarinace
513  fittedCovariance[iv][SymIndex(0,1,1)]=t_fittedCovariance[index][SymIndex(0,1,1)];
514  fittedCovariance[iv][SymIndex(0,2,0)]=t_fittedCovariance[index][SymIndex(0,2,0)];
515  fittedCovariance[iv][SymIndex(0,2,1)]=t_fittedCovariance[index][SymIndex(0,2,1)];
516  fittedCovariance[iv][SymIndex(0,2,2)]=t_fittedCovariance[index][SymIndex(0,2,2)];
517  }
518 // Pseudo-tracks. They are filled based on fitted results for nonmerged vertices
519 // or as sum for merged vertices
520  VectMOM tmpMom{};
521  for(iv=0; iv<(int)cstate.m_cascadeVList.size(); iv++){
522  index=getSimpleVIndex( cstate.m_cascadeVList[iv].vID, cstate ); //index of current vertex in simplified structure
523  int NTrkInVrt=cstate.m_cascadeVList[iv].trkInVrt.size();
524  for(ip=0; ip<(int)cstate.m_cascadeVList[iv].inPointingV.size(); ip++){ //inPointing verties
525  int tmpIndexV=indexInV( cstate.m_cascadeVList[iv].inPointingV[ip], cstate); //index of inPointing vertex in full structure
526  if(cstate.m_cascadeVList[tmpIndexV].mergedTO){ //vertex is merged, so take pseudo-track as a sum
527  tmpMom.Px=tmpMom.Py=tmpMom.Pz=tmpMom.E=0.;
528  for(it=0; it<(int)(cstate.m_cascadeVList[tmpIndexV].trkInVrt.size()+
529  cstate.m_cascadeVList[tmpIndexV].inPointingV.size()); it++){
530  tmpMom.Px += fittedParticles[tmpIndexV][it].Px; tmpMom.Py += fittedParticles[tmpIndexV][it].Py;
531  tmpMom.Pz += fittedParticles[tmpIndexV][it].Pz; tmpMom.E += fittedParticles[tmpIndexV][it].E;
532  }
533  fittedParticles[iv][ip+NTrkInVrt]=tmpMom;
534  }else{
535  int indexS=getSimpleVIndex( cstate.m_cascadeVList[iv].inPointingV[ip], cstate ); //index of inPointing vertex in simplified structure
536  for(itmp=0; itmp<(int)new_cascadeDefinition[index].size(); itmp++) if(indexS==new_cascadeDefinition[index][itmp])break;
537  fittedParticles[iv][ip+NTrkInVrt]=t_fittedParticles[index][itmp+new_vertexDefinition[index].size()];
538  }
539  }
540  }
541  if(msgLvl(MSG::DEBUG)){
542  msg(MSG::DEBUG)<<"Refit cascade momenta"<<endmsg;
543  for(int kv=0; kv<(int)fittedParticles.size(); kv++){
544  for(int kt=0; kt<(int)fittedParticles[kv].size(); kt++)
545  std::cout<<
546  " Px="<<fittedParticles[kv][kt].Px<<" Py="<<fittedParticles[kv][kt].Py<<";";
547  std::cout<<'\n';
548  }
549  }
550 // Covariance matrix for nonmerged vertices is updated.
551 // For merged vertices (both IN and TO ) it's taken from old fit
552 
553  for(iv=0; iv<(int)cstate.m_cascadeVList.size(); iv++){
554  bool isMerged=false;
555  if(cstate.m_cascadeVList[iv].mergedTO)isMerged=true; //vertex is merged
556  index=getSimpleVIndex( cstate.m_cascadeVList[iv].vID, cstate ); //index of current vertex in simplified structure
557  for(ip=0; ip<(int)cstate.m_cascadeVList[iv].inPointingV.size(); ip++){ //inPointing verties
558  int tmpIndexV=indexInV( cstate.m_cascadeVList[iv].inPointingV[ip], cstate); //index of inPointing vertex in full structure
559  if(cstate.m_cascadeVList[tmpIndexV].mergedTO)isMerged=true; //vertex is merged
560  }
561  if(!isMerged){
562  fittedCovariance[iv]=t_fittedCovariance[index]; //copy complete covarinace matrix for nonmerged vertices
563  }
564  }
565  }
566 //
567 //-------------------------------------Saving
568 //
569  ATH_MSG_DEBUG("Now save results");
570  Amg::MatrixX VrtCovMtx(3,3);
571  Trk::Perigee * measPerigee;
572  std::vector<xAOD::Vertex*> xaodVrtList(0);
573  double phi, theta, invP, mom, fullChi2=0.;
574 
575  int NDOF=getCascadeNDoF(cstate); if(NDOFsqueezed) NDOF=NDOFsqueezed;
576  if(primVrt){ if(FirstDecayAtPV){ NDOF+=3; }else{ NDOF+=2; } }
577 
578  for(iv=0; iv<(int)cVertices.size(); iv++){
579  Amg::Vector3D FitVertex(cVertices[iv].X+state.m_refFrameX,cVertices[iv].Y+state.m_refFrameY,cVertices[iv].Z+state.m_refFrameZ);
580  VrtCovMtx(0,0) = covVertices[iv][0]; VrtCovMtx(0,1) = covVertices[iv][1];
581  VrtCovMtx(1,1) = covVertices[iv][2]; VrtCovMtx(0,2) = covVertices[iv][3];
582  VrtCovMtx(1,2) = covVertices[iv][4]; VrtCovMtx(2,2) = covVertices[iv][5];
583  VrtCovMtx(1,0) = VrtCovMtx(0,1);
584  VrtCovMtx(2,0) = VrtCovMtx(0,2);
585  VrtCovMtx(2,1) = VrtCovMtx(1,2);
586  double Chi2=0;
587  for(it=0; it<(int)vertexDefinition[iv].size(); it++) { Chi2 += particleChi2[vertexDefinition[iv][it]];};
588  fullChi2+=Chi2;
589 
590 //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=--=-=-=-=-=-=-=-=-=-= xAOD::Vertex creation
591  xAOD::Vertex * tmpXAODVertex=new xAOD::Vertex();
592  tmpXAODVertex->makePrivateStore();
593  tmpXAODVertex->setPosition(FitVertex);
594  tmpXAODVertex->setFitQuality(Chi2, (float)NDOF);
595  std::vector<VxTrackAtVertex> & tmpVTAV=tmpXAODVertex->vxTrackAtVertex();
596  tmpVTAV.clear();
597 
598  int NRealT=vertexDefinition[iv].size();
599  Amg::MatrixX genCOV( NRealT*3+3, NRealT*3+3 ); // Fill cov. matrix for vertex
600  for( it=0; it<NRealT*3+3; it++){ // (X,Y,Z,px1,py1,....pxn,pyn,pzn)
601  for( jt=0; jt<=it; jt++){ //
602  genCOV(it,jt) = genCOV(jt,it) = fittedCovariance[iv][it*(it+1)/2+jt]; // for real tracks only
603  } } // (first in the list)
604  Amg::MatrixX fullDeriv;
605  if( m_makeExtendedVertex ){
606  //VK fullDeriv=new CLHEP::HepMatrix( NRealT*3+3, NRealT*3+3, 0); // matrix is filled by zeros
607  fullDeriv=Amg::MatrixX::Zero(NRealT*3+3, NRealT*3+3); // matrix is filled by zeros
608  fullDeriv(0,0)=fullDeriv(1,1)=fullDeriv(2,2)=1.;
609  }
610  for( it=0; it<NRealT; it++) {
611  mom= sqrt( fittedParticles[iv][it].Pz*fittedParticles[iv][it].Pz
612  +fittedParticles[iv][it].Py*fittedParticles[iv][it].Py
613  +fittedParticles[iv][it].Px*fittedParticles[iv][it].Px);
614  double Px=fittedParticles[iv][it].Px;
615  double Py=fittedParticles[iv][it].Py;
616  double Pz=fittedParticles[iv][it].Pz;
617  double Pt= sqrt(Px*Px + Py*Py) ;
618  phi=atan2( Py, Px);
619  theta=acos( Pz/mom );
620  invP = - state.m_ich[vertexDefinition[iv][it]] / mom; // Change charge sign according to ATLAS
621 // d(Phi,Theta,InvP)/d(Px,Py,Pz) - Perigee vs summary momentum
622  Amg::MatrixX tmpDeriv( 5, NRealT*3+3);
623  tmpDeriv.setZero(); // matrix is filled by zeros
624  tmpDeriv(0,1) = -sin(phi); // Space derivatives
625  tmpDeriv(0,2) = cos(phi);
626  tmpDeriv(1,1) = -cos(phi)/tan(theta);
627  tmpDeriv(1,2) = -sin(phi)/tan(theta);
628  tmpDeriv(1,3) = 1.;
629  tmpDeriv(2+0,3*it+3+0) = -Py/Pt/Pt; //dPhi/dPx
630  tmpDeriv(2+0,3*it+3+1) = Px/Pt/Pt; //dPhi/dPy
631  tmpDeriv(2+0,3*it+3+2) = 0; //dPhi/dPz
632  tmpDeriv(2+1,3*it+3+0) = Px*Pz/(Pt*mom*mom); //dTheta/dPx
633  tmpDeriv(2+1,3*it+3+1) = Py*Pz/(Pt*mom*mom); //dTheta/dPy
634  tmpDeriv(2+1,3*it+3+2) = -Pt/(mom*mom); //dTheta/dPz
635  tmpDeriv(2+2,3*it+3+0) = -Px/(mom*mom) * invP; //dInvP/dPx
636  tmpDeriv(2+2,3*it+3+1) = -Py/(mom*mom) * invP; //dInvP/dPy
637  tmpDeriv(2+2,3*it+3+2) = -Pz/(mom*mom) * invP; //dInvP/dPz
638 //---------- Here for Eigen block(startrow,startcol,sizerow,sizecol)
639  if( m_makeExtendedVertex )fullDeriv.block(3*it+3+0,3*it+3+0,3,3) = tmpDeriv.block(2,3*it+3+0,3,3);
640 //----------
641  AmgSymMatrix(5) tmpCovMtx ; // New Eigen based EDM
642  tmpCovMtx = genCOV.similarity(tmpDeriv); // New Eigen based EDM
643  measPerigee = new Perigee( 0.,0., phi, theta, invP, PerigeeSurface(FitVertex), std::move(tmpCovMtx) ); // New Eigen based EDM
644  tmpVTAV.emplace_back( particleChi2[vertexDefinition[iv][it]] , measPerigee ) ;
645  }
646  std::vector<float> floatErrMtx;
647  if( m_makeExtendedVertex ) {
648  Amg::MatrixX tmpCovMtx(NRealT*3+3,NRealT*3+3); // New Eigen based EDM
649  tmpCovMtx=genCOV.similarity(fullDeriv);
650  floatErrMtx.resize((NRealT*3+3)*(NRealT*3+3+1)/2);
651  int ivk=0;
652  for(int i=0;i<NRealT*3+3;i++){
653  for(int j=0;j<=i;j++){
654  floatErrMtx.at(ivk++)=tmpCovMtx(i,j);
655  }
656  }
657  }else{
658  floatErrMtx.resize(6);
659  for(int i=0; i<6; i++) floatErrMtx[i]=covVertices[iv][i];
660  }
661  tmpXAODVertex->setCovariance(floatErrMtx);
662  for(int itvk=0; itvk<NRealT; itvk++) {
663  ElementLink<xAOD::TrackParticleContainer> TEL;
664  if(itvk < (int)cstate.m_cascadeVList[iv].trkInVrt.size()){
665  TEL.setElement( cstate.m_partListForCascade[ cstate.m_cascadeVList[iv].trkInVrt[itvk] ] );
666  }else{
667  TEL.setElement( nullptr );
668  }
669  tmpXAODVertex->addTrackAtVertex(TEL,1.);
670  }
671  xaodVrtList.push_back(tmpXAODVertex); //VK Save xAOD::Vertex
672 //
673  }
674 //
675 // Save momenta of all particles including combined at vertex positions
676 //
677  std::vector<TLorentzVector> tmpMoms;
678  std::vector<std::vector<TLorentzVector> > particleMoms;
679  std::vector<Amg::MatrixX> particleCovs;
680  int allFitPrt=0;
681  for(iv=0; iv<(int)cVertices.size(); iv++){
682  tmpMoms.clear();
683  int NTrkF=fittedParticles[iv].size();
684  for(it=0; it< NTrkF; it++) {
685  tmpMoms.emplace_back( fittedParticles[iv][it].Px, fittedParticles[iv][it].Py,
686  fittedParticles[iv][it].Pz, fittedParticles[iv][it].E );
687  }
688  //CLHEP::HepSymMatrix COV( NTrkF*3+3, 0 );
689  Amg::MatrixX COV(NTrkF*3+3,NTrkF*3+3); COV=Amg::MatrixX::Zero(NTrkF*3+3,NTrkF*3+3);
690  for( it=0; it<NTrkF*3+3; it++){
691  for( jt=0; jt<=it; jt++){
692  COV(it,jt) = COV(jt,it) = fittedCovariance[iv][it*(it+1)/2+jt];
693  } }
694  particleMoms.push_back( std::move(tmpMoms) );
695  particleCovs.push_back( std::move(COV) );
696  allFitPrt += NTrkF;
697  }
698 //
699  int NAPAR=(allFitPrt+cVertices.size())*3; //Full size of complete covariance matrix
700  //CLHEP::HepSymMatrix FULL( NAPAR, 0 );
701  Amg::MatrixX FULL(NAPAR,NAPAR); FULL.setZero();
702  if( !NDOFsqueezed ){ //normal cascade
703  for( it=0; it<NAPAR; it++){
704  for( jt=0; jt<=it; jt++){
705  FULL(it,jt) = FULL(jt,it) = fitFullCovariance[it*(it+1)/2+jt];
706  } }
707  }else{ //squeezed cascade
708  //int mcount=1; //Indexing in SUB starts from 1 !!!!
709  int mcount=0; //Indexing in BLOCK starts from 0 !!!!
710  for(iv=0; iv<(int)cstate.m_cascadeVList.size(); iv++){
711  //FULL.sub(mcount,particleCovs[iv]); mcount += particleCovs[iv].num_col();
712  FULL.block(mcount,mcount,particleCovs[iv].rows(),particleCovs[iv].cols())=particleCovs[iv];
713  mcount += particleCovs[iv].rows();
714  }
715  }
716 //
717 //
718 // VxCascadeInfo * recCascade= new VxCascadeInfo(vxVrtList,particleMoms,particleCovs, NDOF ,fullChi2);
719  VxCascadeInfo * recCascade= new VxCascadeInfo(std::move(xaodVrtList),std::move(particleMoms),std::move(particleCovs), NDOF ,fullChi2);
720  recCascade->setFullCascadeCovariance(FULL);
721  CLEANCASCADE();
722  return recCascade;
723 }

◆ getCascadeNDoF()

int Trk::TrkVKalVrtFitter::getCascadeNDoF ( const CascadeState &  cstate)
staticprivate

Definition at line 79 of file TrkCascadeFitter.cxx.

80 {
81 
82 // get Tracks, Vertices and Pointings in cascade
83 //
84  int nTrack = cstate.m_partListForCascade.size();
85  int nVertex = cstate.m_cascadeVList.size();
86 
87  int nPointing = 0;
88  for( int iv=0; iv<nVertex; iv++) nPointing += cstate.m_cascadeVList[iv].inPointingV.size();
89 
90  int nMassCnst = cstate.m_partMassCnstForCascade.size(); // mass cnsts
91 
92  return 2*nTrack - 3*nVertex + 2*nPointing + nMassCnst;
93 }

◆ GetPerigee()

const Perigee * Trk::TrkVKalVrtFitter::GetPerigee ( const TrackParameters *  i_ntrk)
staticprivate

Definition at line 226 of file CvtTrackParticle.cxx.

227  {
228  const Perigee* mPer = nullptr;
229  if(i_ntrk->surfaceType()==Trk::SurfaceType::Perigee && i_ntrk->covariance()!= nullptr ) {
230  mPer = dynamic_cast<const Perigee*> (i_ntrk);
231  }
232  return mPer;
233  }

◆ getSimpleVIndex()

int Trk::TrkVKalVrtFitter::getSimpleVIndex ( const VertexID &  vrt,
const CascadeState &  cstate 
)
staticprivate

Definition at line 804 of file TrkCascadeFitter.cxx.

806 {
807  int NVRT=cstate.m_cascadeVList.size();
808 
809  int iv=indexInV(vrt, cstate);
810  if(iv<0) return -1; //not found
811 
812  int ivv=0;
813  if(cstate.m_cascadeVList[iv].mergedTO){
814  for(ivv=0; ivv<NVRT; ivv++) if(cstate.m_cascadeVList[iv].mergedTO == cstate.m_cascadeVList[ivv].vID) break;
815  if(iv==NVRT) return -1; //not found
816  iv=ivv;
817  }
818  return cstate.m_cascadeVList[iv].indexInSimpleCascade;
819 }

◆ GiveFullMatrix()

Amg::MatrixX * Trk::TrkVKalVrtFitter::GiveFullMatrix ( int  NTrk,
std::vector< double > &  Matrix 
)
staticprivate

Definition at line 649 of file TrkVKalVrtFitter.cxx.

650 {
651  Amg::MatrixX * mtx = new Amg::MatrixX(3+3*NTrk,3+3*NTrk);
652  long int ij=0;
653  for(int i=1; i<=(3+3*NTrk); i++){
654  for(int j=1; j<=i; j++){
655  if(i==j){ (*mtx)(i-1,j-1)=Matrix[ij];}
656  else { (*mtx).fillSymmetric(i-1,j-1,Matrix[ij]);}
657  ij++;
658  }
659  }
660  return mtx;
661 }

◆ indexInV()

int Trk::TrkVKalVrtFitter::indexInV ( const VertexID &  vrt,
const CascadeState &  cstate 
)
staticprivate

Definition at line 823 of file TrkCascadeFitter.cxx.

825 { int icv; int NVRT=cstate.m_cascadeVList.size();
826  for(icv=0; icv<NVRT; icv++) if(vrt==cstate.m_cascadeVList[icv].vID)break;
827  if(icv==NVRT)return -1;
828  return icv;
829 }

◆ initCnstList()

void Trk::TrkVKalVrtFitter::initCnstList ( )
private

◆ initialize()

StatusCode Trk::TrkVKalVrtFitter::initialize ( )
finaloverridevirtual

Definition at line 72 of file TrkVKalVrtFitter.cxx.

73 {
74 
75 // Checking ROBUST algoritms
76  if(m_Robustness<0 || m_Robustness>7 ) m_Robustness=0;
77 
78 
79  if(!m_useFixedField){
80  // Read handle for AtlasFieldCacheCondObj
81  if (!m_fieldCacheCondObjInputKey.key().empty()){
82  if( (m_fieldCacheCondObjInputKey.initialize()).isSuccess() ){
83  m_isAtlasField = true;
84  ATH_MSG_DEBUG( "Found AtlasFieldCacheCondObj with key ="<< m_fieldCacheCondObjInputKey.key());
85  }else{
86  ATH_MSG_INFO( "No AtlasFieldCacheCondObj with key ="<< m_fieldCacheCondObjInputKey.key());
87  ATH_MSG_INFO( "Use fixed magnetic field instead");
88  }
89  }
90  }
91 //
92 // Only here the VKalVrtFitter propagator object is created if ATHENA propagator is provided (see setAthenaPropagator)
93 // In this case the ATHENA propagator can be used via pointers:
94 // m_InDetExtrapolator - direct access
95 // m_fitPropagator - via VKalVrtFitter object VKalExtPropagator
96 // If ATHENA propagator is not provided, only defined object is
97 // myPropagator - extern propagator from TrkVKalVrtCore
98 //
99  if (m_extPropagator.empty()){
100  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<< "External propagator is not supplied - use internal one"<<endmsg;
101  m_extPropagator.disable();
102  }else{
103  if (m_extPropagator.retrieve().isFailure()) {
104  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<< "Could not find external propagator=" <<m_extPropagator<<endmsg;
105  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<< "TrkVKalVrtFitter will uses internal propagator" << endmsg;
106  m_extPropagator.disable();
107  }else{
108  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<< "External propagator="<<m_extPropagator<<" retrieved" << endmsg;
109  setAthenaPropagator(m_extPropagator.get());
110  }
111  }
112 
113 //
114 //
115 //
116  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG)<< "TrkVKalVrtFitter initialize() successful" << endmsg;
117  if(msgLvl(MSG::DEBUG)){
118  msg(MSG::DEBUG)<< "TrkVKalVrtFitter configuration:" << endmsg;
119  msg(MSG::DEBUG)<< " Frozen version for BTagging: "<< m_frozenVersionForBTagging <<endmsg;
120  msg(MSG::DEBUG)<< " Allow ultra displaced vertices: "<< m_allowUltraDisplaced <<endmsg;
121  msg(MSG::DEBUG)<< " A priori vertex constraint: "<< m_useAprioriVertex <<endmsg;
122  msg(MSG::DEBUG)<< " Angle dTheta=0 constraint: "<< m_useThetaCnst <<endmsg;
123  msg(MSG::DEBUG)<< " Angle dPhi=0 constraint: "<< m_usePhiCnst <<endmsg;
124  msg(MSG::DEBUG)<< " Pointing to other vertex constraint: "<< m_usePointingCnst <<endmsg;
125  msg(MSG::DEBUG)<< " ZPointing to other vertex constraint: "<< m_useZPointingCnst <<endmsg;
126  msg(MSG::DEBUG)<< " Comb. particle pass near other vertex:"<< m_usePassNear <<endmsg;
127  msg(MSG::DEBUG)<< " Pass near with comb.particle errors: "<< m_usePassWithTrkErr <<endmsg;
128  if(m_massForConstraint>0){
129  msg(MSG::DEBUG)<< " Mass constraint M="<< m_massForConstraint <<endmsg;
130  msg(MSG::DEBUG)<< " with particles M=";
131  for(int i=0; i<(int)m_c_MassInputParticles.size(); i++) msg(MSG::DEBUG)<<m_c_MassInputParticles[i]<<", ";
132  msg(MSG::DEBUG)<<endmsg; ;
133  }
134  if(m_IterationNumber==0){
135  msg(MSG::DEBUG)<< " Default iteration number limit 50 is used " <<endmsg;
136  } else {
137  msg(MSG::DEBUG)<< " Iteration number limit: "<< m_IterationNumber <<endmsg;
138  }
139 
140  if(m_isAtlasField){ msg(MSG::DEBUG)<< " ATLAS magnetic field is used!"<<endmsg; }
141  else { msg(MSG::DEBUG)<< " Constant magnetic field is used! B="<<m_BMAG<<endmsg; }
142 
143  if(m_InDetExtrapolator){ msg(MSG::DEBUG)<< " InDet extrapolator is used!"<<endmsg; }
144  else { msg(MSG::DEBUG)<< " Internal VKalVrt extrapolator is used!"<<endmsg;}
145 
146  if(m_Robustness) { msg(MSG::DEBUG)<< " VKalVrt uses robust algorithm! Type="<<m_Robustness<<" with Scale="<<m_RobustScale<<endmsg; }
147 
148  if(m_firstMeasuredPoint){ msg(MSG::DEBUG)<< " VKalVrt will use FirstMeasuredPoint strategy in fits with InDetExtrapolator"<<endmsg; }
149  else { msg(MSG::DEBUG)<< " VKalVrt will use Perigee strategy in fits with InDetExtrapolator"<<endmsg; }
150  if(m_firstMeasuredPointLimit){ msg(MSG::DEBUG)<< " VKalVrt will use FirstMeasuredPointLimit strategy "<<endmsg; }
151  }
152 
153 
154  return StatusCode::SUCCESS;
155 }

◆ initState() [1/2]

void Trk::TrkVKalVrtFitter::initState ( const EventContext &  ctx,
State &  state 
) const
private

Definition at line 164 of file TrkVKalVrtFitter.cxx.

166 {
167  //----------------------------------------------------------------------
168  // New magnetic field object is created. It's provided to VKalVrtCore.
169  // VKalVrtFitter must set up Core BEFORE any call required propagation!!!
170  if (m_isAtlasField) {
171  // For the moment, use Gaudi Hive for the event context - would need to be passed in from clients
172  SG::ReadCondHandle<AtlasFieldCacheCondObj> readHandle{m_fieldCacheCondObjInputKey, ctx};
173  const AtlasFieldCacheCondObj* fieldCondObj{*readHandle};
174  if (fieldCondObj == nullptr) {
175  ATH_MSG_ERROR("Failed to retrieve AtlasFieldCacheCondObj with key " << m_fieldCacheCondObjInputKey.key());
176  return;
177  }
178  fieldCondObj->getInitializedCache (state.m_fieldCache);
179  state.m_fitField.setAtlasField(&state.m_fieldCache);
180  } else {
181  state.m_fitField.setAtlasField(m_BMAG);
182  }
183  state.m_eventContext = &ctx;
184  state.m_vkalFitControl.vk_objProp = m_fitPropagator;
185  state.m_useAprioriVertex = m_useAprioriVertex;
186  state.m_useThetaCnst = m_useThetaCnst;
187  state.m_usePhiCnst = m_usePhiCnst;
188  state.m_usePointingCnst = m_usePointingCnst;
189  state.m_useZPointingCnst = m_useZPointingCnst;
190  state.m_usePassNear = m_usePassNear;
191  state.m_usePassWithTrkErr = m_usePassWithTrkErr;
192  state.m_VertexForConstraint = m_c_VertexForConstraint;
193  state.m_CovVrtForConstraint = m_c_CovVrtForConstraint;
194  state.m_massForConstraint = m_massForConstraint;
195  state.m_Robustness = m_Robustness;
196  state.m_RobustScale = m_RobustScale;
197  state.m_MassInputParticles = m_c_MassInputParticles;
198  state.m_frozenVersionForBTagging = m_frozenVersionForBTagging;
199  state.m_allowUltraDisplaced = m_allowUltraDisplaced;
200 }

◆ initState() [2/2]

void Trk::TrkVKalVrtFitter::initState ( State &  state) const
private

Definition at line 158 of file TrkVKalVrtFitter.cxx.

159 {
160  initState(Gaudi::Hive::currentContext(), state);
161 }

◆ makeSimpleCascade()

void Trk::TrkVKalVrtFitter::makeSimpleCascade ( std::vector< std::vector< int > > &  vrtDef,
std::vector< std::vector< int > > &  cascadeDef,
CascadeState &  cstate 
)
staticprivate

Definition at line 180 of file TrkCascadeFitter.cxx.

183 {
184  int iv,ip,it, nVAdd, iva;
185  vrtDef.clear();
186  cascadeDef.clear();
187  int NVC=cstate.m_cascadeVList.size();
188  vrtDef.resize(NVC);
189  cascadeDef.resize(NVC);
190 //
191 //---- First set up position of each vertex in simple structure with merging(!!!)
192 //
193  int vCounter=0;
194  for(iv=0; iv<NVC; iv++){
195  cascadeV &vrt=cstate.m_cascadeVList[iv];
196  vrt.indexInSimpleCascade=-1; // set to -1 for merged vertices not present in simple list
197  if(vrt.mergedTO) continue; // vertex is merged with another one;
198  vrt.indexInSimpleCascade=vCounter; // vertex position in simple cascade structure
199  vCounter++;
200  }
201 //---- Fill vertices in simple structure
202  vCounter=0;
203  for(iv=0; iv<NVC; iv++){
204  const cascadeV &vrt=cstate.m_cascadeVList[iv];
205  if(vrt.mergedTO) continue; // vertex is merged with another one;
206  for(it=0; it<(int)vrt.trkInVrt.size(); it++) vrtDef[vCounter].push_back(vrt.trkInVrt[it]); //copy real tracks
207  for(ip=0; ip<(int)vrt.inPointingV.size(); ip++) {
208  //int indInFull=vrt.inPointingV[ip]; // pointing vertex in full list WRONG!!!
209  int indInFull=indexInV(vrt.inPointingV[ip], cstate); // pointing vertex in full list
210  int indInSimple=cstate.m_cascadeVList[indInFull].indexInSimpleCascade; // its index in simple structure
211  if(indInSimple<0) continue; // merged out vertex. Will be added as tracks
212  cascadeDef[vCounter].push_back(indInSimple);
213  }
214  nVAdd=vrt.mergedIN.size();
215  if( nVAdd ) { //----------------------------- mergedIN(added) vertices exist
216  for(iva=0; iva<nVAdd; iva++){
217  const cascadeV &vrtM=cstate.m_cascadeVList[vrt.mergedIN[iva]]; // merged/added vertex itself
218  for(it=0; it<(int)vrtM.trkInVrt.size(); it++) vrtDef[vCounter].push_back(vrtM.trkInVrt[it]);
219  for(ip=0; ip<(int)vrtM.inPointingV.size(); ip++) {
220  //int indInFull=vrtM.inPointingV[ip]; // pointing vertex in full list WRONG!!!
221  int indInFull=indexInV(vrtM.inPointingV[ip], cstate); // pointing vertex in full list
222  int indInSimple=cstate.m_cascadeVList[indInFull].indexInSimpleCascade; // its index in simple structure
223  if(indInSimple<0) continue; // merged out vertex. Will be added as tracks
224  cascadeDef[vCounter].push_back(indInSimple);
225  }
226  }
227  }
228 
229  vCounter++;
230  }
231  vrtDef.resize(vCounter);
232  cascadeDef.resize(vCounter);
233 }

◆ makeState()

std::unique_ptr< IVKalState > Trk::TrkVKalVrtFitter::makeState ( const EventContext &  ctx) const
finaloverridevirtual

Definition at line 58 of file TrkVKalVrtFitter.cxx.

59 {
60  auto state = std::make_unique<State>();
61  initState(ctx, *state);
62  return state;
63 }

◆ makeXAODVertex()

xAOD::Vertex * Trk::TrkVKalVrtFitter::makeXAODVertex ( int  Neutrals,
const Amg::Vector3D &  Vertex,
const dvect &  fitErrorMatrix,
const dvect &  Chi2PerTrk,
const std::vector< dvect > &  TrkAtVrt,
double  Chi2,
State &  state 
) const
private

Definition at line 665 of file TrkVKalVrtFitter.cxx.

670 {
671  long int NTrk = state.m_FitStatus;
672  long int Ndf = VKalGetNDOF(state)+state.m_planeCnstNDOF;
673 
674  xAOD::Vertex * tmpVertex=new xAOD::Vertex();
675  tmpVertex->makePrivateStore();
676  tmpVertex->setPosition(Vertex);
677  tmpVertex->setFitQuality(Chi2, (float)Ndf);
678 
679  std::vector<VxTrackAtVertex> & tmpVTAV=tmpVertex->vxTrackAtVertex();
680  tmpVTAV.clear();
681  std::vector <double> CovFull;
682  StatusCode sc = VKalGetFullCov( NTrk, CovFull, state);
683  int covarExist=0; if( sc.isSuccess() ) covarExist=1;
684 
685  std::vector<float> floatErrMtx;
686  if( m_makeExtendedVertex && covarExist ) {
687  floatErrMtx.resize(CovFull.size());
688  for(int i=0; i<(int)CovFull.size(); i++) {
689  if( CovFull[i] < std::numeric_limits<float>::max() &&
690  CovFull[i] > std::numeric_limits<float>::lowest() ){
691  floatErrMtx[i]=static_cast<float>(CovFull[i]);
692  } else {
693  floatErrMtx[i]=std::numeric_limits<float>::max();
694  }
695  }
696  }else{
697  floatErrMtx.resize(fitErrorMatrix.size());
698  for(int i=0; i<(int)fitErrorMatrix.size(); i++) {
699  if( fitErrorMatrix[i] < std::numeric_limits<float>::max() &&
700  fitErrorMatrix[i] > std::numeric_limits<float>::lowest() ){
701  floatErrMtx[i]=static_cast<float>(fitErrorMatrix[i]);
702  } else {
703  floatErrMtx[i]=std::numeric_limits<float>::max();
704  }
705  }
706  }
707  tmpVertex->setCovariance(floatErrMtx);
708 
709  for(int ii=0; ii<NTrk ; ii++) {
710  AmgSymMatrix(5) CovMtxP;
711  if(covarExist){ FillMatrixP( ii, CovMtxP, CovFull );}
712  else { CovMtxP.setIdentity();}
713  Perigee * tmpChargPer=nullptr;
714  NeutralPerigee * tmpNeutrPer=nullptr;
715  if(ii<NTrk-Neutrals){
716  tmpChargPer = new Perigee( 0.,0., TrkAtVrt[ii][0],
717  TrkAtVrt[ii][1],
718  TrkAtVrt[ii][2],
719  PerigeeSurface(Vertex), std::move(CovMtxP) );
720  }else{
721  tmpNeutrPer = new NeutralPerigee( 0.,0., TrkAtVrt[ii][0],
722  TrkAtVrt[ii][1],
723  TrkAtVrt[ii][2],
724  PerigeeSurface(Vertex),
725  std::move(CovMtxP) );
726  }
727  tmpVTAV.emplace_back(Chi2PerTrk[ii], tmpChargPer, tmpNeutrPer );
728  }
729 
730  return tmpVertex;
731 }

◆ nextVertex() [1/2]

VertexID Trk::TrkVKalVrtFitter::nextVertex ( const std::vector< const xAOD::TrackParticle * > &  list,
std::span< const double >  particleMass,
const std::vector< VertexID > &  precedingVertices,
IVKalState &  istate,
double  massConstraint = 0. 
) const
finaloverride

Definition at line 144 of file TrkCascadeFitter.cxx.

149 {
150  assert(dynamic_cast<State*> (&istate)!=nullptr);
151  State& state = static_cast<State&> (istate);
152  CascadeState& cstate = *state.m_cascadeState;
153 
154  VertexID vID=nextVertex( list, particleMass, istate, massConstraint);
155 //
156  int lastC=cstate.m_partMassCnstForCascade.size()-1; // Check if full vertex mass constraint exist
157  if( lastC>=0 ){ if( cstate.m_partMassCnstForCascade[lastC].VRT == vID ){
158  for(int iv=0; iv<(int)precedingVertices.size(); iv++){
159  cstate.m_partMassCnstForCascade[lastC].pseudoInVrt.push_back(precedingVertices[iv]); }
160  }
161  }
162 //
163 //-- New vertex structure-----------------------------------
164  int lastV=cstate.m_cascadeVList.size()-1;
165  for(int iv=0; iv<(int)precedingVertices.size(); iv++){
166  cstate.m_cascadeVList[lastV].inPointingV.push_back(precedingVertices[iv]); // fill preceding vertices list
167  }
168 //--
169  return vID;
170 }

◆ nextVertex() [2/2]

VertexID Trk::TrkVKalVrtFitter::nextVertex ( const std::vector< const xAOD::TrackParticle * > &  list,
std::span< const double >  particleMass,
IVKalState &  istate,
double  massConstraint = 0. 
) const
finaloverride

Definition at line 97 of file TrkCascadeFitter.cxx.

101 {
102  assert(dynamic_cast<State*> (&istate)!=nullptr);
103  State& state = static_cast<State&> (istate);
104  CascadeState& cstate = *state.m_cascadeState;
105 
106 //----
107  int NV = cstate.m_cascadeSize++;
108  VertexID new_vID=10000+NV;
109 //----
110  int NTRK = list.size();
111  int presentNT = cstate.m_partListForCascade.size();
112 //----
113 
114  double totMass=0;
115  for(int it=0; it<NTRK; it++){
116  cstate.m_partListForCascade.push_back(list[it]);
117  cstate.m_partMassForCascade.push_back(particleMass[it]);
118  totMass += particleMass[it];
119  }
120 //---------------------- Fill complete vertex mass constraint
121  if(totMass < massConstraint) {
122  PartialMassConstraint tmpMcnst;
123  tmpMcnst.Mass = massConstraint;
124  tmpMcnst.VRT = new_vID;
125  for(int it=0; it<NTRK; it++)tmpMcnst.trkInVrt.push_back(it+presentNT);
126  cstate.m_partMassCnstForCascade.push_back(std::move(tmpMcnst));
127  }
128 //
129 //
130 //-- New vertex structure-----------------------------------
131  cascadeV newV; newV.vID=new_vID;
132  for(int it=0; it<NTRK; it++){
133  newV.trkInVrt.push_back(it+presentNT);
134  }
135  cstate.m_cascadeVList.push_back(std::move(newV));
136 //--------------------------------------------------------------
137  return new_vID;
138 }

◆ printSimpleCascade()

void Trk::TrkVKalVrtFitter::printSimpleCascade ( std::vector< std::vector< int > > &  vrtDef,
std::vector< std::vector< int > > &  cascadeDef,
const CascadeState &  cstate 
)
staticprivate

Definition at line 237 of file TrkCascadeFitter.cxx.

240 {
241  int kk,kkk;
242  for(kk=0; kk<(int)vrtDef.size(); kk++){
243  std::cout<<" Vertex("<<kk<<"):: trk=";
244  for(kkk=0; kkk<(int)vrtDef[kk].size(); kkk++){
245  std::cout<<vrtDef[kk][kkk]<<", ";} std::cout<<" pseu=";
246  for(kkk=0; kkk<(int)cascadeDef[kk].size(); kkk++){
247  std::cout<<cascadeDef[kk][kkk]<<", ";}
248  } std::cout<<'\n';
249 //---
250  for(kk=0; kk<(int)vrtDef.size(); kk++){
251  std::cout<<" Vertex("<<kk<<"):: trkM=";
252  for(kkk=0; kkk<(int)vrtDef[kk].size(); kkk++){
253  std::cout<<cstate.m_partMassForCascade[vrtDef[kk][kkk]]<<", ";}
254  }std::cout<<'\n';
255 //--
256  for (const PartialMassConstraint& c : cstate.m_partMassCnstForCascade) {
257  std::cout<<" MCnst vID=";
258  std::cout<<c.VRT<<" m="<<c.Mass<<" trk=";
259  for(int idx : c.trkInVrt) {
260  std::cout<<idx<<", ";
261  }
262  std::cout<<" pseudo=";
263  for (VertexID id : c.pseudoInVrt) {
264  std::cout<<id<<", ";
265  }
266  std::cout<<'\n';
267  }
268 }

◆ setApproximateVertex()

void Trk::TrkVKalVrtFitter::setApproximateVertex ( double  X,
double  Y,
double  Z,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 115 of file SetFitOptions.cxx.

117  {
118  assert(dynamic_cast<State*> (&istate)!=nullptr);
119  State& state = static_cast<State&> (istate);
120  state.m_ApproximateVertex.assign ({X, Y, Z});
121  }

◆ setAthenaPropagator()

void Trk::TrkVKalVrtFitter::setAthenaPropagator ( const Trk::IExtrapolator *  Pnt)
private

Definition at line 477 of file VKalExtPropagator.cxx.

478  {
479  // Save external propagator in VKalExtPropagator object and send it to TrkVKalVrtCore
480 //
481  if(m_fitPropagator != nullptr) delete m_fitPropagator;
482  m_fitPropagator = new VKalExtPropagator( this );
483  m_fitPropagator->setPropagator(Pnt);
484  m_InDetExtrapolator = Pnt; // Pointer to InDet extrapolator
485  }

◆ setCnstType()

void Trk::TrkVKalVrtFitter::setCnstType ( int  TYPE,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 89 of file SetFitOptions.cxx.

90  {
91  assert(dynamic_cast<State*> (&istate)!=nullptr);
92  State& state = static_cast<State&> (istate);
93  if(TYPE>0)msg(MSG::DEBUG)<< "ConstraintType is changed at execution stage. New type="<<TYPE<< endmsg;
94  if(TYPE<0)TYPE=0;
95  if(TYPE>14)TYPE=0;
96  if( TYPE == 2) state.m_usePointingCnst = true;
97  if( TYPE == 3) state.m_useZPointingCnst = true;
98  if( TYPE == 4) state.m_usePointingCnst = true;
99  if( TYPE == 5) state.m_useZPointingCnst = true;
100  if( TYPE == 6) state.m_useAprioriVertex = true;
101  if( TYPE == 7) state.m_usePassWithTrkErr = true;
102  if( TYPE == 8) state.m_usePassWithTrkErr = true;
103  if( TYPE == 9) state.m_usePassNear = true;
104  if( TYPE == 10) state.m_usePassNear = true;
105  if( TYPE == 11) state.m_usePhiCnst = true;
106  if( TYPE == 12) { state.m_usePhiCnst = true; state.m_useThetaCnst = true;}
107  if( TYPE == 13) { state.m_usePhiCnst = true; state.m_usePassNear = true;}
108  if( TYPE == 14) { state.m_usePhiCnst = true; state.m_useThetaCnst = true; state.m_usePassNear = true;}
109  }

◆ setCovVrtForConstraint()

void Trk::TrkVKalVrtFitter::setCovVrtForConstraint ( double  XX,
double  XY,
double  YY,
double  XZ,
double  YZ,
double  ZZ,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 185 of file SetFitOptions.cxx.

188  {
189  assert(dynamic_cast<State*> (&istate)!=nullptr);
190  State& state = static_cast<State&> (istate);
191  state.m_CovVrtForConstraint.assign ({XX, XY, YY, XZ, YZ, ZZ});
192  }

◆ setMassForConstraint() [1/2]

void Trk::TrkVKalVrtFitter::setMassForConstraint ( double  Mass,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 141 of file SetFitOptions.cxx.

143  {
144  assert(dynamic_cast<State*> (&istate)!=nullptr);
145  State& state = static_cast<State&> (istate);
146  state.m_massForConstraint = MASS;
147  }

◆ setMassForConstraint() [2/2]

void Trk::TrkVKalVrtFitter::setMassForConstraint ( double  Mass,
std::span< const int >  TrkIndex,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 149 of file SetFitOptions.cxx.

152  {
153  assert(dynamic_cast<State*> (&istate)!=nullptr);
154  State& state = static_cast<State&> (istate);
155  state.m_partMassCnst.push_back(MASS);
156  state.m_partMassCnstTrk.emplace_back(TrkIndex.begin(), TrkIndex.end());
157  }

◆ setMassInputParticles()

void Trk::TrkVKalVrtFitter::setMassInputParticles ( const std::vector< double > &  mass,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 194 of file SetFitOptions.cxx.

196  {
197  assert(dynamic_cast<State*> (&istate)!=nullptr);
198  State& state = static_cast<State&> (istate);
199  state.m_MassInputParticles = mass;
200  for (double& m : state.m_MassInputParticles) {
201  m = std::abs(m);
202  }
203  }

◆ setRobustness()

void Trk::TrkVKalVrtFitter::setRobustness ( int  IROB,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 123 of file SetFitOptions.cxx.

124  { if(IROB>0)msg(MSG::DEBUG)<< "Robustness is changed at execution stage "<<m_Robustness<<"=>"<<IROB<< endmsg;
125  assert(dynamic_cast<State*> (&istate)!=nullptr);
126  State& state = static_cast<State&> (istate);
127  state.m_Robustness = IROB;
128  if(state.m_Robustness<0)state.m_Robustness=0;
129  if(state.m_Robustness>7)state.m_Robustness=0;
130  }

◆ setRobustScale()

void Trk::TrkVKalVrtFitter::setRobustScale ( double  Scale,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 132 of file SetFitOptions.cxx.

133  { if(Scale!=m_RobustScale)msg(MSG::DEBUG)<< "Robust Scale is changed at execution stage "<<m_RobustScale<<"=>"<<Scale<< endmsg;
134  assert(dynamic_cast<State*> (&istate)!=nullptr);
135  State& state = static_cast<State&> (istate);
136  state.m_RobustScale = Scale;
137  if(state.m_RobustScale<0.01) state.m_RobustScale=1.;
138  if(state.m_RobustScale>100.) state.m_RobustScale=1.;
139  }

◆ setVertexForConstraint() [1/2]

void Trk::TrkVKalVrtFitter::setVertexForConstraint ( const xAOD::Vertex &  Vrt,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 159 of file SetFitOptions.cxx.

161  {
162  assert(dynamic_cast<State*> (&istate)!=nullptr);
163  State& state = static_cast<State&> (istate);
164  state.m_VertexForConstraint.assign ({Vrt.position().x(),
165  Vrt.position().y(),
166  Vrt.position().z()});
167 
168  state.m_CovVrtForConstraint.assign ({
169  Vrt.covariancePosition()(Trk::x,Trk::x),
170  Vrt.covariancePosition()(Trk::x,Trk::y),
171  Vrt.covariancePosition()(Trk::y,Trk::y),
172  Vrt.covariancePosition()(Trk::x,Trk::z),
173  Vrt.covariancePosition()(Trk::y,Trk::z),
174  Vrt.covariancePosition()(Trk::z,Trk::z)});
175  }

◆ setVertexForConstraint() [2/2]

void Trk::TrkVKalVrtFitter::setVertexForConstraint ( double  X,
double  Y,
double  Z,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 177 of file SetFitOptions.cxx.

179  {
180  assert(dynamic_cast<State*> (&istate)!=nullptr);
181  State& state = static_cast<State&> (istate);
182  state.m_VertexForConstraint.assign ({X, Y, Z});
183  }

◆ startVertex()

VertexID Trk::TrkVKalVrtFitter::startVertex ( const std::vector< const xAOD::TrackParticle * > &  list,
std::span< const double >  particleMass,
IVKalState &  istate,
double  massConstraint = 0. 
) const
finaloverride

Interface for cascade fit.

Definition at line 62 of file TrkCascadeFitter.cxx.

66 {
67  assert(dynamic_cast<State*> (&istate)!=nullptr);
68  State& state = static_cast<State&> (istate);
69  state.m_cascadeState = std::make_unique<CascadeState>();
70  state.m_vkalFitControl.renewCascadeEvent(new CascadeEvent());
71 
72  return nextVertex (list, particleMass, istate, massConstraint);
73 }

◆ VKalGetFullCov()

StatusCode Trk::TrkVKalVrtFitter::VKalGetFullCov ( long int  NTrk,
dvect &  CovMtx,
IVKalState &  istate,
bool  useMom = false 
) const
finaloverridevirtual

Definition at line 450 of file VKalVrtFitSvc.cxx.

453  {
454  assert(dynamic_cast<State*> (&istate)!=nullptr);
455  State& state = static_cast<State&> (istate);
456  if(!state.m_FitStatus) return StatusCode::FAILURE;
457  if(NTrk<1) return StatusCode::FAILURE;
458  if(NTrk>NTrMaxVFit) return StatusCode::FAILURE;
459  if(state.m_ErrMtx.empty()) return StatusCode::FAILURE; //Now error matrix is taken from CORE in VKalVrtFit3.
460 //
461 // ------ Magnetic field access
462 //
463  double fx,fy,BMAG_CUR;
464  state.m_fitField.getMagFld(state.m_save_xyzfit[0],state.m_save_xyzfit[1],state.m_save_xyzfit[2],fx,fy,BMAG_CUR);
465  if(fabs(BMAG_CUR) < 0.01) BMAG_CUR=0.01; // Safety
466 //
467 // ------ Base code
468 //
469  int i,j,ik,jk,ip,iTrk;
470  int DIM=3*NTrk+3; //Current size of full covariance matrix
471  std::vector<std::vector<double> > Deriv (DIM);
472  for (std::vector<double>& v : Deriv) v.resize (DIM);
473  std::vector<double> CovMtxOld(DIM*DIM);
474 
475 
476  CovVrtTrk.resize(DIM*(DIM+1)/2);
477 
478  ip=0;
479  for( i=0; i<DIM;i++) {
480  for( j=0; j<=i; j++) {
481  CovMtxOld[i*DIM+j]=CovMtxOld[j*DIM+i]=state.m_ErrMtx[ip++];
482  }
483  }
484 
485  //delete [] ErrMtx;
486 
487  for(i=0;i<DIM;i++){ for(j=0;j<DIM;j++) {Deriv[i][j]=0.;}}
488  Deriv[0][0]= 1.;
489  Deriv[1][1]= 1.;
490  Deriv[2][2]= 1.;
491 
492  int iSt=0;
493  double Theta,invR,Phi;
494  for( iTrk=0; iTrk<NTrk; iTrk++){
495  Theta=state.m_parfs[iTrk][0];
496  Phi =state.m_parfs[iTrk][1];
497  invR =state.m_parfs[iTrk][2];
498  /*-----------*/
499  /* dNew/dOld */
500  iSt = 3 + iTrk*3;
501  if( !useMom ){
502  Deriv[iSt ][iSt+1] = 1; // Phi <-> Theta
503  Deriv[iSt+1][iSt ] = 1; // Phi <-> Theta
504  Deriv[iSt+2][iSt ] = -(cos(Theta)/(m_CNVMAG*BMAG_CUR)) * invR ; // d1/p / dTheta
505  Deriv[iSt+2][iSt+2] = -(sin(Theta)/(m_CNVMAG*BMAG_CUR)) ; // d1/p / d1/R
506  }else{
507  double pt=(m_CNVMAG*BMAG_CUR)/fabs(invR);
508  double px=pt*cos(Phi);
509  double py=pt*sin(Phi);
510  double pz=pt/tan(Theta);
511  Deriv[iSt ][iSt ]= 0; //dPx/dTheta
512  Deriv[iSt ][iSt+1]= -py; //dPx/dPhi
513  Deriv[iSt ][iSt+2]= -px/invR; //dPx/dinvR
514 
515  Deriv[iSt+1][iSt ]= 0; //dPy/dTheta
516  Deriv[iSt+1][iSt+1]= px; //dPy/dPhi
517  Deriv[iSt+1][iSt+2]= -py/invR; //dPy/dinvR
518 
519  Deriv[iSt+2][iSt ]= -pt/sin(Theta)/sin(Theta); //dPz/dTheta
520  Deriv[iSt+2][iSt+1]= 0; //dPz/dPhi
521  Deriv[iSt+2][iSt+2]= -pz/invR; //dPz/dinvR
522  }
523  }
524 //---------- Only upper half if filled and saved
525  int ipnt=0;
526  double tmp, tmpTmp;
527  for(i=0;i<DIM;i++){
528  for(j=0;j<=i;j++){
529  tmp=0.;
530  for(ik=0;ik<DIM;ik++){
531  if(Deriv[i][ik] == 0.) continue;
532  tmpTmp=0;
533  for(jk=DIM-1;jk>=0;jk--){
534  if(Deriv[j][jk] == 0.) continue;
535  tmpTmp += CovMtxOld[ik*DIM+jk]*Deriv[j][jk];
536  }
537  tmp += Deriv[i][ik]*tmpTmp;
538  }
539  CovVrtTrk[ipnt++]=tmp;
540  }}
541 
542  return StatusCode::SUCCESS;
543 
544  }

◆ VKalGetImpact() [1/4]

double Trk::TrkVKalVrtFitter::VKalGetImpact ( const Trk::Perigee *  InpPerigee,
const Amg::Vector3D &  Vertex,
const long int  Charge,
dvect &  Impact,
dvect &  ImpactError 
) const
finaloverridevirtual

Definition at line 21 of file VKalGetImpact.cxx.

26  {
27  State state;
28  initState (state);
29  return VKalGetImpact (InpPerigee, Vertex, Charge, Impact, ImpactError, state);
30  }

◆ VKalGetImpact() [2/4]

double Trk::TrkVKalVrtFitter::VKalGetImpact ( const Trk::Perigee *  InpPerigee,
const Amg::Vector3D &  Vertex,
const long int  Charge,
dvect &  Impact,
dvect &  ImpactError,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 32 of file VKalGetImpact.cxx.

38  {
39  assert(dynamic_cast<State*> (&istate)!=nullptr);
40  State& state = static_cast<State&> (istate);
41 
42  //
43  //------ Variables and arrays needed for fitting kernel
44  //
45  double SIGNIF=0.;
46  std::vector<const Trk::Perigee*> InpPerigeeList;
47  InpPerigeeList.push_back(InpPerigee);
48 
49  //
50  //------ extract information about selected tracks
51  //
52  int ntrk=0;
53  StatusCode sc = CvtPerigee(InpPerigeeList, ntrk, state);
54  if(sc.isFailure() || ntrk != 1) { //Something is wrong in conversion
55  Impact.assign(5,1.e10);
56  ImpactError.assign(3,1.e20);
57  return 1.e10;
58  }
59  long int vkCharge = state.m_ich[0];
60  if(Charge==0) vkCharge=0;
61 
62  //
63  // Target vertex in ref.frame defined by track themself
64  //
65  double VrtInp[3]={Vertex.x()-state.m_refFrameX,
66  Vertex.y()-state.m_refFrameY,
67  Vertex.z()-state.m_refFrameZ};
68  double VrtCov[6]={0.,0.,0.,0.,0.,0.};
69 
70  Impact.resize(5);
71  ImpactError.resize(3);
72  Trk::cfimp(0, vkCharge, 0,
73  &state.m_apar[0][0], &state.m_awgt[0][0],
74  &VrtInp[0], &VrtCov[0],
75  Impact.data(), ImpactError.data(),
76  &SIGNIF, &state.m_vkalFitControl);
77 
78  return SIGNIF;
79  }

◆ VKalGetImpact() [3/4]

double Trk::TrkVKalVrtFitter::VKalGetImpact ( const xAOD::TrackParticle *  InpTrk,
const Amg::Vector3D &  Vertex,
const long int  Charge,
dvect &  Impact,
dvect &  ImpactError 
) const
finaloverridevirtual

Definition at line 82 of file VKalGetImpact.cxx.

84  {
85  State state;
86  initState (state);
87  return VKalGetImpact (InpTrk, Vertex, Charge, Impact, ImpactError, state);
88  }

◆ VKalGetImpact() [4/4]

double Trk::TrkVKalVrtFitter::VKalGetImpact ( const xAOD::TrackParticle *  InpTrk,
const Amg::Vector3D &  Vertex,
const long int  Charge,
dvect &  Impact,
dvect &  ImpactError,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 91 of file VKalGetImpact.cxx.

94  {
95  assert(dynamic_cast<State*> (&istate)!=nullptr);
96  State& state = static_cast<State&> (istate);
97 //
98 //------ Variables and arrays needed for fitting kernel
99 //
100  double SIGNIF=0.;
101 
102  std::vector<const xAOD::TrackParticle*> InpTrkList(1,InpTrk);
103 //
104 
105 //
106 //------ extract information about selected tracks
107 //
108  int ntrk=0;
109  StatusCode sc = CvtTrackParticle(InpTrkList,ntrk,state);
110  if(sc.isFailure() || ntrk != 1 ) { //Something is wrong in conversion
111  Impact.assign(5,1.e10);
112  ImpactError.assign(3,1.e20);
113  return 1.e10;
114  }
115  double sizeR = state.m_allowUltraDisplaced ? m_MSsizeR : m_IDsizeR;
116  double sizeZ = state.m_allowUltraDisplaced ? m_MSsizeZ : m_IDsizeZ;
117  if (std::abs(Vertex.z()) > sizeZ || Vertex.perp() > sizeR) {
118  Impact.assign(5, 1.e10);
119  ImpactError.assign(3, 1.e20);
120  return 1.e10;
121  }
122  long int vkCharge=state.m_ich[0];
123  if(Charge==0)vkCharge=0;
124 //
125 // Target vertex in ref.frame defined by track itself
126 //
127  double VrtInp[3]={Vertex.x() -state.m_refFrameX, Vertex.y() -state.m_refFrameY, Vertex.z() -state.m_refFrameZ};
128  double VrtCov[6]={0.,0.,0.,0.,0.,0.};
129 //
130 //
131  Impact.resize(5); ImpactError.resize(3);
132  Trk::cfimp( 0, vkCharge, 0, &state.m_apar[0][0], &state.m_awgt[0][0], &VrtInp[0], &VrtCov[0], Impact.data(), ImpactError.data(), &SIGNIF, &state.m_vkalFitControl);
133 
134  return SIGNIF;
135 
136  }

◆ VKalGetMassError()

StatusCode Trk::TrkVKalVrtFitter::VKalGetMassError ( double &  Mass,
double &  MassError,
const IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 550 of file VKalVrtFitSvc.cxx.

552  {
553  assert(dynamic_cast<const State*> (&istate)!=nullptr);
554  const State& state = static_cast<const State&> (istate);
555  if(!state.m_FitStatus) return StatusCode::FAILURE;
556  dM = state.m_vkalFitControl.getVertexMass();
557  MassError = state.m_vkalFitControl.getVrtMassError();
558  return StatusCode::SUCCESS;
559  }

◆ VKalGetNDOF()

int Trk::TrkVKalVrtFitter::VKalGetNDOF ( const State &  state)
staticprivate

Definition at line 578 of file VKalVrtFitSvc.cxx.

579  {
580  if(!state.m_FitStatus) return 0;
581  int NDOF=2*state.m_FitStatus-3;
582  if(state.m_usePointingCnst) { NDOF+=2; }
583  else if(state.m_useZPointingCnst) { NDOF+=1; }
584  if( state.m_usePassNear || state.m_usePassWithTrkErr ) { NDOF+= 2; }
585 
586  if( state.m_massForConstraint>0. ) { NDOF+=1; }
587  if( !state.m_partMassCnst.empty() ) { NDOF+= state.m_partMassCnst.size(); }
588  if( state.m_useAprioriVertex ) { NDOF+= 3; }
589  if( state.m_usePhiCnst ) { NDOF+=1; }
590  if( state.m_useThetaCnst ) { NDOF+=1; }
591  return NDOF;
592  }

◆ VKalGetTrkWeights()

StatusCode Trk::TrkVKalVrtFitter::VKalGetTrkWeights ( dvect &  Weights,
const IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 562 of file VKalVrtFitSvc.cxx.

564  {
565  assert(dynamic_cast<const State*> (&istate)!=nullptr);
566  const State& state = static_cast<const State&> (istate);
567  if(!state.m_FitStatus) return StatusCode::FAILURE; // no fit made
568  trkWeights.clear();
569 
570  int NTRK=state.m_FitStatus;
571 
572  for (int i=0; i<NTRK; i++) trkWeights.push_back(state.m_vkalFitControl.vk_forcft.robres[i]);
573 
574  return StatusCode::SUCCESS;
575  }

◆ VKalToTrkTrack()

void Trk::TrkVKalVrtFitter::VKalToTrkTrack ( double  curBMAG,
double  vp1,
double  vp2,
double  vp3,
double &  tp1,
double &  tp2,
double &  tp3 
) const
private

Definition at line 422 of file VKalVrtFitSvc.cxx.

425  { tp1= vp2; //phi angle
426  tp2= vp1; //theta angle
427  tp3= vp3 * std::sin( vp1 ) /(m_CNVMAG*curBMAG);
428  constexpr double pi = M_PI;
429  // -pi < phi < pi range
430  while ( tp1 > pi) tp1 -= 2.*pi;
431  while ( tp1 <-pi) tp1 += 2.*pi;
432  // 0 < Theta < pi range
433  while ( tp2 > pi) tp2 -= 2.*pi;
434  while ( tp2 <-pi) tp2 += 2.*pi;
435  if ( tp2 < 0.) {
436  tp2 = fabs(tp2); tp1 += pi;
437  while ( tp1 > pi) tp1 -= 2.*pi;
438  }
439 
440  }

◆ VKalTransform()

void Trk::TrkVKalVrtFitter::VKalTransform ( double  MAG,
double  A0V,
double  ZV,
double  PhiV,
double  ThetaV,
double  PInv,
const double  CovTrk[15],
long int &  Charge,
double  VTrkPar[5],
double  VTrkCov[15] 
) const
private

Definition at line 59 of file VKalTransform.cxx.

62  {
63  int i,j,ii,jj;
64  double CnvCst=m_CNVMAG*BMAG_FIXED;
65  double sinT = sin(ThetaV);
66  double cosT = cos(ThetaV);
67 
68  VTrkPar[0] = - A0V ;
69  VTrkPar[1] = ZV ;
70  VTrkPar[2] = ThetaV;
71  VTrkPar[3] = PhiV;
72  VTrkPar[4] = -PInv*CnvCst/sinT ;
73  Charge = PInv > 0 ? -1 : 1;
74 //
75 //
76  double CovI[5][5];
77  double Deriv[5][5] ={{0.,0.,0.,0.,0.},{0.,0.,0.,0.,0.},{0.,0.,0.,0.,0.},
78  {0.,0.,0.,0.,0.},{0.,0.,0.,0.,0.}};
79 
80  CovI[0][0] = CovTrk[0];
81 
82  CovI[1][0] = CovTrk[1];
83  CovI[0][1] = CovTrk[1];
84  CovI[1][1] = CovTrk[2];
85 
86  CovI[0][2] = CovTrk[3];
87  CovI[2][0] = CovTrk[3];
88  CovI[1][2] = CovTrk[4];
89  CovI[2][1] = CovTrk[4];
90  CovI[2][2] = CovTrk[5];
91 
92  CovI[0][3] = CovTrk[6];
93  CovI[3][0] = CovTrk[6];
94  CovI[1][3] = CovTrk[7];
95  CovI[3][1] = CovTrk[7];
96  CovI[2][3] = CovTrk[8];
97  CovI[3][2] = CovTrk[8];
98  CovI[3][3] = CovTrk[9];
99 
100  CovI[0][4] = CovTrk[10] ;
101  CovI[4][0] = CovTrk[10] ;
102  CovI[1][4] = CovTrk[11] ;
103  CovI[4][1] = CovTrk[11] ;
104  CovI[2][4] = CovTrk[12] ;
105  CovI[4][2] = CovTrk[12] ;
106  CovI[3][4] = CovTrk[13] ;
107  CovI[4][3] = CovTrk[13] ;
108  CovI[4][4] = CovTrk[14] ;
109 
110 
111  Deriv[0][0] = -1.;
112  Deriv[1][1] = 1.;
113  Deriv[2][3] = 1.;
114  Deriv[3][2] = 1.;
115  Deriv[4][3] = PInv*CnvCst *(cosT/sinT/sinT) ;
116  Deriv[4][4] = -CnvCst/sinT;
117 
118  double ct;
119  int ipnt=0;
120  for(i=0;i<5;i++){ for(j=0;j<=i;j++){
121  ct=0.;
122  for(ii=4;ii>=0;ii--){
123  if(Deriv[i][ii] == 0.) continue;
124  for(jj=4;jj>=0;jj--){
125  if(Deriv[j][jj] == 0.) continue;
126  ct += CovI[ii][jj]*Deriv[i][ii]*Deriv[j][jj];};};
127  VTrkCov[ipnt++]=ct;
128  };}
129 
130 }

◆ VKalVrtConfigureFitterCore()

void Trk::TrkVKalVrtFitter::VKalVrtConfigureFitterCore ( int  NTRK,
State &  state 
) const
private

Definition at line 18 of file SetFitOptions.cxx.

19  {
20  state.m_FitStatus = 0; // Drop all previous fit results
21  state.m_globalFirstHit = nullptr;
22  state.m_vkalFitControl.vk_forcft = ForCFT();
23 
24  //Set input particle masses
25  for(int it=0; it<NTRK; it++){
26  if( it<(int)state.m_MassInputParticles.size() ) {
27  state.m_vkalFitControl.vk_forcft.wm[it] = (double)(state.m_MassInputParticles[it]);
28  }
29  else { state.m_vkalFitControl.vk_forcft.wm[it]=ParticleConstants::chargedPionMassInMeV; }
30  }
31  // Set reference vertex for different pointing constraints
32  if(state.m_VertexForConstraint.size() >= 3){
33  state.m_vkalFitControl.vk_forcft.vrt[0] =state.m_VertexForConstraint[0] - state.m_refFrameX;
34  state.m_vkalFitControl.vk_forcft.vrt[1] =state.m_VertexForConstraint[1] - state.m_refFrameY;
35  state.m_vkalFitControl.vk_forcft.vrt[2] =state.m_VertexForConstraint[2] - state.m_refFrameZ;
36  }else {for( int i=0; i<3; i++) state.m_vkalFitControl.vk_forcft.vrt[i] = 0.; }
37  // Set covariance matrix for reference vertex
38  if(state.m_CovVrtForConstraint.size() >= 6){
39  for( int i=0; i<6; i++) { state.m_vkalFitControl.vk_forcft.covvrt[i] = (double)(state.m_CovVrtForConstraint[i]); }
40  }else{ for( int i=0; i<6; i++) { state.m_vkalFitControl.vk_forcft.covvrt[i] = 0.; } }
41 
42  // Add global mass constraint if present
43  if(state.m_massForConstraint >= 0.) state.m_vkalFitControl.setMassCnstData(NTRK,state.m_massForConstraint);
44  // Add partial mass constraints if present
45  if(!state.m_partMassCnst.empty()) {
46  for(int ic=0; ic<(int)state.m_partMassCnst.size(); ic++){
47  state.m_vkalFitControl.setMassCnstData(NTRK, state.m_partMassCnstTrk[ic],state.m_partMassCnst[ic]);
48  }
49  }
50  // Set general configuration parameters
51  state.m_vkalFitControl.setRobustness(state.m_Robustness);
52  state.m_vkalFitControl.setRobustScale(state.m_RobustScale);
53  if(!m_firstMeasuredPointLimit)state.m_vkalFitControl.setUsePlaneCnst(0.,0.,0.,0.);
54  else state.m_vkalFitControl.setUsePlaneCnst(state.m_parPlaneCnst[0],state.m_parPlaneCnst[1],
55  state.m_parPlaneCnst[2],state.m_parPlaneCnst[3]);
56  if(m_firstMeasuredRadiusLimit)state.m_vkalFitControl.setUseRadiusCnst(state.m_cnstRadius,state.m_cnstRadiusRef);
57  if(state.m_useAprioriVertex) state.m_vkalFitControl.setUseAprioriVrt();
58  if(state.m_useThetaCnst) state.m_vkalFitControl.setUseThetaCnst();
59  if(state.m_usePhiCnst) state.m_vkalFitControl.setUsePhiCnst();
60  if(state.m_usePointingCnst) state.m_vkalFitControl.setUsePointingCnst(1);
61  if(state.m_useZPointingCnst) state.m_vkalFitControl.setUsePointingCnst(2);
62  if(state.m_usePassNear) state.m_vkalFitControl.setUsePassNear(1);
63  if(state.m_usePassWithTrkErr)state.m_vkalFitControl.setUsePassNear(2);
64 
65  if(state.m_frozenVersionForBTagging)state.m_vkalFitControl.m_frozenVersionForBTagging=true;
66  if(state.m_allowUltraDisplaced)state.m_vkalFitControl.m_allowUltraDisplaced=true;
67 
68  if(m_IterationPrecision>0.) state.m_vkalFitControl.setIterationPrec(m_IterationPrecision);
69  if(m_IterationNumber) state.m_vkalFitControl.setIterationNum(m_IterationNumber);
70 
71  }

◆ VKalVrtCvtTool()

StatusCode Trk::TrkVKalVrtFitter::VKalVrtCvtTool ( const Amg::Vector3D &  Vertex,
const TLorentzVector &  Momentum,
const dvect &  CovVrtMom,
const long int &  Charge,
dvect &  Perigee,
dvect &  CovPerigee,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 380 of file VKalVrtFitSvc.cxx.

387  {
388  assert(dynamic_cast<State*> (&istate)!=nullptr);
389  State& state = static_cast<State&> (istate);
390  int i,j,ij;
391  double Vrt[3],PMom[4],Cov0[21],Per[5],CovPer[15];
392 
393  for(i=0; i<3; i++) Vrt[i]=Vertex[i];
394  for(i=0; i<3; i++) PMom[i]=Momentum[i];
395  for(ij=i=0; i<6; i++){
396  for(j=0; j<=i; j++){
397  Cov0[ij]=CovVrtMom[ij];
398  ij++;
399  }
400  }
401  state.m_refFrameX=state.m_refFrameY=state.m_refFrameZ=0.; //VK Work in ATLAS ref frame ONLY!!!
402  long int vkCharge=-Charge; //VK 30.11.2009 Change sign according to ATLAS
403 //
404 // ------ Magnetic field in vertex
405 //
406  double fx,fy,BMAG_CUR;
407  state.m_fitField.getMagFld(Vrt[0], Vrt[1], Vrt[2] ,fx,fy,BMAG_CUR);
408  if(fabs(BMAG_CUR) < 0.01) BMAG_CUR=0.01; // Safety
409 
410  Trk::xyztrp( vkCharge, Vrt, PMom, Cov0, BMAG_CUR, Per, CovPer );
411 
412  Perigee.clear();
413  CovPerigee.clear();
414 
415  for(i=0; i<5; i++) Perigee.push_back((double)Per[i]);
416  for(i=0; i<15; i++) CovPerigee.push_back((double)CovPer[i]);
417 
418  return StatusCode::SUCCESS;
419  }

◆ VKalVrtFit() [1/3]

virtual StatusCode Trk::TrkVKalVrtFitter::VKalVrtFit ( const std::vector< const Perigee * > &  ,
Amg::Vector3D &  Vertex,
TLorentzVector &  Momentum,
long int &  Charge,
dvect &  ErrorMatrix,
dvect &  Chi2PerTrk,
std::vector< std::vector< double >> &  TrkAtVrt,
double &  Chi2,
IVKalState &  istate,
bool  ifCovV0 = false 
) const
finaloverridevirtual

◆ VKalVrtFit() [2/3]

virtual StatusCode Trk::TrkVKalVrtFitter::VKalVrtFit ( const std::vector< const TrackParameters * > &  ,
const std::vector< const NeutralParameters * > &  ,
Amg::Vector3D &  Vertex,
TLorentzVector &  Momentum,
long int &  Charge,
dvect &  ErrorMatrix,
dvect &  Chi2PerTrk,
std::vector< std::vector< double >> &  TrkAtVrt,
double &  Chi2,
IVKalState &  istate,
bool  ifCovV0 = false 
) const
finaloverridevirtual

◆ VKalVrtFit() [3/3]

virtual StatusCode Trk::TrkVKalVrtFitter::VKalVrtFit ( const std::vector< const xAOD::TrackParticle * > &  ,
const std::vector< const xAOD::NeutralParticle * > &  ,
Amg::Vector3D &  Vertex,
TLorentzVector &  Momentum,
long int &  Charge,
dvect &  ErrorMatrix,
dvect &  Chi2PerTrk,
std::vector< std::vector< double >> &  TrkAtVrt,
double &  Chi2,
IVKalState &  istate,
bool  ifCovV0 = false 
) const
finaloverridevirtual

◆ VKalVrtFit3()

int Trk::TrkVKalVrtFitter::VKalVrtFit3 ( int  ntrk,
Amg::Vector3D &  Vertex,
TLorentzVector &  Momentum,
long int &  Charge,
dvect &  ErrorMatrix,
dvect &  Chi2PerTrk,
std::vector< std::vector< double >> &  TrkAtVrt,
double &  Chi2,
State &  state,
bool  ifCovV0 
) const
private

Definition at line 266 of file VKalVrtFitSvc.cxx.

276 {
277 //
278 //------ Variables and arrays needed for fitting kernel
279 //
280  int ierr,i;
281  double xyz0[3],covf[21],chi2f=-10.;
282  double ptot[4]={0.};
283  double xyzfit[3]={0.};
284 //
285 //--- Set field value at (0.,0.,0.) - some safety
286 //
287  double Bx,By,Bz;
288  state.m_fitField.getMagFld(-state.m_refFrameX,-state.m_refFrameY,-state.m_refFrameZ,Bx,By,Bz);
289 //
290 //------ Fit option setting
291 //
292  VKalVrtConfigureFitterCore(ntrk, state);
293 //
294 //------ Fit itself
295 //
296  state.m_FitStatus=0;
297  state.m_vkalFitControl.renewFullCovariance(nullptr); //
298  state.m_vkalFitControl.setVertexMass(-1.);
299  state.m_vkalFitControl.setVrtMassError(-1.);
300  if(state.m_ApproximateVertex.size()==3 && fabs(state.m_ApproximateVertex[2])<m_IDsizeZ &&
301  sqrt(state.m_ApproximateVertex[0]*state.m_ApproximateVertex[0]+state.m_ApproximateVertex[1]*state.m_ApproximateVertex[1])<m_IDsizeR)
302  {
303  xyz0[0]=(double)state.m_ApproximateVertex[0] - state.m_refFrameX;
304  xyz0[1]=(double)state.m_ApproximateVertex[1] - state.m_refFrameY;
305  xyz0[2]=(double)state.m_ApproximateVertex[2] - state.m_refFrameZ;
306  } else {
307  xyz0[0]=xyz0[1]=xyz0[2]=0.;
308  }
309  double par0[NTrMaxVFit][3]; //used only for fit preparation
310  Trk::cfpest( ntrk, xyz0, state.m_ich, state.m_apar, par0);
311 
312  Chi2PerTrk.resize (ntrk);
313  ierr=Trk::CFit( &state.m_vkalFitControl, ifCovV0, ntrk, state.m_ich, xyz0, par0, state.m_apar, state.m_awgt,
314  xyzfit, state.m_parfs, ptot, covf, chi2f,
315  Chi2PerTrk.data());
316 
317  if(msgLvl(MSG::DEBUG))msg(MSG::DEBUG) << "VKalVrt fit status="<<ierr<<" Chi2="<<chi2f<<endmsg;
318 
319  Chi2 = 100000000.;
320  if(ierr){
321  return ierr;
322  }
323  if(ptot[0]*ptot[0]+ptot[1]*ptot[1] == 0.) return -5; // Bad (divergent) fit
324 //
325 // Postfit operation. Creation of array for different error calculations and full error matrix copy
326 //
327  state.m_FitStatus=ntrk;
328  if(ifCovV0 && state.m_vkalFitControl.getFullCovariance()){ //If full fit error matrix is returned by VKalVrtCORE
329  int SymCovMtxSize=(3*ntrk+3)*(3*ntrk+4)/2;
330  state.m_ErrMtx.assign (state.m_vkalFitControl.getFullCovariance(),
331  state.m_vkalFitControl.getFullCovariance()+SymCovMtxSize);
332  state.m_vkalFitControl.renewFullCovariance(nullptr);
333  ErrorMatrix.clear(); ErrorMatrix.reserve(21); ErrorMatrix.assign(covf,covf+21);
334  } else {
335  ErrorMatrix.clear(); ErrorMatrix.reserve(6); ErrorMatrix.assign(covf,covf+6);
336  }
337 //---------------------------------------------------------------------------
338  Momentum.SetPxPyPzE( ptot[0], ptot[1], ptot[2], ptot[3] );
339  Chi2 = (double) chi2f;
340 
341  Vertex[0]= xyzfit[0] + state.m_refFrameX;
342  Vertex[1]= xyzfit[1] + state.m_refFrameY;
343  Vertex[2]= xyzfit[2] + state.m_refFrameZ;
344 
345  double sizeR = state.m_allowUltraDisplaced ? m_MSsizeR : m_IDsizeR;
346  double sizeZ = state.m_allowUltraDisplaced ? m_MSsizeZ : m_IDsizeZ;
347  if (Vertex.perp() > sizeR || std::abs(Vertex.z()) > sizeZ) return -5; // Solution outside acceptable volume due to divergence
348 
349  state.m_save_xyzfit[0]=xyzfit[0]; // saving of vertex position
350  state.m_save_xyzfit[1]=xyzfit[1]; // for full error matrix
351  state.m_save_xyzfit[2]=xyzfit[2];
352 //
353 // ------ Magnetic field in fitted vertex
354 //
355  double fx,fy,BMAG_CUR;
356  state.m_fitField.getMagFld(xyzfit[0] ,xyzfit[1] ,xyzfit[2] ,fx,fy,BMAG_CUR);
357  if(fabs(BMAG_CUR) < 0.01) BMAG_CUR=0.01; // Safety
358 
359  Charge=0; for(i=0; i<ntrk; i++){Charge+=state.m_ich[i];};
360  Charge=-Charge; //VK 30.11.2009 Change sign acoording to ATLAS
361 
362 
363  TrkAtVrt.clear(); TrkAtVrt.reserve(ntrk);
364  for(i=0; i<ntrk; i++){
365  std::vector<double> TrkPar(3);
366  VKalToTrkTrack(BMAG_CUR,(double)state.m_parfs[i][0],(double)state.m_parfs[i][1],(double) state.m_parfs[i][2],
367  TrkPar[0],TrkPar[1],TrkPar[2]);
368  TrkPar[2] = -TrkPar[2]; // Change of sign needed
369  TrkAtVrt.push_back( TrkPar );
370  }
371  return 0;
372  }

◆ VKalVrtFitFast() [1/3]

virtual StatusCode Trk::TrkVKalVrtFitter::VKalVrtFitFast ( const std::span< const xAOD::TrackParticle *const >  ,
Amg::Vector3D &  Vertex,
IVKalState &  istate 
) const
finaloverridevirtual

◆ VKalVrtFitFast() [2/3]

StatusCode Trk::TrkVKalVrtFitter::VKalVrtFitFast ( const std::vector< const TrackParameters * > &  InpTrk,
Amg::Vector3D &  Vertex,
IVKalState &  istate 
) const
finaloverridevirtual

Definition at line 107 of file VKalVrtFitFastSvc.cxx.

110  {
111  assert(dynamic_cast<State*> (&istate)!=nullptr);
112  State& state = static_cast<State&> (istate);
113 //
114 // Convert particles and setup reference frame
115 //
116  int ntrk=0;
117  StatusCode sc = CvtTrackParameters(InpTrk,ntrk,state);
118  if(sc.isFailure() || ntrk<1 ) return StatusCode::FAILURE;
119  double fx,fy,BMAG_CUR;
120  state.m_fitField.getMagFld(0.,0.,0.,fx,fy,BMAG_CUR);
121  if(fabs(BMAG_CUR) < 0.1) BMAG_CUR=0.1;
122 //
123 //------ Variables and arrays needed for fitting kernel
124 //
125  double out[3];
126  std::vector<double> xx,yy,zz;
127  Vertex[0]=Vertex[1]=Vertex[2]=0.;
128 //
129 //
130  double xyz0[3]={ -state.m_refFrameX, -state.m_refFrameY, -state.m_refFrameZ};
131  if(ntrk==2){
132  Trk::vkvFastV(&state.m_apar[0][0],&state.m_apar[1][0], xyz0, BMAG_CUR, out);
133  } else {
134  for(int i=0;i<ntrk-1; i++){
135  for(int j=i+1; j<ntrk; j++){
136  Trk::vkvFastV(&state.m_apar[i][0],&state.m_apar[j][0], xyz0, BMAG_CUR, out);
137  xx.push_back(out[0]);
138  yy.push_back(out[1]);
139  zz.push_back(out[2]);
140  }
141  }
142  out[0] = median(xx);
143  out[1] = median(yy);
144  out[2] = median(zz);
145 
146  }
147  Vertex[0]= out[0] + state.m_refFrameX;
148  Vertex[1]= out[1] + state.m_refFrameY;
149  Vertex[2]= out[2] + state.m_refFrameZ;
150 
151 
152  return StatusCode::SUCCESS;
153  }

◆ VKalVrtFitFast() [3/3]

StatusCode Trk::TrkVKalVrtFitter::VKalVrtFitFast ( std::span< const xAOD::TrackParticle *const >  InpTrk,
Amg::Vector3D &  Vertex,
double &  minDZ,
IVKalState &  istate 
) const
virtual

Definition at line 56 of file VKalVrtFitFastSvc.cxx.

59  {
60  assert(dynamic_cast<State*> (&istate)!=nullptr);
61  State& state = static_cast<State&> (istate);
62 //
63 // Convert particles and setup reference frame
64 //
65  int ntrk=0;
66  StatusCode sc = CvtTrackParticle(InpTrk,ntrk,state);
67  if(sc.isFailure() || ntrk<1 ) return StatusCode::FAILURE;
68  double fx,fy,BMAG_CUR;
69  state.m_fitField.getMagFld(0.,0.,0.,fx,fy,BMAG_CUR);
70  if(fabs(BMAG_CUR) < 0.1) BMAG_CUR=0.1;
71 //
72 //------ Variables and arrays needed for fitting kernel
73 //
74  double out[3];
75  std::vector<double> xx,yy,zz,difz;
76  Vertex[0]=Vertex[1]=Vertex[2]=0.;
77 //
78 //
79  double xyz0[3]={ -state.m_refFrameX, -state.m_refFrameY, -state.m_refFrameZ};
80  if(ntrk==2){
81  minDZ=Trk::vkvFastV(&state.m_apar[0][0],&state.m_apar[1][0], xyz0, BMAG_CUR, out);
82  } else {
83  for(int i=0;i<ntrk-1; i++){
84  for(int j=i+1; j<ntrk; j++){
85  double dZ=Trk::vkvFastV(&state.m_apar[i][0],&state.m_apar[j][0], xyz0, BMAG_CUR, out);
86  xx.push_back(out[0]);
87  yy.push_back(out[1]);
88  zz.push_back(out[2]);
89  difz.push_back(dZ);
90  }
91  }
92  out[0] = median(xx);
93  out[1] = median(yy);
94  out[2] = median(zz);
95  minDZ = median(difz);
96  }
97  Vertex[0]= out[0] + state.m_refFrameX;
98  Vertex[1]= out[1] + state.m_refFrameY;
99  Vertex[2]= out[2] + state.m_refFrameZ;
100 
101 
102  return StatusCode::SUCCESS;
103  }

Friends And Related Function Documentation

◆ VKalExtPropagator

friend class VKalExtPropagator
friend

Definition at line 68 of file TrkVKalVrtFitter.h.

Member Data Documentation

◆ m_allowUltraDisplaced

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_allowUltraDisplaced {this, "allowUltraDisplaced", false, "Allow ultra displaced vertices"}
private

Definition at line 358 of file TrkVKalVrtFitter.h.

◆ m_BMAG

double Trk::TrkVKalVrtFitter::m_BMAG {1.997}
private

Definition at line 475 of file TrkVKalVrtFitter.h.

◆ m_c_CovVrtForConstraint

Gaudi::Property<std::vector<double> > Trk::TrkVKalVrtFitter::m_c_CovVrtForConstraint {this, "CovVrtForConstraint", {0,0,0,0,0,0}}
private

Definition at line 331 of file TrkVKalVrtFitter.h.

◆ m_c_MassInputParticles

Gaudi::Property<std::vector<double> > Trk::TrkVKalVrtFitter::m_c_MassInputParticles {this, "InputParticleMasses", {}, "List of masses of input particles (pions assumed if absent)"}
private

Definition at line 332 of file TrkVKalVrtFitter.h.

◆ m_c_VertexForConstraint

Gaudi::Property<std::vector<double> > Trk::TrkVKalVrtFitter::m_c_VertexForConstraint {this, "VertexForConstraint", {0,0,0}}
private

Definition at line 330 of file TrkVKalVrtFitter.h.

◆ m_cascadeCnstPrecision

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_cascadeCnstPrecision {this, "CascadeCnstPrecision", 1.e-4}
private

Definition at line 322 of file TrkVKalVrtFitter.h.

◆ m_CNVMAG

double Trk::TrkVKalVrtFitter::m_CNVMAG {0.29979246}
private

Definition at line 476 of file TrkVKalVrtFitter.h.

◆ m_extPropagator

ToolHandle<IExtrapolator> Trk::TrkVKalVrtFitter::m_extPropagator {this, "Extrapolator", "", "External propagator"}
private

Definition at line 334 of file TrkVKalVrtFitter.h.

◆ m_fieldCacheCondObjInputKey

SG::ReadCondHandleKey<AtlasFieldCacheCondObj> Trk::TrkVKalVrtFitter::m_fieldCacheCondObjInputKey
private
Initial value:
{ this,
"AtlasFieldCacheCondObj",
"fieldCondObj",
"Name of the Magnetic Field key" }

Definition at line 337 of file TrkVKalVrtFitter.h.

◆ m_firstMeasuredPoint

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_firstMeasuredPoint {this, "FirstMeasuredPoint", false, "Use FirstMeasuredPoint strategy in fits"}
private

Definition at line 341 of file TrkVKalVrtFitter.h.

◆ m_firstMeasuredPointLimit

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_firstMeasuredPointLimit {this, "FirstMeasuredPointLimit", false, "Use FirstMeasuredPointLimit strategy"}
private

Definition at line 342 of file TrkVKalVrtFitter.h.

◆ m_firstMeasuredRadiusLimit

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_firstMeasuredRadiusLimit
private
Initial value:
{this, "FirstMeasuredRadiusLimit", false,
"Use radius of FirstMeasuredRadiusLimit as maximal vertex radius"}

Definition at line 343 of file TrkVKalVrtFitter.h.

◆ m_fitPropagator

VKalExtPropagator* Trk::TrkVKalVrtFitter::m_fitPropagator {}
private

Definition at line 479 of file TrkVKalVrtFitter.h.

◆ m_frozenVersionForBTagging

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_frozenVersionForBTagging {this, "FrozenVersionForBTagging", false, "Frozen version for BTagging"}
private

Definition at line 357 of file TrkVKalVrtFitter.h.

◆ m_IDsizeR

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_IDsizeR {this, "IDsizeR", 1150.0}
private

Definition at line 326 of file TrkVKalVrtFitter.h.

◆ m_IDsizeZ

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_IDsizeZ {this, "IDsizeZ", 3000.0}
private

Definition at line 327 of file TrkVKalVrtFitter.h.

◆ m_InDetExtrapolator

const IExtrapolator* Trk::TrkVKalVrtFitter::m_InDetExtrapolator {}
private

Pointer to Extrapolator AlgTool.

Definition at line 480 of file TrkVKalVrtFitter.h.

◆ m_isAtlasField

bool Trk::TrkVKalVrtFitter::m_isAtlasField {false}
private

Definition at line 348 of file TrkVKalVrtFitter.h.

◆ m_IterationNumber

Gaudi::Property<int> Trk::TrkVKalVrtFitter::m_IterationNumber {this, "IterationNumber", 0}
private

Definition at line 324 of file TrkVKalVrtFitter.h.

◆ m_IterationPrecision

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_IterationPrecision {this, "IterationPrecision", 0.0}
private

Definition at line 325 of file TrkVKalVrtFitter.h.

◆ m_makeExtendedVertex

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_makeExtendedVertex {this, "MakeExtendedVertex", false, "Return VxCandidate with full covariance matrix"}
private

Definition at line 345 of file TrkVKalVrtFitter.h.

◆ m_massForConstraint

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_massForConstraint {this, "MassForConstraint", -1.0}
private

Definition at line 323 of file TrkVKalVrtFitter.h.

◆ m_MSsizeR

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_MSsizeR {this, "MSsizeR", 8000.0}
private

Definition at line 328 of file TrkVKalVrtFitter.h.

◆ m_MSsizeZ

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_MSsizeZ {this, "MSsizeZ", 10000.0}
private

Definition at line 329 of file TrkVKalVrtFitter.h.

◆ m_Robustness

Gaudi::Property<int> Trk::TrkVKalVrtFitter::m_Robustness {this, "Robustness", 0}
private

Definition at line 320 of file TrkVKalVrtFitter.h.

◆ m_RobustScale

Gaudi::Property<double> Trk::TrkVKalVrtFitter::m_RobustScale {this, "RobustScale", 1.0}
private

Definition at line 321 of file TrkVKalVrtFitter.h.

◆ m_useAprioriVertex

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_useAprioriVertex {this, "useAprioriVertexCnst", false, "Use a priori vertex constraint"}
private

Definition at line 350 of file TrkVKalVrtFitter.h.

◆ m_useFixedField

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_useFixedField {this, "useFixedField", false, "Use fixed magnetic field instead of exact Atlas one"}
private

Definition at line 346 of file TrkVKalVrtFitter.h.

◆ m_usePassNear

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_usePassNear {this, "usePassNearCnst", false, "Use combined particle pass near other vertex constraint"}
private

Definition at line 355 of file TrkVKalVrtFitter.h.

◆ m_usePassWithTrkErr

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_usePassWithTrkErr {this, "usePassWithTrkErrCnst", false, "Use pass near with combined particle errors constraint"}
private

Definition at line 356 of file TrkVKalVrtFitter.h.

◆ m_usePhiCnst

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_usePhiCnst {this, "usePhiCnst", false, "Use angle dPhi=0 constraint"}
private

Definition at line 352 of file TrkVKalVrtFitter.h.

◆ m_usePointingCnst

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_usePointingCnst {this, "usePointingCnst", false, "Use pointing to other vertex constraint"}
private

Definition at line 353 of file TrkVKalVrtFitter.h.

◆ m_useThetaCnst

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_useThetaCnst {this, "useThetaCnst", false, "Use angle dTheta=0 constraint"}
private

Definition at line 351 of file TrkVKalVrtFitter.h.

◆ m_useZPointingCnst

Gaudi::Property<bool> Trk::TrkVKalVrtFitter::m_useZPointingCnst {this, "useZPointingCnst", false, "Use ZPointing to other vertex constraint"}
private

Definition at line 354 of file TrkVKalVrtFitter.h.

The documentation for this class was generated from the following files:
Trk::TrkVKalVrtFitter::m_IterationNumber
Gaudi::Property< int > m_IterationNumber
Definition: TrkVKalVrtFitter.h:324
Matrix
Definition: Trigger/TrigT1/TrigT1RPChardware/TrigT1RPChardware/Matrix.h:15
cmd-l1calo-dq-test.xx
xx
Definition: cmd-l1calo-dq-test.py:16
Trk::y
@ y
Definition: ParamDefs.h:56
Trk::TrkVKalVrtFitter::findPositions
static int findPositions(const std::vector< int > &, const std::vector< int > &, std::vector< int > &)
Definition: TrkCascadeFitter.cxx:787
Trk::TrkVKalVrtFitter::VKalVrtFit
virtual StatusCode VKalVrtFit(const std::vector< const xAOD::TrackParticle * > &, const std::vector< const xAOD::NeutralParticle * > &, Amg::Vector3D &Vertex, TLorentzVector &Momentum, long int &Charge, dvect &ErrorMatrix, dvect &Chi2PerTrk, std::vector< std::vector< double >> &TrkAtVrt, double &Chi2, IVKalState &istate, bool ifCovV0=false) const override final
xAOD::Vertex_v1::x
float x() const
Returns the x position.
Trk::py
@ py
Definition: ParamDefs.h:60
Trk::TrkVKalVrtFitter::m_c_CovVrtForConstraint
Gaudi::Property< std::vector< double > > m_c_CovVrtForConstraint
Definition: TrkVKalVrtFitter.h:331
Trk::TrkVKalVrtFitter::makeXAODVertex
xAOD::Vertex * makeXAODVertex(int, const Amg::Vector3D &, const dvect &, const dvect &, const std::vector< dvect > &, double, State &state) const
Definition: TrkVKalVrtFitter.cxx:665
Trk::VKContraintType::Theta
@ Theta
xAOD::Vertex_v1::setPosition
void setPosition(const Amg::Vector3D &position)
Sets the 3-position.
Trk::NTrMaxVFit
@ NTrMaxVFit
Definition: TrkVKalVrtFitter.h:35
CLEANCASCADE
#define CLEANCASCADE()
Definition: TrkCascadeFitter.cxx:272
xAOD::Vertex_v1::setFitQuality
void setFitQuality(float chiSquared, float numberDoF)
Set the 'Fit Quality' information.
Definition: Vertex_v1.cxx:150
Amg::MatrixX
Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic > MatrixX
Dynamic Matrix - dynamic allocation.
Definition: EventPrimitives.h:27
Trk::TrkVKalVrtFitter::setVertexForConstraint
virtual void setVertexForConstraint(const xAOD::Vertex &, IVKalState &istate) const override final
Definition: SetFitOptions.cxx:159
Trk::TrkVKalVrtFitter::setCovVrtForConstraint
virtual void setCovVrtForConstraint(double XX, double XY, double YY, double XZ, double YZ, double ZZ, IVKalState &istate) const override final
Definition: SetFitOptions.cxx:185
Trk::z
@ z
global position (cartesian)
Definition: ParamDefs.h:57
xAOD::Vertex
Vertex_v1 Vertex
Define the latest version of the vertex class.
Definition: Event/xAOD/xAODTracking/xAODTracking/Vertex.h:16
Trk::TrkVKalVrtFitter::initState
void initState(const EventContext &ctx, State &state) const
Definition: TrkVKalVrtFitter.cxx:164
SG::ReadCondHandle
Definition: ReadCondHandle.h:44
Trk::VertexID
int VertexID
Definition: IVertexCascadeFitter.h:23
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition: AthMsgStreamMacros.h:31
PlotCalibFromCool.yy
yy
Definition: PlotCalibFromCool.py:714
Trk::processCascadePV
int processCascadePV(CascadeEvent &cascadeEvent_, const double *primVrt, const double *primVrtCov)
Definition: CFitCascade.cxx:503
Trk::makeCascade
int makeCascade(VKalVrtControl &FitCONTROL, long int NTRK, const long int *ich, double *wm, double *inp_Trk5, double *inp_CovTrk5, const std::vector< std::vector< int > > &vertexDefinition, const std::vector< std::vector< int > > &cascadeDefinition, double definedCnstAccuracy)
Definition: CascadeDefinition.cxx:116
Trk::TrkVKalVrtFitter::m_InDetExtrapolator
const IExtrapolator * m_InDetExtrapolator
Pointer to Extrapolator AlgTool.
Definition: TrkVKalVrtFitter.h:480
AtlasFieldCacheCondObj
Definition: AtlasFieldCacheCondObj.h:19
Trk::TrkVKalVrtFitter::m_usePhiCnst
Gaudi::Property< bool > m_usePhiCnst
Definition: TrkVKalVrtFitter.h:352
correlationModel::FULL
@ FULL
Definition: AsgElectronEfficiencyCorrectionTool.cxx:49
Base_Fragment.mass
mass
Definition: Sherpa_i/share/common/Base_Fragment.py:59
Trk::SymIndex
int SymIndex(int it, int i, int j)
Definition: TrkCascadeFitter.cxx:271
Trk::TrkVKalVrtFitter::CreatePerigee
virtual std::unique_ptr< Trk::Perigee > CreatePerigee(const std::vector< double > &VKPerigee, const std::vector< double > &VKCov, IVKalState &istate) const override final
Definition: CvtPerigee.cxx:159
Trk::TrkVKalVrtFitter::VKalTransform
void VKalTransform(double MAG, double A0V, double ZV, double PhiV, double ThetaV, double PInv, const double[15], long int &Charge, double[5], double[15]) const
Definition: VKalTransform.cxx:59
Trk::TrkVKalVrtFitter::m_usePassWithTrkErr
Gaudi::Property< bool > m_usePassWithTrkErr
Definition: TrkVKalVrtFitter.h:356
Trk::ParametersBase::position
const Amg::Vector3D & position() const
Access method for the position.
index
Definition: index.py:1
Monitored::Z
@ Z
Definition: HistogramFillerUtils.h:24
max
constexpr double max()
Definition: ap_fixedTest.cxx:33
Trk::ParametersT
Dummy class used to allow special convertors to be called for surfaces owned by a detector element.
Definition: EMErrorDetail.h:25
Trk::TrkVKalVrtFitter::FillMatrixP
static void FillMatrixP(AmgSymMatrix(5)&, std::vector< double > &)
Definition: TrkVKalVrtFitter.cxx:614
Trk::TrkVKalVrtFitter::m_BMAG
double m_BMAG
Definition: TrkVKalVrtFitter.h:475
Trk::CurvilinearParameters
CurvilinearParametersT< TrackParametersDim, Charged, PlaneSurface > CurvilinearParameters
Definition: Tracking/TrkEvent/TrkParameters/TrkParameters/TrackParameters.h:29
Trk::TrkVKalVrtFitter::VKalVrtFitFast
virtual StatusCode VKalVrtFitFast(std::span< const xAOD::TrackParticle *const >, Amg::Vector3D &Vertex, double &minDZ, IVKalState &istate) const
Definition: VKalVrtFitFastSvc.cxx:56
min
constexpr double min()
Definition: ap_fixedTest.cxx:26
Trk::TrkVKalVrtFitter::m_isAtlasField
bool m_isAtlasField
Definition: TrkVKalVrtFitter.h:348
Trk::TrkVKalVrtFitter::makeSimpleCascade
static void makeSimpleCascade(std::vector< std::vector< int > > &, std::vector< std::vector< int > > &, CascadeState &cstate)
Definition: TrkCascadeFitter.cxx:180
CSV_InDetExporter.new
new
Definition: CSV_InDetExporter.py:145
PlotCalibFromCool.begin
begin
Definition: PlotCalibFromCool.py:94
Trk::TrkVKalVrtFitter::nextVertex
VertexID nextVertex(const std::vector< const xAOD::TrackParticle * > &list, std::span< const double > particleMass, IVKalState &istate, double massConstraint=0.) const override final
Definition: TrkCascadeFitter.cxx:97
skel.it
it
Definition: skel.GENtoEVGEN.py:407
Trk::TrkVKalVrtFitter::m_extPropagator
ToolHandle< IExtrapolator > m_extPropagator
Definition: TrkVKalVrtFitter.h:334
Trk::TrkVKalVrtFitter::CvtTrackParameters
StatusCode CvtTrackParameters(const std::vector< const TrackParameters * > &InpTrk, int &ntrk, State &state) const
Definition: CvtParametersBase.cxx:24
test_pyathena.pt
pt
Definition: test_pyathena.py:11
python.AthDsoLogger.out
out
Definition: AthDsoLogger.py:70
M_PI
#define M_PI
Definition: ActiveFraction.h:11
Trk::TrkVKalVrtFitter::m_fitPropagator
VKalExtPropagator * m_fitPropagator
Definition: TrkVKalVrtFitter.h:479
Trk::TrkVKalVrtFitter::m_IDsizeR
Gaudi::Property< double > m_IDsizeR
Definition: TrkVKalVrtFitter.h:326
Trk::TrkVKalVrtFitter::VKalGetImpact
virtual double VKalGetImpact(const xAOD::TrackParticle *, const Amg::Vector3D &Vertex, const long int Charge, dvect &Impact, dvect &ImpactError, IVKalState &istate) const override final
Definition: VKalGetImpact.cxx:91
InDetSecVtxTruthMatchUtils::isMerged
bool isMerged(int matchInfo)
Definition: InDetSecVtxTruthMatchTool.h:79
Trk::getFittedCascade
void getFittedCascade(CascadeEvent &cascadeEvent_, std::vector< Vect3DF > &cVertices, std::vector< std::vector< double > > &covVertices, std::vector< std::vector< VectMOM > > &fittedParticles, std::vector< std::vector< double > > &cascadeCovar, std::vector< double > &particleChi2, std::vector< double > &fullCovar)
Definition: CFitCascade.cxx:685
Trk::TrkVKalVrtFitter::VKalExtPropagator
friend class VKalExtPropagator
Definition: TrkVKalVrtFitter.h:68
Trk::TrkVKalVrtFitter::m_allowUltraDisplaced
Gaudi::Property< bool > m_allowUltraDisplaced
Definition: TrkVKalVrtFitter.h:358
InDet::median
float median(std::vector< float > &Vec)
Definition: BTagVrtSec.cxx:35
Trk::TrkVKalVrtFitter::CvtPerigee
StatusCode CvtPerigee(const std::vector< const Perigee * > &list, int &ntrk, State &state) const
Definition: CvtPerigee.cxx:25
xAOD::Vertex_v1::position
const Amg::Vector3D & position() const
Returns the 3-pos.
Trk::VKalExtPropagator::myExtrapWithMatUpdate
const TrackParameters * myExtrapWithMatUpdate(long int TrkID, const TrackParameters *inpPer, Amg::Vector3D *endPoint, const IVKalState &istate) const
Definition: VKalExtPropagator.cxx:202
Phi
@ Phi
Definition: RPCdef.h:8
ParticleJetParams::kt
@ kt
Definition: ParticleJetParamDefs.h:43
Trk::TrkVKalVrtFitter::m_MSsizeZ
Gaudi::Property< double > m_MSsizeZ
Definition: TrkVKalVrtFitter.h:329
Sign
int Sign(int in)
Definition: JSSTaggerUtils.cxx:23
Trk::cfimp
void cfimp(long int TrkID, long int ich, int IFL, double *par, const double *err, double *vrt, double *vcov, double *rimp, double *rcov, double *sign, VKalVrtControlBase *FitCONTROL)
Definition: cfImp.cxx:43
drawFromPickle.cos
cos
Definition: drawFromPickle.py:36
Trk::TrkVKalVrtFitter::m_useAprioriVertex
Gaudi::Property< bool > m_useAprioriVertex
Definition: TrkVKalVrtFitter.h:350
Trk::TrkVKalVrtFitter::getCascadeNDoF
static int getCascadeNDoF(const CascadeState &cstate)
Definition: TrkCascadeFitter.cxx:79
SG::VarHandleKey::key
const std::string & key() const
Return the StoreGate ID for the referenced object.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:141
const
bool const RAWDATA *ch2 const
Definition: LArRodBlockPhysicsV0.cxx:560
Trk::setCascadeMassConstraint
int setCascadeMassConstraint(CascadeEvent &cascadeEvent_, long int IV, double Mass)
Definition: CascadeDefinition.cxx:241
Trk::pz
@ pz
global momentum (cartesian)
Definition: ParamDefs.h:61
Trk::Perigee
ParametersT< TrackParametersDim, Charged, PerigeeSurface > Perigee
Definition: Tracking/TrkEvent/TrkParameters/TrkParameters/TrackParameters.h:33
Trk::TrkVKalVrtFitter::convertAmg5SymMtx
bool convertAmg5SymMtx(const AmgSymMatrix(5) *, double[15]) const
Definition: VKalTransform.cxx:26
xAOD::Vertex_v1::setCovariance
void setCovariance(const std::vector< float > &value)
Sets the covariance matrix as a simple vector of values.
Trk::TrkVKalVrtFitter::VKalGetNDOF
static int VKalGetNDOF(const State &state)
Definition: VKalVrtFitSvc.cxx:578
Trk::TrkVKalVrtFitter::m_Robustness
Gaudi::Property< int > m_Robustness
Definition: TrkVKalVrtFitter.h:320
Trk::TrkVKalVrtFitter::indexInV
static int indexInV(const VertexID &, const CascadeState &cstate)
Definition: TrkCascadeFitter.cxx:823
Trk::TrkVKalVrtFitter::VKalGetFullCov
virtual StatusCode VKalGetFullCov(long int, dvect &CovMtx, IVKalState &istate, bool=false) const override final
Definition: VKalVrtFitSvc.cxx:450
python.CaloAddPedShiftConfig.type
type
Definition: CaloAddPedShiftConfig.py:42
Trk::TrkVKalVrtFitter::m_MSsizeR
Gaudi::Property< double > m_MSsizeR
Definition: TrkVKalVrtFitter.h:328
pi
#define pi
Definition: TileMuonFitter.cxx:65
AthenaPoolTestRead.sc
sc
Definition: AthenaPoolTestRead.py:27
similarity
Matrix< Scalar, OtherDerived::RowsAtCompileTime, OtherDerived::RowsAtCompileTime > similarity(const MatrixBase< OtherDerived > &m) const
similarity method : yields ms = m*s*m^T
Definition: AmgMatrixBasePlugin.h:133
mergePhysValFiles.end
end
Definition: DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py:92
Monitored::X
@ X
Definition: HistogramFillerUtils.h:24
Trk::processCascade
int processCascade(CascadeEvent &cascadeEvent_)
Definition: CFitCascade.cxx:244
ParticleConstants::PDG2011::chargedPionMassInMeV
constexpr double chargedPionMassInMeV
the mass of the charged pion (in MeV)
Definition: ParticleConstants.h:41
Trk::AmgSymMatrix
AmgSymMatrix(5) &GXFTrackState
Definition: GXFTrackState.h:156
python.setupRTTAlg.size
int size
Definition: setupRTTAlg.py:39
python.changerun.kk
list kk
Definition: changerun.py:39
Trk::vkalPropagator::Propagate
static void Propagate(long int TrkID, long int Charge, double *ParOld, double *CovOld, double *RefStart, double *RefEnd, double *ParNew, double *CovNew, VKalVrtControlBase *FitControl=0)
Definition: Propagator.cxx:127
xAOD::Vertex_v1::addTrackAtVertex
void addTrackAtVertex(const ElementLink< TrackParticleContainer > &tr, float weight=1.0)
Add a new track to the vertex.
Definition: Vertex_v1.cxx:314
Trk::TrkVKalVrtFitter::m_CNVMAG
double m_CNVMAG
Definition: TrkVKalVrtFitter.h:476
ParticleGun_EoverP_Config.mom
mom
Definition: ParticleGun_EoverP_Config.py:63
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition: AthMsgStreamMacros.h:33
ParticleGun_EoverP_Config.momentum
momentum
Definition: ParticleGun_EoverP_Config.py:63
Trk::xyztrp
void xyztrp(const long int ich, double *vrt0, double *pv0, double *covi, double BMAG, double *paro, double *errt)
Definition: XYZtrp.cxx:16
Trk::TrkVKalVrtFitter::setAthenaPropagator
void setAthenaPropagator(const Trk::IExtrapolator *)
Definition: VKalExtPropagator.cxx:477
lumiFormat.i
int i
Definition: lumiFormat.py:85
Trk::TrkVKalVrtFitter::m_cascadeCnstPrecision
Gaudi::Property< double > m_cascadeCnstPrecision
Definition: TrkVKalVrtFitter.h:322
Trk::TrkVKalVrtFitter::printSimpleCascade
static void printSimpleCascade(std::vector< std::vector< int > > &, std::vector< std::vector< int > > &, const CascadeState &cstate)
Definition: TrkCascadeFitter.cxx:237
Trk::TrkVKalVrtFitter::setApproximateVertex
virtual void setApproximateVertex(double X, double Y, double Z, IVKalState &istate) const override final
Definition: SetFitOptions.cxx:115
Trk::theta
@ theta
Definition: ParamDefs.h:66
xAOD::FirstMeasurement
@ FirstMeasurement
Parameter defined at the position of the 1st measurement.
Definition: TrackingPrimitives.h:214
Trk::TrkVKalVrtFitter::m_c_MassInputParticles
Gaudi::Property< std::vector< double > > m_c_MassInputParticles
Definition: TrkVKalVrtFitter.h:332
endmsg
#define endmsg
Definition: AnalysisConfig_Ntuple.cxx:63
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
vector
Definition: MultiHisto.h:13
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition: AthMsgStreamMacros.h:29
Trk::TrkVKalVrtFitter::m_IterationPrecision
Gaudi::Property< double > m_IterationPrecision
Definition: TrkVKalVrtFitter.h:325
AmgVector
AmgVector(4) T2BSTrackFilterTool
Definition: T2BSTrackFilterTool.cxx:114
Trk::NeutralPerigee
ParametersT< NeutralParametersDim, Neutral, PerigeeSurface > NeutralPerigee
Definition: NeutralParameters.h:31
MuonR4::State
CalibratedSpacePoint::State State
Definition: SpacePointCalibrator.cxx:33
Trk::px
@ px
Definition: ParamDefs.h:59
test_pyathena.parent
parent
Definition: test_pyathena.py:15
find_tgc_unfilled_channelids.ip
ip
Definition: find_tgc_unfilled_channelids.py:3
Trk::vkvFastV
double vkvFastV(double *p1, double *p2, const double *vRef, double dbmag, double *out)
Definition: VKvFast.cxx:42
histSizes.list
def list(name, path='/')
Definition: histSizes.py:38
beamspotnt.rows
list rows
Definition: bin/beamspotnt.py:1111
drawFromPickle.tan
tan
Definition: drawFromPickle.py:36
Trk::TrkVKalVrtFitter::m_usePointingCnst
Gaudi::Property< bool > m_usePointingCnst
Definition: TrkVKalVrtFitter.h:353
AnalysisUtils::Delta::R
double R(const INavigable4Momentum *p1, const double v_eta, const double v_phi)
Definition: AnalysisMisc.h:49
Trk::ParametersBase
Definition: ParametersBase.h:55
xAOD::Vertex_v1::z
float z() const
Returns the z position.
xAOD::double
double
Definition: CompositeParticle_v1.cxx:159
Scale
void Scale(TH1 *h, double d=1)
Definition: comparitor.cxx:79
DeMoUpdate.tmp
string tmp
Definition: DeMoUpdate.py:1167
Trk::TrkVKalVrtFitter::m_firstMeasuredRadiusLimit
Gaudi::Property< bool > m_firstMeasuredRadiusLimit
Definition: TrkVKalVrtFitter.h:343
Trk::cfpest
void cfpest(int ntrk, double *xyz, long int *ich, double(*parst)[5], double(*parf)[3])
Definition: cfPEst.cxx:10
dvect
std::vector< double > dvect
Definition: InDetVKalVxInJetTool.h:83
calibdata.ct
ct
Definition: calibdata.py:417
Trk::TrkVKalVrtFitter::m_usePassNear
Gaudi::Property< bool > m_usePassNear
Definition: TrkVKalVrtFitter.h:355
grepfile.ic
int ic
Definition: grepfile.py:33
Monitored::Y
@ Y
Definition: HistogramFillerUtils.h:24
TYPE
#define TYPE(CODE, TYP, IOTYP)
Trk
Ensure that the ATLAS eigen extensions are properly loaded.
Definition: FakeTrackBuilder.h:9
Trk::SurfaceType::Perigee
@ Perigee
Amg
Definition of ATLAS Math & Geometry primitives (Amg)
Definition: AmgStringHelpers.h:19
Trk::TrkVKalVrtFitter::VKalToTrkTrack
void VKalToTrkTrack(double curBMAG, double vp1, double vp2, double vp3, double &tp1, double &tp2, double &tp3) const
Definition: VKalVrtFitSvc.cxx:422
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:240
Trk::TrkVKalVrtFitter::getSimpleVIndex
static int getSimpleVIndex(const VertexID &, const CascadeState &cstate)
Definition: TrkCascadeFitter.cxx:804
Trk::TrkVKalVrtFitter::m_useThetaCnst
Gaudi::Property< bool > m_useThetaCnst
Definition: TrkVKalVrtFitter.h:351
VP1PartSpect::E
@ E
Definition: VP1PartSpectFlags.h:21
Trk::VKalExtPropagator::setPropagator
void setPropagator(const IExtrapolator *)
Definition: VKalExtPropagator.cxx:41
Trk::TrkVKalVrtFitter::m_massForConstraint
Gaudi::Property< double > m_massForConstraint
Definition: TrkVKalVrtFitter.h:323
Trk::TrackParameters
ParametersBase< TrackParametersDim, Charged > TrackParameters
Definition: Tracking/TrkEvent/TrkParameters/TrkParameters/TrackParameters.h:27
Trk::TrkVKalVrtFitter::m_fieldCacheCondObjInputKey
SG::ReadCondHandleKey< AtlasFieldCacheCondObj > m_fieldCacheCondObjInputKey
Definition: TrkVKalVrtFitter.h:337
charge
double charge(const T &p)
Definition: AtlasPID.h:986
SG::CondHandleKey::initialize
StatusCode initialize(bool used=true)
Trk::TrkVKalVrtFitter::m_frozenVersionForBTagging
Gaudi::Property< bool > m_frozenVersionForBTagging
Definition: TrkVKalVrtFitter.h:357
SG::AuxElement::makePrivateStore
void makePrivateStore()
Create a new (empty) private store for this object.
Definition: AuxElement.cxx:192
Amg::Vector3D
Eigen::Matrix< double, 3, 1 > Vector3D
Definition: GeoPrimitives.h:47
Trk::TrkVKalVrtFitter::VKalGetTrkWeights
virtual StatusCode VKalGetTrkWeights(dvect &Weights, const IVKalState &istate) const override final
Definition: VKalVrtFitSvc.cxx:562
Trk::VKalExtPropagator::myExtrapNeutral
const NeutralParameters * myExtrapNeutral(const NeutralParameters *inpPer, Amg::Vector3D *endPoint) const
Definition: VKalExtPropagator.cxx:420
Trk::vertex
@ vertex
Definition: MeasurementType.h:21
DeMoScan.index
string index
Definition: DeMoScan.py:362
xAOD::Vertex_v1::addNeutralAtVertex
void addNeutralAtVertex(const ElementLink< NeutralParticleContainer > &tr, float weight=1.0)
Add a new neutral to the vertex.
Definition: Vertex_v1.cxx:323
Prompt::Def::Pt
@ Pt
Definition: VarHolder.h:76
columnar::empty
bool empty() const noexcept
Definition: ObjectRange.h:163
Trk::TrkVKalVrtFitter::m_firstMeasuredPointLimit
Gaudi::Property< bool > m_firstMeasuredPointLimit
Definition: TrkVKalVrtFitter.h:342
VKalVrtAthena::varHolder_detail::clear
void clear(T &var)
Definition: NtupleVars.h:48
python.CaloAddPedShiftConfig.int
int
Definition: CaloAddPedShiftConfig.py:45
xAOD::Vertex_v1
Class describing a Vertex.
Definition: Vertex_v1.h:42
lumiFormat.fill
fill
Definition: lumiFormat.py:104
DEBUG
#define DEBUG
Definition: page_access.h:11
Trk::TrkVKalVrtFitter::m_useFixedField
Gaudi::Property< bool > m_useFixedField
Definition: TrkVKalVrtFitter.h:346
xAOD::Vertex_v1::y
float y() const
Returns the y position.
if
if(febId1==febId2)
Definition: LArRodBlockPhysicsV0.cxx:567
LArNewCalib_DelayDump_OFC_Cali.idx
idx
Definition: LArNewCalib_DelayDump_OFC_Cali.py:69
Trk::TrkVKalVrtFitter::VKalVrtConfigureFitterCore
void VKalVrtConfigureFitterCore(int NTRK, State &state) const
Definition: SetFitOptions.cxx:18
Trk::TrkVKalVrtFitter::m_useZPointingCnst
Gaudi::Property< bool > m_useZPointingCnst
Definition: TrkVKalVrtFitter.h:354
run_AODTCCLinking.inputList
list inputList
Definition: run_AODTCCLinking.py:93
physics_parameters.parameters
parameters
Definition: physics_parameters.py:144
Trk::TrkVKalVrtFitter::m_RobustScale
Gaudi::Property< double > m_RobustScale
Definition: TrkVKalVrtFitter.h:321
Trk::phi
@ phi
Definition: ParamDefs.h:75
Trk::CFit
int CFit(VKalVrtControl *FitCONTROL, int ifCovV0, int NTRK, long int *ich, double xyz0[3], double(*par0)[3], double(*inp_Trk5)[5], double(*inp_CovTrk5)[15], double xyzfit[3], double(*parfs)[3], double ptot[4], double covf[21], double &chi2, double *chi2tr)
Definition: CFit.cxx:435
Trk::TrkVKalVrtFitter::fit
virtual xAOD::Vertex * fit(const std::vector< const TrackParameters * > &perigeeList, const Amg::Vector3D &startingPoint) const override final
Interface for MeasuredPerigee with starting point.
Definition: TrkVKalVrtFitter.cxx:203
I
#define I(x, y, z)
Definition: MD5.cxx:116
drawFromPickle.sin
sin
Definition: drawFromPickle.py:36
xAOD::Vertex_v1::vxTrackAtVertex
std::vector< Trk::VxTrackAtVertex > & vxTrackAtVertex()
Non-const access to the VxTrackAtVertex vector.
Definition: Vertex_v1.cxx:181
Trk::x
@ x
Definition: ParamDefs.h:55
Trk::TrkVKalVrtFitter::CvtTrackParticle
StatusCode CvtTrackParticle(std::span< const xAOD::TrackParticle *const > list, int &ntrk, State &state) const
Definition: CvtTrackParticle.cxx:27
test_pyathena.counter
counter
Definition: test_pyathena.py:15
Trk::TrkVKalVrtFitter::m_firstMeasuredPoint
Gaudi::Property< bool > m_firstMeasuredPoint
Definition: TrkVKalVrtFitter.h:341
python.compressB64.c
def c
Definition: compressB64.py:93
Trk::TrkVKalVrtFitter::m_c_VertexForConstraint
Gaudi::Property< std::vector< double > > m_c_VertexForConstraint
Definition: TrkVKalVrtFitter.h:330
python.AutoConfigFlags.msg
msg
Definition: AutoConfigFlags.py:7
Trk::TrkVKalVrtFitter::m_makeExtendedVertex
Gaudi::Property< bool > m_makeExtendedVertex
Definition: TrkVKalVrtFitter.h:345
Trk::v
@ v
Definition: ParamDefs.h:78
python.SystemOfUnits.m
float m
Definition: SystemOfUnits.py:106
generate::Zero
void Zero(TH1D *hin)
Definition: generate.cxx:32
Trk::TrkVKalVrtFitter::m_IDsizeZ
Gaudi::Property< double > m_IDsizeZ
Definition: TrkVKalVrtFitter.h:327
Generated on Sun Aug 31 2025 22:11:12 for ATLAS Offline Software by   doxygen 1.8.18