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