TwoBodyDecayMomConstraintProducer Class Reference

#include <RecoTracker/ConstraintProducerTest/src/TwoBodyDecayMomConstraintProducer.cc>

Inheritance diagram for TwoBodyDecayMomConstraintProducer:

Public Member Functions
	TwoBodyDecayMomConstraintProducer (const edm::ParameterSet &)
	~TwoBodyDecayMomConstraintProducer () override=default
Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final
bool	wantsGlobalLuminosityBlocks () const final
bool	wantsGlobalRuns () const final
bool	wantsStreamLuminosityBlocks () const final
bool	wantsStreamRuns () const final
Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
	~EDProducerBase () override
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 ()
std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const
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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
	~ProducerBase () noexcept(false) override
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
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 &	itemsToGetFrom (BranchType iType) 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
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)
virtual	~EDConsumerBase () noexcept(false)

Private Types
enum	MomentumForRefitting { atVertex, atInnermostSurface }

Private Member Functions
std::pair< bool, TrajectoryStateOnSurface >	innermostState (const reco::TransientTrack &ttrack) const
bool	match (const TrajectoryStateOnSurface &newTsos, const TrajectoryStateOnSurface &oldTsos) const
std::pair< double, double >	momentaAtInnermostSurface (const TwoBodyDecay &tbd, const std::vector< reco::TransientTrack > &ttracks, const MagneticField *magField) const
std::pair< double, double >	momentaAtVertex (const TwoBodyDecay &tbd) const
void	produce (edm::StreamID streamid, edm::Event &, const edm::EventSetup &) const override

Static Private Member Functions
static MomentumForRefitting	momentumForRefittingFromString (std::string momentumForRefittingString)

Private Attributes
const edm::InputTag	bsSrcTag_
edm::EDGetTokenT< reco::BeamSpot >	bsToken_
const double	chi2CutValue_
const double	fixedMomentumError_
const MomentumForRefitting	momentumForRefitting_
const double	primaryMass_
const double	primaryWidth_
const double	secondaryMass_
const double	sigmaPositionCutValue_
const edm::InputTag	srcTag_
const TwoBodyDecayFitter	tbdFitter_
edm::EDGetTokenT< reco::TrackCollection >	trackCollToken_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType
Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::global::EDProducerBase
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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
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

Description: Produces track parameter constraints for refitting tracks, according to information TwoBodyDecay kinematic fit.

Original Author: Edmund Widl

Definition at line 40 of file TwoBodyDecayMomConstraintProducer.cc.

Member Enumeration Documentation

enum TwoBodyDecayMomConstraintProducer::MomentumForRefitting

private

Enumerator
atVertex
atInnermostSurface

Definition at line 74 of file TwoBodyDecayMomConstraintProducer.cc.

{ atVertex, atInnermostSurface };

Constructor & Destructor Documentation

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

explicit

Definition at line 87 of file TwoBodyDecayMomConstraintProducer.cc.

References bsSrcTag_, bsToken_, srcTag_, and trackCollToken_.

                                                                                                     :
  srcTag_( iConfig.getParameter<edm::InputTag>( "src" ) ),
  bsSrcTag_( iConfig.getParameter<edm::InputTag>( "beamSpot" ) ),
  tbdFitter_( iConfig ),
  primaryMass_( iConfig.getParameter<double>( "primaryMass" ) ),
  primaryWidth_( iConfig.getParameter<double>( "primaryWidth" ) ),
  secondaryMass_( iConfig.getParameter<double>( "secondaryMass" ) ),
  sigmaPositionCutValue_( iConfig.getParameter<double>( "sigmaPositionCut" ) ),
  chi2CutValue_( iConfig.getParameter<double>( "chi2Cut" ) ),
  fixedMomentumError_( iConfig.getParameter<double>( "fixedMomentumError" ) ),
  momentumForRefitting_( momentumForRefittingFromString( iConfig.getParameter<std::string>( "momentumForRefitting" ) ) )
{
  trackCollToken_ = consumes<reco::TrackCollection>(edm::InputTag(srcTag_));
  bsToken_ = consumes<reco::BeamSpot>(edm::InputTag(bsSrcTag_));

  produces< std::vector<MomentumConstraint> >();
  produces<TrackMomConstraintAssociationCollection>();

//   //debug
//   histos_["deltaEta1"] = new TH1F( "deltaEta1", "deltaEta1", 200, -1., 1. );
//   histos_["deltaP1"] = new TH1F( "deltaP1", "deltaP1", 200, -50., 50. );

//   histos_["deltaEta2"] = new TH1F( "deltaEta2", "deltaEta2", 200, -1., 1. );
//   histos_["deltaP2"] = new TH1F( "deltaP2", "deltaP2", 200, -50., 50. );
}

