/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/QuickAna/QuickAna/JetTool.h

#ifndef QUICK_ANA__JET_TOOL_H
#define QUICK_ANA__JET_TOOL_H

//        Copyright Iowa State University 2014.
//                  Author: Nils Krumnack
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

// Please feel free to contact me (nils.erik.krumnack@cern.ch) for bug
// reports, feature suggestions, praise and complaints.


// This module still needs to be documented.  The interface provided
// in this module is intended for experts only.  The module is
// considered to be in the pre-alpha stage.



#include <QuickAna/Global.h>

#include <QuickAna/AnaToolCorrect.h>
#include <AsgTools/AnaToolHandle.h>
#include <QuickAna/AnaToolSelect.h>
#include <QuickAna/AnaToolWeight.h>
#include <QuickAna/SelectionCut.h>

// Forward declarations
class IJetCalibrationTool;
class ICPJetUncertaintiesTool;
class IJERTool;
class IJERSmearingTool;
class IJetSelector;
class IBTaggingEfficiencyTool;
class IBTaggingSelectionTool;
class IJetUpdateJvt;

namespace CP {
  class IJetJvtEfficiency;
}

namespace ana
{

  class JetToolCorrect : public AnaToolCorrect<xAOD::JetContainer>
  {

    ASG_TOOL_CLASS (JetToolCorrect, ana::IAnaTool)

  public:
    //
    // public interface
    //

    JetToolCorrect (const std::string& name);

    virtual StatusCode
    useInitialConfiguration (const InternalConfiguration& conf) override;

    StatusCode initialize() override;

    virtual StatusCode
    correctObject (xAOD::Jet& jet) override;

  private:

    asg::AnaToolHandle<IJetCalibrationTool> m_calibration_tool;

    asg::AnaToolHandle<ICPJetUncertaintiesTool> m_uncertainties_tool;

    asg::AnaToolHandle<IJERTool> m_resolution_tool;

    asg::AnaToolHandle<IJERSmearingTool> m_smearing_tool;

    asg::AnaToolHandle<IJetUpdateJvt> m_jvt_tool;

    asg::AnaToolHandle<CP::IJetJvtEfficiency> m_jvtEffTool;

    asg::AnaToolHandle<IBTaggingSelectionTool> m_bsel_tool;
    asg::AnaToolHandle<IBTaggingSelectionTool> m_bsel_OR_tool;

    asg::AnaToolHandle<IJetSelector> m_cleaning_tool;


  private:

    bool m_isData = false;
    bool m_isAFII = false;
    std::string m_jetContainer;


  private:

    bool m_enableBTagging;
    std::string m_btagger;
    std::string m_btagWP;
    std::string m_btagWP_OR;


  private:
    SelectionCut cut_calibration_tool;
    SelectionCut cut_uncertainties_tool;
    SelectionCut cut_smearing_tool;
    SelectionCut cut_cleaning_tool;
  };



  class JetToolSelect : public AnaToolSelect<xAOD::JetContainer>
  {

    ASG_TOOL_CLASS (JetToolSelect, ana::IAnaTool)

  public:
    //
    // public interface
    //

    JetToolSelect (const std::string& name);

    StatusCode initialize() override;

    virtual StatusCode
    selectObject (xAOD::Jet& jet) override;

  public:

    SelectionStep m_jvt_cut_step;


  private:
    SelectionCut cut_jvt;
  };



  class JetToolWeight : public AnaToolWeight<xAOD::JetContainer>
  {

    ASG_TOOL_CLASS (JetToolWeight, ana::IAnaTool)

  public:
    //
    // public interface
    //

    JetToolWeight (const std::string& name);

    virtual unsigned inputTypes () const override;

    virtual unsigned outputTypes () const override;

    StatusCode initialize() override;

    virtual StatusCode execute (IEventObjects& objects) override;

    virtual StatusCode
    objectWeight (const xAOD::Jet& jet, float& weight) override;

  private:

    asg::AnaToolHandle<IBTaggingEfficiencyTool> m_btagging_eff_tool;

    asg::AnaToolHandle<CP::IJetJvtEfficiency> m_jvtEffTool;


  private:
    SG::AuxElement::ConstAccessor<SelectType> m_anaSelect;
    SG::AuxElement::Accessor<float> m_anaWeight;

    std::string m_btagger;
    std::string m_btagWP;

  };



  // Maker function for jet tools
  StatusCode makeJetTool (DefinitionArgs& args,
                          const std::string& collection,
                          const SelectionStep& jvt_step=SelectionStep::MET,
                          const std::string& btagWP="");
}

#endif

---