MuonSeedFromRecHits.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackingTools_MuonSeedFromRecHits_H
#define RecoMuon_TrackingTools_MuonSeedFromRecHits_H

#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "RecoMuon/TransientTrackingRecHit/interface/MuonTransientTrackingRecHit.h"
#include "MagneticField/Engine/interface/MagneticField.h"
class MuonSeedPtExtractor;

class MuonSeedFromRecHits {
public:
  MuonSeedFromRecHits();
  virtual ~MuonSeedFromRecHits() {}

  void setBField(const MagneticField* field) { theField = field; }
  void setPtExtractor(const MuonSeedPtExtractor* extractor) { thePtExtractor = extractor; }

  void add(MuonTransientTrackingRecHit::MuonRecHitPointer hit) { theRhits.push_back(hit); }
  MuonTransientTrackingRecHit::ConstMuonRecHitPointer firstRecHit() const { return theRhits.front(); }
  unsigned int nrhit() const { return theRhits.size(); }
  void clear() { theRhits.clear(); }

  TrajectorySeed createSeed(float ptmean,
                            float sptmean,
                            MuonTransientTrackingRecHit::ConstMuonRecHitPointer last) const;

protected:
  typedef MuonTransientTrackingRecHit::MuonRecHitContainer MuonRecHitContainer;
  typedef MuonTransientTrackingRecHit::MuonRecHitPointer MuonRecHitPointer;
  typedef MuonTransientTrackingRecHit::ConstMuonRecHitPointer ConstMuonRecHitPointer;

  MuonTransientTrackingRecHit::MuonRecHitContainer theRhits;
  const MagneticField* theField;
  const MuonSeedPtExtractor* thePtExtractor;
};

#endif