TwoBodyDecayMomConstraintProducer::~TwoBodyDecayMomConstraintProducer ( )

overridedefault

Member Function Documentation

std::pair< bool, TrajectoryStateOnSurface > TwoBodyDecayMomConstraintProducer::innermostState ( const reco::TransientTrack &  ttrack ) const

private

Definition at line 237 of file TwoBodyDecayMomConstraintProducer.cc.

References TrajectoryStateOnSurface::globalPosition(), reco::TransientTrack::innermostMeasurementState(), insideOut, reco::TransientTrack::outermostMeasurementState(), and PV3DBase< T, PVType, FrameType >::perp().

Referenced by momentaAtInnermostSurface().

{
  double outerR = ttrack.outermostMeasurementState().globalPosition().perp();
  double innerR = ttrack.innermostMeasurementState().globalPosition().perp();
  bool insideOut = ( outerR > innerR );
  return std::make_pair( insideOut, insideOut ? ttrack.innermostMeasurementState() : ttrack.outermostMeasurementState() );
}

bool TwoBodyDecayMomConstraintProducer::match ( const TrajectoryStateOnSurface &  newTsos, const TrajectoryStateOnSurface &  oldTsos  ) const

private

Definition at line 247 of file TwoBodyDecayMomConstraintProducer.cc.

References TrajectoryStateOnSurface::localPosition(), sigmaPositionCutValue_, PV3DBase< T, PVType, FrameType >::x(), and PV3DBase< T, PVType, FrameType >::y().

Referenced by momentaAtInnermostSurface().

{
  LocalPoint lp1 = newTsos.localPosition();
  LocalPoint lp2 = oldTsos.localPosition();

  double deltaX = lp1.x() - lp2.x();
  double deltaY = lp1.y() - lp2.y();

  return ( ( fabs(deltaX) < sigmaPositionCutValue_ ) && ( fabs(deltaY) < sigmaPositionCutValue_ ) );
}

std::pair< double, double > TwoBodyDecayMomConstraintProducer::momentaAtInnermostSurface ( const TwoBodyDecay &  tbd, const std::vector< reco::TransientTrack > &  ttracks, const MagneticField *  magField  ) const

private

Get the innermost trajectory states

Construct the TBD trajectory states

Match the old and the new estimates for the trajectory state

Definition at line 209 of file TwoBodyDecayMomConstraintProducer.cc.

References innermostState(), TwoBodyDecayTrajectoryState::isValid(), match(), mps_fire::result, secondaryMass_, and TwoBodyDecayTrajectoryState::trajectoryStates().

Referenced by produce().

{
  std::pair<double, double> result = std::make_pair( -1., -1. );

  std::pair<bool, TrajectoryStateOnSurface> oldInnermostState1 = innermostState( ttracks[0] );
  std::pair<bool, TrajectoryStateOnSurface> oldInnermostState2 = innermostState( ttracks[1] );
  if ( !oldInnermostState1.second.isValid() || !oldInnermostState2.second.isValid() ) return result;

  TwoBodyDecayTrajectoryState::TsosContainer trackTsos( oldInnermostState1.second, oldInnermostState2.second );
  TwoBodyDecayTrajectoryState tbdTrajState( trackTsos, tbd, secondaryMass_, magField, true );
  if ( !tbdTrajState.isValid() ) return result;

  bool match1 = match( tbdTrajState.trajectoryStates(true).first, oldInnermostState1.second );
  bool match2 = match( tbdTrajState.trajectoryStates(true).second, oldInnermostState2.second );
  if ( !match1 || !match2 ) return result;

  result = std::make_pair( fabs( 1./tbdTrajState.trajectoryStates(true).first.localParameters().qbp() ),
               fabs( 1./tbdTrajState.trajectoryStates(true).second.localParameters().qbp() ) );
  return result;
}

std::pair< double, double > TwoBodyDecayMomConstraintProducer::momentaAtVertex ( const TwoBodyDecay &  tbd ) const

private

Definition at line 197 of file TwoBodyDecayMomConstraintProducer.cc.

