ConversionProducer.h

Go to the documentation of this file.

#ifndef RecoEgamma_EgammaPhotonProducers_ConversionProducer_h
#define RecoEgamma_EgammaPhotonProducers_ConversionProducer_h

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

//ECAL clusters
#include "DataFormats/EgammaReco/interface/BasicCluster.h"
#include "DataFormats/EgammaReco/interface/BasicClusterFwd.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/CaloRecHit/interface/CaloCluster.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "DataFormats/GeometrySurface/interface/BoundCylinder.h"
#include "DataFormats/GeometrySurface/interface/BoundDisk.h"
#include "DataFormats/GeometrySurface/interface/SimpleCylinderBounds.h"
#include "DataFormats/GeometrySurface/interface/SimpleDiskBounds.h"

#include "DataFormats/GeometrySurface/interface/ReferenceCounted.h"
#include "RecoVertex/VertexPrimitives/interface/VertexState.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicVertex.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicParticle.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicTree.h"
#include "RecoVertex/KinematicFitPrimitives/interface/TransientTrackKinematicParticle.h"
#include "RecoVertex/KinematicFit/interface/KinematicParticleVertexFitter.h"

//Tracker tracks
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"

//photon data format
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/EgammaCandidates/interface/Conversion.h"
#include "DataFormats/EgammaCandidates/interface/ConversionFwd.h"
#include "DataFormats/EgammaTrackReco/interface/ConversionTrackFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

class TransientTrackBuilder;
class ConversionVertexFinder;

class ConversionProducer : public edm::stream::EDProducer<> {
public:
  explicit ConversionProducer(const edm::ParameterSet&);
  ~ConversionProducer() override;

private:
  void produce(edm::Event&, const edm::EventSetup&) override;

  void buildSuperAndBasicClusterGeoMap(const edm::Event&,
                                       std::multimap<double, reco::CaloClusterPtr>& basicClusterPtrs,
                                       std::multimap<double, reco::CaloClusterPtr>& superClusterPtrs);

  // ----------member data ---------------------------
  std::string algoName_;

  typedef math::XYZPointF Point;
  typedef std::vector<Point> PointCollection;

  edm::EDGetTokenT<edm::View<reco::ConversionTrack> > src_;

  edm::EDGetTokenT<edm::View<reco::CaloCluster> > scBarrelProducer_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > scEndcapProducer_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > bcBarrelCollection_;
  edm::EDGetTokenT<edm::View<reco::CaloCluster> > bcEndcapCollection_;
  std::string ConvertedPhotonCollection_;

  bool allowD0_, allowDeltaPhi_, allowTrackBC_, allowDeltaCot_, allowMinApproach_, allowOppCharge_, allowVertex_;

  bool bypassPreselGsf_, bypassPreselEcal_, bypassPreselEcalEcal_;

  bool usePvtx_;  //if use primary vertices
  edm::EDGetTokenT<reco::VertexCollection> vertexProducer_;
  ConversionVertexFinder* theVertexFinder_;

  const TransientTrackBuilder* thettbuilder_;

  double deltaEta_;

  double halfWayEta_, halfWayPhi_;  //halfway open angle to search in basic clusters
  unsigned int maxNumOfTrackInPU_;
  double maxTrackZ_;
  double maxTrackRho_;
  double minSCEt_;
  double dEtacutForSCmatching_;
  double dPhicutForSCmatching_;
  double energyBC_;             //1.5GeV for track BC selection
  double energyTotalBC_;        //5GeV for track pair BC selection
  double d0Cut_;                //0 for d0*charge cut
  double dzCut_;                //innerposition of z diff cut
  double dEtaTkBC_, dPhiTkBC_;  //0.06 0.6 for track and BC matching

  double maxChi2Left_, maxChi2Right_;  //5. 5. for track chi2 quality
  double minHitsLeft_, minHitsRight_;  //5 2 for track hits quality

