#include <JpsiExample.h>

Inheritance diagram for JpsiExample:

Collaboration diagram for JpsiExample:

Public Member Functions
	JpsiExample (const std::string &name, ISvcLocator *pSvcLocator)
StatusCode	initialize () override
StatusCode	execute () override
StatusCode	finalize () override
virtual StatusCode	sysInitialize () override
	Override sysInitialize.
virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies.
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)
void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed.

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
void	initializeBranches (void)
void	clearBranches (void)
TVector3	trackMomentum (const xAOD::Vertex *vxCandidate, uint trkIndex) const
TLorentzVector	track4Momentum (const xAOD::Vertex *vxCandidate, int trkIndex, double mass) const
TVector3	origTrackMomentum (const xAOD::Vertex *vxCandidate, int trkIndex) const
TLorentzVector	origTrack4Momentum (const xAOD::Vertex *vxCandidate, int trkIndex, double mass) const
double	invariantMassError (const xAOD::Vertex *vxCandidate, const std::vector< double > &masses) const
double	massErrorVKalVrt (const xAOD::Vertex *vxCandidate, const std::vector< double > &masses) const
double	trackCharge (const xAOD::Vertex *vxCandidate, int i) const
Amg::MatrixX *	convertVKalCovMatrix (int NTrk, const std::vector< float > &Matrix) const
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
double	m_muonMass
std::string	m_userFName
int	m_eventCntr
int	m_jpsiCntr
std::vector< double > *	m_jpsiMass
std::vector< double > *	m_jpsiMassError
std::vector< double > *	m_jpsiMassRec
std::vector< double > *	m_jpsiMassPullRec
std::vector< double > *	m_jpsiMassPullMC
std::vector< double > *	m_jpsiChi2
std::vector< double > *	m_trkRefitPx1
std::vector< double > *	m_trkRefitPy1
std::vector< double > *	m_trkRefitPz1
std::vector< double > *	m_trkRefitPx2
std::vector< double > *	m_trkRefitPy2
std::vector< double > *	m_trkRefitPz2
std::vector< double > *	m_vx
std::vector< double > *	m_vy
std::vector< double > *	m_vz
std::vector< double > *	m_trkOrigCharge1
std::vector< double > *	m_trkOrigPx1
std::vector< double > *	m_trkOrigPy1
std::vector< double > *	m_trkOrigPz1
std::vector< double > *	m_trkOrigCharge2
std::vector< double > *	m_trkOrigPx2
std::vector< double > *	m_trkOrigPy2
std::vector< double > *	m_trkOrigPz2
TFile *	m_outputFile
TTree *	m_auxTree
SG::ReadHandleKey< xAOD::VertexContainer >	m_JpsiCandidatesKey
	Name of J/psi container.
SG::ReadDecorHandleKey< xAOD::VertexContainer >	m_refPX
SG::ReadDecorHandleKey< xAOD::VertexContainer >	m_refPY
SG::ReadDecorHandleKey< xAOD::VertexContainer >	m_refPZ
DataObjIDColl	m_extendedExtraObjects
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default)
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 22 of file JpsiExample.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Algorithm > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ JpsiExample()

JpsiExample::JpsiExample	(	const std::string &	name,
		ISvcLocator *	pSvcLocator )

Definition at line 29 of file JpsiExample.cxx.

                                                                        :
  AthAlgorithm(name, pSvcLocator),
  m_muonMass(ParticleConstants::muonMassInMeV)
{
  
  // Declare user-defined properties - cuts and vertexing method
  declareProperty("muonMass"        , m_muonMass);
  declareProperty("outputNTupleName", m_userFName);
  declareProperty("JpsiCandidates"  ,m_JpsiCandidatesKey = "JpsiCandidates");
 
  declareProperty("RefTrackPx", m_refPX);
  declareProperty("RefTrackPy", m_refPY);
  declareProperty("RefTrackPz", m_refPZ);
 
  // Global Counters; for truth statistics
  m_eventCntr = 0;
  m_jpsiCntr = 0;
 
  m_jpsiMass = 0;
  m_jpsiMassError = 0;
  m_jpsiMassRec = 0;
  m_jpsiMassPullRec = 0;
  m_jpsiMassPullMC = 0;
  m_jpsiChi2 = 0;
 
  m_trkRefitPx1 = 0;
  m_trkRefitPy1 = 0;
  m_trkRefitPz1 = 0;
  m_trkRefitPx2 = 0;
  m_trkRefitPy2 = 0;
  m_trkRefitPz2 = 0;
 
  m_vx = 0;
  m_vy = 0;
  m_vz = 0;
 
  m_trkOrigCharge1 = 0;
  m_trkOrigPx1 = 0;
  m_trkOrigPy1 = 0;
  m_trkOrigPz1 = 0;
  m_trkOrigCharge2 = 0;
  m_trkOrigPx2 = 0;
  m_trkOrigPy2 = 0;
  m_trkOrigPz2 = 0;
 
  m_outputFile = 0;
  m_auxTree = 0;
}

