#include <MuonErrorMatrixAdjuster.h>

Inheritance diagram for MuonErrorMatrixAdjuster:

Public Member Functions
	MuonErrorMatrixAdjuster (const edm::ParameterSet &)
	constructor More...

	~MuonErrorMatrixAdjuster ()
	destructor More...

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector < edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector < ProductResolverIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
bool	attachRecHits (const reco::Track &recotrack_orig, reco::Track &recotrack, reco::TrackExtra &trackextra, TrackingRecHitCollection &RHcol, const TrackerTopology &ttopo)
	attached rechits to the newly created reco::Track and reco::TrackExtra More...

virtual void	beginJob ()
	framework method More...

bool	divide (reco::TrackBase::CovarianceMatrix &num_matrix, const reco::TrackBase::CovarianceMatrix &denom_matrix)
	divide the num_matrix (first argument) by second matrix, TERM by TERM More...

virtual void	endJob ()

reco::TrackBase::CovarianceMatrix	fix_cov_matrix (const reco::TrackBase::CovarianceMatrix &error_matrix, const GlobalVector &momentum)
	return a corrected error matrix More...

reco::Track	makeTrack (const reco::Track &recotrack_orig, const FreeTrajectoryState &PCAstate)
	create a corrected reco::Track from itself and trajectory state (redundant information) More...

reco::TrackExtra *	makeTrackExtra (const reco::Track &recotrack_orig, reco::Track &recotrack, reco::TrackExtraCollection &TEcol)
	create a track extra for the newly created recotrack, scaling the outer/inner measurment error matrix by the scale matrix recotrack/recotrack_orig More...

void	multiply (reco::TrackBase::CovarianceMatrix &revised_matrix, const reco::TrackBase::CovarianceMatrix &scale_matrix)
	mutliply revised_matrix (first argument) by second matrix TERM by TERM More...

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

bool	selectTrack (const reco::Track &recotrack_orig)
	make a selection on the reco:Track. (dummy for the moment) More...

Private Attributes
std::string	theCategory
	log category: MuonErrorMatrixAdjuster More...

edm::ESHandle< MagneticField >	theField
	hold on to the magnetic field More...

std::string	theInstanceName
	instrance name of the created track collecion. rechit and trackextra have no instance name More...

MuonErrorMatrix *	theMatrixProvider

edm::ParameterSet	theMatrixProvider_pset
	holds the error matrix parametrization More...

edm::RefProd < TrackingRecHitCollection >	theRefprodRH
	get reference before put rechit to the event, in order to create edm::Ref More...

edm::RefProd < reco::TrackExtraCollection >	theRefprodTE
	get reference before put track extra to the event, in order to create edm::Ref More...

bool	theRescale
	select the rescaling or replacing method to correct the error matrix More...

edm::Ref < TrackingRecHitCollection > ::key_type	theRHi

edm::Ref < reco::TrackExtraCollection > ::key_type	theTEi

edm::InputTag	theTrackLabel
	input tag of the reco::Track collection to be corrected More...

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

EDProducer which duplicatesa collection of track, adjusting their error matrix

track collection is retrieve from the event, duplicated, while the error matrix is corrected rechit are copied into a new collection track extra is also copied and error matrix are corrected by the same scale factors

Author: Jean-Roch Vlimant UCSB; Finn Rebassoo UCSB

Definition at line 45 of file MuonErrorMatrixAdjuster.h.

Constructor & Destructor Documentation

MuonErrorMatrixAdjuster::MuonErrorMatrixAdjuster ( const edm::ParameterSet & iConfig )

explicit

constructor

Definition at line 19 of file MuonErrorMatrixAdjuster.cc.

References edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, theCategory, theInstanceName, theMatrixProvider_pset, theRescale, and theTrackLabel.

 {
   theCategory="MuonErrorMatrixAdjuster";
   theInstanceName = iConfig.getParameter<std::string>("instanceName");
   //register your products
   produces<reco::TrackCollection>( theInstanceName);
   produces<TrackingRecHitCollection>(); 
   produces<reco::TrackExtraCollection>();
           
   theTrackLabel = iConfig.getParameter<edm::InputTag>("trackLabel");
   theRescale = iConfig.getParameter<bool>("rescale");
 
   theMatrixProvider_pset = iConfig.getParameter<edm::ParameterSet>("errorMatrix_pset");
 }

