/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/JetMomentTools/JetMomentTools/JetVertexFractionTool.h

// JetVertexFractionTool.h

#ifndef JETMOMENTTOOLS_JETVERTEXFRACTIONTOOL_H
#define JETMOMENTTOOLS_JETVERTEXFRACTIONTOOL_H



#include "AsgTools/ToolHandle.h"

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

#include "JetInterface/IJetTrackSelector.h"
#include "JetRec/JetModifierBase.h"
#include "JetEDM/TrackVertexAssociation.h"

#include <vector>
#include <string>

class JetVertexFractionTool : public JetModifierBase {
  ASG_TOOL_CLASS(JetVertexFractionTool,IJetModifier)

public:  // methods

  // Constructor from tool name
  JetVertexFractionTool(const std::string& name);

  // Initialization.
  StatusCode initialize();

  // Inherited methods to modify a jet
  // Calls getJetVertexFraction and puts the result in the jet
  virtual int modifyJet(xAOD::Jet& jet) const;

  // Local method to calculate and return the JVF vector
  const std::vector<float> getJetVertexFraction(const xAOD::VertexContainer*,
                                                const std::vector<const xAOD::TrackParticle*>&,
                                                const jet::TrackVertexAssociation*) const;

  // Local method to calculate the JVF for a given vertex
  float getJetVertexFraction(const xAOD::Vertex*,
                             const std::vector<const xAOD::TrackParticle*>&,
                             const jet::TrackVertexAssociation*) const;

  // Local method to determine the highest JVF vertex and get an ElementLink to it
  ElementLink<xAOD::VertexContainer> getMaxJetVertexFraction(const xAOD::VertexContainer*,
                                                                   const std::vector<float>&) const;

private:  // data

  // Configurable parameters
  std::string m_verticesName;
  std::string m_assocTracksName;
  std::string m_tvaName;
  ToolHandle<IJetTrackSelector> m_htsel;
  std::string m_jvfname;

private:  // methods

  std::vector<float> getEmptyJetVertexFraction(const xAOD::VertexContainer*) const;

};


#endif

---