PFCluster.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_PFCluster_h
#define DataFormats_ParticleFlowReco_PFCluster_h

#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "DataFormats/CaloRecHit/interface/CaloCluster.h"

#include "Math/GenVector/PositionVector3D.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "Rtypes.h" 

#include "DataFormats/ParticleFlowReco/interface/PFRecHitFraction.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHit.h"
#include "DataFormats/ParticleFlowReco/interface/PFLayer.h"

#include <iostream>
#include <vector>
#include <algorithm>
#if !defined(__CINT__) && !defined(__MAKECINT__) && !defined(__REFLEX__)
#include <atomic>
#endif


class PFClusterAlgo;

namespace reco {

  class PFCluster : public CaloCluster {
  public:

    typedef std::vector<std::pair<CaloClusterPtr::key_type,edm::Ptr<PFCluster> > > EEtoPSAssociation;
    typedef ROOT::Math::PositionVector3D<ROOT::Math::CylindricalEta3D<Double32_t> > REPPoint;
  
    PFCluster() : CaloCluster(CaloCluster::particleFlow), time_(-99.0), layer_(PFLayer::NONE), color_(1) {}

    PFCluster(PFLayer::Layer layer, double energy,
          double x, double y, double z );

    void reset();

    void resetHitsAndFractions();
    
    void addRecHitFraction( const reco::PFRecHitFraction& frac);
    
    const std::vector< reco::PFRecHitFraction >& recHitFractions() const 
      { return rechits_; }    
    
    void setLayer( PFLayer::Layer layer);
    
    PFLayer::Layer  layer() const;
    
    double        energy() const {return energy_;}

    double        time() const {return time_;}

    void         setTime(double time) {time_ = time;}
    
    const REPPoint&       positionREP() const {return posrep_;}
    
    void calculatePositionREP() {
      posrep_.SetCoordinates( position_.Rho(), 
                  position_.Eta(), 
                  position_.Phi() ); 
    }
    
    static double getDepthCorrection(double energy, bool isBelowPS = false,
                     bool isHadron = false);
    
    void         setColor(int color) {color_ = color;}
    
    int          color() const {return color_;}
    
    
    PFCluster& operator=(const PFCluster&);
    
    friend    std::ostream& operator<<(std::ostream& out, 
                       const PFCluster& cluster);

    static void setDepthCorParameters(int mode, 
                      double a, double b, 
                      double ap, double bp ) {
      depthCorMode_ = mode;
      depthCorA_ = a; 
      depthCorB_ = b; 
      depthCorAp_ = ap; 
      depthCorBp_ = bp; 
    } 
    

    
    double charge() const { return 0;}

    double pt() const { 
      return (energy() * sin(position_.theta()));
    }

    double theta() const { 
      return position_.theta();
    }
    
    math::XYZPoint const & vertex() const { 
      return dummyVtx_;      
    }
    double vx() const { return vertex().x(); }
    double vy() const { return vertex().y(); }
    double vz() const { return vertex().z(); }    

#if !defined(__CINT__) && !defined(__MAKECINT__) && !defined(__REFLEX__)
    template<typename pruner>
      void pruneUsing(pruner prune) {
      hitsAndFractions_.clear();
      std::vector<reco::PFRecHitFraction>::iterator iter = 
    std::stable_partition(rechits_.begin(),rechits_.end(),prune);
      rechits_.erase(iter,rechits_.end());
      hitsAndFractions_.reserve(rechits_.size());
      for( const auto& hitfrac : rechits_ ) {
    hitsAndFractions_.emplace_back(hitfrac.recHitRef()->detId(),
                       hitfrac.fraction());
      }
    }
#endif
    
  private:
    
    std::vector< reco::PFRecHitFraction >  rechits_;
    
    REPPoint            posrep_;

    double time_;

    PFLayer::Layer layer_; 

#if !defined(__CINT__) && !defined(__MAKECINT__) && !defined(__REFLEX__)
    static std::atomic<int>    depthCorMode_;

    static std::atomic<double> depthCorA_;

    static std::atomic<double> depthCorB_ ;

    static std::atomic<double> depthCorAp_;

    static std::atomic<double> depthCorBp_;
#else
    static int    depthCorMode_;
    
    static double depthCorA_;
    
    static double depthCorB_ ;
    
    static double depthCorAp_;
    
    static double depthCorBp_;
#endif
    
    static const math::XYZPoint dummyVtx_;

    int                 color_;
  };
}

#endif