Member Function Documentation

◆ clearBranches()

void JpsiExample::clearBranches ( void )

private

Definition at line 230 of file JpsiExample.cxx.

                                {
 
  m_jpsiMass->clear();
  m_jpsiMassError->clear();
  m_jpsiMassRec->clear();
  m_jpsiMassPullRec->clear();
  m_jpsiMassPullMC->clear();
  m_jpsiChi2->clear();
 
//  m_trkRefitCharge1->clear();
  m_trkRefitPx1->clear();
  m_trkRefitPy1->clear();
  m_trkRefitPz1->clear();
//  m_trkRefitCharge2->clear();
  m_trkRefitPx2->clear();
  m_trkRefitPy2->clear();
  m_trkRefitPz2->clear();
 
  m_vx->clear();
  m_vy->clear();
  m_vz->clear();
 
  m_trkOrigCharge1->clear();
  m_trkOrigPx1->clear();
  m_trkOrigPy1->clear();
  m_trkOrigPz1->clear();
  m_trkOrigCharge2->clear();
  m_trkOrigPx2->clear();
  m_trkOrigPy2->clear();
  m_trkOrigPz2->clear();
 
//  m_rxyError->clear();
 
  return;
 
}

◆ convertVKalCovMatrix()

Amg::MatrixX * JpsiExample::convertVKalCovMatrix	(	int	NTrk,
		const std::vector< float > &	Matrix ) const

private

Definition at line 497 of file JpsiExample.cxx.

{
  // stolen from VKalVrtFitter    
  Amg::MatrixX * mtx = new Amg::MatrixX(3+3*NTrk,3+3*NTrk);
  long int ij=0;
  for(int i=1; i<=(3+3*NTrk); i++){
    for(int j=1; j<=i; j++){
      if(i==j) { 
        (*mtx)(i-1,j-1)=Matrix[ij];
      }else{ 
        (*mtx).fillSymmetric(i-1,j-1,Matrix[ij]);
      }
      ij++;
    }
  }
  
  return mtx;
}

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty	(	Gaudi::Property< T, V, H > &	hndl,
		const SG::VarHandleKeyType &	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

  {
    return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                       hndl.value(), 
                                                       hndl.documentation());
 
  }

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                     {
    typedef typename SG::HandleClassifier<T>::type htype;
    return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
  }

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Algorithm > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

95{ return m_detStore; }

AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

StoreGateSvc_t m_detStore

Definition AthCommonDataStore.h:393

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Algorithm > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

85{ return m_evtStore; }

AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

StoreGateSvc_t m_evtStore

Definition AthCommonDataStore.h:390

◆ execute()

StatusCode JpsiExample::execute ( )

override