References TwoBodyDecayModel::cartesianSecondaryMomenta(), TwoBodyDecayParameters::mass, and secondaryMass_.

Referenced by produce().

{
  // construct global trajectory parameters at the starting point
  TwoBodyDecayModel tbdDecayModel( tbd[TwoBodyDecayParameters::mass], secondaryMass_ );
  std::pair< AlgebraicVector, AlgebraicVector > secondaryMomenta = tbdDecayModel.cartesianSecondaryMomenta( tbd );

  return std::make_pair( secondaryMomenta.first.norm(),
             secondaryMomenta.second.norm() );
}

TwoBodyDecayMomConstraintProducer::MomentumForRefitting TwoBodyDecayMomConstraintProducer::momentumForRefittingFromString ( std::string  momentumForRefittingString )

staticprivate

Definition at line 260 of file TwoBodyDecayMomConstraintProducer.cc.

References atInnermostSurface, atVertex, DEFINE_FWK_MODULE, and Exception.

                                                                                                   {
  if ( strMomentumForRefitting == "atVertex" ) {
    return atVertex;
  } else if ( strMomentumForRefitting == "atInnermostSurface" ) {
    return atInnermostSurface;
  } else {
    throw cms::Exception("TwoBodyDecayMomConstraintProducer") << "value of config  variable 'momentumForRefitting': "
                                                              << "has to be 'atVertex' or 'atInnermostSurface'";
  }
}

void TwoBodyDecayMomConstraintProducer::produce ( edm::StreamID  streamid, edm::Event &  iEvent, const edm::EventSetup &  iSetup  ) const

overrideprivate

Construct virtual measurement (for TBD)

Get transient tracks from track collection

Fit the TBD

Definition at line 116 of file TwoBodyDecayMomConstraintProducer.cc.

References atInnermostSurface, atVertex, ecalDrivenElectronSeedsParameters_cff::beamSpot, bsToken_, TwoBodyDecay::chi2(), chi2CutValue_, TwoBodyDecayFitter::estimate(), fixedMomentumError_, edm::EventSetup::get(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), TwoBodyDecay::isValid(), momentaAtInnermostSurface(), momentaAtVertex(), momentumForRefitting_, eostools::move(), or, convertSQLitetoXML_cfg::output, primaryMass_, primaryWidth_, edm::Handle< T >::product(), edm::ESHandle< T >::product(), edm::Event::put(), secondaryMass_, reco::TransientTrack::setES(), tbdFitter_, pfDisplacedTrackerVertex_cfi::trackColl, and trackCollToken_.

