TemplateTagger.hh

//  Template Tagger Package
//  Version 0.3.0 (December 1, 2012)
//  Questions/Comments?  jose.juknevich@weizmann.ac.il

// Copyright (c) 2011-12, Jose Juknevich, Mihailo Backovic, and Raz Alon.
//
//----------------------------------------------------------------------
// This file is part of the Template Tagger package ("Template Overlap").
//
//  Template Tagger 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.
//----------------------------------------------------------------------


#ifndef __TEMPLATE_TAGGER_HH__
#define __TEMPLATE_TAGGER_HH__

#include "matching.hh"

#include "fastjet/FunctionOfPseudoJet.hh"
#include <fastjet/Selector.hh>
#include <fastjet/CompositeJetStructure.hh> // to derive the FilterStructure from CompositeJetStructure
#include <fastjet/tools/Transformer.hh>     // to derive Filter from Transformer



#include <string>
#include <climits>
#include <utility>


using namespace TemplateOverlap;



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


class TemplateTagger;
class TemplateTaggerStructure;


class Noverlap : public FunctionOfPseudoJet<templ_t> {
public:

  Noverlap(int N , double sigma, double R0, const string & templateFile)
  : _N(N), _templateFinder(templateFile, Settings(R0, sigma)), _R0(R0), _sigma(sigma)
{}
   
  templ_t result(const PseudoJet& jet) const {return _templateFinder.getOv(jet);}

private:

   int _N;
   mutable TemplateOverlap::MatchingMethod _templateFinder; 
   double _R0, _sigma;
   
};


class TemplateTagger : public Transformer {
public:
  
   TemplateTagger(FunctionOfPseudoJet<templ_t> *Ov_func, Selector selector, const double R1, const double ovcut ) : 
    _Ovfunc(Ov_func), _selector(selector), _R1(R1), _ovcut(ovcut){};
    
  virtual std::string description() const;

  virtual PseudoJet result(const PseudoJet & jet) const;

  typedef TemplateTaggerStructure StructureType;

protected:
  
  FunctionOfPseudoJet<templ_t> *_Ovfunc; 
  mutable Selector _selector;  
  double _R1; 
  double _ovcut;                 
  
};

class TemplateTaggerStructure : public CompositeJetStructure{
public:
  TemplateTaggerStructure(const std::vector<PseudoJet> & pieces_in) :
    CompositeJetStructure(pieces_in), _ov(0.0){}

  inline double ov() const{return _ov;}
  inline std::vector<PseudoJet> maxTempl() const {return _peak_template;}
  
  virtual std::string description() const { return "Template Tagger"; }


protected:
  double _ov;      
  std::vector<PseudoJet> _peak_template; 

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

//----------------------------------------------------------------------
// TemplateTagger class implementation
//----------------------------------------------------------------------

// class description
string TemplateTagger::description() const {
  ostringstream ostr;
  ostr << "Filter with subjet_def = ";
  if (_Ovfunc)
    ostr << "Template Tagger with cone";
  ostr<< ", selection " << _selector.description();
  
  return ostr.str();
}


//----------------------------------------------------------------------

// return a vector of subjets, which are the ones that would be kept
// by the filtering
PseudoJet TemplateTagger::result(const PseudoJet &jet) const {

  templ_t my_result = (*_Ovfunc)(jet);
  if (my_result.first < _ovcut || ! _selector.pass(jet) ) return PseudoJet();
     
  std::vector<fastjet::PseudoJet> particles =  jet.constituents();
  fastjet::Selector sel = fastjet::SelectorCircle(_R1);
  std::vector<fastjet::PseudoJet> subjets;
  for (unsigned int i=0; i< my_result.second.size(); ++i){
    sel.set_reference(my_result.second[i]);
    fastjet::PseudoJet subjet = fastjet::join(sel(particles));
    subjets.push_back(subjet);
  }
  PseudoJet result = join<StructureType>(subjets);
  StructureType * s = (StructureType *) result.structure_non_const_ptr();
  s->_ov = my_result.first;
  s->_peak_template = my_result.second;
  return result;
}





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

#endif  // __TEMPLATE_TAGGER_HH__