PFCandidatePhotonExtra.h

Go to the documentation of this file.

#ifndef ParticleFlowCandidate_PFCandidatePhotonExtra_h
#define ParticleFlowCandidate_PFCandidatePhotonExtra_h

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/EgammaCandidates/interface/ConversionFwd.h"

#include <iosfwd>

namespace reco {
  class PFCandidatePhotonExtra {
  public:
    PFCandidatePhotonExtra();
    PFCandidatePhotonExtra(const reco::SuperClusterRef&);
    ~PFCandidatePhotonExtra() { ; }

    // variables for the single conversion identification

    reco::SuperClusterRef superClusterRef() const { return scRef_; }

    void addSingleLegConvTrackRef(const reco::TrackRef& trackref);

    const std::vector<reco::TrackRef>& singleLegConvTrackRef() const { return assoSingleLegRefTrack_; }

    void addSingleLegConvMva(float& mvasingleleg);

    const std::vector<float>& singleLegConvMva() const { return assoSingleLegMva_; }

    void addConversionRef(const reco::ConversionRef& convref);

    reco::ConversionRefVector conversionRef() const { return assoConversionsRef_; }

    //from Mustache Id:
    void setMustache_Et(float Must_Et) { Mustache_Et_ = Must_Et; }
    void setExcludedClust(int excluded) { Excluded_clust_ = excluded; }
    float Mustache_Et() const { return Mustache_Et_; }
    int ExcludedClust() const { return Excluded_clust_; }

    //MVA Energy Regression:
    void setMVAGlobalCorrE(float GCorr) { GlobalCorr_ = GCorr; }
    float MVAGlobalCorrE() const { return GlobalCorr_; }

    void setMVAGlobalCorrEError(float GCorr) { GlobalCorrEError_ = GCorr; }
    float MVAGlobalCorrEError() const { return GlobalCorrEError_; }

    void addLCorrClusEnergy(float LCorrE);
    const std::vector<float>& LCorrClusEnergy() const { return LocalCorr_; }

    void SetPFPhotonRes(float Res) { MVAResolution_ = Res; }
    float PFPhotonRes() const { return MVAResolution_; }

  private:
    reco::SuperClusterRef scRef_;

    std::vector<reco::TrackRef> assoSingleLegRefTrack_;

    std::vector<float> assoSingleLegMva_;

    reco::ConversionRefVector assoConversionsRef_;

    //for Mustache_Id
    float Mustache_Et_;
    int Excluded_clust_;

    //for MVA Regression Energy
    std::vector<float> LocalCorr_;
    float GlobalCorr_;
    float GlobalCorrEError_;
    float MVAResolution_;
  };
}  // namespace reco
#endif