MuonTrackFinder.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackingTools_MuonTrackFinder_H
#define RecoMuon_TrackingTools_MuonTrackFinder_H

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackExtraFwd.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "RecoMuon/TrackingTools/interface/MuonCandidate.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"

#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"

#include <vector>

class MuonTrajectoryBuilder;
class MuonTrajectoryCleaner;
class MuonTrackLoader;

class MuonTrackFinder {
public:
  typedef MuonCandidate::TrajectoryContainer TrajectoryContainer;
  typedef MuonCandidate::CandidateContainer CandidateContainer;
  typedef std::pair<const Trajectory*, reco::TrackRef> TrackCand;

public:
  MuonTrackFinder(std::unique_ptr<MuonTrajectoryBuilder> ConcreteMuonTrajectoryBuilder,
                  std::unique_ptr<MuonTrackLoader> trackLoader,
                  edm::ConsumesCollector iC);

  MuonTrackFinder(std::unique_ptr<MuonTrajectoryBuilder> ConcreteMuonTrajectoryBuilder,
                  std::unique_ptr<MuonTrackLoader> trackLoader,
                  std::unique_ptr<MuonTrajectoryCleaner> cleaner,
                  edm::ConsumesCollector iC);

  virtual ~MuonTrackFinder();

  edm::OrphanHandle<reco::TrackCollection> reconstruct(const edm::Handle<edm::View<TrajectorySeed> >&,
                                                       edm::Event&,
                                                       const edm::EventSetup&);

  void reconstruct(const std::vector<TrackCand>&, edm::Event&, const edm::EventSetup&);

private:
  void setEvent(const edm::Event&);

  edm::OrphanHandle<reco::TrackCollection> load(TrajectoryContainer&, edm::Event&, const TrackerTopology& ttopo);

  void load(CandidateContainer&, edm::Event&, const TrackerTopology& ttopo);

private:
  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> theTtopoToken;

  std::unique_ptr<MuonTrajectoryBuilder> theTrajBuilder;

  std::unique_ptr<MuonTrajectoryCleaner> theTrajCleaner;

  std::unique_ptr<MuonTrackLoader> theTrackLoader;
};
#endif