Conversion.h

Go to the documentation of this file.

#ifndef EgammaCandidates_Conversion_h
#define EgammaCandidates_Conversion_h

#include <bitset>
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h" 
#include "DataFormats/GeometryVector/interface/GlobalVector.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1DFloat.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/CaloRecHit/interface/CaloCluster.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"


namespace reco {
    class Conversion  {
  public:

      enum ConversionAlgorithm {undefined=0, 
                ecalSeeded=1, 
                trackerOnly=2, 
                mixed=3, 
                                pflow=4,
                algoSize=5}; 

      enum ConversionQuality {generalTracksOnly=0, 
                  arbitratedEcalSeeded=1, 
                  arbitratedMerged=2,
                  arbitratedMergedEcalGeneral=3,
                  highPurity=8, 
                  highEfficiency=9,
                  ecalMatched1Track=10,
                  ecalMatched2Track=11};

      static const std::string algoNames[];      

      // Default constructor
      Conversion();
      
      Conversion( const reco::CaloClusterPtrVector clu, 
          const std::vector<edm::RefToBase<reco::Track> > tr,
          const std::vector<math::XYZPointF> trackPositionAtEcal , 
          const reco::Vertex  &  convVtx,
          const std::vector<reco::CaloClusterPtr> & matchingBC,
          const float DCA,        
          const std::vector<math::XYZPointF> & innPoint,
          const std::vector<math::XYZVectorF> & trackPin ,
          const std::vector<math::XYZVectorF> & trackPout,
                  const std::vector<uint8_t> nHitsBeforeVtx,
                  const std::vector<Measurement1DFloat> & dlClosestHitToVtx,
                  uint8_t nSharedHits,
                  const float mva,
          ConversionAlgorithm=undefined);


      Conversion( const reco::CaloClusterPtrVector clu, 
          const std::vector<reco::TrackRef> tr,
          const std::vector<math::XYZPointF> trackPositionAtEcal , 
          const reco::Vertex  &  convVtx,
          const std::vector<reco::CaloClusterPtr> & matchingBC,
          const float DCA,        
          const std::vector<math::XYZPointF> & innPoint,
          const std::vector<math::XYZVectorF> & trackPin ,
          const std::vector<math::XYZVectorF> & trackPout,
                  const float mva,
          ConversionAlgorithm=undefined);
      


      
      Conversion( const reco::CaloClusterPtrVector clu, 
          const std::vector<reco::TrackRef> tr,
          const reco::Vertex  &  convVtx,
          ConversionAlgorithm=undefined);
      
      Conversion( const reco::CaloClusterPtrVector clu, 
          const std::vector<edm::RefToBase<reco::Track> > tr,
          const reco::Vertex  &  convVtx,
          ConversionAlgorithm=undefined);
      
           
      
      virtual ~Conversion();
      Conversion * clone() const;
      reco::CaloClusterPtrVector caloCluster() const {return caloCluster_ ;}
      std::vector<edm::RefToBase<reco::Track> > tracks() const ; 
      const reco::Vertex & conversionVertex() const  { return theConversionVertex_ ; }
      bool isConverted() const;
      unsigned int nTracks() const {return  tracks().size(); }
      double MVAout() const { return theMVAout_;}
      std::vector<float> const oneLegMVA() {return theOneLegMVA_;}
      double pairInvariantMass() const;
      double pairCotThetaSeparation() const;
      math::XYZVectorF  pairMomentum() const;
      math::XYZTLorentzVectorF   refittedPair4Momentum() const;
      math::XYZVectorF  refittedPairMomentum() const;
      double EoverP() const;
      double EoverPrefittedTracks() const;
      // Dist of minimum approach between tracks
      double distOfMinimumApproach() const {return  theMinDistOfApproach_;}
      // deltaPhi tracks at innermost point
      double dPhiTracksAtVtx() const;
      // deltaPhi tracks at ECAl
      double dPhiTracksAtEcal() const;
      // deltaEta tracks at ECAl
      double dEtaTracksAtEcal() const;

      //impact parameter and decay length computed with respect to given beamspot or vertex
      //computed from refittedPairMomentum

