MuonErrorMatrixAdjuster.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackingTools_MuonErrorMatrixAdjuster_H
#define RecoMuon_TrackingTools_MuonErrorMatrixAdjuster_H
 
#include <memory>
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include <FWCore/Framework/interface/EventSetup.h>
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include <DataFormats/TrackReco/interface/Track.h>
#include <DataFormats/TrackReco/interface/TrackExtra.h>
#include <DataFormats/TrackingRecHit/interface/TrackingRecHit.h>
 
#include "FWCore/Utilities/interface/InputTag.h"
 
class FreeTrajectoryState;
class MuonErroMatrix;
class MagneticField;
class MuonErrorMatrix;
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
 
#include <FWCore/Framework/interface/ESHandle.h>
 
//
// class decleration
//
 
class MuonErrorMatrixAdjuster : public edm::EDProducer {
public:
  explicit MuonErrorMatrixAdjuster(const edm::ParameterSet&);
  ~MuonErrorMatrixAdjuster() override;
 
private:
  void beginJob() override;
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endJob() override;
 
  reco::TrackBase::CovarianceMatrix fix_cov_matrix(const reco::TrackBase::CovarianceMatrix& error_matrix,
                                                   const GlobalVector& momentum);
  void multiply(reco::TrackBase::CovarianceMatrix& revised_matrix,
                const reco::TrackBase::CovarianceMatrix& scale_matrix);
  bool divide(reco::TrackBase::CovarianceMatrix& num_matrix, const reco::TrackBase::CovarianceMatrix& denom_matrix);
 
  reco::Track makeTrack(const reco::Track& recotrack_orig, const FreeTrajectoryState& PCAstate);
 
  bool selectTrack(const reco::Track& recotrack_orig);
 
  reco::TrackExtra* makeTrackExtra(const reco::Track& recotrack_orig,
                                   reco::Track& recotrack,
                                   reco::TrackExtraCollection& TEcol);
 
  bool attachRecHits(const reco::Track& recotrack_orig,
                     reco::Track& recotrack,
                     reco::TrackExtra& trackextra,
                     TrackingRecHitCollection& RHcol,
                     const TrackerTopology& ttopo);
 
  // ----------member data ---------------------------
  std::string theCategory;
 
  edm::InputTag theTrackLabel;
 
  std::string theInstanceName;
 
  bool theRescale;
 
  edm::ParameterSet theMatrixProvider_pset;
  MuonErrorMatrix* theMatrixProvider;
 
  edm::ESHandle<MagneticField> theField;
 
  edm::RefProd<reco::TrackExtraCollection> theRefprodTE;
  edm::Ref<reco::TrackExtraCollection>::key_type theTEi;
 
  edm::RefProd<TrackingRecHitCollection> theRefprodRH;
  edm::Ref<TrackingRecHitCollection>::key_type theRHi;
};
 
#endif