SimpleVertexTree.h

Go to the documentation of this file.

#ifndef SimpleVertexTree_H
#define SimpleVertexTree_H
 
#include <string>
 
#include "RecoVertex/VertexPrimitives/interface/TransientVertex.h"
#include "RecoVertex/KalmanVertexFit/interface/VertexFitterResult.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertex.h"
#include "MagneticField/Engine/interface/MagneticField.h"
 
#include "TString.h"
 
class TFile;
class TTree;
 
class SimpleVertexTree {
public:
  SimpleVertexTree(const char* fitterName = "VertexFitter", const MagneticField* magField = nullptr);
  virtual ~SimpleVertexTree();
 
  void fill(const TransientVertex& recv,
            const TrackingVertex* simv = nullptr,
            reco::RecoToSimCollection* recSimColl = nullptr,
            const float& time = 0.);
 
  void fill(const TransientVertex& recv, const TrackingVertex* simv = nullptr, const float& time = 0.);
 
  void fill(const TransientVertex& recv, const float& time = 0.);
 
  void fill(const TrackingVertex* simv);
 
  //   void fill(const TransientVertex & recVertex, const std::vector < RecTrack > & recTrackV,
  //            const SimVertex * simv, const float &time);
  //
  //   void fill(const std::vector < RecTrack > & recTrackV, const TkSimVertex * simv = 0,
  //            const float &time = 0.);
 
  void fill();
 
private:
  void defineTrackBranch(const TString& prefix,
                         const TString& type,
                         const float* (VertexFitterResult::*pfunc)(const int)const,
                         const TString& index);
 
  float simPos[3];
  float recPos[3];
  float recErr[3];
  float chiTot, ndf, chiProb;
  int numberOfVertices;
  TTree* vertexTree;
  VertexFitterResult* result;
  TString theFitterName;
 
  bool trackTest;
  int maxTrack;
  TString* parameterNames[5];
};
#endif