BPHDecayMomentum.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_RecoDecay_BPHDecayMomentum_h
#define HeavyFlavorAnalysis_RecoDecay_BPHDecayMomentum_h

//----------------------
// Base Class Headers --
//----------------------


//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoCandidatePtr.h"
#include "DataFormats/PatCandidates/interface/CompositeCandidate.h"
class BPHRecoBuilder;

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

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

class BPHDecayMomentum {

  friend class BPHRecoBuilder;

 public:

  virtual ~BPHDecayMomentum();

  virtual const pat::CompositeCandidate& composite() const;

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

  virtual const std::vector<std::string>& compNames() const;

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

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

  virtual const reco::Candidate* originalReco(
          const reco::Candidate* daug ) const;

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

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

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

 protected:

  struct Component {
    const reco::Candidate* cand;
    double mass;
    double msig;
    std::string searchList;
  };

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

  // get an object filled in the constructor
  // to be used in the creation of other bases of BPHRecoCandidate
  const std::vector<Component>& componentList() const;

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

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

 private:

  // object filled in the constructor
  // to be used in the creation of other bases of BPHRecoCandidate
  std::vector<Component> compList;

  // names used for simple and previously reconstructed particles
  std::vector<std::string> nList;
  std::vector<std::string> nComp;

  // pointers to simple and previously reconstructed particles
  // (clones stored for simple particles)
  std::vector<const reco::Candidate*> dList;
  std::vector<BPHRecoConstCandPtr   > cList;

  // maps linking names to decay products
  // (simple and previously reconstructed particles)
  std::map<std::string,const reco::Candidate*> dMap;
  std::map<std::string,BPHRecoConstCandPtr   > cMap;

  // map linking cloned particles to original ones
  std::map<const reco::Candidate*, const reco::Candidate*> clonesMap;

  // reconstruction results cache
  mutable bool oldMom;
  mutable std::vector<const reco::Candidate*> dFull;
  mutable pat::CompositeCandidate compCand;

  // create clones of simple particles, store them and their names
  void clonesList( const std::map<std::string,Component>& daugMap );

  // fill lists of previously reconstructed particles and their names
  // and retrieve cascade decay products
  void dCompList();

  // compute the total momentum of simple particles, produced
  // directly or in cascade decays
  virtual void sumMomentum(
               const std::vector<const reco::Candidate*> dl ) const;

  // recursively fill the list of simple particles, produced
  // directly or in cascade decays
  virtual void fillDaug( std::vector<const reco::Candidate*>& ad ) const;

  // compute the total momentum and cache it
  virtual void computeMomentum() const;

};


#endif