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/PFLayer.h"

#include <iostream>
#include <vector>



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), color_(1) {}

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

    void reset();
    
    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_;}
    
    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(); }    

  private:
    
    std::vector< reco::PFRecHitFraction >  rechits_;
    
    REPPoint            posrep_;
    
    
    static int    depthCorMode_;
    
    static double depthCorA_;
    
    static double depthCorB_ ;
    
    static double depthCorAp_;
    
    static double depthCorBp_;
    
    static const math::XYZPoint dummyVtx_;

    int                 color_;
    
    friend class ::PFClusterAlgo;
  };
}

#endif