/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_14/src/DataFormats/MuonReco/interface/MuonCocktails.h

Go to the documentation of this file.

#ifndef MuonReco_MuonCocktails_h
#define MuonReco_MuonCocktails_h

#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackToTrackMap.h"

namespace muon {
  
  reco::Muon::MuonTrackTypePair tevOptimized(const reco::TrackRef& combinedTrack,
                                             const reco::TrackRef& trackerTrack,
                                             const reco::TrackRef& tpfmsTrack,
                                             const reco::TrackRef& pickyTrack,
                                             const double ptThreshold = 200.,
                                             const double tune1 = 4.,
                                             const double tune2 = 6.,
                                             const double dptcut = -1.);

  // Version for convenience. (NB: can be used with pat::Muon, even
  // with embedded tracks, equally conveniently!)
  inline reco::Muon::MuonTrackTypePair tevOptimized(const reco::Muon& muon,
                                                    const double ptThreshold = 200.,
                                                    const double tune1 = 4.,
                                                    const double tune2 = 6.,
                                                    const double dptcut = -1.) {
    return tevOptimized(muon.globalTrack(),
                        muon.innerTrack(),
                        muon.tpfmsTrack(),
                        muon.pickyTrack(),
                        ptThreshold,
                        tune1,
                        tune2,
                        dptcut);  
  }

  reco::TrackRef getTevRefitTrack(const reco::TrackRef& combinedTrack,
                                  const reco::TrackToTrackMap& map);
  
  // The next two versions of tevOptimized are for backward
  // compatibility; TrackToTrackMaps are to be removed from the
  // EventContent, so these versions will go away (along with the
  // helper getter function) after a deprecation period. Since they
  // are just for backward compatibility and not for new code, we
  // don't bother to expose the tune parameters.

  inline reco::Muon::MuonTrackTypePair tevOptimized(const reco::TrackRef& combinedTrack,
                                                    const reco::TrackRef& trackerTrack,
                                                    const reco::TrackToTrackMap& tevMap1,
                                                    const reco::TrackToTrackMap& tevMap2,
                                                    const reco::TrackToTrackMap& tevMap3,
                                                    const double ptThreshold = 200.,
                                                    const double tune1 = 4.,
                                                    const double tune2 = 6.,
                                                    const double dptcut = -1.) {
    return tevOptimized(combinedTrack,
                        trackerTrack,
                        getTevRefitTrack(combinedTrack, tevMap2),
                        getTevRefitTrack(combinedTrack, tevMap3),
                        ptThreshold,
                        tune1,
                        tune2,
                        dptcut);
  }
  
  inline reco::Muon::MuonTrackTypePair tevOptimized(const reco::Muon& muon,
                                                    const reco::TrackToTrackMap& tevMap1,
                                                    const reco::TrackToTrackMap& tevMap2,
                                                    const reco::TrackToTrackMap& tevMap3 ) {
    return tevOptimized(muon.combinedMuon(),
                        muon.track(),
                        getTevRefitTrack(muon.combinedMuon(), tevMap2),
                        getTevRefitTrack(muon.combinedMuon(), tevMap3));
  }
  
  // The cocktail used as the soon-to-be-old default momentum
  // assignment for the reco::Muon.
  reco::Muon::MuonTrackTypePair sigmaSwitch(const reco::TrackRef& combinedTrack,
                                            const reco::TrackRef& trackerTrack,
                                            const double nSigma = 2.,
                                            const double ptThreshold = 200.);
  
  // Convenience version of the above.
  inline reco::Muon::MuonTrackTypePair sigmaSwitch(const reco::Muon& muon,
                                                   const double nSigma = 2.,
                                                   const double ptThreshold = 200.) {
    return muon::sigmaSwitch(muon.globalTrack(),
                             muon.innerTrack(),
                             nSigma,
                             ptThreshold);
  }

  // "Truncated muon reconstructor": the first cocktail, between just
  // tracker-only and TPFMS. Similar to tevOptimized.
  reco::Muon::MuonTrackTypePair TMR(const reco::TrackRef& trackerTrack,
                                    const reco::TrackRef& fmsTrack,
                                    const double tune=4.);
  
  double trackProbability(const reco::TrackRef track);
 
  inline bool cocktailInputIsOK(const reco::TrackRef& combinedTrack,
                                 const reco::TrackRef& trackerTrack,
                                const reco::TrackRef& tpfmsTrack,
                                const reco::TrackRef& pickyTrack){
    
    return combinedTrack.isNonnull() && trackerTrack.isNonnull() && tpfmsTrack.isNonnull() && pickyTrack.isNonnull();
  }
  
  inline bool cocktailInputIsOK(const reco::Muon& muon){
    return cocktailInputIsOK(muon.globalTrack(),
                             muon.innerTrack(),
                             muon.tpfmsTrack(),
                             muon.pickyTrack());
  }
  
  reco::Muon::MuonTrackTypePair muonBestTrack(const reco::Muon& muon, reco::TunePType tunePType);
}
#endif