SuperCluster.h

Go to the documentation of this file.

#ifndef EgammaReco_SuperCluster_h
#define EgammaReco_SuperCluster_h

#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/CaloRecHit/interface/CaloCluster.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "DataFormats/DetId/interface/DetId.h"


namespace reco {
  class SuperCluster : public CaloCluster {
  public:
    typedef math::XYZPoint Point;

    SuperCluster() : CaloCluster(0., Point(0.,0.,0.)), preshowerEnergy_(0), rawEnergy_(-1.), phiWidth_(0), etaWidth_(0), preshowerEnergy1_(0), preshowerEnergy2_(0) {}

    SuperCluster( double energy, const Point& position );

    SuperCluster( double energy, const Point& position,
                  const CaloClusterPtr & seed,
                  const CaloClusterPtrVector& clusters,
          double Epreshower=0.,
          double phiWidth=0., double etaWidth=0.,
          double Epreshower1=0., double Epreshower2=0.);

    // to be merged in the previous one? -- FIXME
    SuperCluster( double energy, const Point& position,
                  const CaloClusterPtr & seed,
                  const CaloClusterPtrVector& clusters,
                  const CaloClusterPtrVector& preshowerClusters,
          double Epreshower=0.,
          double phiWidth=0., double etaWidth=0.,
          double Epreshower1=0., double Epreshower2=0.);


    double rawEnergy() const { return rawEnergy_; }

    double preshowerEnergy() const { return preshowerEnergy_; }
    double preshowerEnergyPlane1() const {return preshowerEnergy1_; }
    double preshowerEnergyPlane2() const {return preshowerEnergy2_; }

    double phiWidth() const { return phiWidth_; }
    double etaWidth() const { return etaWidth_; }

    //Assign new variables to supercluster
    void setPreshowerEnergy( double preshowerEnergy ) { preshowerEnergy_ = preshowerEnergy; };
    void setPreshowerEnergyPlane1( double preshowerEnergy1 ) { preshowerEnergy1_ = preshowerEnergy1; }; 
    void setPreshowerEnergyPlane2( double preshowerEnergy2 ) { preshowerEnergy2_ = preshowerEnergy2; }; 
    void setPhiWidth( double pw ) { phiWidth_ = pw; }
    void setEtaWidth( double ew ) { etaWidth_ = ew; }

    const CaloClusterPtr & seed() const { return seed_; }

    const CaloClusterPtrVector& clusters() const { return clusters_; }

    const CaloClusterPtrVector& preshowerClusters() const { return preshowerClusters_; }

    CaloCluster_iterator clustersBegin() const { return clusters_.begin(); }

    CaloCluster_iterator clustersEnd() const { return clusters_.end(); }

    CaloCluster_iterator preshowerClustersBegin() const { return preshowerClusters_.begin(); }

    CaloCluster_iterator preshowerClustersEnd() const { return preshowerClusters_.end(); }    

    size_t clustersSize() const { return clusters_.size(); }

    size_t preshowerClustersSize() const { return preshowerClusters_.size(); }

    //std::vector<DetId> getHitsByDetId() const { return usedHits_; }

    void setSeed( const CaloClusterPtr & r ) { seed_ = r; }

    //(re)-set clusters
    void setClusters(const CaloClusterPtrVector &clusters) { 
      clusters_ = clusters;
      computeRawEnergy();
    }
    
    //(re)-set preshower clusters
    void setPreshowerClusters(const CaloClusterPtrVector &clusters) { preshowerClusters_ = clusters; }
    
    //clear hits and fractions vector (for slimming)
    void clearHitsAndFractions() { hitsAndFractions_.clear(); }
    
    void addCluster( const CaloClusterPtr & r ) { 
      clusters_.push_back( r ); 
      computeRawEnergy();
    }    

    void addPreshowerCluster( const CaloClusterPtr & r ) { preshowerClusters_.push_back( r ); }    

    void setPreshowerPlanesStatus(const uint32_t& status){
      uint32_t flags = flags_ & flagsMask_;
      flags_= flags | (status << flagsOffset_);
    }

    const int  getPreshowerPlanesStatus() const {
      return (flags_>>flagsOffset_);
    }

  private:

    void computeRawEnergy();

    CaloClusterPtr seed_;

    CaloClusterPtrVector clusters_;

    CaloClusterPtrVector preshowerClusters_;

    //std::vector<DetId> usedHits_;

    double preshowerEnergy_;

    double rawEnergy_;
    
    double phiWidth_;
    double etaWidth_;

    double preshowerEnergy1_;
    double preshowerEnergy2_;
  };

}
#endif