HEPTopTaggerWrapper.h

Go to the documentation of this file.

//  2011 Christopher Vermilion
//
//----------------------------------------------------------------------
//  This file 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 3 of the License, or
//  (at your option) any later version.
//
//  This file 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.
//
//  The GNU General Public License is available at
//  http://www.gnu.org/licenses/gpl.html or you can write to the Free Software
//  Foundation, Inc.:
//      59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//----------------------------------------------------------------------

#ifndef __HEPTOPTAGGER_WRAPPER_HH__
#define __HEPTOPTAGGER_WRAPPER_HH__

#include <fastjet/tools/TopTaggerBase.hh>
#include <fastjet/CompositeJetStructure.hh>
#include <sstream>

FASTJET_BEGIN_NAMESPACE

class HEPTopTaggerStructure;


class HEPTopTagger : public TopTaggerBase {
public:
  HEPTopTagger(double mass_drop_threshold=0.8, double max_subjet_mass=30.,
               bool use_subjet_mass_cuts=false)
    : _mass_drop_threshold(mass_drop_threshold),
      _max_subjet_mass(max_subjet_mass),
      _use_subjet_mass_cuts(use_subjet_mass_cuts)
  {}

  virtual std::string description() const;

  virtual PseudoJet result(const PseudoJet & jet) const;

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

private:
  double _mass_drop_threshold;
  double _max_subjet_mass;
  bool _use_subjet_mass_cuts; 
};



//BEGIN COMMENTING OUT BY CHRISTOPHER SILKWORTH
/*
class HEPTopTaggerStructure : public JHTopTaggerStructure {
public:
  HEPTopTaggerStructure(std::vector<PseudoJet> pieces,
      const JetDefinition::Recombiner *recombiner = 0)
    : JHTopTaggerStructure(pieces, recombiner) {}

protected:
  friend class HEPTopTagger;
};
*/
//END COMMENTING OUT

//BEGIN ADDED BY CHRISTOPHER SILKWORTH


class HEPTopTaggerStructure : public CompositeJetStructure, public TopTaggerBaseStructure {
 public:
   HEPTopTaggerStructure(const std::vector<PseudoJet>& pieces_in,
                  const JetDefinition::Recombiner *recombiner = 0) :
      CompositeJetStructure(pieces_in, recombiner), _cos_theta_w(0.0),W_rec(recombiner), 
      rW_()
    {}
 
      inline PseudoJet const & W() const{ 
         rW_ = join(_pieces[0], _pieces[1], *W_rec);
         return rW_;
      }
 
      
      inline PseudoJet  W1() const{
         assert(W().pieces().size()>0);
         return W().pieces()[0];
      }
      
      inline PseudoJet W2() const{
         assert(W().pieces().size()>1);
         return W().pieces()[1];
      }


   inline const PseudoJet & non_W() const{ 
     return _pieces[2];
   }
 
   inline double cos_theta_W() const {return _cos_theta_w;}
 
 //  /// returns the original jet (before tagging)
 //  const PseudoJet & original() const {return _original_jet;}

 
 
 protected:
      double _cos_theta_w; 
      const JetDefinition::Recombiner  * W_rec;
   //PseudoJet _W;             ///< the tagged W
   //PseudoJet _non_W;         ///< the remaining pieces
 //  PseudoJet _original_jet;  ///< the original jet (before tagging)
 
      mutable PseudoJet rW_;

   // allow the tagger to set these
   friend class HEPTopTagger;
 };


//END ADDED BY CHRISTOPHER SILKWORTH


//------------------------------------------------------------------------
// description of the tagger
inline std::string HEPTopTagger::description() const{ 
  std::ostringstream oss;
  oss << "HEPTopTagger with {max. subjet mass = " << _max_subjet_mass
      << ", mass-drop threshold = " << _mass_drop_threshold
      << ", and " << (_use_subjet_mass_cuts ? "using" : "not using") << " subjet mass cuts" << std::endl;
  oss << description_of_selectors();
  return oss.str();
}


FASTJET_END_NAMESPACE

#endif // __HEPTOPTAGGER_HH__