JetVertexTaggerTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// JetVertexTaggerTool.h
 
#ifndef JETMOMENTTOOLS_JETVERTEXTAGGERTOOL_H
#define JETMOMENTTOOLS_JETVERTEXTAGGERTOOL_H
 
 
#include "AsgTools/ToolHandle.h"
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
#include "AsgDataHandles/ReadDecorHandleKey.h"
#include "AsgDataHandles/WriteDecorHandleKey.h"
#include "xAODTracking/Vertex.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
 
#include "JetRec/JetModifierBase.h"
#include "JetEDM/TrackVertexAssociation.h"
#include "JetInterface/IJetDecorator.h"
#include "JetInterface/IJetUpdateJvt.h"
 
#include <vector>
#include <string>
 
#include <TFile.h>
#include <TString.h>
#include <TH1D.h>
#include <TH2D.h>
 
class JetVertexTaggerTool
: public asg::AsgTool,
  virtual public IJetDecorator,
  virtual public IJetUpdateJvt {
  ASG_TOOL_CLASS2(JetVertexTaggerTool,IJetDecorator,IJetUpdateJvt)
 
public:
  // Constructor from tool name
  JetVertexTaggerTool(const std::string& name);
 
  // Destructor
  virtual ~JetVertexTaggerTool();
 
  // Initialization.
  StatusCode initialize() override;
 
  // Inherited method to decorate a jet container
  virtual StatusCode decorate(const xAOD::JetContainer& jetCont) const override;
 
  // Evaluate JVT from Rpt and JVFcorr.
  float evaluateJvt(float rpt, float jvfcorr) const;
 
  // Update JVT using new jet pt
  float updateJvt(const xAOD::Jet& jet) const override;
 
  // Local method to return the HS vertex - that of type PriVtx
  const xAOD::Vertex* findHSVertex() const;
 
private:  // data
 
  // Configurable parameters
  Gaudi::Property<std::string> m_jetContainerName{this,"JetContainer", "", "SG key for the input jet container"};
  Gaudi::Property<std::string> m_jvtlikelihoodHistName{this, "JVTLikelihoodHistName", "JVTRootCore_kNN100trim_pt20to50_Likelihood", "JVT likelihood histogram name"};
  Gaudi::Property<std::string> m_jvtfileName{this, "JVTFileName", "JetMomentTools/JVTlikelihood_20140805.root", "JVT likelihood file name"};
  Gaudi::Property<bool> m_suppressInputDeps{this, "SuppressInputDependence", false, "Will JVFCorr and SumPtTrk be created in the same algorithm that uses this tool?"};
 
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainer_key{this, "VertexContainer", "PrimaryVertices", "SG key for input vertex container"};
  SG::ReadDecorHandleKey<xAOD::JetContainer> m_jvfCorrKey{this, "JVFCorrName", "JVFCorr", "SG key for input JVFCorr decoration"};
  SG::ReadDecorHandleKey<xAOD::JetContainer> m_jvfCorrVtxKey; // Set automatically if running by-vertex jet reconstruction
  SG::ReadDecorHandleKey<xAOD::JetContainer> m_sumPtTrkKey{this, "SumPtTrkName", "SumPtTrkPt500", "SG key for input SumPtTrk decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_jvtKey{this, "JVTName", "Jvt", "SG key for output JVT decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_jvtVecKey; // Set automatically if running by-vertex jet reconstruction
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_rptKey{this, "RpTName", "JvtRpt", "SG key for output RpT decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_rptVecKey; // Set automatically if running by-vertex jet reconstruction
 
  Gaudi::Property<bool> m_useOriginVertex = {this, "UseOriginVertex", false, "use origin vertex for each jet"};
 
  // Internal objects
  TString m_fn;
  TH2F * m_jvthisto;
 
};
 
 
#endif