#include <BPHKinematicFit.h>

Inheritance diagram for BPHKinematicFit:

Public Member Functions
double	constrMass () const
	retrieve the constraint More...

double	constrSigma () const

virtual const RefCountedKinematicVertex	currentDecayVertex () const

virtual const RefCountedKinematicParticle	currentParticle () const

virtual bool	isEmpty () const

virtual bool	isValidFit () const

virtual const RefCountedKinematicTree &	kinematicTree () const
	perform the kinematic fit and get the result More...

virtual const RefCountedKinematicTree &	kinematicTree (const std::string &name, double mass, double sigma) const

virtual const RefCountedKinematicTree &	kinematicTree (const std::string &name, double mass) const

virtual const RefCountedKinematicTree &	kinematicTree (const std::string &name, KinematicConstraint *kc) const

virtual const RefCountedKinematicTree &	kinematicTree (const std::string &name, MultiTrackKinematicConstraint *kc) const

virtual const std::vector< RefCountedKinematicParticle > &	kinParticles () const
	get kinematic particles More...

virtual std::vector< RefCountedKinematicParticle >	kinParticles (const std::vector< std::string > &names) const

virtual ParticleMass	mass () const

virtual const math::XYZTLorentzVector &	p4 () const
	compute total momentum after the fit More...

virtual void	resetKinematicFit () const
	reset the kinematic fit More...

void	setConstraint (double mass, double sigma)
	apply a mass constraint More...

	~BPHKinematicFit () override

Public Member Functions inherited from BPHDecayVertex
const reco::Track *	getTrack (const reco::Candidate *cand) const
	get Track for a daughter More...

const std::string &	getTrackSearchList (const reco::Candidate *cand) const
	retrieve track search list More...

reco::TransientTrack *	getTransientTrack (const reco::Candidate *cand) const
	get TransientTrack for a daughter More...

const std::vector< const reco::Track * > &	tracks () const
	get list of Tracks More...

const std::vector< reco::TransientTrack > &	transientTracks () const
	get list of TransientTracks More...

virtual bool	validTracks () const
	check for valid reconstructed vertex More...

virtual bool	validVertex () const

virtual const reco::Vertex &	vertex () const
	get reconstructed vertex More...

	~BPHDecayVertex () override

Public Member Functions inherited from BPHDecayMomentum
virtual const std::vector< std::string > &	compNames () const

virtual const pat::CompositeCandidate &	composite () const
	get a composite by the simple sum of simple particles More...

virtual const std::vector< BPHRecoConstCandPtr > &	daughComp () const

virtual const std::vector< const reco::Candidate * > &	daughFull () const

virtual const std::vector< const reco::Candidate * > &	daughters () const

virtual const std::vector< std::string > &	daugNames () const

virtual BPHRecoConstCandPtr	getComp (const std::string &name) const

virtual const reco::Candidate *	getDaug (const std::string &name) const

virtual const reco::Candidate *	originalReco (const reco::Candidate *daug) const
	get the original particle from the clone More...

virtual	~BPHDecayMomentum ()

Protected Member Functions
virtual void	addK (const std::string &name, const reco::Candidate *daug, double mass=-1.0, double sigma=-1.0)

virtual void	addK (const std::string &name, const reco::Candidate *daug, const std::string &searchList, double mass=-1.0, double sigma=-1.0)

virtual void	addK (const std::string &name, const BPHRecoConstCandPtr &comp)
	add a previously reconstructed particle giving it a name More...

	BPHKinematicFit ()

	BPHKinematicFit (const BPHKinematicFit *ptr)

void	setNotUpdated () const override

Protected Member Functions inherited from BPHDecayVertex
virtual void	addV (const std::string &name, const reco::Candidate *daug, const std::string &searchList, double mass)

virtual void	addV (const std::string &name, const BPHRecoConstCandPtr &comp)
	add a previously reconstructed particle giving it a name More...

	BPHDecayVertex (const edm::EventSetup *es)

	BPHDecayVertex (const BPHDecayVertex ptr, const edm::EventSetup es)

Protected Member Functions inherited from BPHDecayMomentum
virtual void	addP (const std::string &name, const reco::Candidate *daug, double mass=-1.0)

