KVFTest.h

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    KVFTest
// Class:      KVFTest
//
// system include files
#include <memory>
 
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimDataFormats/Vertex/interface/SimVertex.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"
#include "SimDataFormats/Associations/interface/TrackToTrackingParticleAssociator.h"
#include "RecoVertex/KalmanVertexFit/interface/SimpleVertexTree.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include <TFile.h>
 
class KVFTest : public edm::EDAnalyzer {
public:
  explicit KVFTest(const edm::ParameterSet&);
  ~KVFTest() override;
 
  void analyze(const edm::Event&, const edm::EventSetup&) override;
 
  void beginJob() override;
  void endJob() override;
 
private:
  TrackingVertex getSimVertex(const edm::Event& iEvent) const;
 
  edm::ParameterSet theConfig;
  edm::ParameterSet kvfPSet;
  std::unique_ptr<SimpleVertexTree> tree;
  TFile* rootFile_;
 
  std::string outputFile_;  // output file
  edm::EDGetTokenT<reco::TrackCollection> token_tracks;
  edm::EDGetTokenT<TrackingParticleCollection> token_TrackTruth;
  edm::EDGetTokenT<TrackingVertexCollection> token_VertexTruth;
  edm::EDGetTokenT<reco::TrackToTrackingParticleAssociator> token_associatorForParamAtPca;
};