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 "FWCore/Framework/interface/ESHandle.h"
 
#include <vector>
 
namespace edm {
  class ParameterSet;
  class Event;
  class EventSetup;
}  // namespace edm
class TrackerTopology;
 
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);
 
  MuonTrackFinder(std::unique_ptr<MuonTrajectoryBuilder> ConcreteMuonTrajectoryBuilder,
                  std::unique_ptr<MuonTrackLoader> trackLoader,
                  std::unique_ptr<MuonTrajectoryCleaner> cleaner);
 
  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:
  std::unique_ptr<MuonTrajectoryBuilder> theTrajBuilder;
 
  std::unique_ptr<MuonTrajectoryCleaner> theTrajCleaner;
 
  std::unique_ptr<MuonTrackLoader> theTrackLoader;
};
#endif