virtual void	addP (const std::string &name, const BPHRecoConstCandPtr &comp)
	add a previously reconstructed particle giving it a name More...

	BPHDecayMomentum ()

	BPHDecayMomentum (const std::map< std::string, Component > &daugMap)

	BPHDecayMomentum (const std::map< std::string, Component > &daugMap, const std::map< std::string, BPHRecoConstCandPtr > compMap)

const std::vector< Component > &	componentList () const

Private Member Functions
virtual void	buildParticles () const

virtual void	fitMomentum () const

Private Attributes
std::vector< RefCountedKinematicParticle >	allParticles

std::map< const reco::Candidate *, double >	dMSig

std::map< const reco::Candidate *, RefCountedKinematicParticle >	kinMap

RefCountedKinematicTree	kinTree

double	massConst

double	massSigma

bool	oldFit

bool	oldKPs

bool	oldMom

math::XYZTLorentzVector	totalMomentum

Detailed Description

Description: Highest-level base class to encapsulate kinematic fit operations

Author: Paolo Ronchese INFN Padova high-level base class to perform a kinematic fit

Definition at line 35 of file BPHKinematicFit.h.

Constructor & Destructor Documentation

BPHKinematicFit::~BPHKinematicFit ( )

override

Constructor is protected this object can exist only as part of a derived classDestructor

Definition at line 87 of file BPHKinematicFit.cc.

87 {

88 }

BPHKinematicFit::BPHKinematicFit ( )

protected

Definition at line 43 of file BPHKinematicFit.cc.

                                 :
  BPHDecayVertex( nullptr ),
  massConst( -1.0 ),
  massSigma( -1.0 ),
  oldKPs( true ),
  oldFit( true ),
  oldMom( true ),
  kinTree( nullptr ) {
 }

BPHKinematicFit::BPHKinematicFit ( const BPHKinematicFit * ptr )

protected

Definition at line 54 of file BPHKinematicFit.cc.

References EnergyCorrector::c, BPHDecayMomentum::Component::cand, BPHDecayMomentum::componentList(), BPHDecayMomentum::daughComp(), BPHDecayMomentum::daughters(), BPHDecayMomentum::dMap, dMSig, mps_fire::i, list(), funct::m, BPHDecayMomentum::Component::msig, gen::n, and BPHDecayMomentum::originalReco().

                                                             :
  BPHDecayVertex( ptr, nullptr ),
  massConst( -1.0 ),
  massSigma( -1.0 ),
  oldKPs( true ),
  oldFit( true ),
  oldMom( true ),
  kinTree( nullptr ) {
   map<const reco::Candidate*,const reco::Candidate*> iMap;
   const vector<const reco::Candidate*>& daug = daughters();
   const vector<Component>& list = ptr->componentList();
   int i;
   int n = daug.size();
   for ( i = 0; i < n; ++i ) {
     const reco::Candidate* cand = daug[i];
     iMap[originalReco( cand )] = cand;
   }
   for ( i = 0; i < n; ++i ) {
     const Component& c = list[i];
     dMSig[iMap[c.cand]] = c.msig;
   }
   const vector<BPHRecoConstCandPtr>& dComp = daughComp();
   int j;
   int m = dComp.size();
   for ( j = 0; j < m; ++j ) {
     const map<const reco::Candidate*,double>& dMap = dComp[j]->dMSig;
     dMSig.insert( dMap.begin(), dMap.end() );
   }
 }

Member Function Documentation

virtual void BPHKinematicFit::addK	(	const std::string &	name,
		const reco::Candidate *	daug,
		double	mass = `-1.0`,
		double	sigma = `-1.0`
	)

protectedvirtual

add a simple particle giving it a name particles are cloned, eventually specifying a different mass and a sigma

Referenced by p4(), and BPHPlusMinusCandidate::~BPHPlusMinusCandidate().

virtual void BPHKinematicFit::addK	(	const std::string &	name,
		const reco::Candidate *	daug,
		const std::string &	searchList,
		double	mass = `-1.0`,
		double	sigma = `-1.0`
	)

protectedvirtual

add a simple particle and specify a criterion to search for the associated track

