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