BPHKinematicFit.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_RecoDecay_BPHKinematicFit_h
#define HeavyFlavorAnalysis_RecoDecay_BPHKinematicFit_h

//----------------------
// Base Class Headers --
//----------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHDecayVertex.h"

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicParticle.h"
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"

class MultiTrackKinematicConstraint;
class KinematicConstraint;

//---------------
// C++ Headers --
//---------------
#include <map>

//              ---------------------
//              -- Class Interface --
//              ---------------------

class BPHKinematicFit: public virtual BPHDecayVertex {

 public:

  ~BPHKinematicFit() override;

  void setConstraint( double mass, double sigma );
  double constrMass() const;
  double constrSigma() const;

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

  virtual const RefCountedKinematicTree& kinematicTree() const;
  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 void resetKinematicFit() const;

  // get current particle
  virtual bool isEmpty() const;
  virtual bool isValidFit() const;
  virtual const RefCountedKinematicParticle currentParticle   () const;
  virtual const RefCountedKinematicVertex   currentDecayVertex() const;
  virtual ParticleMass                      mass              () const;

  virtual const math::XYZTLorentzVector& p4() const;

 protected:

  // constructors
  BPHKinematicFit();
  // pointer used to retrieve informations from other bases
  BPHKinematicFit( const BPHKinematicFit* ptr );

  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 );

  // utility function used to cash reconstruction results
  void setNotUpdated() const override;

 private:

  // mass constraint
  double massConst;
  double massSigma;

  // map linking daughters to mass sigma
  std::map<const reco::Candidate*,double> dMSig;

  // reconstruction results cache
  mutable bool oldKPs;
  mutable bool oldFit;
  mutable bool oldMom;
  mutable std::map   <const reco::Candidate*,
                      RefCountedKinematicParticle> kinMap;
  mutable std::vector<RefCountedKinematicParticle> allParticles;
  mutable RefCountedKinematicTree kinTree;
  mutable math::XYZTLorentzVector totalMomentum;

  // build kin particles, perform the fit and compute the total momentum
  virtual void buildParticles() const;
  virtual void fitMomentum() const;

};


#endif