ConstrainedTreeBuilder.h

Go to the documentation of this file.

#ifndef ConstrainedTreeBuilder_H
#define ConstrainedTreeBuilder_H

#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"
#include "RecoVertex/KinematicFitPrimitives/interface/VirtualKinematicParticleFactory.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicVertexFactory.h"

class ConstrainedTreeBuilder {
public:
  ConstrainedTreeBuilder();

  ~ConstrainedTreeBuilder();

  RefCountedKinematicTree buildTree(const std::vector<RefCountedKinematicParticle>& initialParticles,
                                    const std::vector<KinematicState>& finalStates,
                                    const RefCountedKinematicVertex vtx,
                                    const AlgebraicMatrix& fCov) const;

private:
  RefCountedKinematicTree buildTree(const RefCountedKinematicParticle virtualParticle,
                                    const RefCountedKinematicVertex vtx,
                                    const std::vector<RefCountedKinematicParticle>& particles) const;

  AlgebraicMatrix covarianceMatrix(const std::vector<RefCountedKinematicParticle>& rPart,
                                   const AlgebraicVector7& newPar,
                                   const AlgebraicMatrix& fitCov) const;

  VirtualKinematicParticleFactory* pFactory;
  KinematicVertexFactory* vFactory;
};
#endif