CombinedSVSoftLeptonComputer.h

Go to the documentation of this file.

#ifndef RecoBTag_SecondaryVertex_CombinedSVSoftLeptonComputer_h
#define RecoBTag_SecondaryVertex_CombinedSVSoftLeptonComputer_h
 
#include "DataFormats/BTauReco/interface/CandSoftLeptonTagInfo.h"
#include "DataFormats/BTauReco/interface/TaggingVariable.h"
#include "DataFormats/BTauReco/interface/TemplatedSoftLeptonTagInfo.h"
#include "DataFormats/BTauReco/interface/VertexTypes.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "RecoBTag/SecondaryVertex/interface/CombinedSVComputer.h"
 
class CombinedSVSoftLeptonComputer : public CombinedSVComputer {
public:
  explicit CombinedSVSoftLeptonComputer(const edm::ParameterSet &params);
  ~CombinedSVSoftLeptonComputer() override = default;
  double flipSoftLeptonValue(double value) const;
 
  template <class IPTI, class SVTI>
  reco::TaggingVariableList operator()(const IPTI &ipInfo,
                                       const SVTI &svInfo,
                                       const reco::CandSoftLeptonTagInfo &muonInfo,
                                       const reco::CandSoftLeptonTagInfo &elecInfo) const;
 
private:
  bool SoftLeptonFlip;
};
 
double CombinedSVSoftLeptonComputer::flipSoftLeptonValue(double value) const { return SoftLeptonFlip ? -value : value; }
 
template <class IPTI, class SVTI>
reco::TaggingVariableList CombinedSVSoftLeptonComputer::operator()(const IPTI &ipInfo,
                                                                   const SVTI &svInfo,
                                                                   const reco::CandSoftLeptonTagInfo &muonInfo,
                                                                   const reco::CandSoftLeptonTagInfo &elecInfo) const {
  using namespace reco;
 
  // call the inherited operator()
  TaggingVariableList vars = CombinedSVComputer::operator()(ipInfo, svInfo);
 
  //Jets with vtxCategory 99 cause problems
  unsigned int vtxType =
      (vars.checkTag(reco::btau::vertexCategory) ? (unsigned int)(vars.get(reco::btau::vertexCategory)) : 99);
  if (vtxType == 99)
    return vars;
 
  // the following is specific to soft leptons
  int leptonCategory = 0;  // 0 = no lepton, 1 = muon, 2 = electron
 
  for (unsigned int i = 0; i < muonInfo.leptons();
       ++i)  // loop over all muons, not optimal -> find the best or use ranking from best to worst
  {
    leptonCategory = 1;  // muon category
    const SoftLeptonProperties &propertiesMuon = muonInfo.properties(i);
    vars.insert(btau::leptonPtRel, propertiesMuon.ptRel, true);
    vars.insert(btau::leptonSip3d, flipSoftLeptonValue(propertiesMuon.sip3d), true);
    vars.insert(btau::leptonDeltaR, propertiesMuon.deltaR, true);
    vars.insert(btau::leptonRatioRel, propertiesMuon.ratioRel, true);
    vars.insert(btau::leptonEtaRel, propertiesMuon.etaRel, true);
    vars.insert(btau::leptonRatio, propertiesMuon.ratio, true);
  }
 
  if (leptonCategory != 1)  // no soft muon found, try soft electron
  {
    for (unsigned int i = 0; i < elecInfo.leptons();
         ++i)  // loop over all electrons, not optimal -> find the best or use ranking from best to worst
    {
      leptonCategory = 2;  // electron category
      const SoftLeptonProperties &propertiesElec = elecInfo.properties(i);
      vars.insert(btau::leptonPtRel, propertiesElec.ptRel, true);
      vars.insert(btau::leptonSip3d, flipSoftLeptonValue(propertiesElec.sip3d), true);
      vars.insert(btau::leptonDeltaR, propertiesElec.deltaR, true);
      vars.insert(btau::leptonRatioRel, propertiesElec.ratioRel, true);
      vars.insert(btau::leptonEtaRel, propertiesElec.etaRel, true);
      vars.insert(btau::leptonRatio, propertiesElec.ratio, true);
    }
  }
 
  // set the default value for vertexLeptonCategory to 2 (= NoVertexNoSoftLepton)
  int vertexLepCat = 2;
 
  if (leptonCategory == 0)  // no soft lepton
  {
    if (vtxType == (unsigned int)(btag::Vertices::RecoVertex))
      vertexLepCat = 0;
    else if (vtxType == (unsigned int)(btag::Vertices::PseudoVertex))
      vertexLepCat = 1;
    else
      vertexLepCat = 2;
  } else if (leptonCategory == 1)  // soft muon
  {
    if (vtxType == (unsigned int)(btag::Vertices::RecoVertex))
      vertexLepCat = 3;
    else if (vtxType == (unsigned int)(btag::Vertices::PseudoVertex))
      vertexLepCat = 4;
    else
      vertexLepCat = 5;
  } else if (leptonCategory == 2)  // soft electron
  {
    if (vtxType == (unsigned int)(btag::Vertices::RecoVertex))
      vertexLepCat = 6;
    else if (vtxType == (unsigned int)(btag::Vertices::PseudoVertex))
      vertexLepCat = 7;
    else
      vertexLepCat = 8;
  }
  vars.insert(btau::vertexLeptonCategory, vertexLepCat, true);
 
  vars.finalize();
  return vars;
}
 
#endif  // RecoBTag_SecondaryVertex_CombinedSVSoftLeptonComputer_h