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