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

#ifndef TAUREC_TAUTRACKFINDER_H
#define TAUREC_TAUTRACKFINDER_H

#include "tauRecTools/TauRecToolBase.h"
#include "GaudiKernel/ToolHandle.h"
#include "ITrackToVertex/ITrackToVertex.h"

#include "xAODTracking/Vertex.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"

#include "VxVertex/RecVertex.h"

namespace Trk {
  class ITrackSelectorTool;
  class IParticleCaloExtensionTool;
}


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

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    enum TauTrackType
    {
        TauTrackCore  = 0,
        TauTrackWide  = 1,
        TauTrackOther = 2,
        NotTauTrack   = 3
    };

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    virtual StatusCode initialize();
    virtual StatusCode eventInitialize();
    virtual StatusCode execute(xAOD::TauJet& pTau);
    virtual StatusCode eventFinalize();
    virtual StatusCode finalize();
    
    virtual void print() const { }
    
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    StatusCode extrapolateToCaloSurface(xAOD::TauJet& pTau);

    TauTrackType tauTrackType( const xAOD::TauJet& tauJet,
            const xAOD::TrackParticle& trackParticle,
            const xAOD::Vertex* primaryVertex);

    void getTauTracksFromPV( const xAOD::TauJet& tauJet,
            const xAOD::TrackParticleContainer& trackParticleCont,
            const xAOD::Vertex* primaryVertex,
            std::vector<const xAOD::TrackParticle*> &tauTracks,
            std::vector<const xAOD::TrackParticle*> &wideTracks,
            std::vector<const xAOD::TrackParticle*> &otherTracks);

    // old style AOD version
    void removeOffsideTracksWrtLeadTrk(std::vector<const Rec::TrackParticle*> &tauTracks,
                                           std::vector<const Rec::TrackParticle*> &wideTracks,
                                           std::vector<const Rec::TrackParticle*> &otherTracks,
                                           const Trk::RecVertex* tauOrigin,
                                           double maxDeltaZ0);

    // new xAOD version
    void removeOffsideTracksWrtLeadTrk(std::vector<const xAOD::TrackParticle*> &tauTracks,
                                           std::vector<const xAOD::TrackParticle*> &wideTracks,
                                           std::vector<const xAOD::TrackParticle*> &otherTracks,
                                           const xAOD::Vertex* tauOrigin,
                                           double maxDeltaZ0);

    void  getDeltaZ0Values(std::vector<float>& vDeltaZ0coreTrks, std::vector<float>& vDeltaZ0wideTrks);
    void  resetDeltaZ0Cache();

private:
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    float getZ0(const Rec::TrackParticle* track, const Trk::RecVertex* vertex);  //AOD version
    float getZ0(const xAOD::TrackParticle* track, const xAOD::Vertex* vertex);   //xAOD version

private:
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    std::string m_inputTauJetContainerName;
    std::string m_inputTrackParticleContainerName;
    std::string m_inputPrimaryVertexContainerName;

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    ToolHandle< Trk::IParticleCaloExtensionTool >  m_caloExtensionTool;
    ToolHandle<Trk::ITrackSelectorTool> m_trackSelectorTool_tau;
    ToolHandle<Reco::ITrackToVertex> m_trackToVertexTool;
    
    //-------------------------------------------------------------
    //-------------------------------------------------------------
    double  m_maxJetDr_tau;
    double  m_maxJetDr_wide;
    
    //-------------------------------------------------------------
    // z0 cuts
    //-------------------------------------------------------------
    float m_z0maxDelta;
    bool m_applyZ0cut;
    bool m_storeInOtherTrks;
    std::vector<float> m_vDeltaZ0coreTrks;
    std::vector<float> m_vDeltaZ0wideTrks;

    //-------------------------------------------------------------
    // Bypass TrackSelectorTool / Extrapolation
    //-------------------------------------------------------------

    bool m_bypassSelector;
    bool m_bypassExtrapolator;

    //-------------------------------------------------------------
    //-------------------------------------------------------------
    template <class T>
    bool openContainer(T* &container, std::string containerName, bool printFATAL=false);

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

};

#endif

---