Definition at line 128 of file JpsiExample.cxx.

                                {
 
 
  ATH_MSG_DEBUG("in execute()");
 
  // Read in the Jpsis from StoreGate 
  const xAOD::VertexContainer*    jpsiContainer(0);
  SG::ReadHandle<xAOD::VertexContainer> handle(m_JpsiCandidatesKey);
  jpsiContainer = handle.cptr();
 
  if (!jpsiContainer) {
    ATH_MSG_ERROR("No Jpsi Container Found, skipping event");
    return StatusCode::RECOVERABLE;
  } else {
    ATH_MSG_DEBUG("Obtained jpsiContainer");
  }
  ATH_MSG_DEBUG("number of jpsi candidates " << jpsiContainer->size());
  m_jpsiCntr += jpsiContainer->size(); // Count the Jpsis
 
 
  // Set the auxilliary branches of the ntuple to zero
  // to avoid accumulation 
  clearBranches();
 
  ++m_eventCntr;     // Increment event counter
 
  // Extract information from the Jpsi candidates
  for (const xAOD::Vertex* jpsiCandidate : *jpsiContainer) {
    //ATH_MSG_INFO("Cov. matrix " << jpsiCandidate->covariance().size());
 
    // refitted track parameters
    TLorentzVector refTrk1 = track4Momentum(jpsiCandidate, 0, m_muonMass);
    m_trkRefitPx1->push_back(refTrk1.Px());
    m_trkRefitPy1->push_back(refTrk1.Py());
    m_trkRefitPz1->push_back(refTrk1.Pz());
 
    TLorentzVector refTrk2 = track4Momentum(jpsiCandidate, 1, m_muonMass);
    m_trkRefitPx2->push_back(refTrk2.Px());
    m_trkRefitPy2->push_back(refTrk2.Py());
    m_trkRefitPz2->push_back(refTrk2.Pz());
 
    //vertex position
    m_vx->push_back(jpsiCandidate->x());
    m_vy->push_back(jpsiCandidate->y());
    m_vz->push_back(jpsiCandidate->z());
 
    TLorentzVector origTrk1 = origTrack4Momentum(jpsiCandidate, 0, m_muonMass);
    m_trkOrigCharge1->push_back(trackCharge(jpsiCandidate, 0));
    m_trkOrigPx1->push_back(origTrk1.Px());
    m_trkOrigPy1->push_back(origTrk1.Py());
    m_trkOrigPz1->push_back(origTrk1.Pz());
 
    TLorentzVector origTrk2 = origTrack4Momentum(jpsiCandidate, 1, m_muonMass);
    m_trkOrigCharge2->push_back(trackCharge(jpsiCandidate, 1));
    m_trkOrigPx2->push_back(origTrk2.Px());
    m_trkOrigPy2->push_back(origTrk2.Py());
    m_trkOrigPz2->push_back(origTrk2.Pz());
    
    // Calculate the invariant masses and their errors
    double orig_mass = (origTrk1+origTrk2).M();                      // mass from original tracks
    double mass      = (refTrk1+refTrk2).M();                        // mass from refitted tracks
    double error     = invariantMassError(jpsiCandidate, std::vector<double>(2, m_muonMass));  // invariant mass error
 
    m_jpsiMass->push_back(mass);
    m_jpsiMassError->push_back(error);
    m_jpsiChi2->push_back(jpsiCandidate->chiSquared());
    m_jpsiMassRec->push_back(orig_mass);
    m_jpsiMassPullRec->push_back((mass-orig_mass)/error);
    m_jpsiMassPullMC->push_back((mass-ParticleConstants::JpsiMassInMeV)/error);
    
  }
     
  // Write all data to the nTuple
  m_auxTree->Fill();
 
  // END OF ANALYSIS
  return StatusCode::SUCCESS;
}

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.cxx.

{
  // If we didn't find any symlinks to add, just return the collection
  // from the base class.  Otherwise, return the extended collection.
  if (!m_extendedExtraObjects.empty()) {
    return m_extendedExtraObjects;
  }
  return Algorithm::extraOutputDeps();
}

◆ finalize()

StatusCode JpsiExample::finalize ( )

override

Definition at line 209 of file JpsiExample.cxx.

                                 {
 
  ATH_MSG_DEBUG("in finalize()");
 
  std::cout << "===================" << std::endl;
  std::cout << "SUMMARY OF ANALYSIS" << std::endl;
  std::cout << "===================" << std::endl;
  std::cout << " " << std::endl;
  std::cout << "Total number of events analysed: " << m_eventCntr << std::endl;
  std::cout << "Total number of jpsi candidates: " << m_jpsiCntr << std::endl;
  // Save auxiliary n-tuple
  m_outputFile->cd();
  m_auxTree->Write();
  m_auxTree->Print(); 
  // Close file
  m_outputFile->Close();
  return StatusCode::SUCCESS;
}

◆ initialize()

StatusCode JpsiExample::initialize ( )

override