virtual void BPHKinematicFit::addK	(	const std::string &	name,
		const BPHRecoConstCandPtr &	comp
	)

protectedvirtual

add a previously reconstructed particle giving it a name

void BPHKinematicFit::buildParticles ( ) const

privatevirtual

Definition at line 380 of file BPHKinematicFit.cc.

References allParticles, BPHDecayMomentum::daughFull(), dMSig, BPHDecayVertex::getTransientTrack(), kinMap, mass(), reco::Candidate::mass(), gen::n, oldKPs, KinematicParticleFactoryFromTransientTrack::particle(), and groupFilesInBlocks::tt.

Referenced by kinParticles().

                                            {
   kinMap.clear();
   allParticles.clear();
   const vector<const reco::Candidate*>& daug = daughFull();
   KinematicParticleFactoryFromTransientTrack pFactory;
   int n = daug.size();
   allParticles.reserve( n );
   float chi = 0.0;
   float ndf = 0.0;
   while ( n-- ) {
     const reco::Candidate* cand = daug[n];
     ParticleMass mass = cand->mass();
     float sigma = dMSig.find( cand )->second;
     if ( sigma < 0 ) sigma = 1.0e-7;
     reco::TransientTrack* tt = getTransientTrack( cand );
     if ( tt != nullptr ) allParticles.push_back( kinMap[cand] =
                                            pFactory.particle( *tt, 
                                            mass, chi, ndf, sigma ) );
   }
   oldKPs = false;
   return;
 }

double BPHKinematicFit::constrMass ( ) const

retrieve the constraint

Definition at line 101 of file BPHKinematicFit.cc.

References massConst.

                                          {
   return massConst;
 }

double BPHKinematicFit::constrSigma ( ) const

Definition at line 106 of file BPHKinematicFit.cc.

References massSigma.

                                           {
   return massSigma;
 }

const RefCountedKinematicVertex BPHKinematicFit::currentDecayVertex ( ) const

virtual

Definition at line 325 of file BPHKinematicFit.cc.

References isEmpty(), and kinTree.

                                                                           {
   if ( isEmpty() ) return RefCountedKinematicVertex( nullptr );
   return kinTree->currentDecayVertex();
 }

const RefCountedKinematicParticle BPHKinematicFit::currentParticle ( ) const

virtual

Definition at line 319 of file BPHKinematicFit.cc.

References isEmpty(), and kinTree.

Referenced by fitMomentum(), isValidFit(), and mass().

                                                                          {
   if ( isEmpty() ) return RefCountedKinematicParticle( nullptr );
   return kinTree->currentParticle();
 }

void BPHKinematicFit::fitMomentum ( ) const

privatevirtual

Definition at line 404 of file BPHKinematicFit.cc.

References AlCaHLTBitMon_QueryRunRegistry::comp, currentParticle(), BPHDecayMomentum::daughComp(), BPHDecayMomentum::daughters(), MillePedeFileConverter_cfg::e, KinematicState::globalMomentum(), isValidFit(), funct::m, KinematicState::mass(), gen::n, oldMom, mathSSE::sqrt(), totalMomentum, x, PV3DBase< T, PVType, FrameType >::x(), y, PV3DBase< T, PVType, FrameType >::y(), z, and PV3DBase< T, PVType, FrameType >::z().

Referenced by p4().

                                         {
   if ( isValidFit() ) {
     const KinematicState& ks = currentParticle()->currentState();
     GlobalVector tm = ks.globalMomentum();
     double x = tm.x();
     double y = tm.y();
     double z = tm.z();
     double m = ks.mass();
     double e = sqrt( ( x * x ) + ( y * y ) + ( z * z ) + ( m * m ) );
     totalMomentum.SetPxPyPzE( x, y, z, e );
   }
   else {
     edm::LogPrint( "FitNotFound" )
                 << "BPHKinematicFit::fitMomentum: "
                 << "simple momentum sum computed";
     math::XYZTLorentzVector tm;
     const vector<const reco::Candidate*>& daug = daughters();
     int n = daug.size();
     while ( n-- ) tm += daug[n]->p4();
     const vector<BPHRecoConstCandPtr>& comp = daughComp();
     int m = comp.size();
     while ( m-- ) tm += comp[m]->p4();
     totalMomentum = tm;
   }
   oldMom = false;
   return;
 }

