BPHDecayVertex.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_RecoDecay_BPHDecayVertex_h
#define HeavyFlavorAnalysis_RecoDecay_BPHDecayVertex_h

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

namespace edm {
  class EventSetup;
}

namespace reco {
  class TransientTrack;
  class Vertex;
}

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "DataFormats/VertexReco/interface/Vertex.h"

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

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

class BPHDecayVertex: public virtual BPHDecayMomentum {

 public:

  virtual ~BPHDecayVertex();

  virtual bool validTracks() const;
  virtual bool validVertex() const;

  virtual const reco::Vertex& vertex() const;

  const std::vector<const reco::Track*>& tracks() const;

  const reco::Track* getTrack( const reco::Candidate* cand ) const;

  const std::vector<reco::TransientTrack>& transientTracks() const;

  reco::TransientTrack* getTransientTrack( const reco::Candidate* cand ) const;

  const std::string& getTrackSearchList( const reco::Candidate* cand ) const;

 protected:

  // constructor
  BPHDecayVertex( const edm::EventSetup* es );
  // pointer used to retrieve informations from other bases
  BPHDecayVertex( const BPHDecayVertex* ptr,
                  const edm::EventSetup* es );

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

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

 private:

  // EventSetup needed to build TransientTrack
  const edm::EventSetup* evSetup;

  // map linking particles to associated track search list
  std::map<const reco::Candidate*,std::string> searchMap;

  // reconstruction results cache
  mutable bool oldTracks;
  mutable bool oldVertex;
  mutable bool validTks;
  mutable std::vector<const    reco::Track*> rTracks;
  mutable std::vector<reco::TransientTrack> trTracks;
  mutable std::map<const reco::Candidate*,const    reco::Track*> tkMap;
  mutable std::map<const reco::Candidate*,reco::TransientTrack*> ttMap;
  mutable reco::Vertex fittedVertex;

  // create TransientTrack and fit vertex
  virtual void tTracks() const;
  virtual void fitVertex() const;

};


#endif