      //transverse impact parameter
      double dxy(const math::XYZPoint& myBeamSpot = math::XYZPoint()) const;
      //longitudinal impact parameter
      double dz(const math::XYZPoint& myBeamSpot = math::XYZPoint()) const;
      //transverse decay length
      double lxy(const math::XYZPoint& myBeamSpot = math::XYZPoint()) const;
      //longitudinal decay length
      double lz(const math::XYZPoint& myBeamSpot = math::XYZPoint()) const;
      //z position of intersection with beamspot in rz plane (possible tilt of beamspot is neglected)
      double zOfPrimaryVertexFromTracks(const math::XYZPoint& myBeamSpot = math::XYZPoint()) const { return dz(myBeamSpot) + myBeamSpot.z(); }

      const std::vector<math::XYZPointF> & ecalImpactPosition() const  {return thePositionAtEcal_;}
      //  pair of BC matching a posteriori the tracks
      const std::vector<reco::CaloClusterPtr>&  bcMatchingWithTracks() const { return theMatchingBCs_;}
      std::vector<double> tracksSigned_d0() const ;
      const std::vector<math::XYZPointF>& tracksInnerPosition() const {return theTrackInnerPosition_;}
      const std::vector<math::XYZVectorF>& tracksPout() const {return theTrackPout_;}
      const std::vector<math::XYZVectorF>& tracksPin() const  {return theTrackPin_;}
      const std::vector<uint8_t> &nHitsBeforeVtx() const { return nHitsBeforeVtx_; }
      const std::vector<Measurement1DFloat> &dlClosestHitToVtx() const { return dlClosestHitToVtx_; }
      uint8_t nSharedHits() const { return nSharedHits_; }


      void setMVAout(const float& mva) { theMVAout_=mva;}
      void setOneLegMVA(const std::vector<float>& mva) { theOneLegMVA_=mva;}
      // Set the ptr to the Super cluster if not set in the constructor 
      void setMatchingSuperCluster ( const  reco::CaloClusterPtrVector& sc) { caloCluster_= sc;}
      void setConversionAlgorithm(const ConversionAlgorithm a, bool set=true) { if (set) algorithm_=a; else algorithm_=undefined;}
      ConversionAlgorithm algo() const ;
      std::string algoName() const;
      static std::string algoName(ConversionAlgorithm );
      static ConversionAlgorithm  algoByName(const std::string &name);      

      bool quality(ConversionQuality q) const { return  (qualityMask_ & (1<<q))>>q; }
      void setQuality(ConversionQuality q, bool b);


      
    private:
      
      reco::CaloClusterPtrVector caloCluster_;
      std::vector<reco::TrackRef>  tracks_;
      mutable std::vector<edm::RefToBase<reco::Track> >  trackToBaseRefs_;
      std::vector<math::XYZPointF>  thePositionAtEcal_;
      reco::Vertex theConversionVertex_;
      std::vector<reco::CaloClusterPtr> theMatchingBCs_;
      float theMinDistOfApproach_;
      std::vector<math::XYZPointF> theTrackInnerPosition_;    
      std::vector<math::XYZVectorF> theTrackPin_;    
      std::vector<math::XYZVectorF> theTrackPout_;    
      std::vector<uint8_t> nHitsBeforeVtx_;
      std::vector<Measurement1DFloat> dlClosestHitToVtx_;
      uint8_t nSharedHits_;
      float theMVAout_;
      std::vector<float>  theOneLegMVA_;
      uint8_t algorithm_;
      uint16_t qualityMask_;


  };

    inline Conversion::ConversionAlgorithm Conversion::algo() const {
      return (ConversionAlgorithm) algorithm_;
    }
    
    
    inline std::string Conversion::algoName() const{
            
      switch(algorithm_)
    {
    case undefined: return "undefined";
    case ecalSeeded: return "ecalSeeded";
    case trackerOnly: return "trackerOnly";
    case mixed: return "mixed";
    case pflow: return "pflow";

    }
      return "undefined";
    }

    inline std::string Conversion::algoName(ConversionAlgorithm a){
      if(int(a) < int(algoSize) && int(a)>0) return algoNames[int(a)];
      return "undefined";
    }

    inline void Conversion::setQuality(ConversionQuality q, bool b){
      if (b)//regular OR if setting value to true
        qualityMask_ |= (1<<q) ;
      else // doing "half-XOR" if unsetting value
        qualityMask_ &= (~(1<<q));

    }
  
}

#endif