#include <JpsiPlusPsiCascade.h>

Inheritance diagram for DerivationFramework::JpsiPlusPsiCascade:

Collaboration diagram for DerivationFramework::JpsiPlusPsiCascade:

Public Member Functions
	JpsiPlusPsiCascade (const std::string &t, const std::string &n, const IInterface *p)

virtual	~JpsiPlusPsiCascade ()=default

virtual StatusCode	initialize () override

StatusCode	performSearch (std::vector< Trk::VxCascadeInfo * > cascadeinfoContainer, std::vector< Trk::VxCascadeInfo > *cascadeinfoContainer_noConstr) const

virtual StatusCode	addBranches () const override
	Pass the thinning service More...

ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`. More...

virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm. More...

virtual StatusCode	sysStart () override
	Handle START transition. More...

virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...

virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...

Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T > &t)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...

void	updateVHKA (Gaudi::Details::PropertyBase &)

MsgStream &	msg () const

MsgStream &	msg (const MSG::Level lvl) const

bool	msgLvl (const MSG::Level lvl) const

Static Public Member Functions
static const InterfaceID &	interfaceID ()

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...

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. More...

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Private Attributes
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainerKey

SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexPsiContainerKey

std::vector< std::string >	m_vertexJpsiHypoNames

std::vector< std::string >	m_vertexPsiHypoNames

SG::WriteHandleKeyArray< xAOD::VertexContainer >	m_cascadeOutputsKeys

SG::ReadHandleKey< xAOD::VertexContainer >	m_VxPrimaryCandidateName
	Name of primary vertex container. More...

SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_trackContainerName

SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfo_key

double	m_jpsiMassLower

double	m_jpsiMassUpper

double	m_diTrackMassLower

double	m_diTrackMassUpper

double	m_psiMassLower

double	m_psiMassUpper

double	m_jpsi2MassLower

double	m_jpsi2MassUpper

double	m_MassLower

double	m_MassUpper

int	m_vtx1Daug_num

double	m_vtx1Daug1MassHypo

double	m_vtx1Daug2MassHypo

double	m_vtx1Daug3MassHypo

double	m_vtx1Daug4MassHypo

double	m_vtx2Daug1MassHypo

double	m_vtx2Daug2MassHypo

double	m_mass_jpsi

double	m_mass_diTrk

double	m_mass_psi

double	m_mass_jpsi2

bool	m_constrPsi

bool	m_constrJpsi

bool	m_constrDiTrk

bool	m_constrJpsi2

double	m_chi2cut_Psi

double	m_chi2cut_Jpsi

double	m_chi2cut

unsigned int	m_maxCandidates

ToolHandle< Trk::TrkVKalVrtFitter >	m_iVertexFitter

ToolHandle< Analysis::PrimaryVertexRefitter >	m_pvRefitter

ToolHandle< Trk::V0Tools >	m_V0Tools

ToolHandle< DerivationFramework::CascadeTools >	m_CascadeTools

ServiceHandle< IPartPropSvc >	m_partPropSvc {this, "PartPropSvc", "PartPropSvc"}

bool	m_refitPV

SG::WriteHandleKey< xAOD::VertexContainer >	m_refPVContainerName

std::string	m_hypoName

int	m_PV_max

int	m_DoVertexType

size_t	m_PV_minNTracks

StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...

StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...

std::vector< SG::VarHandleKeyArray * >	m_vhka

bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 33 of file JpsiPlusPsiCascade.h.

Member Typedef Documentation

◆ StoreGateSvc_t

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

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ JpsiPlusPsiCascade()

DerivationFramework::JpsiPlusPsiCascade::JpsiPlusPsiCascade	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 252 of file JpsiPlusPsiCascade.cxx.

                                                                                                                : AthAlgTool(type,name,parent),
     m_vertexContainerKey(""),
     m_vertexPsiContainerKey(""),
     m_cascadeOutputsKeys({"JpsiPlusPsiCascadeVtx1", "JpsiPlusPsiCascadeVtx2", "JpsiPlusPsiCascadeVtx3"}),
     m_VxPrimaryCandidateName("PrimaryVertices"),
     m_trackContainerName("InDetTrackParticles"),
     m_eventInfo_key("EventInfo"),
     m_jpsiMassLower(0.0),
     m_jpsiMassUpper(20000.0),
     m_diTrackMassLower(-1.0),
     m_diTrackMassUpper(-1.0),
     m_psiMassLower(0.0),
     m_psiMassUpper(25000.0),
     m_jpsi2MassLower(0.0),
     m_jpsi2MassUpper(20000.0),
     m_MassLower(0.0),
     m_MassUpper(31000.0),
     m_vtx1Daug_num(4),
     m_vtx1Daug1MassHypo(-1),
     m_vtx1Daug2MassHypo(-1),
     m_vtx1Daug3MassHypo(-1),
     m_vtx1Daug4MassHypo(-1),
     m_vtx2Daug1MassHypo(-1),
     m_vtx2Daug2MassHypo(-1),
     m_mass_jpsi(-1),
     m_mass_diTrk(-1),
     m_mass_psi(-1),
     m_mass_jpsi2(-1),
     m_constrPsi(false),
     m_constrJpsi(false),
     m_constrDiTrk(false),
     m_constrJpsi2(false),
     m_chi2cut_Psi(-1.0),
     m_chi2cut_Jpsi(-1.0),
     m_chi2cut(-1.0),
     m_maxCandidates(0),
     m_iVertexFitter("Trk::TrkVKalVrtFitter"),
     m_pvRefitter("Analysis::PrimaryVertexRefitter", this),
     m_V0Tools("Trk::V0Tools"),
     m_CascadeTools("DerivationFramework::CascadeTools")
   {
     declareProperty("JpsiVertices",               m_vertexContainerKey);
     declareProperty("PsiVertices",                m_vertexPsiContainerKey);
     declareProperty("JpsiVtxHypoNames",           m_vertexJpsiHypoNames);
     declareProperty("PsiVtxHypoNames",            m_vertexPsiHypoNames);
     declareProperty("VxPrimaryCandidateName",     m_VxPrimaryCandidateName);
     declareProperty("TrackContainerName",         m_trackContainerName);
     declareProperty("RefPVContainerName",         m_refPVContainerName = "RefittedPrimaryVertices");
     declareProperty("JpsiMassLowerCut",           m_jpsiMassLower);
     declareProperty("JpsiMassUpperCut",           m_jpsiMassUpper);
     declareProperty("DiTrackMassLower",           m_diTrackMassLower); // only effective when m_vtx1Daug_num=4
     declareProperty("DiTrackMassUpper",           m_diTrackMassUpper); // only effective when m_vtx1Daug_num=4
     declareProperty("PsiMassLowerCut",            m_psiMassLower);
     declareProperty("PsiMassUpperCut",            m_psiMassUpper);
     declareProperty("Jpsi2MassLowerCut",          m_jpsi2MassLower);
     declareProperty("Jpsi2MassUpperCut",          m_jpsi2MassUpper);
     declareProperty("MassLowerCut",               m_MassLower);
     declareProperty("MassUpperCut",               m_MassUpper);
     declareProperty("HypothesisName",             m_hypoName = "TQ");
     declareProperty("NumberOfPsiDaughters",       m_vtx1Daug_num); // 3 or 4 only
     declareProperty("Vtx1Daug1MassHypo",          m_vtx1Daug1MassHypo);
     declareProperty("Vtx1Daug2MassHypo",          m_vtx1Daug2MassHypo);
     declareProperty("Vtx1Daug3MassHypo",          m_vtx1Daug3MassHypo);
     declareProperty("Vtx1Daug4MassHypo",          m_vtx1Daug4MassHypo);
     declareProperty("Vtx2Daug1MassHypo",          m_vtx2Daug1MassHypo);
     declareProperty("Vtx2Daug2MassHypo",          m_vtx2Daug2MassHypo);
     declareProperty("JpsiMass",                   m_mass_jpsi);
     declareProperty("DiTrackMass",                m_mass_diTrk);
     declareProperty("PsiMass",                    m_mass_psi);
     declareProperty("Jpsi2Mass",                  m_mass_jpsi2);
     declareProperty("ApplyJpsiMassConstraint",    m_constrJpsi);
     declareProperty("ApplyDiTrackMassConstraint", m_constrDiTrk); // only effective when m_vtx1Daug_num=4
     declareProperty("ApplyPsiMassConstraint",     m_constrPsi);
     declareProperty("ApplyJpsi2MassConstraint",   m_constrJpsi2);
     declareProperty("Chi2CutPsi",                 m_chi2cut_Psi);
     declareProperty("Chi2CutJpsi",                m_chi2cut_Jpsi);
     declareProperty("Chi2Cut",                    m_chi2cut);
     declareProperty("MaxCandidates",              m_maxCandidates);
     declareProperty("RefitPV",                    m_refitPV         = true);
     declareProperty("MaxnPV",                     m_PV_max          = 1000);
     declareProperty("MinNTracksInPV",             m_PV_minNTracks   = 0);
     declareProperty("DoVertexType",               m_DoVertexType    = 7);
     declareProperty("TrkVertexFitterTool",        m_iVertexFitter);
     declareProperty("PVRefitter",                 m_pvRefitter);
     declareProperty("V0Tools",                    m_V0Tools);
     declareProperty("CascadeTools",               m_CascadeTools);
     declareProperty("CascadeVertexCollections",   m_cascadeOutputsKeys);
   }

◆ ~JpsiPlusPsiCascade()

virtual DerivationFramework::JpsiPlusPsiCascade::~JpsiPlusPsiCascade ( )

virtualdefault

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::JpsiPlusPsiCascade::addBranches ( ) const

overridevirtual

Pass the thinning service

Implements DerivationFramework::IAugmentationTool.

Definition at line 62 of file JpsiPlusPsiCascade.cxx.

                                                    {
     if (m_vtx1Daug_num != 3 && m_vtx1Daug_num != 4) {
       ATH_MSG_FATAL("Incorrect number of Psi daughters (should be 3 or 4)");
       return StatusCode::FAILURE;
     }
  
     constexpr int topoN = 3;
     if(m_cascadeOutputsKeys.size() != topoN) {
       ATH_MSG_FATAL("Incorrect number of VtxContainers");
       return StatusCode::FAILURE;
     }
     std::array<SG::WriteHandle<xAOD::VertexContainer>, topoN> VtxWriteHandles; int ikey(0);
     for(const SG::WriteHandleKey<xAOD::VertexContainer>& key : m_cascadeOutputsKeys) {
       VtxWriteHandles[ikey] = SG::WriteHandle<xAOD::VertexContainer>(key);
       ATH_CHECK( VtxWriteHandles[ikey].record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>()) );
       ikey++;
     }
  
     //----------------------------------------------------
     // retrieve primary vertices
     //----------------------------------------------------
     SG::ReadHandle<xAOD::VertexContainer> pvContainer(m_VxPrimaryCandidateName);
     ATH_CHECK( pvContainer.isValid() );
     if (pvContainer.cptr()->size()==0) {
       ATH_MSG_WARNING("You have no primary vertices: " << pvContainer.cptr()->size());
       return StatusCode::RECOVERABLE;
     }
  
     //----------------------------------------------------
     // Record refitted primary vertices
     //----------------------------------------------------
     SG::WriteHandle<xAOD::VertexContainer> refPvContainer;
     if(m_refitPV) {
       refPvContainer = SG::WriteHandle<xAOD::VertexContainer>(m_refPVContainerName);
       ATH_CHECK( refPvContainer.record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>()) );
     }
  
     std::vector<Trk::VxCascadeInfo*> cascadeinfoContainer;
     std::vector<Trk::VxCascadeInfo*> cascadeinfoContainer_noConstr;
     ATH_CHECK(performSearch(&cascadeinfoContainer,&cascadeinfoContainer_noConstr));
  
     SG::ReadHandle<xAOD::EventInfo> evt(m_eventInfo_key);
     ATH_CHECK( evt.isValid() );
     BPhysPVCascadeTools helper(&(*m_CascadeTools), evt.cptr());
     helper.SetMinNTracksInPV(m_PV_minNTracks);
  
     // Decorators for the vertices
     SG::AuxElement::Decorator<VertexLinkVector> CascadeLinksDecor("CascadeVertexLinks"); 
     SG::AuxElement::Decorator<VertexLinkVector> PsiLinksDecor("PsiVertexLinks"); 
     SG::AuxElement::Decorator<VertexLinkVector> JpsiLinksDecor("JpsiVertexLinks"); 
     SG::AuxElement::Decorator<float> chi2_decor("ChiSquared");
     SG::AuxElement::Decorator<int> ndof_decor("nDoF");
     SG::AuxElement::Decorator<float> chi2_nc_decor("ChiSquared_nc");
     SG::AuxElement::Decorator<int> ndof_nc_decor("nDoF_nc");
     SG::AuxElement::Decorator<float> Pt_decor("Pt");
     SG::AuxElement::Decorator<float> PtErr_decor("PtErr");
     SG::AuxElement::Decorator<float> chi2_SV1_decor("ChiSquared_SV1");
     SG::AuxElement::Decorator<float> chi2_nc_SV1_decor("ChiSquared_nc_SV1");
     SG::AuxElement::Decorator<float> chi2_V1_decor("ChiSquared_V1");
     SG::AuxElement::Decorator<int> ndof_V1_decor("nDoF_V1");
     SG::AuxElement::Decorator<float> lxy_SV1_decor("lxy_SV1");
     SG::AuxElement::Decorator<float> lxyErr_SV1_decor("lxyErr_SV1");
     SG::AuxElement::Decorator<float> a0xy_SV1_decor("a0xy_SV1");
     SG::AuxElement::Decorator<float> a0xyErr_SV1_decor("a0xyErr_SV1");
     SG::AuxElement::Decorator<float> a0z_SV1_decor("a0z_SV1");
     SG::AuxElement::Decorator<float> a0zErr_SV1_decor("a0zErr_SV1");
     SG::AuxElement::Decorator<float> chi2_SV2_decor("ChiSquared_SV2");
     SG::AuxElement::Decorator<float> chi2_nc_SV2_decor("ChiSquared_nc_SV2");
     SG::AuxElement::Decorator<float> chi2_V2_decor("ChiSquared_V2");
     SG::AuxElement::Decorator<int> ndof_V2_decor("nDoF_V2");
     SG::AuxElement::Decorator<float> lxy_SV2_decor("lxy_SV2");
     SG::AuxElement::Decorator<float> lxyErr_SV2_decor("lxyErr_SV2");
     SG::AuxElement::Decorator<float> a0xy_SV2_decor("a0xy_SV2");
     SG::AuxElement::Decorator<float> a0xyErr_SV2_decor("a0xyErr_SV2");
     SG::AuxElement::Decorator<float> a0z_SV2_decor("a0z_SV2");
     SG::AuxElement::Decorator<float> a0zErr_SV2_decor("a0zErr_SV2");
  
     // Get the containers and identify the input Jpsi and Psi
     SG::ReadHandle<xAOD::VertexContainer> psiContainer(m_vertexPsiContainerKey);
     ATH_CHECK( psiContainer.isValid() );
     SG::ReadHandle<xAOD::VertexContainer> jpsiContainer(m_vertexContainerKey);
     ATH_CHECK( jpsiContainer.isValid() );
  
     for(size_t ic=0; ic<cascadeinfoContainer.size(); ic++) {
       Trk::VxCascadeInfo* cascade_info = cascadeinfoContainer[ic];
       if(cascade_info==nullptr) {
         ATH_MSG_ERROR("CascadeInfo is null");
         continue;
       }
  
       Trk::VxCascadeInfo* cascade_info_noConstr = cascadeinfoContainer_noConstr[ic];
  
       const std::vector<xAOD::Vertex*> &cascadeVertices = cascade_info->vertices();
       if(cascadeVertices.size() != topoN) ATH_MSG_ERROR("Incorrect number of vertices");
       if(cascadeVertices[0]==nullptr || cascadeVertices[1]==nullptr || cascadeVertices[2]==nullptr) ATH_MSG_ERROR("Error null vertex");
       // Keep vertices
       for(int i=0; i<topoN; i++) VtxWriteHandles[i].ptr()->push_back(cascadeVertices[i]);
  
       cascade_info->setSVOwnership(false); // Prevent Container from deleting vertices
       const auto mainVertex = cascadeVertices[2]; // this is the mother vertex
       const std::vector< std::vector<TLorentzVector> > &moms = cascade_info->getParticleMoms();
  
       // Set links to cascade vertices
       std::vector<VertexLink> precedingVertexLinks;
       VertexLink vertexLink1;
       vertexLink1.setElement(cascadeVertices[0]);
       vertexLink1.setStorableObject(*VtxWriteHandles[0].ptr());
       if( vertexLink1.isValid() ) precedingVertexLinks.push_back( vertexLink1 );
       VertexLink vertexLink2;
       vertexLink2.setElement(cascadeVertices[1]);
       vertexLink2.setStorableObject(*VtxWriteHandles[1].ptr());
       if( vertexLink2.isValid() ) precedingVertexLinks.push_back( vertexLink2 );
       CascadeLinksDecor(*mainVertex) = precedingVertexLinks;
  
       // Identify the input Jpsi
       const xAOD::Vertex* jpsiVertex = BPhysPVCascadeTools::FindVertex<2>(jpsiContainer.cptr(), cascadeVertices[1]);
       // Identify the input Psi
       const xAOD::Vertex* psiVertex(0);
       if(m_vtx1Daug_num==4) psiVertex = BPhysPVCascadeTools::FindVertex<4>(psiContainer.cptr(), cascadeVertices[0]);
       else psiVertex = BPhysPVCascadeTools::FindVertex<3>(psiContainer.cptr(), cascadeVertices[0]);
  
       // Set links to input vertices
       std::vector<const xAOD::Vertex*> jpsiVerticestoLink;
       if(jpsiVertex) jpsiVerticestoLink.push_back(jpsiVertex);
       else ATH_MSG_WARNING("Could not find linking Jpsi");
       if(!BPhysPVCascadeTools::LinkVertices(JpsiLinksDecor, jpsiVerticestoLink, jpsiContainer.cptr(), mainVertex)) ATH_MSG_ERROR("Error decorating with Jpsi vertex");
  
       std::vector<const xAOD::Vertex*> psiVerticestoLink;
       if(psiVertex) psiVerticestoLink.push_back(psiVertex);
       else ATH_MSG_WARNING("Could not find linking Psi");
       if(!BPhysPVCascadeTools::LinkVertices(PsiLinksDecor, psiVerticestoLink, psiContainer.cptr(), mainVertex)) ATH_MSG_ERROR("Error decorating with Psi vertex");
  
       xAOD::BPhysHypoHelper vtx(m_hypoName, mainVertex);
  
       // Get refitted track momenta from all vertices, charged tracks only
       BPhysPVCascadeTools::SetVectorInfo(vtx, cascade_info);
       vtx.setPass(true);
  
       //
       // Decorate main vertex
       //
       // mass, mass error
       // https://gitlab.cern.ch/atlas/athena/-/blob/21.2/Tracking/TrkVertexFitter/TrkVKalVrtFitter/TrkVKalVrtFitter/VxCascadeInfo.h
       BPHYS_CHECK( vtx.setMass(m_CascadeTools->invariantMass(moms[2])) );
       BPHYS_CHECK( vtx.setMassErr(m_CascadeTools->invariantMassError(moms[2],cascade_info->getCovariance()[2])) );
       // pt and pT error (the default pt of mainVertex is != the pt of the full cascade fit!)
       Pt_decor(*mainVertex)       = m_CascadeTools->pT(moms[2]);
       PtErr_decor(*mainVertex)    = m_CascadeTools->pTError(moms[2],cascade_info->getCovariance()[2]);
       // chi2 and ndof (the default chi2 of mainVertex is != the chi2 of the full cascade fit!)
       chi2_decor(*mainVertex)     = cascade_info->fitChi2();
       ndof_decor(*mainVertex)     = cascade_info->nDoF();
       chi2_nc_decor(*mainVertex)  = cascade_info_noConstr ? cascade_info_noConstr->fitChi2() : -999999.;
       ndof_nc_decor(*mainVertex)  = cascade_info_noConstr ? cascade_info_noConstr->nDoF() : -1;
  
       // decorate the Psi vertex
       chi2_SV1_decor(*cascadeVertices[0])    = m_V0Tools->chisq(cascadeVertices[0]);
       chi2_nc_SV1_decor(*cascadeVertices[0]) = cascade_info_noConstr ? m_V0Tools->chisq(cascade_info_noConstr->vertices()[0]) : -999999.;
       chi2_V1_decor(*cascadeVertices[0])     = m_V0Tools->chisq(psiVertex);
       ndof_V1_decor(*cascadeVertices[0])     = m_V0Tools->ndof(psiVertex);
       lxy_SV1_decor(*cascadeVertices[0])     = m_CascadeTools->lxy(moms[0],cascadeVertices[0],mainVertex);
       lxyErr_SV1_decor(*cascadeVertices[0])  = m_CascadeTools->lxyError(moms[0],cascade_info->getCovariance()[0],cascadeVertices[0],mainVertex);
       a0z_SV1_decor(*cascadeVertices[0])     = m_CascadeTools->a0z(moms[0],cascadeVertices[0],mainVertex);
       a0zErr_SV1_decor(*cascadeVertices[0])  = m_CascadeTools->a0zError(moms[0],cascade_info->getCovariance()[0],cascadeVertices[0],mainVertex);
       a0xy_SV1_decor(*cascadeVertices[0])    = m_CascadeTools->a0xy(moms[0],cascadeVertices[0],mainVertex);
       a0xyErr_SV1_decor(*cascadeVertices[0]) = m_CascadeTools->a0xyError(moms[0],cascade_info->getCovariance()[0],cascadeVertices[0],mainVertex);
  
       // decorate the Jpsi vertex
       chi2_SV2_decor(*cascadeVertices[1])    = m_V0Tools->chisq(cascadeVertices[1]);
       chi2_nc_SV2_decor(*cascadeVertices[1]) = cascade_info_noConstr ? m_V0Tools->chisq(cascade_info_noConstr->vertices()[1]) : -999999.;
       chi2_V2_decor(*cascadeVertices[1])     = m_V0Tools->chisq(jpsiVertex);
       ndof_V2_decor(*cascadeVertices[1])     = m_V0Tools->ndof(jpsiVertex);
       lxy_SV2_decor(*cascadeVertices[1])     = m_CascadeTools->lxy(moms[1],cascadeVertices[1],mainVertex);
       lxyErr_SV2_decor(*cascadeVertices[1])  = m_CascadeTools->lxyError(moms[1],cascade_info->getCovariance()[1],cascadeVertices[1],mainVertex);
       a0z_SV2_decor(*cascadeVertices[1])     = m_CascadeTools->a0z(moms[1],cascadeVertices[1],mainVertex);
       a0zErr_SV2_decor(*cascadeVertices[1])  = m_CascadeTools->a0zError(moms[1],cascade_info->getCovariance()[1],cascadeVertices[1],mainVertex);
       a0xy_SV2_decor(*cascadeVertices[1])    = m_CascadeTools->a0xy(moms[1],cascadeVertices[1],mainVertex);
       a0xyErr_SV2_decor(*cascadeVertices[1]) = m_CascadeTools->a0xyError(moms[1],cascade_info->getCovariance()[1],cascadeVertices[1],mainVertex);
  
       double Mass_Moth = m_CascadeTools->invariantMass(moms[2]); //size=2
       ATH_CHECK(helper.FillCandwithRefittedVertices(m_refitPV, pvContainer.cptr(), m_refitPV ? refPvContainer.ptr() : 0, &(*m_pvRefitter), m_PV_max, m_DoVertexType, cascade_info, 2, Mass_Moth, vtx));
     } // loop over cascadeinfoContainer
  
     // Deleting cascadeinfo since this won't be stored.
     // Vertices have been kept in m_cascadeOutputs and should be owned by their container
     for (auto cascade_info : cascadeinfoContainer) delete cascade_info;
     for (auto cascade_info_noConstr : cascadeinfoContainer_noConstr) delete cascade_info_noConstr;
  
     return StatusCode::SUCCESS;
   }

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyArrayType &
	)

inlineprivateinherited

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

Definition at line 170 of file AthCommonDataStore.h.

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

◆ declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	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());
  
   }

◆ declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleType &
	)

inlineprivateinherited

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

Definition at line 184 of file AthCommonDataStore.h.

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

◆ declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty	(	Gaudi::Property< T > &	t,
		const SG::NotHandleType &
	)

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(t);
   }

◆ declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleBase &	hndl,
		const std::string &	doc,
		const SG::VarHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

   {
     this->declare(hndl.vhKey());
     hndl.vhKey().setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKey &	hndl,
		const std::string &	doc,
		const SG::VarHandleKeyType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

   {
     this->declare(hndl);
     hndl.setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKeyArray &	hndArr,
		const std::string &	doc,
		const SG::VarHandleKeyArrayType &
	)

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

   {
  
     // std::ostringstream ost;
     // ost << Algorithm::name() << " VHKA declareProp: " << name 
     //     << " size: " << hndArr.keys().size() 
     //     << " mode: " << hndArr.mode() 
     //     << "  vhka size: " << m_vhka.size()
     //     << "\n";
     // debug() << ost.str() << endmsg;
  
     hndArr.setOwner(this);
     m_vhka.push_back(&hndArr);
  
     Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
     if (p != 0) {
       p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
     } else {
       ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                     << name);
     }
  
     return p;
  
   }

◆ declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc,
		const SG::NotHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(name, property, doc);
   }

◆ declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc = `"none"`
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

   {
     typedef typename SG::HandleClassifier<T>::type htype;
     return declareProperty (name, property, doc, htype());
   }

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T > & 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< AlgTool > >::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; }

◆ evtStore() [1/2]

ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::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; }

◆ evtStore() [2/2]

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( ) const

inlineinherited

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

Definition at line 90 of file AthCommonDataStore.h.

90 { return m_evtStore; }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::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

◆ initialize()

StatusCode DerivationFramework::JpsiPlusPsiCascade::initialize ( )

overridevirtual

Definition at line 22 of file JpsiPlusPsiCascade.cxx.

                                             {
     // retrieving vertex Fitter
     ATH_CHECK( m_iVertexFitter.retrieve() );
  
     // retrieve PV refitter
     ATH_CHECK( m_pvRefitter.retrieve() );
  
     // retrieving the V0 tools
     ATH_CHECK( m_V0Tools.retrieve() );
  
     // retrieving the Cascade tools
     ATH_CHECK( m_CascadeTools.retrieve() );
  
     ATH_CHECK( m_vertexContainerKey.initialize() );
     ATH_CHECK( m_vertexPsiContainerKey.initialize() );
     ATH_CHECK( m_VxPrimaryCandidateName.initialize() );
     ATH_CHECK( m_trackContainerName.initialize() );
     ATH_CHECK( m_refPVContainerName.initialize() );
     ATH_CHECK( m_cascadeOutputsKeys.initialize() );
     ATH_CHECK( m_eventInfo_key.initialize() );
  
     ATH_CHECK( m_partPropSvc.retrieve() );
     auto pdt = m_partPropSvc->PDT();
  
     // retrieve particle masses
     // https://gitlab.cern.ch/atlas/athena/-/blob/main/Generators/TruthUtils/TruthUtils/AtlasPID.h
     if(m_mass_jpsi < 0.) m_mass_jpsi = BPhysPVCascadeTools::getParticleMass(pdt, MC::JPSI);
     if(m_mass_jpsi2 < 0.) m_mass_jpsi2 = BPhysPVCascadeTools::getParticleMass(pdt, MC::JPSI);
     if(m_mass_psi < 0.) m_mass_psi = BPhysPVCascadeTools::getParticleMass(pdt, MC::PSI2S);
  
     if(m_vtx1Daug1MassHypo < 0.) m_vtx1Daug1MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::MUON);
     if(m_vtx1Daug2MassHypo < 0.) m_vtx1Daug2MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::MUON);
     if(m_vtx1Daug3MassHypo < 0.) m_vtx1Daug3MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::PIPLUS);
     if(m_vtx1Daug4MassHypo < 0.) m_vtx1Daug4MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::PIPLUS);
     if(m_vtx2Daug1MassHypo < 0.) m_vtx2Daug1MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::MUON);
     if(m_vtx2Daug2MassHypo < 0.) m_vtx2Daug2MassHypo = BPhysPVCascadeTools::getParticleMass(pdt, MC::MUON);
  
     return StatusCode::SUCCESS;
   }

◆ inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

◆ interfaceID()

static const InterfaceID& DerivationFramework::JpsiPlusPsiCascade::interfaceID ( )

inlinestatic

Definition at line 36 of file JpsiPlusPsiCascade.h.

36 { return IID_JpsiPlusPsiCascade;}

◆ msg() [1/2]

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< AlgTool >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

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

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                                {
     return this->msgLevel(lvl);
   }

◆ outputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

◆ performSearch()

StatusCode DerivationFramework::JpsiPlusPsiCascade::performSearch	(	std::vector< Trk::VxCascadeInfo * > *	cascadeinfoContainer,
		std::vector< Trk::VxCascadeInfo * > *	cascadeinfoContainer_noConstr
	)		const

Definition at line 341 of file JpsiPlusPsiCascade.cxx.

                                                                                                                                                                     {
     ATH_MSG_DEBUG( "JpsiPlusPsiCascade::performSearch" );
     assert(cascadeinfoContainer!=nullptr && cascadeinfoContainer_noConstr!=nullptr);
  
     // Get TrackParticle container (for setting links to the original tracks)
     SG::ReadHandle<xAOD::TrackParticleContainer> trackContainer(m_trackContainerName);
     ATH_CHECK( trackContainer.isValid() );
  
     std::vector<const xAOD::TrackParticle*> tracksJpsi;
     std::vector<const xAOD::TrackParticle*> tracksDiTrk;
     std::vector<const xAOD::TrackParticle*> tracksPsi;
     std::vector<const xAOD::TrackParticle*> tracksJpsi2;
     std::vector<double> massesPsi;
     massesPsi.push_back(m_vtx1Daug1MassHypo);
     massesPsi.push_back(m_vtx1Daug2MassHypo);
     massesPsi.push_back(m_vtx1Daug3MassHypo);
     if(m_vtx1Daug_num==4) massesPsi.push_back(m_vtx1Daug4MassHypo);
     std::array<double,2> massesJpsi2{m_vtx2Daug1MassHypo, m_vtx2Daug2MassHypo};
  
     // Get Psi container
     SG::ReadHandle<xAOD::VertexContainer> psiContainer(m_vertexPsiContainerKey);
     ATH_CHECK( psiContainer.isValid() );
  
     // Get Jpsi container
     SG::ReadHandle<xAOD::VertexContainer> jpsiContainer(m_vertexContainerKey);
     ATH_CHECK( jpsiContainer.isValid() );
  
     // Select the J/psi candidates before calling cascade fit
     std::vector<const xAOD::Vertex*> selectedJpsiCandidates;
     for(auto vxcItr=jpsiContainer.cptr()->cbegin(); vxcItr!=jpsiContainer.cptr()->cend(); ++vxcItr) {
       // Check the passed flag first
       const xAOD::Vertex* vtx = *vxcItr;
       bool passed = false;
       for(size_t i=0; i<m_vertexJpsiHypoNames.size(); i++) {
     SG::AuxElement::Accessor<Char_t> flagAcc("passed_"+m_vertexJpsiHypoNames[i]);
     if(flagAcc.isAvailable(*vtx) && flagAcc(*vtx)) {
       passed |= 1;
     }
       }
       if(m_vertexJpsiHypoNames.size() && !passed) continue;
  
       // Check Jpsi candidate invariant mass and skip if need be
       double mass_jpsi2 = m_V0Tools->invariantMass(*vxcItr, massesJpsi2);
       if (mass_jpsi2 < m_jpsi2MassLower || mass_jpsi2 > m_jpsi2MassUpper) continue;
  
       double chi2DOF = (*vxcItr)->chiSquared()/(*vxcItr)->numberDoF();
       if(m_chi2cut_Jpsi>0 && chi2DOF>m_chi2cut_Jpsi) continue;
  
       selectedJpsiCandidates.push_back(*vxcItr);
     }
     if(selectedJpsiCandidates.size()==0) return StatusCode::SUCCESS;
  
     // Select the Psi candidates before calling cascade fit
     std::vector<const xAOD::Vertex*> selectedPsiCandidates;
     for(auto vxcItr=psiContainer.cptr()->cbegin(); vxcItr!=psiContainer.cptr()->cend(); ++vxcItr) {
       // Check the passed flag first
       const xAOD::Vertex* vtx = *vxcItr;
       bool passed = false;
       for(size_t i=0; i<m_vertexPsiHypoNames.size(); i++) {
     SG::AuxElement::Accessor<Char_t> flagAcc("passed_"+m_vertexPsiHypoNames[i]);
     if(flagAcc.isAvailable(*vtx) && flagAcc(*vtx)) {
       passed |= 1;
     }
       }
       if(m_vertexPsiHypoNames.size() && !passed) continue;
  
       // Check Psi candidate invariant mass and skip if need be
       double mass_psi = m_V0Tools->invariantMass(*vxcItr,massesPsi);
       if(mass_psi < m_psiMassLower || mass_psi > m_psiMassUpper) continue;
  
       // Add loose cut on Jpsi mass from Psi -> Jpsi pi+ pi-, or on phi mass from Ds+ -> phi pi+
       TLorentzVector p4_mu1, p4_mu2;
       p4_mu1.SetPtEtaPhiM( (*vxcItr)->trackParticle(0)->pt(),
                (*vxcItr)->trackParticle(0)->eta(),
                (*vxcItr)->trackParticle(0)->phi(), m_vtx1Daug1MassHypo);
       p4_mu2.SetPtEtaPhiM( (*vxcItr)->trackParticle(1)->pt(),
                (*vxcItr)->trackParticle(1)->eta(),
                (*vxcItr)->trackParticle(1)->phi(), m_vtx1Daug2MassHypo);
       double mass_jpsi = (p4_mu1 + p4_mu2).M();
       if (mass_jpsi < m_jpsiMassLower || mass_jpsi > m_jpsiMassUpper) continue;
  
       if(m_vtx1Daug_num==4 && m_diTrackMassLower>=0 && m_diTrackMassUpper>=0) {
     TLorentzVector p4_trk1, p4_trk2;
     p4_trk1.SetPtEtaPhiM( (*vxcItr)->trackParticle(2)->pt(),
                   (*vxcItr)->trackParticle(2)->eta(),
                   (*vxcItr)->trackParticle(2)->phi(), m_vtx1Daug3MassHypo);
     p4_trk2.SetPtEtaPhiM( (*vxcItr)->trackParticle(3)->pt(),
                   (*vxcItr)->trackParticle(3)->eta(),
                   (*vxcItr)->trackParticle(3)->phi(), m_vtx1Daug4MassHypo);
     double mass_diTrk = (p4_trk1 + p4_trk2).M();
     if (mass_diTrk < m_diTrackMassLower || mass_diTrk > m_diTrackMassUpper) continue;
       }
  
       double chi2DOF = (*vxcItr)->chiSquared()/(*vxcItr)->numberDoF();
       if(m_chi2cut_Psi>0 && chi2DOF>m_chi2cut_Psi) continue;
  
       selectedPsiCandidates.push_back(*vxcItr);
     }
     if(selectedPsiCandidates.size()==0) return StatusCode::SUCCESS;
  
     std::vector<std::pair<const xAOD::Vertex*, const xAOD::Vertex*> > candidatePairs;
     for(auto jpsiItr=selectedJpsiCandidates.cbegin(); jpsiItr!=selectedJpsiCandidates.cend(); ++jpsiItr) {
       tracksJpsi2.clear();
       for(size_t i=0; i<(*jpsiItr)->nTrackParticles(); i++) tracksJpsi2.push_back((*jpsiItr)->trackParticle(i));
       for(auto psiItr=selectedPsiCandidates.cbegin(); psiItr!=selectedPsiCandidates.cend(); ++psiItr) {
     bool skip = false;
     for(size_t j=0; j<(*psiItr)->nTrackParticles(); j++) {
       if(std::find(tracksJpsi2.cbegin(), tracksJpsi2.cend(), (*psiItr)->trackParticle(j)) != tracksJpsi2.cend()) { skip = true; break; }
     }
     if(skip) continue;
     candidatePairs.push_back(std::pair<const xAOD::Vertex*, const xAOD::Vertex*>(*jpsiItr,*psiItr));
       }
     }
  
     std::sort( candidatePairs.begin(), candidatePairs.end(), [](std::pair<const xAOD::Vertex*, const xAOD::Vertex*> a, std::pair<const xAOD::Vertex*, const xAOD::Vertex*> b) { return a.first->chiSquared()/a.first->numberDoF()+a.second->chiSquared()/a.second->numberDoF() < b.first->chiSquared()/b.first->numberDoF()+b.second->chiSquared()/b.second->numberDoF(); } );
     if(m_maxCandidates>0 && candidatePairs.size()>m_maxCandidates) {
       candidatePairs.erase(candidatePairs.begin()+m_maxCandidates, candidatePairs.end());
     }
  
     for(size_t ic=0; ic<candidatePairs.size(); ic++) {
       const xAOD::Vertex* jpsiVertex = candidatePairs[ic].first;
       const xAOD::Vertex* psiVertex = candidatePairs[ic].second;
  
       tracksJpsi2.clear();
       for(size_t it=0; it<jpsiVertex->nTrackParticles(); it++) tracksJpsi2.push_back(jpsiVertex->trackParticle(it));
       if (tracksJpsi2.size() != 2 || massesJpsi2.size() != 2) {
     ATH_MSG_ERROR("Problems with Jpsi input: number of tracks or track mass inputs is not 2!");
       }
       tracksPsi.clear();
       for(size_t it=0; it<psiVertex->nTrackParticles(); it++) tracksPsi.push_back(psiVertex->trackParticle(it));
       if (tracksPsi.size() != massesPsi.size()) {
     ATH_MSG_ERROR("Problems with Psi input: number of tracks or track mass inputs is not correct!");
       }
       
       tracksJpsi.clear();
       tracksJpsi.push_back(psiVertex->trackParticle(0));
       tracksJpsi.push_back(psiVertex->trackParticle(1));
       tracksDiTrk.clear();
       if(m_vtx1Daug_num==4) {
     tracksDiTrk.push_back(psiVertex->trackParticle(2));
     tracksDiTrk.push_back(psiVertex->trackParticle(3));
       }
  
       TLorentzVector p4_moth;
       TLorentzVector tmp;
       for(size_t it=0; it<jpsiVertex->nTrackParticles(); it++) {
     tmp.SetPtEtaPhiM(jpsiVertex->trackParticle(it)->pt(),jpsiVertex->trackParticle(it)->eta(),jpsiVertex->trackParticle(it)->phi(),massesJpsi2[it]);
     p4_moth += tmp;
       }
       for(size_t it=0; it<psiVertex->nTrackParticles(); it++) {
     tmp.SetPtEtaPhiM(psiVertex->trackParticle(it)->pt(),psiVertex->trackParticle(it)->eta(),psiVertex->trackParticle(it)->phi(),massesPsi[it]);
     p4_moth += tmp;
       }
       if (p4_moth.M() < m_MassLower || p4_moth.M() > m_MassUpper) continue;
  
       // Apply the user's settings to the fitter
       std::unique_ptr<Trk::IVKalState> state = m_iVertexFitter->makeState();
       // Robustness: http://cdsweb.cern.ch/record/685551
       int robustness = 0;
       m_iVertexFitter->setRobustness(robustness, *state);
       // Build up the topology
       // Vertex list
       std::vector<Trk::VertexID> vrtList;
       // Psi vertex
       Trk::VertexID vID1;
       // https://gitlab.cern.ch/atlas/athena/-/blob/21.2/Tracking/TrkVertexFitter/TrkVKalVrtFitter/TrkVKalVrtFitter/IVertexCascadeFitter.h
       if (m_constrPsi) {
     vID1 = m_iVertexFitter->startVertex(tracksPsi,massesPsi,*state,m_mass_psi);
       } else {
     vID1 = m_iVertexFitter->startVertex(tracksPsi,massesPsi,*state);
       }
       vrtList.push_back(vID1);
       // Jpsi vertex
       Trk::VertexID vID2;
       if (m_constrJpsi2) {
     vID2 = m_iVertexFitter->nextVertex(tracksJpsi2,massesJpsi2,*state,m_mass_jpsi2);
       } else {
     vID2 = m_iVertexFitter->nextVertex(tracksJpsi2,massesJpsi2,*state);
       }
       vrtList.push_back(vID2);
       // Mother vertex including Jpsi and Psi
       std::vector<const xAOD::TrackParticle*> tp; tp.clear();
       std::vector<double> tp_masses; tp_masses.clear();
       m_iVertexFitter->nextVertex(tp,tp_masses,vrtList,*state);
       if (m_constrJpsi) {
     std::vector<Trk::VertexID> cnstV; cnstV.clear();
     if ( !m_iVertexFitter->addMassConstraint(vID1,tracksJpsi,cnstV,*state,m_mass_jpsi).isSuccess() ) {
       ATH_MSG_WARNING("addMassConstraint for Jpsi failed");
     }
       }
       if (m_constrDiTrk && m_vtx1Daug_num==4 && m_mass_diTrk>0) {
     std::vector<Trk::VertexID> cnstV; cnstV.clear();
     if ( !m_iVertexFitter->addMassConstraint(vID1,tracksDiTrk,cnstV,*state,m_mass_diTrk).isSuccess() ) {
       ATH_MSG_WARNING("addMassConstraint for DiTrk failed");
     }
       }
       // Do the work
       std::unique_ptr<Trk::VxCascadeInfo> result(m_iVertexFitter->fitCascade(*state));
  
       bool pass = false;
       if (result != nullptr) {
     for(auto v : result->vertices()) {
       if(v->nTrackParticles()==0) {
         std::vector<ElementLink<xAOD::TrackParticleContainer> > nullLinkVector;
         v->setTrackParticleLinks(nullLinkVector);
       }
     }
     // reset links to original tracks
     BPhysPVCascadeTools::PrepareVertexLinks(result.get(), trackContainer.cptr());
  
     // necessary to prevent memory leak
     result->setSVOwnership(true);
  
     // Chi2/DOF cut
     double chi2DOF = result->fitChi2()/result->nDoF();
     bool chi2CutPassed = (m_chi2cut <= 0.0 || chi2DOF < m_chi2cut);
  
     if(chi2CutPassed) {
       cascadeinfoContainer->push_back(result.release());
       pass = true;
     }
       }
  
       // do cascade fit again without any mass constraints
       if(pass) {
     if(m_constrJpsi || m_constrPsi || m_constrJpsi2 || (m_constrDiTrk && m_vtx1Daug_num==4 && m_mass_diTrk>0)) {
       std::unique_ptr<Trk::IVKalState> state (m_iVertexFitter->makeState());
       m_iVertexFitter->setRobustness(robustness, *state);
       std::vector<Trk::VertexID> vrtList_nc;
       // Psi vertex
       Trk::VertexID vID1_nc = m_iVertexFitter->startVertex(tracksPsi,massesPsi,*state);
       vrtList_nc.push_back(vID1_nc);
       Trk::VertexID vID2_nc = m_iVertexFitter->nextVertex(tracksJpsi2,massesJpsi2,*state);
       vrtList_nc.push_back(vID2_nc);
       // Mother vertex including Jpsi and Psi
       std::vector<const xAOD::TrackParticle*> tp; tp.clear();
       std::vector<double> tp_masses; tp_masses.clear();
       m_iVertexFitter->nextVertex(tp,tp_masses,vrtList_nc,*state);
       // Do the work
       std::unique_ptr<Trk::VxCascadeInfo> result_nc(m_iVertexFitter->fitCascade(*state));
  
       if (result_nc != nullptr) {
         for(auto v : result_nc->vertices()) {
           if(v->nTrackParticles()==0) {
         std::vector<ElementLink<xAOD::TrackParticleContainer> > nullLinkVector;
         v->setTrackParticleLinks(nullLinkVector);
           }
         }
         // reset links to original tracks
         BPhysPVCascadeTools::PrepareVertexLinks(result_nc.get(), trackContainer.cptr());
  
         // necessary to prevent memory leak
         result_nc->setSVOwnership(true);
         cascadeinfoContainer_noConstr->push_back(result_nc.release());
       }
       else cascadeinfoContainer_noConstr->push_back(0);
     }
     else cascadeinfoContainer_noConstr->push_back(0);
       }
     } //Iterate over candidatePairs
  
     return StatusCode::SUCCESS;
   }

◆ 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< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

◆ renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in DerivationFramework::CfAthAlgTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and asg::AsgMetadataTool.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::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_cascadeOutputsKeys

SG::WriteHandleKeyArray<xAOD::VertexContainer> DerivationFramework::JpsiPlusPsiCascade::m_cascadeOutputsKeys

private

Definition at line 48 of file JpsiPlusPsiCascade.h.

◆ m_CascadeTools

ToolHandle< DerivationFramework::CascadeTools > DerivationFramework::JpsiPlusPsiCascade::m_CascadeTools

private

Definition at line 87 of file JpsiPlusPsiCascade.h.

◆ m_chi2cut

double DerivationFramework::JpsiPlusPsiCascade::m_chi2cut

private

Definition at line 81 of file JpsiPlusPsiCascade.h.

◆ m_chi2cut_Jpsi

double DerivationFramework::JpsiPlusPsiCascade::m_chi2cut_Jpsi

private

Definition at line 80 of file JpsiPlusPsiCascade.h.

◆ m_chi2cut_Psi

double DerivationFramework::JpsiPlusPsiCascade::m_chi2cut_Psi

private

Definition at line 79 of file JpsiPlusPsiCascade.h.

◆ m_constrDiTrk

bool DerivationFramework::JpsiPlusPsiCascade::m_constrDiTrk

private

Definition at line 77 of file JpsiPlusPsiCascade.h.

◆ m_constrJpsi

bool DerivationFramework::JpsiPlusPsiCascade::m_constrJpsi

private

Definition at line 76 of file JpsiPlusPsiCascade.h.

◆ m_constrJpsi2

bool DerivationFramework::JpsiPlusPsiCascade::m_constrJpsi2

private

Definition at line 78 of file JpsiPlusPsiCascade.h.

◆ m_constrPsi

bool DerivationFramework::JpsiPlusPsiCascade::m_constrPsi

private

Definition at line 75 of file JpsiPlusPsiCascade.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_diTrackMassLower

double DerivationFramework::JpsiPlusPsiCascade::m_diTrackMassLower

private

Definition at line 55 of file JpsiPlusPsiCascade.h.

◆ m_diTrackMassUpper

double DerivationFramework::JpsiPlusPsiCascade::m_diTrackMassUpper

private

Definition at line 56 of file JpsiPlusPsiCascade.h.

◆ m_DoVertexType

int DerivationFramework::JpsiPlusPsiCascade::m_DoVertexType

private

Definition at line 94 of file JpsiPlusPsiCascade.h.

◆ m_eventInfo_key

SG::ReadHandleKey<xAOD::EventInfo> DerivationFramework::JpsiPlusPsiCascade::m_eventInfo_key

private

Definition at line 51 of file JpsiPlusPsiCascade.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_hypoName

std::string DerivationFramework::JpsiPlusPsiCascade::m_hypoName

private

Definition at line 92 of file JpsiPlusPsiCascade.h.

◆ m_iVertexFitter

ToolHandle< Trk::TrkVKalVrtFitter > DerivationFramework::JpsiPlusPsiCascade::m_iVertexFitter

private

Definition at line 84 of file JpsiPlusPsiCascade.h.

◆ m_jpsi2MassLower

double DerivationFramework::JpsiPlusPsiCascade::m_jpsi2MassLower

private

Definition at line 59 of file JpsiPlusPsiCascade.h.

◆ m_jpsi2MassUpper

double DerivationFramework::JpsiPlusPsiCascade::m_jpsi2MassUpper

private

Definition at line 60 of file JpsiPlusPsiCascade.h.

◆ m_jpsiMassLower

double DerivationFramework::JpsiPlusPsiCascade::m_jpsiMassLower

private

Definition at line 53 of file JpsiPlusPsiCascade.h.

◆ m_jpsiMassUpper

double DerivationFramework::JpsiPlusPsiCascade::m_jpsiMassUpper

private

Definition at line 54 of file JpsiPlusPsiCascade.h.

◆ m_mass_diTrk

double DerivationFramework::JpsiPlusPsiCascade::m_mass_diTrk

private

Definition at line 72 of file JpsiPlusPsiCascade.h.

◆ m_mass_jpsi

double DerivationFramework::JpsiPlusPsiCascade::m_mass_jpsi

private

Definition at line 71 of file JpsiPlusPsiCascade.h.

◆ m_mass_jpsi2

double DerivationFramework::JpsiPlusPsiCascade::m_mass_jpsi2

private

Definition at line 74 of file JpsiPlusPsiCascade.h.

◆ m_mass_psi

double DerivationFramework::JpsiPlusPsiCascade::m_mass_psi

private

Definition at line 73 of file JpsiPlusPsiCascade.h.

◆ m_MassLower

double DerivationFramework::JpsiPlusPsiCascade::m_MassLower

private

Definition at line 61 of file JpsiPlusPsiCascade.h.

◆ m_MassUpper

double DerivationFramework::JpsiPlusPsiCascade::m_MassUpper

private

Definition at line 62 of file JpsiPlusPsiCascade.h.

◆ m_maxCandidates

unsigned int DerivationFramework::JpsiPlusPsiCascade::m_maxCandidates

private

Definition at line 82 of file JpsiPlusPsiCascade.h.

◆ m_partPropSvc

ServiceHandle<IPartPropSvc> DerivationFramework::JpsiPlusPsiCascade::m_partPropSvc {this, "PartPropSvc", "PartPropSvc"}

private

Definition at line 88 of file JpsiPlusPsiCascade.h.

◆ m_psiMassLower

double DerivationFramework::JpsiPlusPsiCascade::m_psiMassLower

private

Definition at line 57 of file JpsiPlusPsiCascade.h.

◆ m_psiMassUpper

double DerivationFramework::JpsiPlusPsiCascade::m_psiMassUpper

private

Definition at line 58 of file JpsiPlusPsiCascade.h.

◆ m_PV_max

int DerivationFramework::JpsiPlusPsiCascade::m_PV_max

private

Definition at line 93 of file JpsiPlusPsiCascade.h.

◆ m_PV_minNTracks

size_t DerivationFramework::JpsiPlusPsiCascade::m_PV_minNTracks

private

Definition at line 95 of file JpsiPlusPsiCascade.h.

◆ m_pvRefitter

ToolHandle< Analysis::PrimaryVertexRefitter > DerivationFramework::JpsiPlusPsiCascade::m_pvRefitter

private

Definition at line 85 of file JpsiPlusPsiCascade.h.

◆ m_refitPV

bool DerivationFramework::JpsiPlusPsiCascade::m_refitPV

private

Definition at line 90 of file JpsiPlusPsiCascade.h.

◆ m_refPVContainerName

SG::WriteHandleKey<xAOD::VertexContainer> DerivationFramework::JpsiPlusPsiCascade::m_refPVContainerName

private

Definition at line 91 of file JpsiPlusPsiCascade.h.

◆ m_trackContainerName

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::JpsiPlusPsiCascade::m_trackContainerName

private

Definition at line 50 of file JpsiPlusPsiCascade.h.

◆ m_V0Tools

ToolHandle< Trk::V0Tools > DerivationFramework::JpsiPlusPsiCascade::m_V0Tools

private

Definition at line 86 of file JpsiPlusPsiCascade.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vertexContainerKey

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::JpsiPlusPsiCascade::m_vertexContainerKey

private

Definition at line 44 of file JpsiPlusPsiCascade.h.

◆ m_vertexJpsiHypoNames

std::vector<std::string> DerivationFramework::JpsiPlusPsiCascade::m_vertexJpsiHypoNames

private

Definition at line 46 of file JpsiPlusPsiCascade.h.

◆ m_vertexPsiContainerKey

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::JpsiPlusPsiCascade::m_vertexPsiContainerKey

private

Definition at line 45 of file JpsiPlusPsiCascade.h.

◆ m_vertexPsiHypoNames

std::vector<std::string> DerivationFramework::JpsiPlusPsiCascade::m_vertexPsiHypoNames

private

Definition at line 47 of file JpsiPlusPsiCascade.h.

◆ m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_vtx1Daug1MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx1Daug1MassHypo

private

Definition at line 64 of file JpsiPlusPsiCascade.h.

◆ m_vtx1Daug2MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx1Daug2MassHypo

private

Definition at line 65 of file JpsiPlusPsiCascade.h.

◆ m_vtx1Daug3MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx1Daug3MassHypo

private

Definition at line 66 of file JpsiPlusPsiCascade.h.

◆ m_vtx1Daug4MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx1Daug4MassHypo

private

Definition at line 67 of file JpsiPlusPsiCascade.h.

◆ m_vtx1Daug_num

int DerivationFramework::JpsiPlusPsiCascade::m_vtx1Daug_num

private

Definition at line 63 of file JpsiPlusPsiCascade.h.

◆ m_vtx2Daug1MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx2Daug1MassHypo

private

Definition at line 68 of file JpsiPlusPsiCascade.h.

◆ m_vtx2Daug2MassHypo

double DerivationFramework::JpsiPlusPsiCascade::m_vtx2Daug2MassHypo

private

Definition at line 69 of file JpsiPlusPsiCascade.h.

◆ m_VxPrimaryCandidateName

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::JpsiPlusPsiCascade::m_VxPrimaryCandidateName

private

Name of primary vertex container.

Definition at line 49 of file JpsiPlusPsiCascade.h.

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

Public Member Functions

Static Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ JpsiPlusPsiCascade()

◆ ~JpsiPlusPsiCascade()

Member Function Documentation

◆ addBranches()

◆ declareGaudiProperty() [1/4]

◆ declareGaudiProperty() [2/4]

◆ declareGaudiProperty() [3/4]

◆ declareGaudiProperty() [4/4]

◆ declareProperty() [1/6]

◆ declareProperty() [2/6]

◆ declareProperty() [3/6]

◆ declareProperty() [4/6]

◆ declareProperty() [5/6]

◆ declareProperty() [6/6]

◆ detStore()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ extraDeps_update_handler()

◆ initialize()

◆ inputHandles()

◆ interfaceID()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ outputHandles()

◆ performSearch()

◆ renounce()

◆ renounceArray()

◆ sysInitialize()

◆ sysStart()

◆ updateVHKA()

Member Data Documentation

◆ m_cascadeOutputsKeys

◆ m_CascadeTools

◆ m_chi2cut

◆ m_chi2cut_Jpsi

◆ m_chi2cut_Psi

◆ m_constrDiTrk

◆ m_constrJpsi

◆ m_constrJpsi2

◆ m_constrPsi

◆ m_detStore

◆ m_diTrackMassLower

◆ m_diTrackMassUpper

◆ m_DoVertexType

◆ m_eventInfo_key

◆ m_evtStore

◆ m_hypoName

◆ m_iVertexFitter

◆ m_jpsi2MassLower

◆ m_jpsi2MassUpper

◆ m_jpsiMassLower

◆ m_jpsiMassUpper

◆ m_mass_diTrk

◆ m_mass_jpsi

◆ m_mass_jpsi2

◆ m_mass_psi

◆ m_MassLower

◆ m_MassUpper

◆ m_maxCandidates

◆ m_partPropSvc

◆ m_psiMassLower

◆ m_psiMassUpper

◆ m_PV_max

◆ m_PV_minNTracks

◆ m_pvRefitter

◆ m_refitPV

◆ m_refPVContainerName

◆ m_trackContainerName

◆ m_V0Tools