JetClusterer.h

Go to the documentation of this file.

// this file is -*- C++ -*-
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETREC_JETCLUSTERER_H
#define JETREC_JETCLUSTERER_H
 
 
#include "xAODEventInfo/EventInfo.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/WriteHandleKey.h"
#include "AthContainers/AuxElement.h"
 
#include "JetInterface/IJetProvider.h"
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
#include "JetRec/PseudoJetContainer.h"
#include "JetRec/JetFromPseudojet.h"
#include "JetRec/PseudoJetTranslator.h"
#include "JetEDM/PseudoJetVector.h"
#include "JetEDM/ClusterSequence.h"
 
#include "fastjet/PseudoJet.hh"
#include "fastjet/AreaDefinition.hh"
#include "fastjet/JetDefinition.hh"
 
#include "xAODJet/JetAuxContainer.h"
#include "xAODTracking/VertexContainer.h"
 
 
class JetClusterer
: public asg::AsgTool,
  virtual public JetProvider<xAOD::JetAuxContainer> {
  ASG_TOOL_CLASS(JetClusterer, IJetProvider)
 
public:
 
  // Can't use "using ctor" because of incompatiblity with pyroot in AnalysisBase
  JetClusterer(const std::string &name)
  : AsgTool(name),
    m_jetRankAccessor("jetRank")
  {}
 
  StatusCode initialize() override;

  std::pair<std::unique_ptr<xAOD::JetContainer>, std::unique_ptr<SG::IAuxStore> > getJets() const override;
 
 
protected:

  fastjet::AreaDefinition buildAreaDefinition(bool &seedsok) const ;

  std::unique_ptr<fastjet::ClusterSequence> buildClusterSequence(const PseudoJetVector* pseudoJetvector) const ;
    
  void processPseudoJet(const fastjet::PseudoJet& pj, const PseudoJetContainer& pjCont, xAOD::JetContainer* jets, const xAOD::Vertex* vertex) const;

  SG::ReadHandleKey<xAOD::EventInfo> m_eventinfokey{"EventInfo"};

  SG::ReadHandleKey<PseudoJetContainer> m_inputPseudoJets {this, "InputPseudoJets", "inputpseudojet", "input constituents"};

  SG::WriteHandleKey<PseudoJetVector> m_finalPseudoJets {this, "FinalPseudoJets_DONOTSET", "", "output pseudojets -- autoconfigured name"};
  SG::WriteHandleKey<jet::ClusterSequence> m_clusterSequence {this, "ClusterSequence_DONOTSET", "", "output pseudojets -- autoconfigured name"};
  Gaudi::Property<std::string> m_jetRank {this, "jetRankName", "jetRank", "name for accessor for jet rank"};
  
  // Job options.
  Gaudi::Property<std::string>  m_jetalg {this, "JetAlgorithm", "AntiKt", "alg type : AntiKt, Kt, CA..."};
  Gaudi::Property<float> m_jetrad        {this, "JetRadius", 0.4 , "jet size parameter"}; 
  Gaudi::Property<float> m_ptmin         {this, "PtMin", 0.0, "pT min in MeV"};
  Gaudi::Property<float> m_ghostarea     {this, "GhostArea", 0.0, "Area for ghosts. 0==>no ghosts."};
  Gaudi::Property<int> m_ranopt          {this, "RandomOption", 1, "Rand option: 0=fj default, 1=run/event"};
 
  Gaudi::Property<int> m_inputType       {this, "JetInputType", 0, "input type as in xAOD::JetInput (see xAODJet/JetContainerInfo.h)"};
 
  
  // internal data set from properties during initialize()
  fastjet::JetAlgorithm m_fjalg;
  bool m_useArea{};
  
  Gaudi::Property<float> m_minrad        {this, "VariableRMinRadius", -1.0, "Variable-R min radius" };
  Gaudi::Property<float> m_massscale     {this, "VariableRMassScale", -1.0, "Variable-R mass scale" };
  bool m_isVariableR{};  
  bool isVariableR() const { return m_isVariableR;}
 
  SG::AuxElement::Accessor<int> m_jetRankAccessor;
 
};
 
 
namespace JetClustererHelper {

  void seedsFromEventInfo(const xAOD::EventInfo* ei, std::vector<int> &seeds);
  
}
 
#endif