RecoMuon/TrackingTools/interface/MuonSeedFromRecHits.h - master - githubmirror/cmssw

RecoMuon/TrackingTools/interface/MuonSeedFromRecHits.h

46 строк2 KB

Cms Build

Clang-Format

31 май 2019, 13:18

31 май 2019, 13:18ae1abd0

#ifndef RecoMuon_TrackingTools_MuonSeedFromRecHits_H
#define RecoMuon_TrackingTools_MuonSeedFromRecHits_H

/**  \class MuonSeedFromRecHits
 *
 *  \author A. Vitelli - INFN Torino
 *  
 *  \author porting R.Bellan - INFN Torino
 *
 *   Generate a seed starting from a list of RecHits 
 *
 */

#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