MuonErrorMatrixAdjuster::~MuonErrorMatrixAdjuster ( )

destructor

Definition at line 35 of file MuonErrorMatrixAdjuster.cc.

 {
  
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)
 
 }

Member Function Documentation

bool MuonErrorMatrixAdjuster::attachRecHits	(	const reco::Track &	recotrack_orig,
		reco::Track &	recotrack,
		reco::TrackExtra &	trackextra,
		TrackingRecHitCollection &	RHcol,
		const TrackerTopology &	ttopo
	)

private

attached rechits to the newly created reco::Track and reco::TrackExtra

Definition at line 139 of file MuonErrorMatrixAdjuster.cc.

References reco::TrackBase::appendHitPattern(), edm::OwnVector< T, P >::push_back(), reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), reco::TrackExtraBase::setHits(), theRefprodRH, and theRHi.

Referenced by produce().

                                                                     {
   //loop over the hits of the original track
   trackingRecHit_iterator recHit = recotrack_orig.recHitsBegin();
   auto const firstHitIndex = theRHi;
   for (; recHit!=recotrack_orig.recHitsEnd();++recHit){
     //clone it. this is meandatory
     TrackingRecHit * hit = (*recHit)->clone();
     
     //put it on the new track
     recotrack.appendHitPattern(*hit, ttopo);
     //copy them in the new collection
     RHcol.push_back(hit);
     ++theRHi;
 
   }//loop over original rechits
   //do something with the trackextra 
   trackextra.setHits(theRefprodRH, firstHitIndex, theRHi-firstHitIndex);
   
   return true; //if nothing fails
 }

void MuonErrorMatrixAdjuster::beginJob ( void )

privatevirtual

framework method

Reimplemented from edm::EDProducer.

Definition at line 241 of file MuonErrorMatrixAdjuster.cc.

References theMatrixProvider, and theMatrixProvider_pset.

 {
   theMatrixProvider = new MuonErrorMatrix(theMatrixProvider_pset);
 }

bool MuonErrorMatrixAdjuster::divide	(	reco::TrackBase::CovarianceMatrix &	num_matrix,
		const reco::TrackBase::CovarianceMatrix &	denom_matrix
	)

private

divide the num_matrix (first argument) by second matrix, TERM by TERM

Definition at line 69 of file MuonErrorMatrixAdjuster.cc.

References i, and j.

Referenced by cuy.FindIssue::__init__(), and makeTrackExtra().

                                                                                                                                   {
   //divide term by term the matrix
   // the true type of the matrix is such that [i][j] is the same memory object as [j][i]: looping i:0-5, j:0-5 double multiply the terms
   // need to loop only on i:0-5, j:i-5
   for(int i = 0;i!=5;i++){for(int j = i;j!=5;j++){
       if (denom_matrix(i,j)==0) return false;
       num_matrix(i,j)/=denom_matrix(i,j);
    }}
   return true;
 }

void MuonErrorMatrixAdjuster::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 248 of file MuonErrorMatrixAdjuster.cc.

References theMatrixProvider.

                                 {
   if ( theMatrixProvider) delete theMatrixProvider;
 }

reco::TrackBase::CovarianceMatrix MuonErrorMatrixAdjuster::fix_cov_matrix	(	const reco::TrackBase::CovarianceMatrix &	error_matrix,
		const GlobalVector &	momentum
	)

private

return a corrected error matrix

Definition at line 49 of file MuonErrorMatrixAdjuster.cc.

References MuonErrorMatrix::get(), multiply(), theMatrixProvider, and theRescale.

Referenced by makeTrack().

 {   
   //CovarianceMatrix is template for SMatrix
   reco::TrackBase::CovarianceMatrix revised_matrix( theMatrixProvider->get(momentum));
    
   if( theRescale){
     //rescale old error matrix up by a factor
     multiply(revised_matrix,error_matrix); 
   }
   return revised_matrix;
 }

reco::Track MuonErrorMatrixAdjuster::makeTrack	(	const reco::Track &	recotrack_orig,
		const FreeTrajectoryState &	PCAstate
	)

