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