Definition at line 80 of file JpsiExample.cxx.

                                  {
 
 
  ATH_MSG_DEBUG("in initialize()");
 
  // Open n-tuple file
  m_outputFile = new TFile(m_userFName.c_str(),"recreate");
  // Auxiliary n-tuples not covered by the automatic n-tuple makers
  // Decay angles
  initializeBranches(); // 
  m_auxTree = new TTree("AUX","AUX");
  m_auxTree->Branch("jpsiMass", &m_jpsiMass);
  m_auxTree->Branch("jpsiMassError", &m_jpsiMassError);
  m_auxTree->Branch("jpsiMassRec", &m_jpsiMassRec);
  m_auxTree->Branch("jpsiMassPullRec", &m_jpsiMassPullRec);
  m_auxTree->Branch("jpsiMassPullMC", &m_jpsiMassPullMC);
  m_auxTree->Branch("jpsiChi2", &m_jpsiChi2);
  
//  m_auxTree->Branch("trkRefitCharge1",  &m_trkRefitCharge1);
  m_auxTree->Branch("trkRefitPx1",  &m_trkRefitPx1);
  m_auxTree->Branch("trkRefitPy1", &m_trkRefitPy1);
  m_auxTree->Branch("trkRefitPz1", &m_trkRefitPz1);
//  m_auxTree->Branch("trkRefitCharge2",  &m_trkRefitCharge2);
  m_auxTree->Branch("trkRefitPx2",  &m_trkRefitPx2);
  m_auxTree->Branch("trkRefitPy2", &m_trkRefitPy2);
  m_auxTree->Branch("trkRefitPz2", &m_trkRefitPz2);
 
  m_auxTree->Branch("vx",m_vx);
  m_auxTree->Branch("vy",m_vy);
  m_auxTree->Branch("vz",m_vz);
 
  m_auxTree->Branch("trkOrigCharge1",  &m_trkOrigCharge1);
  m_auxTree->Branch("trkOrigPx1",  &m_trkOrigPx1);
  m_auxTree->Branch("trkOrigPy1", &m_trkOrigPy1);
  m_auxTree->Branch("trkOrigPz1", &m_trkOrigPz1);
  m_auxTree->Branch("trkOrigCharge2",  &m_trkOrigCharge2);
  m_auxTree->Branch("trkOrigPx2",  &m_trkOrigPx2);
  m_auxTree->Branch("trkOrigPy2", &m_trkOrigPy2);
  m_auxTree->Branch("trkOrigPz2", &m_trkOrigPz2);
 
//  m_auxTree->Branch("rxyError", &m_rxyError);
  ATH_CHECK(m_JpsiCandidatesKey.initialize());
  return StatusCode::SUCCESS;
  
}

◆ initializeBranches()

void JpsiExample::initializeBranches ( void )

private

Definition at line 272 of file JpsiExample.cxx.

                                         {
 
  m_jpsiMass = new std::vector<double>;
  m_jpsiMassError = new std::vector<double>;
  m_jpsiMassRec = new std::vector<double>;
  m_jpsiMassPullRec = new std::vector<double>;
  m_jpsiMassPullMC = new std::vector<double>;
  m_jpsiChi2 = new std::vector<double>;
  
//  m_trkRefitCharge1 = new std::vector<double>;
  m_trkRefitPx1 = new std::vector<double>;
  m_trkRefitPy1 = new std::vector<double>;
  m_trkRefitPz1 = new std::vector<double>;
//  m_trkRefitCharge2 = new std::vector<double>;
  m_trkRefitPx2 = new std::vector<double>;
  m_trkRefitPy2 = new std::vector<double>;
  m_trkRefitPz2 = new std::vector<double>;
  
  m_vx = new std::vector<double>;
  m_vy = new std::vector<double>;
  m_vz = new std::vector<double>;
 
  m_trkOrigCharge1 = new std::vector<double>;
  m_trkOrigPx1 = new std::vector<double>;
  m_trkOrigPy1 = new std::vector<double>;
  m_trkOrigPz1 = new std::vector<double>;
  m_trkOrigCharge2 = new std::vector<double>;
  m_trkOrigPx2 = new std::vector<double>;
  m_trkOrigPy2 = new std::vector<double>;
  m_trkOrigPz2 = new std::vector<double>;
 
//  m_rxyError = new std::vector<double>;
  
  return;
 
}

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Algorithm > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ invariantMassError()

