MuonSeedTrack.h

Go to the documentation of this file.

#ifndef Validation_RecoMuon_MuonSeedTrack_H
#define Validation_RecoMuon_MuonSeedTrack_H

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"

#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"

namespace reco {class Track;}

class MuonServiceProxy;
class TrajectorySeed;
class MuonUpdatorAtVertex;

//
// class decleration
//

class MuonSeedTrack : public edm::EDProducer {
 public:
  explicit MuonSeedTrack(const edm::ParameterSet&);

  ~MuonSeedTrack();
  
 private:
  virtual void beginJob() ;

  virtual void produce(edm::Event&, const edm::EventSetup&);

  virtual void endJob() ;
  
  TrajectoryStateOnSurface getSeedTSOS(const TrajectorySeed& seed) const;

  void setAlias( std::string alias ){
    alias.erase( alias.size() - 1, alias.size() );
    theAlias=alias;
  }

  double computeNDOF(const TrajectorySeed&) const;

  std::pair<bool,reco::Track> buildTrackAtPCA(const TrajectorySeed&) const;
  
  // ----------member data ---------------------------
  
  MuonServiceProxy* theService;

  MuonUpdatorAtVertex *theUpdatorAtVtx;

  std::string theAlias;

  //std::string theSeedPropagatorName;

  edm::InputTag theSeedsLabel;
  edm::EDGetTokenT<TrajectorySeedCollection>  theSeedsToken;
  
  bool theAllowNoVtxFlag;      
};

#endif