private

create a corrected reco::Track from itself and trajectory state (redundant information)

Definition at line 80 of file MuonErrorMatrixAdjuster.cc.

References RecoTauCleanerPlugins::charge, reco::TrackBase::charge(), reco::TrackBase::chi2(), beam_dqm_sourceclient-live_cfg::chi2, reco::TrackBase::covariance(), fix_cov_matrix(), LogDebug, FreeTrajectoryState::momentum(), reco::TrackBase::momentum(), ndof, reco::TrackBase::ndof(), reco::TrackBase::referencePoint(), and theCategory.

Referenced by produce().

                                                            {
   //get the parameters of the track so I can reconstruct it
   double chi2 = recotrack_orig.chi2();
   double ndof = recotrack_orig.ndof();
   math::XYZPoint refpos = recotrack_orig.referencePoint();
   math::XYZVector mom = recotrack_orig.momentum();
   int charge = recotrack_orig.charge();
   
   reco::TrackBase::CovarianceMatrix covariance_matrix = fix_cov_matrix(recotrack_orig.covariance(),PCAstate.momentum());
 
   LogDebug(theCategory)<<"chi2: "<<chi2
                <<"\n ndof: "<<ndof
                <<"\n refpos: "<<refpos
                <<"\n mom: "<<mom
                <<"\n charge: "<<charge
                <<"\n covariance:\n"<<recotrack_orig.covariance()
                <<"\n replaced by:\n"<<covariance_matrix;
 
   return reco::Track(chi2,ndof,refpos,mom,charge,covariance_matrix);
 }

reco::TrackExtra * MuonErrorMatrixAdjuster::makeTrackExtra	(	const reco::Track &	recotrack_orig,
		reco::Track &	recotrack,
		reco::TrackExtraCollection &	TEcol
	)

private

create a track extra for the newly created recotrack, scaling the outer/inner measurment error matrix by the scale matrix recotrack/recotrack_orig

Definition at line 102 of file MuonErrorMatrixAdjuster.cc.

References reco::TrackBase::covariance(), divide(), reco::Track::extra(), edm::Ref< C, T, F >::isNull(), multiply(), reco::Track::setExtra(), theCategory, theRefprodTE, and theTEi.

Referenced by produce().

                                                               { 
   //get the 5x5 matrix of recotrack/recotrack_orig
   reco::TrackBase::CovarianceMatrix scale_matrix(recotrack.covariance());
   if (!divide(scale_matrix,recotrack_orig.covariance())){
     edm::LogError( theCategory)<<"original track error matrix has term ==0... skipping.";
     return 0; }
   
   const reco::TrackExtraRef & trackExtra_orig = recotrack_orig.extra();
   if (trackExtra_orig.isNull()) {
     edm::LogError( theCategory)<<"original track has no track extra... skipping.";
     return 0;}
 
   //copy the outer state. rescaling the error matrix
   reco::TrackBase::CovarianceMatrix outerCov(trackExtra_orig->outerStateCovariance());
   multiply(outerCov,scale_matrix);
 
   //copy the inner state, rescaling the error matrix
   reco::TrackBase::CovarianceMatrix innerCov(trackExtra_orig->innerStateCovariance());
   multiply(innerCov,scale_matrix);
 
   //put the trackExtra
   TEcol.push_back(reco::TrackExtra (trackExtra_orig->outerPosition(), trackExtra_orig->outerMomentum(), true,
                     trackExtra_orig->innerPosition(), trackExtra_orig->innerMomentum(), true,
                     outerCov, trackExtra_orig->outerDetId(),
                     innerCov, trackExtra_orig->innerDetId(),
                     trackExtra_orig->seedDirection()));
 
   //add a reference to the trackextra on the track
   recotrack.setExtra(edm::Ref<reco::TrackExtraCollection>( theRefprodTE, theTEi++));
   
   //return the reference to the last inserted then
   return &(TEcol.back());
 }

void MuonErrorMatrixAdjuster::multiply	(	reco::TrackBase::CovarianceMatrix &	revised_matrix,
		const reco::TrackBase::CovarianceMatrix &	scale_matrix
	)

private

mutliply revised_matrix (first argument) by second matrix TERM by TERM

