JetModifiedMassDrop.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// JetModifiedMassDrop.h
 
#ifndef JetModifiedMassDrop_H
#define JetModifiedMassDrop_H
 
// Joe Taenzer
// October 2016
//
// Tool to apply the modified mass drop tagger (mMDT) and groom jets by filtering with mMDT
// mMDT is described in this paper: arXiv:1307.0007
// See also: http://fastjet.hepforge.org/svn/contrib/contribs/RecursiveTools/tags/1.0.0/ModifiedMassDropTagger.hh       
 
#include "AsgTools/AsgTool.h"
#include "JetInterface/IJetGroomer.h"
#include "JetInterface/IJetFromPseudojet.h"
#include "AsgTools/ToolHandle.h"
 
namespace JetRecGroomingParams{
  enum SymmetryMeasure : int;
  enum RecursionChoice : int;
}
 
 
class JetModifiedMassDrop
: public asg::AsgTool,
//  public fastjet::contrib::RecursiveSymmetryCutBase,
  virtual public IJetGroomer {
ASG_TOOL_CLASS(JetModifiedMassDrop, IJetGroomer)
 
public:
 
  // Equivalent enums to those defined in RecursiveSymmetryToolBase.hh
  // Redefined here so they can be used in the python configuration
  // See: http://fastjet.hepforge.org/svn/contrib/contribs/RecursiveTools/tags/1.0.0/RecursiveSymmetryCutBase.hh
  enum SymmetryMeasure{scalar_z, 
                              vector_z, 
                              y         
                       
  };

  enum RecursionChoice{larger_pt, 
                              larger_mt, 
                              larger_m   
  };
 
  // Ctor.
  JetModifiedMassDrop(const std::string& name);
 
  // Dtor.
  ~JetModifiedMassDrop();
 
  // Initilization.
  virtual StatusCode initialize() override;
  
  // Groom a jet and add result to a container.
  virtual int groom(const xAOD::Jet& jin,
                    const PseudoJetContainer&,
                    xAOD::JetContainer& jout) const override;
 
  // Dump to log.
  virtual void print() const override;
 
private:  // data
 
  // Job options.
  float m_zcut;                         // Cut on the symmetry measure
  float m_mu;                           // Mass drop, mu = m_heavy/m_parent
  bool m_doFilt;                        // Filter the jet after mass drop grooming
  double m_filtR;                       // Filter radius to use
  int m_filtNSub;                       // Number of subjets to keep in the filter
 
  // Symmetry measure to use, either y or scalar_z, default is y
  int m_sym;
  // Recursion choice, larger_pt or larger_mt, default is larger_pt
  int m_recursion;
 
  ToolHandle<IJetFromPseudojet> m_bld;  // Tool to build jets.
 
};
 
#endif