---

PFCluster.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_PFCluster_h
#define DataFormats_ParticleFlowReco_PFCluster_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 ROOT::Math::PositionVector3D<ROOT::Math::CylindricalEta3D<Double32_t> > REPPoint;
  
    PFCluster() :  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 unsigned     instanceCounter_;

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

  private:
  
    std::vector< reco::PFRecHitFraction >  rechits_;

    REPPoint            posrep_;


    static int    depthCorMode_;
    static double depthCorA_;
    static double depthCorB_ ;
    static double depthCorAp_;
    static double depthCorBp_;


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

#endif

---