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

// JetIsolationTool.h 
//
// Author: P-A. Delsart
//
// Modified Feb 2015 by D. Adams.
//    It now obtains its inputs from a pseudojet getter instead of a specified
// container of ATLAS objects.

#ifndef JetIsolationTool_H
#define JetIsolationTool_H


#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "JetRec/JetModifierBase.h"
#include "JetUtils/TiledEtaPhiMap.h"
#include "JetInterface/IPseudoJetGetter.h"
#include "fastjet/PseudoJet.hh"
#include "JetEDM/IConstituentUserInfo.h"

namespace jet {

namespace JetIsolation {
  class IsolationCalculator;
}

struct ParticlePosition {
  ParticlePosition(const fastjet::PseudoJet* p)
  : m_eta(p->eta()), m_phi(p->phi()), m_part(p) { };
  ParticlePosition(const xAOD::IParticle* p)
  : m_eta(p->eta()), m_phi(p->phi()), m_part(nullptr) { };
  ParticlePosition(double x=0, double y=0)
  : m_eta(x), m_phi(y), m_part(nullptr) { }
  double x() const {return m_eta;}
  double y() const {return m_phi;}
  double px() const { return m_part->px(); }
  double py() const { return m_part->py(); }
  double pz() const { return m_part->pz(); }
  double e() const { return m_part->e(); }
  double pt() const { return m_part->perp(); }
  void setX(double x){m_eta=x;}
  void setY(double x){m_phi=x;}

  const fastjet::PseudoJet* particle() const {return m_part;}

  struct DR2 {
    double operator()(const ParticlePosition &p1,const ParticlePosition &p2) const {
      return JetTiledMap::utils::DR2(p1.x(),p1.y(), p2.x(), p2.y() );
    }
  };

protected:
  double m_eta,m_phi;
  const fastjet::PseudoJet* m_part ;
};

}  // end namespace jet

//**********************************************************************

class JetIsolationTool : public JetModifierBase { 
  ASG_TOOL_CLASS0(JetIsolationTool)

public: 

  friend class jet::JetIsolation::IsolationCalculator;

  JetIsolationTool(const std::string &myname);

  ~JetIsolationTool(); 

  // Athena algtool Hooks
  StatusCode  initialize();
  StatusCode  finalize();

  // Jet Modifier methods.
  int modify(xAOD::JetContainer& jets) const;
  int modifyJet(xAOD::Jet& ) const{return 0;};

private: 

  std::vector<std::string> m_isolationCodes;
  ToolHandle<IPseudoJetGetter> m_hpjg;

  // and adapted to the jet object, see implementation)
  std::vector<jet::JetIsolation::IsolationCalculator*> m_isoCalculators;

}; 

#endif

---