bool BPHKinematicFit::isEmpty ( void ) const

virtual

Definition at line 305 of file BPHKinematicFit.cc.

References kinematicTree(), kinTree, and nullptr.

Referenced by plotting.Plot::clone(), currentDecayVertex(), and currentParticle().

                                     {
   kinematicTree();
   if ( kinTree.get() == nullptr ) return true;
   return kinTree->isEmpty();
 }

bool BPHKinematicFit::isValidFit ( ) const

virtual

Definition at line 312 of file BPHKinematicFit.cc.

References currentParticle(), and nullptr.

Referenced by fitMomentum().

                                        {
   const RefCountedKinematicParticle kPart = currentParticle();
   if ( kPart.get() == nullptr ) return false;
   return kPart->currentState().isValid();
 }

const RefCountedKinematicTree & BPHKinematicFit::kinematicTree ( ) const

virtual

perform the kinematic fit and get the result

Definition at line 162 of file BPHKinematicFit.cc.

References allParticles, AlCaHLTBitMon_QueryRunRegistry::comp, BPHDecayMomentum::daughFull(), BPHDecayMomentum::daugNames(), cppFunctionSkipper::exception, KinematicConstrainedVertexFitter::fit(), KinematicParticleFitter::fit(), KinematicParticleVertexFitter::fit(), BPHDecayMomentum::getComp(), kinParticles(), kinTree, mass(), massConst, massSigma, CaloTowersParam_cfi::mc, dataset::name, names, groupFilesInBlocks::nn, and oldFit.

Referenced by isEmpty(), and BPHMassFitSelect::~BPHMassFitSelect().

                                                                     {
   if ( oldFit ) return kinematicTree( "", massConst, massSigma );
   return kinTree;
 }

virtual const RefCountedKinematicTree& BPHKinematicFit::kinematicTree	(	const std::string &	name,
		double	mass,
		double	sigma
	)		const

virtual

virtual const RefCountedKinematicTree& BPHKinematicFit::kinematicTree	(	const std::string &	name,
		double	mass
	)		const

virtual

virtual const RefCountedKinematicTree& BPHKinematicFit::kinematicTree	(	const std::string &	name,
		KinematicConstraint *	kc
	)		const

virtual

virtual const RefCountedKinematicTree& BPHKinematicFit::kinematicTree	(	const std::string &	name,
		MultiTrackKinematicConstraint *	kc
	)		const

virtual

const vector< RefCountedKinematicParticle > & BPHKinematicFit::kinParticles ( ) const

virtual

get kinematic particles

Definition at line 111 of file BPHKinematicFit.cc.

References allParticles, buildParticles(), BPHDecayMomentum::daughFull(), BPHDecayMomentum::getDaug(), mps_fire::i, kinMap, kp, funct::m, genParticles_cff::map, gen::n, names, oldKPs, BPhysicsValidation_cfi::pname, and muonDTDigis_cfi::pset.

Referenced by kinematicTree().

                                                                   {
   if ( oldKPs ) buildParticles();
   return allParticles;
 }

virtual std::vector<RefCountedKinematicParticle> BPHKinematicFit::kinParticles ( const std::vector< std::string > & names ) const

virtual

ParticleMass BPHKinematicFit::mass ( ) const

virtual

Definition at line 331 of file BPHKinematicFit.cc.

References currentParticle(), KinematicState::isValid(), KinematicState::mass(), and nullptr.

Referenced by Particle.Particle::__str__(), DiObject.DiMuon::__str__(), BPHPlusMinusCandidate::build(), BPHRecoCandidate::build(), buildParticles(), kinematicTree(), p4(), setConstraint(), and BPHPlusMinusCandidate::~BPHPlusMinusCandidate().

                                          {
   const RefCountedKinematicParticle kPart = currentParticle();
   if ( kPart.get() == nullptr ) return -1.0;
   const KinematicState kStat = kPart->currentState();
   if ( kStat.isValid() ) return kStat.mass();
   return -1.0;
 }

