Tau3MuReco.h

Go to the documentation of this file.

#ifndef TAU3MURECO_H
#define TAU3MURECO_H

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include <DataFormats/TrackReco/interface/Track.h>
#include <DataFormats/MuonReco/interface/Muon.h>
#include <DataFormats/MuonReco/interface/MuonFwd.h>

class Tau3MuReco {
public:
  Tau3MuReco(const edm::ParameterSet& iConfig, edm::ConsumesCollector&& iC);
  ~Tau3MuReco();

  bool doTau3MuReco(const edm::Event& iEvent,
                    const edm::EventSetup& iSetup,
                    reco::MuonCollection* muonCollection,
                    reco::TrackCollection* trackCollection);

private:
  bool check4MuonTrack(
      const reco::Track& track);  //compares track with reconstructed muons and return true if they are equal
  bool find3rdTrack(
      const edm::Event& iEvent,
      const edm::EventSetup& iSetup,
      const reco::TrackCollection& Tracks);  //try to find a 3rd muon in tracks, if this was not detected as a muon
  bool findCorrectPairing();                 //find the correct 3 muons, if more than 3 muons has been reconstructed
  double getInvariantMass(const reco::TrackCollection* tracks, const double MuonMass = 0.106);
  double getDeltaR(const reco::Track& track1, const reco::Track& track2);
  bool removeIncorrectMuon();  //try to remove one muon, which seems to come not from a tau->3Mu decay

  const double m_kMatchingDeltaR;
  const double m_kMatchingPt;
  const double m_kTauMassCut;
  const double m_kTauMass;
  const double m_kMuonMass;

  const edm::EDGetTokenT<reco::MuonCollection> m_kMuonSourceToken;
  const edm::EDGetTokenT<reco::TrackCollection> m_kTrackSourceToken;

  reco::MuonCollection* m_MuonCollection;
  reco::TrackCollection* m_TrackCollection;
};

#endif