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(MuonTrajectoryBuilder* ConcreteMuonTrajectoryBuilder, MuonTrackLoader* trackLoader);

  MuonTrackFinder(MuonTrajectoryBuilder* ConcreteMuonTrajectoryBuilder,
                  MuonTrackLoader* trackLoader,
                  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(const TrajectoryContainer&, edm::Event&, const TrackerTopology& ttopo);

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

private:
  MuonTrajectoryBuilder* theTrajBuilder;

  MuonTrajectoryCleaner* theTrajCleaner;

  MuonTrackLoader* theTrackLoader;
};
#endif