JetIsolationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// JetIsolationTool.h 
//
// Author: P-A. Delsart
//
 
#ifndef JetIsolationTool_H
#define JetIsolationTool_H

 
#include "AsgTools/AsgTool.h"
#include "JetUtils/TiledEtaPhiMap.h"
#include "JetInterface/IJetDecorator.h"
#include "fastjet/PseudoJet.hh"
#include "JetEDM/IConstituentUserInfo.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "StoreGate/ReadDecorHandle.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "StoreGate/WriteDecorHandleKeyArray.h"
#include "StoreGate/WriteDecorHandle.h"
#include "xAODJet/JetContainer.h"
#include "xAODBase/IParticleContainer.h"
#include "xAODEventShape/EventShape.h"
 
namespace jet {
 
namespace JetIsolation {
  class IsolationCalculator;
}
 
 
}  // end namespace jet
 
//**********************************************************************
 
class JetIsolationTool : public asg::AsgTool,
                         virtual public IJetDecorator { 
  ASG_TOOL_CLASS0(JetIsolationTool)
 
public: 
 
  friend class jet::JetIsolation::IsolationCalculator;

  JetIsolationTool(const std::string &myname);

  virtual ~JetIsolationTool(); 
 
  // Athena algtool Hooks
  virtual StatusCode  initialize() override;
  virtual StatusCode  finalize() override;
 
  // Jet Modifier methods.
  virtual StatusCode decorate(const xAOD::JetContainer& jets) const override;
 
private: 
  Gaudi::Property<std::vector<std::string>> m_isolationCodes{this, "IsolationCalculations", {}, "Isolation calculations to be performed"};
  Gaudi::Property<std::string> m_jetContainerName{this, "JetContainer", "", "SG key for the input jet container"};
 
  SG::ReadHandleKey<xAOD::IParticleContainer> m_inputConstitKey{this, "InputConstitContainer", "", "Constituent container to read (should be configured automatically from a JetDefinition)"};
 
 
  SG::WriteDecorHandleKeyArray<xAOD::JetContainer> m_decorKeys{this, "DecorationKeys", {}, "SG key for output momentum decoration (not to be configured manually!)"};
 
  SG::ReadHandleKey<xAOD::EventShape> m_rhoKey{this, "RhoKey", "", "EventDensity for this jet collection. Required only if PtPUsub calculation is done"};
  
  // and adapted to the jet object, see implementation)
  std::vector<jet::JetIsolation::IsolationCalculator*> m_isoCalculators;
 
 
  
}; 
 
#endif