PFTrackTransformer.h

Go to the documentation of this file.

#ifndef PFTrackTransformer_H
#define PFTrackTransformer_H

#include "RecoParticleFlow/PFTracking/interface/PFGeometry.h"
#include "DataFormats/GeometrySurface/interface/Surface.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrack.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "TrackingTools/GsfTools/interface/MultiTrajectoryStateTransform.h"
#include "TrackingTools/GsfTools/interface/MultiTrajectoryStateMode.h"


class Trajectory;
class PFTrackTransformer {
public:
  PFTrackTransformer(const math::XYZVector&);
  ~PFTrackTransformer();

  bool addPoints(reco::PFRecTrack& pftrack,
                 const reco::Track& track,
                 const Trajectory& traj,
                 bool msgwarning = true) const;

  bool addPointsAndBrems(reco::GsfPFRecTrack& pftrack,
                         const reco::Track& track,
                         const Trajectory& traj,
                         const bool& GetMode) const;

  bool addPointsAndBrems(reco::GsfPFRecTrack& pftrack,
                         const reco::GsfTrack& track,
                         const MultiTrajectoryStateTransform& mtjstate) const;

  void OnlyProp() { onlyprop_ = true; }
  bool onlyprop_;

private:
  math::XYZVector B_;
  PFGeometry pfGeometry_;
  static constexpr double cos2ThetaV_Endcap_HiEnd_ = 0.99014;
};

#endif