CMSBoostedTauSeedingAlgorithm.h

Go to the documentation of this file.

#ifndef __RecoJets_JetAlgorithms_CMSBoostedTauSeedingAlgorithm_h__
#define __RecoJets_JetAlgorithms_CMSBoostedTauSeedingAlgorithm_h__

// CMSBoostedTau Package
//
// Find subjets corresponding to decay products of tau lepton pair
// and produce data-formats neccessary to seed tau reconstruction.
//
// Questions/Comments?
//    for physics : Christian.Veelken@cern.ch
//    for implementation : Salvatore.Rappoccio@cern.ch
//
//----------------------------------------------------------------------
// This file is part of FastJet contrib.
//
// It is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2 of the License, or (at
// your option) any later version.
//
// It is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
// License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this code. If not, see <http://www.gnu.org/licenses/>.
//----------------------------------------------------------------------

#include <fastjet/internal/base.hh>
#include <fastjet/tools/Transformer.hh>
#include <fastjet/CompositeJetStructure.hh>

#include <fastjet/ClusterSequence.hh>
#include <fastjet/Error.hh>
#include <fastjet/JetDefinition.hh>
//#include "fastjet/FunctionOfPseudoJet.hh"

#include <map>
#include <sstream>
#include <string>

FASTJET_BEGIN_NAMESPACE  // defined in fastjet/internal/base.hh

    namespace contrib {
  class CMSBoostedTauSeedingAlgorithmStructure;

  //------------------------------------------------------------------------
  class CMSBoostedTauSeedingAlgorithm : public Transformer {
  public:
    // constructors
    CMSBoostedTauSeedingAlgorithm(double ptMin,
                                  double muMin,
                                  double muMax,
                                  double yMin,
                                  double yMax,
                                  double dRMin,
                                  double dRMax,
                                  int maxDepth,
                                  int verbosity = 0);

    // destructor
    ~CMSBoostedTauSeedingAlgorithm() override {}

    // standard usage
    std::string description() const override;

    PseudoJet result(const PseudoJet& jet) const override;

    // the type of the associated structure
    typedef CMSBoostedTauSeedingAlgorithmStructure StructureType;

  protected:
    void dumpSubJetStructure(const fastjet::PseudoJet& jet,
                             int depth,
                             int maxDepth,
                             const std::string& depth_and_idx_string) const;
    std::pair<PseudoJet, PseudoJet> findSubjets(const PseudoJet& jet, int depth, bool& subjetsFound) const;

  private:
    double ptMin_;  
    double muMin_;  
    double muMax_;  
    double yMin_;   
    double yMax_;   
    double dRMin_;  
    double dRMax_;  
    int maxDepth_;  

    int verbosity_;  
  };

  //------------------------------------------------------------------------
  class CMSBoostedTauSeedingAlgorithmStructure : public CompositeJetStructure {
  public:
    CMSBoostedTauSeedingAlgorithmStructure(const PseudoJet& result_jet, const JetDefinition::Recombiner* rec = nullptr)
        : CompositeJetStructure(result_jet.pieces(), rec), _mu(0.0), _y(0.0), _dR(0.0), _pt(0.0) {}

    inline double mu() const { return _mu; }

    inline double y() const { return _y; }

    inline double dR() const { return _dR; }

    inline double pt() const { return _pt; }

    // /// returns the original jet (before tagging)
    //const PseudoJet& original() const { return _original_jet; }

  protected:
    double _mu;  
    double _y;   
    double _dR;  
    double _pt;  
    // allow the tagger to set these
    friend class CMSBoostedTauSeedingAlgorithm;
  };

}  // namespace contrib

FASTJET_END_NAMESPACE

#endif  // __FASTJET_CONTRIB_CMSBOOSTEDTAUSEEDINGALGORITHM_HH__