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/one/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"
#include "DQMServices/Core/interface/DQMStore.h"

#include <DQMServices/Core/interface/DQMEDAnalyzer.h>

namespace reco {
  class Track;
}

class MuonServiceProxy;
class TrajectorySeed;
class MuonUpdatorAtVertex;
//
// class decleration
//

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

  ~MuonSeedTrack() override;

private:
  void beginJob() override;

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

  void endJob() override;

  TrajectoryStateOnSurface getSeedTSOS(const TrajectorySeed& seed) const;

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

  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