KinematicTree.h

Go to the documentation of this file.

#ifndef KinematicTree_H
#define KinematicTree_H
 
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicParticle.h"
#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicVertex.h"
#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "DataFormats/Math/interface/Graph.h"
#include "DataFormats/Math/interface/GraphWalker.h"
 
class KinematicTree : public ReferenceCounted {
public:
  KinematicTree();
 
  ~KinematicTree() override;
 
  bool isEmpty() const;
  bool isValid() const { return !empt; }
 
  bool isConsistent() const;
 
  void addParticle(RefCountedKinematicVertex prodVtx,
                   RefCountedKinematicVertex decVtx,
                   RefCountedKinematicParticle part);
 
  std::vector<RefCountedKinematicParticle> finalStateParticles() const;
 
  RefCountedKinematicParticle topParticle() const;
 
  RefCountedKinematicVertex currentDecayVertex() const;
 
  RefCountedKinematicVertex currentProductionVertex() const;
 
  RefCountedKinematicParticle currentParticle() const;
 
  std::pair<bool, RefCountedKinematicParticle> motherParticle() const;
 
  std::vector<RefCountedKinematicParticle> daughterParticles() const;
 
  void movePointerToTheTop() const;
 
  bool movePointerToTheMother() const;
 
  bool movePointerToTheFirstChild() const;
 
  bool movePointerToTheNextChild() const;
 
  bool findParticle(const RefCountedKinematicParticle part) const;
 
  bool findDecayVertex(const RefCountedKinematicVertex vert) const;
  bool findDecayVertex(KinematicVertex* vert) const;
 
  void replaceCurrentParticle(RefCountedKinematicParticle newPart) const;
 
  void replaceCurrentVertex(RefCountedKinematicVertex newVert) const;
 
  void addTree(RefCountedKinematicVertex vtx, KinematicTree* tr);
 
private:
  bool leftBranchSearch(RefCountedKinematicParticle part) const;
 
  bool leftBranchVertexSearch(RefCountedKinematicVertex vtx) const;
 
  bool leftFinalParticle() const;
 
  void leftBranchAdd(KinematicTree* otherTree, RefCountedKinematicVertex vtx);
 
  mutable bool empt;
 
  mutable math::Graph<RefCountedKinematicVertex, RefCountedKinematicParticle> treeGraph;
  mutable math::GraphWalker<RefCountedKinematicVertex, RefCountedKinematicParticle>* treeWalker;
};
#endif