ReVertex.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
// ****************************************************************************
// ----------------------------------------------------------------------------
// ReVertex header file
//
// Konstantin Beloborodov <Konstantin.Beloborodov@cern.ch>
//
// ----------------------------------------------------------------------------
// ****************************************************************************
 
#ifndef DERIVATIONFRAMEWORK_ReVertex_H
#define DERIVATIONFRAMEWORK_ReVertex_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "xAODTracking/Vertex.h"
#include "xAODEventInfo/EventInfo.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "xAODTracking/TrackParticleContainerFwd.h"

namespace Trk {
    class IVertexFitter;
    class TrkV0VertexFitter;
    class ITrackSelectorTool;
    class V0Tools;
    class TrkVKalVrtFitter;
}
 
namespace Analysis {
    class PrimaryVertexRefitter;
}
 
namespace InDet { class VertexPointEstimator; }
 
namespace DerivationFramework {
 
class ReVertex : public extends<AthAlgTool, IAugmentationTool> {
public:
 
    ReVertex(const std::string& t, const std::string& n, const IInterface* p);
 
    virtual StatusCode initialize() override;
 
    virtual StatusCode addBranches(const EventContext& ctx) const override;
 
    void fitAndStore(const EventContext& ctx,
            xAOD::VertexContainer* vtxContainer,
            const xAOD::Vertex* v,
            const xAOD::VertexContainer    *InVtxContainer,
            const std::vector<const xAOD::TrackParticle*> &inputTracks,
            const xAOD::TrackParticleContainer* importedTrackCollection,
            const xAOD::VertexContainer* pvContainer) const;
            std::unique_ptr<xAOD::Vertex> fit(const EventContext& ctx,
             const std::vector<const xAOD::TrackParticle*> &inputTracks,
             const xAOD::TrackParticleContainer* importedTrackCollection,
             const xAOD::Vertex* pv) const;
private:
    std::vector<int> m_TrackIndices;
    ToolHandle < InDet::VertexPointEstimator > m_vertexEstimator;
    ToolHandle < Trk::IVertexFitter > m_iVertexFitter;
    Trk::TrkVKalVrtFitter* m_VKVFitter{};
    SG::WriteHandleKey<xAOD::VertexContainer> m_OutputContainerName;
    SG::ReadHandleKey<xAOD::VertexContainer> m_inputContainerName;
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackContainer;
    SG::WriteHandleKey<xAOD::VertexContainer> m_refPVContainerName;
    SG::ReadHandleKey<xAOD::VertexContainer> m_pvContainerName;
    SG::ReadHandleKey<xAOD::VertexContainer> m_defaultPVContainerName;
 
    std::vector<double> m_trkMasses;
    std::vector<int> m_indices;
    double m_massConst{};
    double m_totalMassConst{};
    std::vector<std::string> m_hypoNames;
 
    PublicToolHandle<Trk::V0Tools>                    m_v0Tools;
    PublicToolHandle<Analysis::PrimaryVertexRefitter> m_pvRefitter;
    SG::ReadHandleKey<xAOD::EventInfo> m_eventInfo_key{this, "EventInfo", "EventInfo", "Input event information"};
    int         m_PV_max{};
    int         m_DoVertexType{};
    size_t      m_PV_minNTracks{};
    bool        m_do3d{};
    bool        m_AddPVData{};
    bool        m_refitPV{};
    bool        m_doMassConst{};
    bool        m_startingpoint0{};
    
    bool m_vertexFittingWithPV{};
 
   double m_BMassUpper{};
   double m_BMassLower{};
   double m_chi2cut{};                 // chi2/Ndof of the final veretx
   double m_trkDeltaZ{};               // DeltaZ between the JPsi vertex and hadronic tracks Z0
 
   bool m_useAdditionalTrack{};
 
   SG::ReadHandleKeyArray<xAOD::VertexContainer> m_CollectionsToCheck{this, "CheckVertexContainers", {}};
   SG::ReadHandleKeyArray<xAOD::TrackParticleContainer> m_RelinkContainers{this, "RelinkTracks", {}, "Track Containers if they need to be relinked through indirect use" };
 
};
}
 
#endif