const math::XYZTLorentzVector & BPHKinematicFit::p4 ( ) const

virtual

compute total momentum after the fit

Definition at line 340 of file BPHKinematicFit.cc.

References addK(), BPHDecayVertex::addV(), AlCaHLTBitMon_QueryRunRegistry::comp, BPHDecayMomentum::daughters(), BPHDecayMomentum::dMap, dMSig, fitMomentum(), mass(), dataset::name, oldMom, and totalMomentum.

Referenced by Tau.Tau::dxy_approx(), Tau.Tau::dz(), Lepton.Lepton::p4WithFSR(), and BPHMassFitSelect::~BPHMassFitSelect().

                                                        {
   if ( oldMom ) fitMomentum();
   return totalMomentum;
 }

void BPHKinematicFit::resetKinematicFit ( ) const

virtual

reset the kinematic fit

Definition at line 299 of file BPHKinematicFit.cc.

References oldFit, oldKPs, and oldMom.

Referenced by setNotUpdated().

                                               {
   oldKPs = oldFit = oldMom = true;
   return;
 }

void BPHKinematicFit::setConstraint	(	double	mass,
		double	sigma
	)

apply a mass constraint

Operations

Definition at line 93 of file BPHKinematicFit.cc.

References mass(), massConst, massSigma, oldFit, and oldMom.

Referenced by BPHOniaToMuMuBuilder::extractList().

                                                                {
   oldFit = oldMom = true;
   massConst = mass;
   massSigma = sigma;
   return;
 }

void BPHKinematicFit::setNotUpdated ( ) const

overrideprotectedvirtual

Reimplemented from BPHDecayVertex.

Reimplemented in BPHPlusMinusCandidate.

Definition at line 373 of file BPHKinematicFit.cc.

References resetKinematicFit(), and BPHDecayVertex::setNotUpdated().

Referenced by BPHPlusMinusCandidate::setNotUpdated().

                                           {
   BPHDecayVertex::setNotUpdated();
   resetKinematicFit();
   return;
 }

Member Data Documentation

std::vector<RefCountedKinematicParticle> BPHKinematicFit::allParticles

mutableprivate

Definition at line 126 of file BPHKinematicFit.h.

Referenced by buildParticles(), kinematicTree(), and kinParticles().

std::map<const reco::Candidate*,double> BPHKinematicFit::dMSig

private

Definition at line 118 of file BPHKinematicFit.h.

Referenced by BPHKinematicFit(), buildParticles(), and p4().

std::map<const reco::Candidate*, RefCountedKinematicParticle> BPHKinematicFit::kinMap

mutableprivate

Definition at line 125 of file BPHKinematicFit.h.

Referenced by buildParticles(), and kinParticles().

RefCountedKinematicTree BPHKinematicFit::kinTree

mutableprivate

Definition at line 127 of file BPHKinematicFit.h.

Referenced by currentDecayVertex(), currentParticle(), isEmpty(), and kinematicTree().

double BPHKinematicFit::massConst

private

Definition at line 114 of file BPHKinematicFit.h.

Referenced by constrMass(), kinematicTree(), and setConstraint().

double BPHKinematicFit::massSigma

private

Definition at line 115 of file BPHKinematicFit.h.

Referenced by constrSigma(), kinematicTree(), and setConstraint().

bool BPHKinematicFit::oldFit

mutableprivate

Definition at line 122 of file BPHKinematicFit.h.

Referenced by kinematicTree(), resetKinematicFit(), and setConstraint().

bool BPHKinematicFit::oldKPs

mutableprivate

Definition at line 121 of file BPHKinematicFit.h.

Referenced by buildParticles(), kinParticles(), and resetKinematicFit().

bool BPHKinematicFit::oldMom

mutableprivate

Definition at line 123 of file BPHKinematicFit.h.

Referenced by fitMomentum(), p4(), resetKinematicFit(), and setConstraint().

math::XYZTLorentzVector BPHKinematicFit::totalMomentum

mutableprivate

Definition at line 128 of file BPHKinematicFit.h.

Referenced by fitMomentum(), and p4().

Public Member Functions

Protected Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation