FinalTreeBuilder.h

Go to the documentation of this file.

#ifndef FinalTreeBuilder_H
#define FinalTreeBuilder_H

#include "RecoVertex/KinematicFitPrimitives/interface/RefCountedKinematicTree.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicVertexFactory.h"
#include "RecoVertex/KinematicFitPrimitives/interface/VirtualKinematicParticleFactory.h"
#include "RecoVertex/VertexPrimitives/interface/CachingVertex.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicRefittedTrackState.h"
#include "RecoVertex/KinematicFitPrimitives/interface/Matrices.h"

class FinalTreeBuilder {
public:
  FinalTreeBuilder();

  ~FinalTreeBuilder();

  RefCountedKinematicTree buildTree(const CachingVertex<6>& vtx,
                                    const std::vector<RefCountedKinematicParticle>& input) const;

private:
  typedef ReferenceCountingPointer<VertexTrack<6> > RefCountedVertexTrack;
  typedef ReferenceCountingPointer<LinearizedTrackState<6> > RefCountedLinearizedTrackState;
  typedef ReferenceCountingPointer<RefittedTrackState<6> > RefCountedRefittedTrackState;

  //internal calculation and helper methods
  AlgebraicMatrix momentumPart(const CachingVertex<6>& vtx, const AlgebraicVector7& par) const;

  KinematicVertexFactory* kvFactory;
  VirtualKinematicParticleFactory* pFactory;
};

#endif