BPhysPVTools.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_PVTOOLS_H
#define DERIVATIONFRAMEWORK_PVTOOLS_H
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthMessaging.h"
#include "xAODBPhys/BPhysHelper.h"
#include <vector>
#include "xAODEventInfo/EventInfo.h"
  
// Author: Adam Barton <abarton@SPAMMENOTTtttcern.ch>
namespace Trk {
  class V0Tools;
}
namespace Analysis{
  class PrimaryVertexRefitter;
}
 
 
namespace DerivationFramework {
  
  class BPhysPVTools : public AthMessaging {
 
  private:
       const Trk::V0Tools *m_v0Tools;
       const xAOD::EventInfo *m_EvtData;

       size_t m_PV_minNTracks;
 
       bool m_3dCalc;
       
  public:
  
       BPhysPVTools(const Trk::V0Tools *v0Tools);
       BPhysPVTools(const Trk::V0Tools *v0Tools, const xAOD::EventInfo*);
       void SetSave3d(bool v) { m_3dCalc =v; }
       StatusCode FillCandExistingVertices(xAOD::VertexContainer* vtxContainer, const xAOD::VertexContainer* pvContainer, int DoVertexType);
       
       void FillBPhysHelperNULL(xAOD::BPhysHelper &vtx, const xAOD::VertexContainer* PvContainer,
           xAOD::BPhysHelper::pv_type pvtype, bool do3d = false) const;
       
       StatusCode FillCandwithRefittedVertices(xAOD::VertexContainer* vtxContainer, const xAOD::VertexContainer* pvContainer,xAOD::VertexContainer* refPvContainer, const Analysis::PrimaryVertexRefitter* , size_t in_PV_max, int DoVertexType);
       
       void DecorateWithNULL(xAOD::VertexContainer* vtxContainer,const xAOD::VertexContainer* pvContainer, int DoVertexType) const;
       
       void DecorateWithDummyVertex(xAOD::VertexContainer* vtxContainer, const xAOD::VertexContainer* pvContainer, const xAOD::Vertex* Dummy, int DoVertexType, bool SetOrignal) const;
       
       void FillBPhysHelper(xAOD::BPhysHelper &vtx, const xAOD::Vertex* refPV,const xAOD::VertexContainer* refPvContainer,
                    xAOD::BPhysHelper::pv_type pvtype, int) const;
    
       size_t FindLowZIndex(const xAOD::BPhysHelper &Obj,
                const std::vector<const xAOD::Vertex*> &PVlist,
                const size_t PV_minNTracks=0) const;
       size_t FindLowA0Index(const xAOD::BPhysHelper &Obj,
                 const std::vector<const xAOD::Vertex*> &PVlist,
                 const size_t PV_minNTracks=0) const;
       
       size_t FindHighPtIndex(const std::vector<const xAOD::Vertex*> &PVlist) const;
       
       static std::vector<const xAOD::Vertex*> GetGoodPV(const xAOD::VertexContainer* pvContainer);
       
       void SetMinNTracksInPV(size_t PV_minNTracks);

       [[nodiscard]] Amg::Vector3D GetBeamSpot() const;

       size_t FindLowZ0BAIndex(xAOD::BPhysHelper &obj,
                   const std::vector<const xAOD::Vertex*> &PVlist,
                   const size_t PV_minNTracks=0) const;
       double DistInZtoDOCA(xAOD::BPhysHelper &obj,
                const xAOD::Vertex* vertex) const;
       Amg::Vector3D DocaExtrapToBeamSpot(xAOD::BPhysHelper &obj) const;
 
       static void PrepareVertexLinks(xAOD::Vertex* theResult,
               const xAOD::TrackParticleContainer* importedTrackCollection);
  }; // class BPhysPVTools
 
} // namespace DerivationFramework
 
 
#endif // DERIVATIONFRAMEWORK_PVTOOLS_H