JetBadChanCorrTool.h

Go to the documentation of this file.

// this file is -*- C++ -*-
 
/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAOD_ANALYSIS
 
#ifndef JETMOMENTTOOLS_JETBADCHANCORRTOOL_H
#define JETMOMENTTOOLS_JETBADCHANCORRTOOL_H

 
#include "AsgTools/AsgTool.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "JetInterface/IJetDecorator.h"
 
#include "JetRecCalo/MissingCellListTool.h"
 
#include "CaloIdentifier/CaloCell_ID.h"
 
#include "TH1D.h"
#include <utility>
 
 
class Identifier;
 
class ITileBadChanTool;
class ITHistSvc;
 
class JetBadChanCorrTool: public asg::AsgTool,
                          virtual public IJetDecorator
{
  ASG_TOOL_CLASS0(JetBadChanCorrTool)
public:
  JetBadChanCorrTool(const std::string& name);
 
  virtual ~JetBadChanCorrTool();
 
  virtual StatusCode initialize() override;
 
  virtual StatusCode decorate(const xAOD::JetContainer& jets) const override;
 
  virtual StatusCode setupEvent();
 
 
protected:
  // These two apply the moments they compute as decorations to the jet collection
  StatusCode correctionFromClustersBadCells(const xAOD::JetContainer& jets) const;
  StatusCode correctionFromCellsInJet(const xAOD::JetContainer& jets, const jet::CaloCellFastMap * badCellMap) const;
  // This one computes the moment without applying it
  float correctionFromCellsInCone(const xAOD::Jet* jet, const jet::CaloCellFastMap * badCellMap) const;
 
 private:
 
  ServiceHandle<ITHistSvc> m_thistSvc;
 
  Gaudi::Property<std::string> m_jetContainerName{this, "JetContainer", "", "SG key for the input jet container"};
  Gaudi::Property<int> m_nBadCellLimit{this, "NBadCellLimit", 10000, "Limit to calculate moments"};
 
  // for jet-level correction
  Gaudi::Property<std::string> m_streamName{this, "StreamName", "/JetBadChanCorrTool/", "Stream name"};
  Gaudi::Property<std::string> m_profileName{this, "ProfileName", "JetBadChanCorrTool.root", "Profile name"};
  Gaudi::Property<std::string> m_profileTag{this, "ProfileTag", "", "Profile tag"};
 
  Gaudi::Property<bool> m_useCone{this, "UseCone", true, "Use cone?"};
  Gaudi::Property<bool> m_useClusters{this, "UseClusters", false, "Use clusters?"};
 
  SG::ReadHandleKey<jet::CaloCellFastMap> m_badCellMap_key{this, "MissingCellMap", "MissingCaloCellsMap", "SG key for missing cell map"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_corrCellKey{this, "CorrCellDecorKey", "BchCorrCell", "SG key for cell level decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_corrDotxKey{this, "CorrDotxDecorKey", "BchCorrDotx", "SG key for DOTX decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_corrJetKey{this, "CorrJetDecorKey", "BchCorrJet", "SG key for jet level decoration"};
  SG::WriteDecorHandleKey<xAOD::JetContainer> m_corrJetForCellKey{this, "CorrJetForCellDecorKey", "BchCorrJetForCell", "SG key for JetForCell decoration"};
 
  // jet profiles
  class ProfileData {
  public:
    ProfileData(TH1* th, int sample,
    double ptMin=0, double ptMax=9999,
    double etaMin=0, double etaMax=5.0,
    double phiMin=-M_PI, double phiMax=M_PI):
      m_th(th), m_sample(sample), 
      m_ptMin(ptMin), m_ptMax(ptMax), 
      m_etaMin(etaMin), m_etaMax(etaMax),
      m_phiMin(phiMin), m_phiMax(phiMax) {}
 
    virtual ~ProfileData(){}
 
    bool match(double pt, int sample, double eta, double phi) const {
      return ( pt>=m_ptMin && pt<m_ptMax 
         && sample==m_sample
         && fabs(eta)>=m_etaMin && fabs(eta)<m_etaMax 
         && phi>=m_phiMin && phi<m_phiMax);
    }
 
    double frac(double dr) const {
      int idr = std::as_const(m_th)->FindBin(dr);
      return m_th->GetBinContent(idr);
    }
  private:
    TH1* m_th;
    int m_sample;
    double m_ptMin;
    double m_ptMax;
    double m_etaMin;
    double m_etaMax;
    double m_phiMin;
    double m_phiMax;
 
  };
  std::vector<ProfileData> m_profileDatas[CaloCell_ID::Unknown];//24
 
  double getProfile(double pt, double dr, int sample, double eta, double phi) const;
};
#endif
 
#endif
 
// DoxygenDocumentation