JetFitterInitializationHelper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrkJetFitter_JetFitterInitializationHelper_H
#define TrkJetFitter_JetFitterInitializationHelper_H
 
#include "GaudiKernel/ToolHandle.h"
#include <utility>
#include <vector>
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkTrackLink/ITrackLink.h"
 
 
namespace Trk {
  class TrackParticleBase;
  class Track;
  class VxJetCandidate;
  class RecVertex;
  class IVertexLinearizedTrackFactory;
 
  static const InterfaceID IID_JetFitterInitializationHelper("Trk::JetFitterInitializationHelper", 1, 0);
 
 
  class JetFitterInitializationHelper : public AthAlgTool
  {
 
  public:
 
    static const InterfaceID& interfaceID() {
      return IID_JetFitterInitializationHelper;
    }
 
    virtual StatusCode initialize() override;
 
    JetFitterInitializationHelper(const std::string& t, const std::string& n, const IInterface*  p);
 
    ~JetFitterInitializationHelper();
 
    VxJetCandidate * initializeJetCandidate(const std::vector<const Trk::ITrackLink*> & vectorOfLink,
                        const RecVertex* primaryVertex,
                        const Amg::Vector3D* jetdirection=0,
                                            const Amg::Vector3D* linearizationjetdirection=0) const;
 
 
    VxJetCandidate * initializeJetCandidate(const std::vector<const Trk::Track*> & vectorOfT,
                        const RecVertex* primaryVertex,
                        const Amg::Vector3D* jetdirection=0,
                                            const Amg::Vector3D* linearizationjetdirection=0) const;
    
    VxJetCandidate * initializeJetCandidate(const std::vector<const Trk::TrackParticleBase*> & vectorOfTP,
                        const RecVertex* primaryVertex,
                        const Amg::Vector3D* jetdirection=0,
                                            const Amg::Vector3D* linearizationjetdirection=0) const;
    
    static void updateTrackNumbering(VxJetCandidate*) ;
 
    void linearizeAllTracks(VxJetCandidate*,
                bool signfliptreatment=false,
                double maxdistance=1.) const;    //CLHEP::mm
 
  private:
 
    VxJetCandidate * initializeJetClusters(VxJetCandidate* myJetCandidate,
                       const RecVertex* primaryVertex,
                       const Amg::Vector3D* jetdirection=0,
                                           const Amg::Vector3D* linearizationjetdirection=0) const;
                       
 
    ToolHandle<IVertexLinearizedTrackFactory> m_linearizedFactory;
 
    float m_errphiJetAxis;
 
    float m_erretaJetAxis;
 
  };//end class declaration
 
}//end namespace Trk
 
 
#endif