/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_7/src/RecoTauTag/ImpactParameter/src/ImpactParameterAlgorithm.cc

Go to the documentation of this file.

#include "RecoTauTag/ImpactParameter/interface/ImpactParameterAlgorithm.h"

#include "RecoBTag/BTagTools/interface/SignedTransverseImpactParameter.h"
#include "RecoBTag/BTagTools/interface/SignedImpactParameter3D.h"

#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"

ImpactParameterAlgorithm::ImpactParameterAlgorithm(){
        ip_min   = -9999;
        ip_max   = 9999;
        sip_min  = 0;
        use_sign = false;
        use3D    = false; 
}

ImpactParameterAlgorithm::ImpactParameterAlgorithm(const edm::ParameterSet & parameters){
        ip_min       = parameters.getParameter<double>("TauImpactParameterMin");
        ip_max       = parameters.getParameter<double>("TauImpactParameterMax");
        sip_min      = parameters.getParameter<double>("TauImpactParameterSignificanceMin");
        use_sign     = parameters.getParameter<bool>("UseTauImpactParameterSign");
        use3D        = parameters.getParameter<bool>("UseTau3DImpactParameter");

}

void ImpactParameterAlgorithm::setTransientTrackBuilder(const TransientTrackBuilder * builder) { 
        transientTrackBuilder = builder; 
}


std::pair<float,reco::TauImpactParameterInfo> ImpactParameterAlgorithm::tag(const reco::IsolatedTauTagInfoRef & tauRef, const reco::Vertex & pv) {

        if(transientTrackBuilder == 0){
             std::cout << "Transient track builder is 0. abort!" << std::endl;
             abort(); //FIXME: trow an exception here
        }

        reco::TauImpactParameterInfo resultExtended;
        resultExtended.setIsolatedTauTag(tauRef);

        const reco::Jet* jet = tauRef->jet().get();
        GlobalVector direction(jet->px(),jet->py(),jet->pz());

        const reco::TrackRefVector& tracks = tauRef->selectedTracks();

        edm::RefVector<reco::TrackCollection>::const_iterator iTrack;
        for(iTrack = tracks.begin(); iTrack!= tracks.end(); iTrack++){

          const reco::TransientTrack transientTrack = (transientTrackBuilder->build(&(**iTrack)));

          SignedTransverseImpactParameter stip;
          Measurement1D ip = stip.apply(transientTrack,direction,pv).second;

          SignedImpactParameter3D signed_ip3D;
          Measurement1D ip3D = signed_ip3D.apply(transientTrack,direction,pv).second;
          //std::cout << "check pv,ip3d,track z " << pv.z() << " " << ip3D.value() << " " << transientTrack->dz() << std::endl;
          if(!use_sign){
            Measurement1D tmp2D(fabs(ip.value()),ip.error());
            ip = tmp2D;

            Measurement1D tmp3D(fabs(ip3D.value()),ip3D.error());
            ip3D = tmp3D;
          }

          reco::TauImpactParameterTrackData theData;

          theData.transverseIp = ip;
          theData.ip3D = ip3D;
          resultExtended.storeTrackData(*iTrack,theData);

        }

        float discriminator = resultExtended.discriminator(ip_min,ip_max,sip_min,use_sign,use3D);

        return std::make_pair( discriminator, resultExtended );
}