#include <JetProbabilityComputer.h>

Inheritance diagram for JetProbabilityComputer:

Public Member Functions
float	discriminator (const TagInfoHelper &ti) const

double	jetProbability (const std::vector< float > &v) const

	JetProbabilityComputer (const edm::ParameterSet &parameters)

Public Member Functions inherited from JetTagComputer
const std::vector< std::string > &	getInputLabels () const

virtual void	initialize (const JetTagComputerRecord &)

	JetTagComputer ()

	JetTagComputer (const edm::ParameterSet &configuration)

float	operator() (const reco::BaseTagInfo &info) const

float	operator() (const TagInfoHelper &helper) const

void	setupDone ()

virtual	~JetTagComputer ()

Private Attributes
double	m_cutMaxDecayLen

double	m_cutMaxDistToAxis

double	m_deltaR

int	m_ipType

double	m_minTrackProb

reco::TrackBase::TrackQuality	m_trackQuality

int	m_trackSign

bool	m_useAllQualities

bool	useVariableJTA_

reco::TrackIPTagInfo::variableJTAParameters	varJTApars

Additional Inherited Members
Protected Member Functions inherited from JetTagComputer
virtual float	discriminator (const reco::BaseTagInfo &) const

void	uses (unsigned int id, const std::string &label)

void	uses (const std::string &label)

Detailed Description

Definition at line 14 of file JetProbabilityComputer.h.

Constructor & Destructor Documentation

JetProbabilityComputer::JetProbabilityComputer ( const edm::ParameterSet & parameters )

inline

Definition at line 17 of file JetProbabilityComputer.h.

References edm::ParameterSet::getParameter(), m_cutMaxDecayLen, m_cutMaxDistToAxis, m_deltaR, m_ipType, m_minTrackProb, m_trackQuality, m_trackSign, m_useAllQualities, reco::TrackBase::qualityByName(), AlCaHLTBitMon_QueryRunRegistry::string, JetTagComputer::uses(), useVariableJTA_, and varJTApars.

   { 
      m_ipType           = parameters.getParameter<int>("impactParameterType");
      m_minTrackProb     = parameters.getParameter<double>("minimumProbability");
      m_deltaR           = parameters.getParameter<double>("deltaR");
      m_trackSign        = parameters.getParameter<int>("trackIpSign");
      m_cutMaxDecayLen   = parameters.getParameter<double>("maximumDecayLength"); 
      m_cutMaxDistToAxis = parameters.getParameter<double>("maximumDistanceToJetAxis"); 
      //
      // access track quality class; "any" takes everything
      //
      std::string trackQualityType = parameters.getParameter<std::string>("trackQualityClass"); //used
      m_trackQuality =  reco::TrackBase::qualityByName(trackQualityType);
      m_useAllQualities = false;
      if (trackQualityType == "any" || 
      trackQualityType == "Any" || 
      trackQualityType == "ANY" ) m_useAllQualities = true;
 
      useVariableJTA_    = parameters.getParameter<bool>("useVariableJTA");
      if(useVariableJTA_) 
        varJTApars = {
      parameters.getParameter<double>("a_dR"),
      parameters.getParameter<double>("b_dR"),
      parameters.getParameter<double>("a_pT"),
      parameters.getParameter<double>("b_pT"),
      parameters.getParameter<double>("min_pT"),  
      parameters.getParameter<double>("max_pT"),
      parameters.getParameter<double>("min_pT_dRcut"),  
      parameters.getParameter<double>("max_pT_dRcut"),
      parameters.getParameter<double>("max_pT_trackPTcut") };
      
      uses("ipTagInfos");
   }

Member Function Documentation

float JetProbabilityComputer::discriminator ( const TagInfoHelper & ti ) const

inlinevirtual

Reimplemented from JetTagComputer.

Definition at line 51 of file JetProbabilityComputer.h.