double JpsiExample::invariantMassError	(	const xAOD::Vertex *	vxCandidate,
		const std::vector< double > &	masses ) const

private

Definition at line 390 of file JpsiExample.cxx.

{
  uint NTrk = vxCandidate->vxTrackAtVertex().size();
  if (masses.size() != NTrk) {
    ATH_MSG_WARNING("The provided number of masses does not match the number of tracks in the vertex");
    return -999999.;
  }
  
  uint ndim    = vxCandidate->covariance().size();
  uint ndimExp = (3*NTrk+3)*(3*NTrk+3+1)/2;
  if (ndim == ndimExp) {
    return massErrorVKalVrt(vxCandidate,masses);
  } else {
    ATH_MSG_WARNING("Unknown covariance matrix dimension: " << ndim << ", expected: " << ndimExp);
    return -999999.;
  }
}

◆ massErrorVKalVrt()

double JpsiExample::massErrorVKalVrt	(	const xAOD::Vertex *	vxCandidate,
		const std::vector< double > &	masses ) const

private

Definition at line 412 of file JpsiExample.cxx.

{
  unsigned int NTrk = vxCandidate->vxTrackAtVertex().size();
  
  // construct the full covariance matrix    
  Amg::MatrixX* fullCov = convertVKalCovMatrix(NTrk, vxCandidate->covariance());
  
  std::vector<CLHEP::HepLorentzVector> particleMom(NTrk);
  std::vector<CLHEP::HepMatrix> particleDeriv(NTrk);
  CLHEP::HepLorentzVector totalMom;
  CLHEP::HepMatrix tmpDeriv(3,3);
 
  double phi=0.,theta=0.,invP=0.;
  for( unsigned int it=0; it<NTrk; it++){
    const Trk::TrackParameters*  bPer = vxCandidate->vxTrackAtVertex()[it].perigeeAtVertex();
    phi   =  bPer->parameters()[Trk::phi];
    theta =  bPer->parameters()[Trk::theta];
    invP  =  bPer->parameters()[Trk::qOverP];
    CLHEP::HepLorentzVector tmp( cos(phi)*sin(theta)/fabs(invP),
                          sin(phi)*sin(theta)/fabs(invP),
                                   cos(theta)/fabs(invP) );
    double esq = tmp.px()*tmp.px() + tmp.py()*tmp.py() + tmp.pz()*tmp.pz() + masses[it]*masses[it];
    double e = (esq>0.) ? sqrt(esq) : 0.;
    tmp.setE(e);
    particleMom[it] = tmp;
    totalMom += tmp;
 
//  d(Px,Py,Pz)/d(Phi,Theta,InvP)
    tmpDeriv(1,1) = - tmp.py();
    tmpDeriv(2,1) =   tmp.px();
    tmpDeriv(3,1) =   0.;
    tmpDeriv(1,2) =   cos(phi) * tmp.pz();
    tmpDeriv(2,2) =   sin(phi) * tmp.pz();
    tmpDeriv(3,2) = - sin(theta)/fabs(invP);
    tmpDeriv(1,3) = - tmp.px()/invP;
    tmpDeriv(2,3) = - tmp.py()/invP;
    tmpDeriv(3,3) = - tmp.pz()/invP;
    particleDeriv[it] = tmpDeriv;
  }
 
  double dMdPx=0., dMdPy=0., dMdPz=0., dMdPhi=0., dMdTheta=0., dMdInvP=0.;
  std::vector<double> Deriv(3*NTrk+3, 0.);
  for(unsigned int it=0; it<NTrk; it++){
    dMdPx = ( totalMom.e() * particleMom[it].px()/particleMom[it].e() - totalMom.px() ) / totalMom.m();
    dMdPy = ( totalMom.e() * particleMom[it].py()/particleMom[it].e() - totalMom.py() ) / totalMom.m();
    dMdPz = ( totalMom.e() * particleMom[it].pz()/particleMom[it].e() - totalMom.pz() ) / totalMom.m();
 
    dMdPhi   = dMdPx*particleDeriv[it](1,1) + dMdPy*particleDeriv[it](2,1) + dMdPz*particleDeriv[it](3,1);
    dMdTheta = dMdPx*particleDeriv[it](1,2) + dMdPy*particleDeriv[it](2,2) + dMdPz*particleDeriv[it](3,2);
    dMdInvP  = dMdPx*particleDeriv[it](1,3) + dMdPy*particleDeriv[it](2,3) + dMdPz*particleDeriv[it](3,3);
 
    Deriv[3*it + 3 + 0] = dMdPhi;    Deriv[3*it + 3 + 1] = dMdTheta; Deriv[3*it + 3 + 2] = dMdInvP;
  }
 
  double err = 0;
  for(unsigned int i=0; i<3*NTrk+3; i++){
    for(unsigned int j=0; j<3*NTrk+3; j++){
      err += Deriv[i]*( (*fullCov)(i,j))*Deriv[j];
    }
  }
  
  // cleanup
  delete fullCov;
  
  return (err>0.) ? sqrt(err) : 0.;
}

