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