/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/tauRecTools/src/TauVertexFinder.h

#ifndef TAUREC_TAUVERTEXFINDER_H
#define TAUREC_TAUVERTEXFINDER_H

#include "tauRecTools/TauRecToolBase.h"
#include "GaudiKernel/ToolHandle.h"
#include "xAODTracking/VertexContainer.h"
#include "JetEDM/TrackVertexAssociation.h"


class TauVertexFinder : virtual public TauRecToolBase {
public:
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    TauVertexFinder(const std::string& name);
    ASG_TOOL_CLASS2(TauVertexFinder, TauRecToolBase, ITauToolBase);
    ~TauVertexFinder();

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    virtual StatusCode initialize();
    virtual StatusCode eventInitialize();
    virtual StatusCode execute(xAOD::TauJet& pTau);
    virtual StatusCode eventFinalize();
    virtual StatusCode finalize();

    virtual void cleanup(xAOD::TauJet* ) { }
    virtual void print() const { }

    ElementLink<xAOD::VertexContainer> getPV_TJVA(const xAOD::TauJet& tauJet, const xAOD::VertexContainer& vertices);

private:
    float getJetVertexFraction(const xAOD::Vertex* vertex, const std::vector<const xAOD::TrackParticle*>& tracks, const jet::TrackVertexAssociation* tva) const;
        
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    template <class T>
    bool openContainer(T* &container, std::string containerName, bool printFATAL=false);

    template <class T>
    bool retrieveTool(T &tool);

private:
    bool m_printMissingContainerINFO;
    float m_maxJVF;

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    bool m_useTJVA;
    std::string m_inputPrimaryVertexContainerName;
    std::string m_assocTracksName;
    std::string m_trackVertexAssocName;    
};

#endif // not TAUREC_TAUVERTEXFINDER_H

---