  double deltaCotTheta_, deltaPhi_, minApproachLow_,
      minApproachHigh_;  //0.02 0.2 for track pair open angle and > -0.1 cm

  double r_cut;     //cross_r cut
  double vtxChi2_;  //vertex chi2 probablity cut

  bool allowSingleLeg_;  //if single track conversion ?
  bool rightBC_;         //if right leg requires matching BC?

  void buildCollection(edm::Event& iEvent,
                       const edm::EventSetup& iSetup,
                       const std::multimap<float, edm::Ptr<reco::ConversionTrack> >& allTracks,
                       const std::multimap<double, reco::CaloClusterPtr>& superClusterPtrs,
                       const std::multimap<double, reco::CaloClusterPtr>& basicClusterPtrs,
                       const reco::Vertex& the_pvtx,
                       reco::ConversionCollection& outputConvPhotonCollection);

  //track quality cut, returns pass or no
  inline bool trackQualityFilter(const edm::RefToBase<reco::Track>& ref, bool isLeft);
  inline bool trackD0Cut(const edm::RefToBase<reco::Track>& ref);
  inline bool trackD0Cut(const edm::RefToBase<reco::Track>& ref, const reco::Vertex& the_pvtx);

  //track impact point at ECAL wall, returns validity to access position ew
  bool getTrackImpactPosition(const reco::Track* tk_ref,
                              const TrackerGeometry* trackerGeom,
                              const MagneticField* magField,
                              math::XYZPointF& ew);

  //distance at min approaching point, returns distance
  //      double getMinApproach(const edm::RefToBase<reco::Track>& ll, const edm::RefToBase<reco::Track>& rr,
  //          const MagneticField* magField);

  bool preselectTrackPair(const reco::TransientTrack& ttk_l, const reco::TransientTrack& ttk_r, double& appDist);

  //cut-based selection, TODO remove global cut variables
  bool checkTrackPair(const std::pair<edm::RefToBase<reco::Track>, reco::CaloClusterPtr>& ll,
                      const std::pair<edm::RefToBase<reco::Track>, reco::CaloClusterPtr>& rr);

  //kinematic vertex fitting, return true for valid vertex
  bool checkVertex(const reco::TransientTrack& ttk_l,
                   const reco::TransientTrack& ttk_r,
                   const MagneticField* magField,
                   reco::Vertex& the_vertex);
  bool checkPhi(const edm::RefToBase<reco::Track>& tk_l,
                const edm::RefToBase<reco::Track>& tk_r,
                const TrackerGeometry* trackerGeom,
                const MagneticField* magField,
                const reco::Vertex& the_vertex);

  //check the closest BC, returns true for found a BC
  bool getMatchedBC(const std::multimap<double, reco::CaloClusterPtr>& bcMap,
                    const math::XYZPointF& trackImpactPosition,
                    reco::CaloClusterPtr& closestBC);

  // finds the super cluster matching with at least one track in the pair
  bool matchingSC(const std::multimap<double, reco::CaloClusterPtr>& scMap,
                  reco::Conversion& conv,
                  reco::CaloClusterPtrVector& mSC);

  double etaTransformation(float EtaParticle, float Zvertex);

  math::XYZPointF toFConverterP(const math::XYZPoint& val) { return math::XYZPointF(val.x(), val.y(), val.z()); }

  math::XYZVectorF toFConverterV(const math::XYZVector& val) { return math::XYZVectorF(val.x(), val.y(), val.z()); }
};

inline const GeomDet* recHitDet(const TrackingRecHit& hit, const TrackingGeometry* geom) {
  return geom->idToDet(hit.geographicalId());
}

inline const BoundPlane& recHitSurface(const TrackingRecHit& hit, const TrackingGeometry* geom) {
  return recHitDet(hit, geom)->surface();
}

inline LocalVector toLocal(const reco::Track::Vector& v, const Surface& s) {
  return s.toLocal(GlobalVector(v.x(), v.y(), v.z()));
}

#endif