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

 
 virtual ~KinematicTree();
 
 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