◆ msg()

MsgStream & AthCommonMsg< Algorithm >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

◆ msgLvl()

bool AthCommonMsg< Algorithm >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

◆ origTrack4Momentum()

TLorentzVector JpsiExample::origTrack4Momentum	(	const xAOD::Vertex *	vxCandidate,
		int	trkIndex,
		double	mass ) const

private

Definition at line 378 of file JpsiExample.cxx.

{
  TVector3 mom = origTrackMomentum(vxCandidate, trkIndex);
  TLorentzVector lorentz;
  lorentz.SetVectM(mom, mass);
  return lorentz;
}

◆ origTrackMomentum()

TVector3 JpsiExample::origTrackMomentum	(	const xAOD::Vertex *	vxCandidate,
		int	trkIndex ) const

private

Definition at line 362 of file JpsiExample.cxx.

{
  TVector3 mom;
  const xAOD::TrackParticle* origTrack = vxCandidate->trackParticle(trkIndex);
  if(origTrack==NULL) {
    return mom;
  }
  
  mom.SetPtEtaPhi(origTrack->pt(), origTrack->eta(), origTrack->phi());
  return mom;
}

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Algorithm > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< Algorithm > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

  {
    h.renounce();
    PBASE::renounce (h);
  }

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

◆ sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )

overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, AthHistogramAlgorithm, and PyAthena::Alg.

Definition at line 66 of file AthAlgorithm.cxx.

                                       {
  StatusCode sc=AthCommonDataStore<AthCommonMsg<Algorithm>>::sysInitialize();
 
  if (sc.isFailure()) {
    return sc;
  }
  ServiceHandle<ICondSvc> cs("CondSvc",name());
  for (auto h : outputHandles()) {
    if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
      // do this inside the loop so we don't create the CondSvc until needed
      if ( cs.retrieve().isFailure() ) {
        ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
        return StatusCode::SUCCESS;
      }
      if (cs->regHandle(this,*h).isFailure()) {
        sc = StatusCode::FAILURE;
        ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
                      << " with CondSvc");
      }
    }
  }
  return sc;
}

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Algorithm > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ track4Momentum()

TLorentzVector JpsiExample::track4Momentum	(	const xAOD::Vertex *	vxCandidate,
		int	trkIndex,
		double	mass ) const

private

Definition at line 349 of file JpsiExample.cxx.

{
  TVector3 mom = trackMomentum(vxCandidate, trkIndex);
  TLorentzVector lorentz;
  lorentz.SetVectM(mom, mass);
  return lorentz;
}

◆ trackCharge()

double JpsiExample::trackCharge	(	const xAOD::Vertex *	vxCandidate,
		int	i ) const

private

Definition at line 483 of file JpsiExample.cxx.

{
  if(vxCandidate!=NULL && vxCandidate->trackParticle(i)!=NULL) {
    return vxCandidate->trackParticle(i)->charge();
  }
  
  return -999999;
}

◆ trackMomentum()

TVector3 JpsiExample::trackMomentum	(	const xAOD::Vertex *	vxCandidate,
		uint	trkIndex ) const

private

Definition at line 314 of file JpsiExample.cxx.

