JetRecTool.h

Go to the documentation of this file.

// JetRecTool.h. This file is -*- C++ -*-
 
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef JetRecTool_H
#define JetRecTool_H

 
#include "AsgTools/ToolHandle.h"
#include "AsgTools/AsgTool.h"
#include "JetInterface/IJetBuildTool.h"
#include "JetInterface/IJetExecuteTool.h"
#include "JetInterface/IJetFinder.h"
#include "JetInterface/IJetGroomer.h"
#include "JetInterface/IJetModifier.h"
#include "JetInterface/IJetPseudojetRetriever.h"
#include "JetInterface/IJetConsumer.h"
#include "AsgDataHandles/ReadHandleKeyArray.h"
#include "AsgDataHandles/WriteHandleKey.h"
#include "JetEDM/PseudoJetVector.h"
#include "JetRec/PseudoJetContainer.h"
#if !defined(GENERATIONBASE) && !defined(XAOD_ANALYSIS)
  #include "AthenaMonitoringKernel/GenericMonitoringTool.h"
#endif
 
class JetRecTool
: public asg::AsgTool,
  virtual public IJetBuildTool,
  virtual public IJetExecuteTool {
ASG_TOOL_CLASS2(JetRecTool, IJetBuildTool, IJetExecuteTool)
 
public:
 
  JetRecTool(const std::string& myname);

  StatusCode initialize() override;

  virtual const xAOD::JetContainer* build() const override;

  int execute() const override;

  void print() const override;

  int inputContainerNames(std::vector<std::string>& connames) override;

  int outputContainerNames(std::vector<std::string>& connames) override;
 

  void setInputJetContainer(const xAOD::JetContainer* cont);
 
private:
 
  std::unique_ptr<xAOD::JetContainer> fillOutputContainer() const;
  // Collect all input PseudoJetContainers into a single container
  std::unique_ptr<PseudoJetContainer> collectPseudoJets() const;
  const xAOD::JetContainer* getOldJets() const;
  std::unique_ptr<xAOD::JetContainer> makeOutputContainer() const;
 
  std::unique_ptr<xAOD::JetContainer> findJets() const;
  std::unique_ptr<xAOD::JetContainer> groomJets() const;
  std::unique_ptr<xAOD::JetContainer> copyJets() const;
 
 
  // Properties.
  SG::WriteHandleKey<xAOD::JetContainer> m_outcoll {this, "OutputContainer", ""};
  SG::ReadHandleKey<xAOD::JetContainer> m_incoll {this, "InputContainer", ""};
  // The template argument should become PseudoJetContainer
  SG::ReadHandleKeyArray<PseudoJetContainer> m_psjsin {this, "InputPseudoJets", {}};
 
  ToolHandle<IJetExecuteTool> m_intool;
  ToolHandle<IJetPseudojetRetriever> m_hpjr;
  ToolHandle<IJetFinder> m_finder;
  ToolHandle<IJetGroomer> m_groomer;
  ToolHandleArray<IJetModifier> m_modifiers {this, "JetModifiers", {}};
  ToolHandleArray<IJetConsumer> m_consumers {this, "JetConsumers", {}};
  bool m_trigger{};
 
  // Cached state.
  int m_initCount{};
  bool m_find{};
  bool m_groom{};
  bool m_copy{};
  xAOD::JetInput::Type m_inputtype;
  IJetFinder::NameList m_ghostlabs;
  std::vector<std::string> m_incolls;
  std::vector<std::string> m_outcolls;
  const IJetPseudojetRetriever* m_ppjr{};
 
  // trigger hacks
  const xAOD::JetContainer* m_trigInputJetsForGrooming{}; // used in trigger context only
 
#if !defined (GENERATIONBASE) && !defined (XAOD_ANALYSIS)
  ToolHandle<GenericMonitoringTool> m_monTool{this,"MonTool","","Monitoring tool"};
#endif
 
  // not used in AnalysisBase
#ifdef __clang__
  [[maybe_unused]]
#endif
  const double m_mevtogev = 0.001;
 
};
 
#endif