{
  using namespace edm;

  Handle<reco::TrackCollection> trackColl;
  iEvent.getByToken(trackCollToken_, trackColl);

  Handle<reco::BeamSpot> beamSpot;
  iEvent.getByToken(bsToken_, beamSpot);

  ESHandle<MagneticField> magField;
  iSetup.get<IdealMagneticFieldRecord>().get( magField );

  edm::RefProd<std::vector<MomentumConstraint> > rPairs = iEvent.getRefBeforePut<std::vector<MomentumConstraint> >();
  std::unique_ptr<std::vector<MomentumConstraint> > pairs(new std::vector<MomentumConstraint>);
  std::unique_ptr<TrackMomConstraintAssociationCollection> output(new TrackMomConstraintAssociationCollection(trackColl, rPairs));
    
  if ( trackColl->size() == 2 )
  {
    TwoBodyDecayVirtualMeasurement vm( primaryMass_, primaryWidth_, secondaryMass_, *beamSpot.product() );

    std::vector<reco::TransientTrack> ttracks(2);
    ttracks[0] = reco::TransientTrack( reco::TrackRef( trackColl, 0 ), magField.product() );
    ttracks[0].setES( iSetup );
    ttracks[1] = reco::TransientTrack( reco::TrackRef( trackColl, 1 ), magField.product() );
    ttracks[1].setES( iSetup );

    TwoBodyDecay tbd = tbdFitter_.estimate( ttracks, vm );

    if ( !tbd.isValid() or ( tbd.chi2() > chi2CutValue_ ) ) return;

    std::pair<double, double> fitMomenta;
    if ( momentumForRefitting_ == atVertex ) {
      fitMomenta = momentaAtVertex( tbd );
    } else if ( momentumForRefitting_ == atInnermostSurface ) {
      fitMomenta = momentaAtInnermostSurface( tbd, ttracks, magField.product() );
    } // no other possibility!!!

    if ( ( fitMomenta.first > 0. ) and ( fitMomenta.second > 0. ) )
    {
      // produce constraint for first track
      MomentumConstraint constraint1( fitMomenta.first, fixedMomentumError_ );
      pairs->push_back( constraint1 );
      output->insert(reco::TrackRef(trackColl, 0), edm::Ref<std::vector<MomentumConstraint> >(rPairs, 0));

      // produce constraint for second track
      MomentumConstraint constraint2( fitMomenta.second, fixedMomentumError_ );
      pairs->push_back( constraint2 );
      output->insert(reco::TrackRef(trackColl, 1), edm::Ref<std::vector<MomentumConstraint> >(rPairs, 1));
    }

//     // debug
//     if ( tbd.isValid() and ( fitMomenta.first > 0. ) and ( fitMomenta.second > 0. ) ) {
//       TwoBodyDecayModel model;
//       const std::pair< AlgebraicVector, AlgebraicVector > fitMomenta = model.cartesianSecondaryMomenta( tbd );

//       TLorentzVector recoMomentum1( ttracks[0].track().px(), ttracks[0].track().py(), ttracks[0].track().pz(),
//                  sqrt((ttracks[0].track().p()*ttracks[0].track().p())+0.105658*0.105658) );
//       TLorentzVector fitMomentum1( fitMomenta.first[0], fitMomenta.first[1], fitMomenta.first[2],
//                 sqrt( fitMomenta.first.normsq()+0.105658*0.105658) );
//       histos_["deltaP1"]->Fill( recoMomentum1.P() - fitMomentum1.P() );
//       histos_["deltaEta1"]->Fill( recoMomentum1.Eta() - fitMomentum1.Eta() );

//       TLorentzVector recoMomentum2( ttracks[1].track().px(), ttracks[1].track().py(), ttracks[1].track().pz(),
//                  sqrt((ttracks[1].track().p()*ttracks[1].track().p())+0.105658*0.105658) );
//       TLorentzVector fitMomentum2( fitMomenta.second[0], fitMomenta.second[1], fitMomenta.second[2],
//                 sqrt( fitMomenta.second.normsq()+0.105658*0.105658) );
//       histos_["deltaP2"]->Fill( recoMomentum2.P() - fitMomentum2.P() );
//       histos_["deltaEta2"]->Fill( recoMomentum2.Eta() - fitMomentum2.Eta() );
//     }
  }
  
  iEvent.put(std::move(pairs));
  iEvent.put(std::move(output));
}

Member Data Documentation

const edm::InputTag TwoBodyDecayMomConstraintProducer::bsSrcTag_

private

Definition at line 62 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by TwoBodyDecayMomConstraintProducer().

edm::EDGetTokenT<reco::BeamSpot> TwoBodyDecayMomConstraintProducer::bsToken_

private

Definition at line 80 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce(), and TwoBodyDecayMomConstraintProducer().

const double TwoBodyDecayMomConstraintProducer::chi2CutValue_

private

Definition at line 71 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

const double TwoBodyDecayMomConstraintProducer::fixedMomentumError_

private

Definition at line 72 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

const MomentumForRefitting TwoBodyDecayMomConstraintProducer::momentumForRefitting_

private

Definition at line 75 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

const double TwoBodyDecayMomConstraintProducer::primaryMass_

private

Definition at line 66 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

const double TwoBodyDecayMomConstraintProducer::primaryWidth_

private

Definition at line 67 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

const double TwoBodyDecayMomConstraintProducer::secondaryMass_

private

Definition at line 68 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by momentaAtInnermostSurface(), momentaAtVertex(), and produce().

const double TwoBodyDecayMomConstraintProducer::sigmaPositionCutValue_

private

Definition at line 70 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by match().

const edm::InputTag TwoBodyDecayMomConstraintProducer::srcTag_

private

Definition at line 61 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by TwoBodyDecayMomConstraintProducer().

const TwoBodyDecayFitter TwoBodyDecayMomConstraintProducer::tbdFitter_

private

Definition at line 64 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce().

edm::EDGetTokenT<reco::TrackCollection> TwoBodyDecayMomConstraintProducer::trackCollToken_

private

Definition at line 79 of file TwoBodyDecayMomConstraintProducer.cc.

Referenced by produce(), and TwoBodyDecayMomConstraintProducer().