VertexProcessor.h

#ifndef TRACKING_RECO_VERTEXPROCESSOR_H_
#define TRACKING_RECO_VERTEXPROCESSOR_H_
 
//--- Framework ---//
#include "Framework/Configure/Parameters.h"
#include "Framework/EventProcessor.h"
 
// --- Tracking --- //
#include "Tracking/Event/Track.h"
#include "Tracking/Sim/BFieldXYZUtils.h"
#include "Tracking/Sim/TrackingUtils.h"
 
// --- ACTS --- //
 
// Propagator
 
#include "Acts/Propagator/Propagator.hpp"
#include "Tracking/EigenStepper.h"
// #include "Acts/Propagator/Navigator.hpp"
// #include "Acts/Propagator/StandardAborters.hpp"
 
// Vertexing
 
#include "Acts/Vertexing/FullBilloirVertexFitter.hpp"
#include "Acts/Vertexing/HelicalTrackLinearizer.hpp"
#include "Acts/Vertexing/Vertex.hpp"
 
// Magfield
 
#include "Acts/MagneticField/MagneticFieldContext.hpp"
#include "Acts/MagneticField/MagneticFieldProvider.hpp"
 
// Geometry
#include "Acts/Surfaces/PerigeeSurface.hpp"
 
// Root
#include "TFile.h"
#include "TH1F.h"
#include "TLorentzVector.h"
 
// Propagator with void navigator
using VoidPropagator = Acts::Propagator<Acts::EigenStepper<>>;
 
namespace tracking {
namespace reco {
 
class VertexProcessor : public framework::Producer {
 public:
  VertexProcessor(const std::string& name, framework::Process& process);
 
  virtual ~VertexProcessor() = default;
 
  void onProcessStart() override;
  void onProcessEnd() override;
 
  void configure(framework::config::Parameters& parameters) override;
 
  void produce(framework::Event& event) override;
 
 private:
  Acts::GeometryContext gctx_;
  Acts::MagneticFieldContext bctx_;
 
  // Event counter
  int nevents_{0};
 
  // The interpolated bfield
  std::shared_ptr<InterpolatedMagneticField3> sp_interpolated_b_field_;
 
  std::string field_map_{""};
 
  // Track collection name
 
  std::string trk_coll_name_{"Tracks"};
 
  std::string seeds_coll_name_{"RecoilTruthSeeds"};
 
  std::string input_pass_name_{""};
 
  // The propagator
  std::shared_ptr<VoidPropagator> propagator_;
 
  // Processing time counter
  double processing_time_{0.};
 
  TH1F* h_m_;
  TH1F* h_m_truth_filter_;
  TH1F* h_m_truth_;
};
 
}  // namespace reco
}  // namespace tracking
 
#endif  // TRACKING_RECO_VERTEXPROCESSOR_H_