{
  float px = 0., py = 0., pz = 0.;
//  assert(vxCandidate!=0);
//  assert(i<vxCandidate->vxTrackAtVertex().size());
  
//  if (0 != vxCandidate) {
//    const Trk::TrackParameters* aPerigee = vxCandidate->vxTrackAtVertex()[trkIndex].perigeeAtVertex();
//    px = aPerigee->momentum()[Trk::px];
//    py = aPerigee->momentum()[Trk::py];
//    pz = aPerigee->momentum()[Trk::pz];
//  }
  
  SG::ReadDecorHandle<xAOD::VertexContainer, std::vector<float>> hx (m_refPX);
  SG::ReadDecorHandle<xAOD::VertexContainer, std::vector<float>> hy (m_refPY);
  SG::ReadDecorHandle<xAOD::VertexContainer, std::vector<float>> hz (m_refPZ);
 
  const std::vector<float>& refTrackPx = hx(*vxCandidate);
  const std::vector<float>& refTrackPy = hy(*vxCandidate);
  const std::vector<float>& refTrackPz = hz(*vxCandidate);
 
  if(trkIndex < refTrackPx.size() && refTrackPx.size() == refTrackPy.size() && refTrackPz.size()) {
    px = refTrackPx[trkIndex];
    py = refTrackPy[trkIndex];
    pz = refTrackPz[trkIndex];
  }
  
  TVector3 mom(px,py,pz);
  return mom;
}

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
    //         << "  size: " << m_vhka.size() << endmsg;
    for (auto &a : m_vhka) {
      std::vector<SG::VarHandleKey*> keys = a->keys();
      for (auto k : keys) {
        k->setOwner(this);
      }
    }
  }

Member Data Documentation

◆ m_auxTree

TTree* JpsiExample::m_auxTree

private

Definition at line 84 of file JpsiExample.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_eventCntr

int JpsiExample::m_eventCntr

private

Definition at line 48 of file JpsiExample.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects

privateinherited

Definition at line 79 of file AthAlgorithm.h.

◆ m_JpsiCandidatesKey

SG::ReadHandleKey<xAOD::VertexContainer> JpsiExample::m_JpsiCandidatesKey

private

Name of J/psi container.

Definition at line 86 of file JpsiExample.h.

◆ m_jpsiChi2

std::vector<double>* JpsiExample::m_jpsiChi2

private

Definition at line 56 of file JpsiExample.h.

◆ m_jpsiCntr

int JpsiExample::m_jpsiCntr

private

Definition at line 49 of file JpsiExample.h.

◆ m_jpsiMass

std::vector<double>* JpsiExample::m_jpsiMass

private

Definition at line 51 of file JpsiExample.h.

◆ m_jpsiMassError

std::vector<double>* JpsiExample::m_jpsiMassError

private

Definition at line 52 of file JpsiExample.h.

◆ m_jpsiMassPullMC

std::vector<double>* JpsiExample::m_jpsiMassPullMC

private

Definition at line 55 of file JpsiExample.h.

◆ m_jpsiMassPullRec

std::vector<double>* JpsiExample::m_jpsiMassPullRec

private

Definition at line 54 of file JpsiExample.h.

◆ m_jpsiMassRec

std::vector<double>* JpsiExample::m_jpsiMassRec

private

Definition at line 53 of file JpsiExample.h.

◆ m_muonMass

double JpsiExample::m_muonMass

private

Definition at line 45 of file JpsiExample.h.

◆ m_outputFile

TFile* JpsiExample::m_outputFile

private

Definition at line 83 of file JpsiExample.h.

◆ m_refPX

SG::ReadDecorHandleKey<xAOD::VertexContainer> JpsiExample::m_refPX

private

Definition at line 87 of file JpsiExample.h.

◆ m_refPY

SG::ReadDecorHandleKey<xAOD::VertexContainer> JpsiExample::m_refPY

private

Definition at line 88 of file JpsiExample.h.

◆ m_refPZ

SG::ReadDecorHandleKey<xAOD::VertexContainer> JpsiExample::m_refPZ

private

Definition at line 89 of file JpsiExample.h.

◆ m_trkOrigCharge1

std::vector<double>* JpsiExample::m_trkOrigCharge1

private

Definition at line 71 of file JpsiExample.h.

◆ m_trkOrigCharge2

std::vector<double>* JpsiExample::m_trkOrigCharge2

private

Definition at line 75 of file JpsiExample.h.

◆ m_trkOrigPx1

std::vector<double>* JpsiExample::m_trkOrigPx1

