CombinedSVSoftLeptonComputer.h

Go to the documentation of this file.

#ifndef RecoBTag_SecondaryVertex_CombinedSVSoftLeptonComputer_h
#define RecoBTag_SecondaryVertex_CombinedSVSoftLeptonComputer_h

#include "DataFormats/BTauReco/interface/SoftLeptonTagInfo.h"

#include "RecoBTag/SecondaryVertex/interface/CombinedSVComputer.h"

//(z)
#include "DataFormats/BTauReco/interface/CandSoftLeptonTagInfo.h"
#include "DataFormats/BTauReco/src/classes.h"
#include "DataFormats/PatCandidates/interface/Jet.h"


class CombinedSVSoftLeptonComputer : public CombinedSVComputer {
    public:
    explicit CombinedSVSoftLeptonComputer(const edm::ParameterSet &params);
    
    template <class IPTI,class SVTI>
    reco::TaggingVariableList
    operator () (const IPTI &ipInfo, const SVTI &svInfo,
             const reco::CandSoftLeptonTagInfo &muonInfo,
             const reco::CandSoftLeptonTagInfo &elecInfo ) const;
};

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,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,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