Definition at line 61 of file MuonErrorMatrixAdjuster.cc.

References i, and j.

Referenced by fix_cov_matrix(), and makeTrackExtra().

                                                                                                                                         {
   //scale term by term the matrix
   // the true type of the matrix is such that [i][j] is the same memory object as [j][i]: looping i:0-5, j:0-5 double multiply the terms
   // need to loop only on i:0-5, j:i-5
   for(int i = 0;i!=5;i++){for(int j = i;j!=5;j++){
       revised_matrix(i,j)*=scale_matrix(i,j);
     }}
 }

void MuonErrorMatrixAdjuster::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Implements edm::EDProducer.

Definition at line 168 of file MuonErrorMatrixAdjuster.cc.

References attachRecHits(), edm::EventSetup::get(), edm::Event::getByLabel(), edm::Event::getRefBeforePut(), trajectoryStateTransform::initialFreeState(), LogDebug, PV3DBase< T, PVType, FrameType >::mag(), makeTrack(), makeTrackExtra(), eostools::move(), FreeTrajectoryState::position(), edm::ESHandle< class >::product(), edm::Event::put(), Reconstruction_cff::recotrack, selectTrack(), theCategory, theField, theInstanceName, theRefprodRH, theRefprodTE, theRHi, theTEi, theTrackLabel, and testEve_cfg::tracks.

 {
   using namespace edm;
 
   //open a collection of track
   edm::Handle<reco::TrackCollection> tracks;
   iEvent.getByLabel( theTrackLabel,tracks);
   LogDebug( theCategory)<<"considering: "<<tracks->size()<<" uncorrected reco::Track from the event.("<< theTrackLabel<<")";
   
   //get the mag field
   iSetup.get<IdealMagneticFieldRecord>().get( theField);
 
   edm::ESHandle<TrackerTopology> httopo;
   iSetup.get<TrackerTopologyRcd>().get(httopo);
   const TrackerTopology& ttopo = *httopo;
 
   //prepare the output collection
   auto Toutput = std::make_unique<reco::TrackCollection>();
   auto TRHoutput = std::make_unique<TrackingRecHitCollection>();
   auto TEoutput = std::make_unique<reco::TrackExtraCollection>();
   theRefprodTE = iEvent.getRefBeforePut<reco::TrackExtraCollection>();
   theTEi=0;
   theRefprodRH =iEvent.getRefBeforePut<TrackingRecHitCollection>();
   theRHi=0;
   
   
   
   for(unsigned int it=0;it!=tracks->size();it++)
     {
       const reco::Track & recotrack_orig = (*tracks)[it];
       FreeTrajectoryState PCAstate = trajectoryStateTransform::initialFreeState(recotrack_orig, theField.product());
       if (PCAstate.position().mag()==0)
     {edm::LogError( theCategory)<<"invalid state from track initial state in "<< theTrackLabel <<". skipping.";continue;}
 
       //create a reco::Track
       reco::Track recotrack = makeTrack(recotrack_orig,PCAstate);
       
       //make a selection on the create reco::Track
       if (!selectTrack(recotrack)) continue;
       
       Toutput->push_back(recotrack);
       reco::Track & recotrackref = Toutput->back();
       
       //build the track extra
       reco::TrackExtra * extra = makeTrackExtra(recotrack_orig,recotrackref,*TEoutput);
       if (!extra){
     edm::LogError( theCategory)<<"cannot create the track extra for this track.";
     //pop the inserted track
     Toutput->pop_back();
     continue;}
       
       //attach the collection of rechits
       if (!attachRecHits(recotrack_orig,recotrackref,*extra,*TRHoutput, ttopo)){
     edm::LogError( theCategory)<<"cannot attach any rechits on this track";
     //pop the inserted track
     Toutput->pop_back();
     //pop the track extra
     TEoutput->pop_back();
     theTEi--;
     continue;}
       
     }//loop over the original reco tracks
   
   LogDebug( theCategory)<<"writing: "<<Toutput->size()<<" corrected reco::Track to the event.";
   
   iEvent.put(std::move(Toutput), theInstanceName);
   iEvent.put(std::move(TEoutput));
   iEvent.put(std::move(TRHoutput));
 
 }