private

Definition at line 72 of file JpsiExample.h.

◆ m_trkOrigPx2

std::vector<double>* JpsiExample::m_trkOrigPx2

private

Definition at line 76 of file JpsiExample.h.

◆ m_trkOrigPy1

std::vector<double>* JpsiExample::m_trkOrigPy1

private

Definition at line 73 of file JpsiExample.h.

◆ m_trkOrigPy2

std::vector<double>* JpsiExample::m_trkOrigPy2

private

Definition at line 77 of file JpsiExample.h.

◆ m_trkOrigPz1

std::vector<double>* JpsiExample::m_trkOrigPz1

private

Definition at line 74 of file JpsiExample.h.

◆ m_trkOrigPz2

std::vector<double>* JpsiExample::m_trkOrigPz2

private

Definition at line 78 of file JpsiExample.h.

◆ m_trkRefitPx1

std::vector<double>* JpsiExample::m_trkRefitPx1

private

Definition at line 59 of file JpsiExample.h.

◆ m_trkRefitPx2

std::vector<double>* JpsiExample::m_trkRefitPx2

private

Definition at line 63 of file JpsiExample.h.

◆ m_trkRefitPy1

std::vector<double>* JpsiExample::m_trkRefitPy1

private

Definition at line 60 of file JpsiExample.h.

◆ m_trkRefitPy2

std::vector<double>* JpsiExample::m_trkRefitPy2

private

Definition at line 64 of file JpsiExample.h.

◆ m_trkRefitPz1

std::vector<double>* JpsiExample::m_trkRefitPz1

private

Definition at line 61 of file JpsiExample.h.

◆ m_trkRefitPz2

std::vector<double>* JpsiExample::m_trkRefitPz2

private

Definition at line 65 of file JpsiExample.h.

◆ m_userFName

std::string JpsiExample::m_userFName

private

Definition at line 46 of file JpsiExample.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< Algorithm > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_vx

std::vector<double>* JpsiExample::m_vx

private

Definition at line 67 of file JpsiExample.h.

◆ m_vy

std::vector<double>* JpsiExample::m_vy

private

Definition at line 68 of file JpsiExample.h.

◆ m_vz

std::vector<double>* JpsiExample::m_vz

private

Definition at line 69 of file JpsiExample.h.

The documentation for this class was generated from the following files:

Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ JpsiExample()

Member Function Documentation

◆ clearBranches()

◆ convertVKalCovMatrix()

◆ declareGaudiProperty()

◆ declareProperty()

◆ detStore()

◆ evtStore()

◆ execute()

◆ extraDeps_update_handler()

◆ extraOutputDeps()

◆ finalize()

◆ initialize()

◆ initializeBranches()

◆ inputHandles()

◆ invariantMassError()

◆ massErrorVKalVrt()

◆ msg()

◆ msgLvl()

◆ origTrack4Momentum()

◆ origTrackMomentum()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ sysInitialize()

◆ sysStart()

◆ track4Momentum()

◆ trackCharge()

◆ trackMomentum()

◆ updateVHKA()

Member Data Documentation

◆ m_auxTree

◆ m_detStore

◆ m_eventCntr

◆ m_evtStore

◆ m_extendedExtraObjects

◆ m_JpsiCandidatesKey

◆ m_jpsiChi2

◆ m_jpsiCntr

◆ m_jpsiMass

◆ m_jpsiMassError

◆ m_jpsiMassPullMC

◆ m_jpsiMassPullRec

◆ m_jpsiMassRec

◆ m_muonMass

◆ m_outputFile

◆ m_refPX

◆ m_refPY

◆ m_refPZ

◆ m_trkOrigCharge1

◆ m_trkOrigCharge2

◆ m_trkOrigPx1

◆ m_trkOrigPx2

◆ m_trkOrigPy1

◆ m_trkOrigPy2

◆ m_trkOrigPz1

◆ m_trkOrigPz2

◆ m_trkRefitPx1

◆ m_trkRefitPx2

◆ m_trkRefitPy1

◆ m_trkRefitPy2

◆ m_trkRefitPz1

◆ m_trkRefitPz2

◆ m_userFName

◆ m_varHandleArraysDeclared

◆ m_vhka

◆ m_vx

◆ m_vy

◆ m_vz