References JetTagComputer::TagInfoHelper::get(), i, reco::TrackIPTagInfo::impactParameterData(), PyquenDefaultSettings_cff::impactParameters, edm::Ref< C, T, F >::isNull(), reco::JTATagInfo::jet(), jetProbability(), m_ipType, m_useAllQualities, AlCaHLTBitMon_ParallelJobs::p, p4, reco::TrackIPTagInfo::primaryVertex(), reco::TrackIPTagInfo::probabilities(), reco::TrackIPTagInfo::selectedTracks(), testEve_cfg::tracks, useVariableJTA_, reco::TrackIPTagInfo::variableJTA(), and varJTApars.

    {
       const reco::TrackIPTagInfo & tkip = ti.get<reco::TrackIPTagInfo>();
 
       const edm::RefVector<reco::TrackCollection> & tracks(tkip.selectedTracks());
       const std::vector<float> & allProbabilities((tkip.probabilities(m_ipType)));
       const std::vector<reco::TrackIPTagInfo::TrackIPData> & impactParameters((tkip.impactParameterData()));
     
       if (tkip.primaryVertex().isNull()) return 0 ; 
     
       GlobalPoint pv(tkip.primaryVertex()->position().x(),tkip.primaryVertex()->position().y(),tkip.primaryVertex()->position().z());
 
       std::vector<float> probabilities;
       int i=0;
       for(std::vector<float>::const_iterator it = allProbabilities.begin(); it!=allProbabilities.end(); ++it, i++)
        {
         if ( fabs(impactParameters[i].distanceToJetAxis.value()) < m_cutMaxDistToAxis  &&        // distance to JetAxis
              (impactParameters[i].closestToJetAxis - pv).mag() < m_cutMaxDecayLen  &&      // max decay len
              (m_useAllQualities  == true || (*tracks[i]).quality(m_trackQuality)) // use selected track qualities
         )
          {
           float p;
           if(m_trackSign ==0 )
           { 
            if (*it >=0){p=*it/2.;}else{p=1.+*it/2.;}
           }
           else if(m_trackSign > 0)
           {
             if(*it >=0 ) p=*it; else continue; 
           } else
           {
            if(*it <=0 ) p= -*it; else continue; 
           } 
       if (useVariableJTA_) {
         if (tkip.variableJTA( varJTApars )[i])
           probabilities.push_back(p);
       }else{
         if(m_deltaR <= 0  || ROOT::Math::VectorUtil::DeltaR((*tkip.jet()).p4().Vect(), (*tracks[i]).momentum()) < m_deltaR )
           probabilities.push_back(p);
       }
      }
        }
       return jetProbability(probabilities); 
    }

double JetProbabilityComputer::jetProbability ( const std::vector< float > & v ) const

inline

Definition at line 96 of file JetProbabilityComputer.h.

References gather_cfg::cout, create_public_lumi_plots::exp, prof2calltree::l, log, m_minTrackProb, bookConverter::max, min(), and lumiQueryAPI::q.

Referenced by discriminator().

 {
    int ngoodtracks=v.size();
    double SumJet=0.;
 
   for(std::vector<float>::const_iterator q = v.begin(); q != v.end(); q++){
     SumJet+=(*q>m_minTrackProb)?log(*q):log(m_minTrackProb);
   }
 
   double ProbJet;
   double Loginvlog=0;
 
   if(SumJet<0.){
     if(ngoodtracks>=2){
       Loginvlog=log(-SumJet);
     }
     double Prob=1.;
     double lfact=1.;
     for(int l=1; l!=ngoodtracks; l++){
        lfact*=l;
       Prob+=exp(l*Loginvlog-log(1.*lfact));
     }
     double LogProb=log(Prob);
     ProbJet=
       std::min(exp(std::max(LogProb+SumJet,-30.)),1.);
   }else{
     ProbJet=1.;
   }
   if(ProbJet>1)
    std::cout << "ProbJet too high: "  << ProbJet << std::endl;
 
   //double LogProbJet=-log(ProbJet);
   //  //return 1.-ProbJet;
       return -log10(ProbJet)/4.;
   }