PFCandidateElectronExtra.h

Go to the documentation of this file.

#ifndef ParticleFlowCandidate_PFCandidateElectronExtra_h
#define ParticleFlowCandidate_PFCandidateElectronExtra_h
 
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"
 
#include <iosfwd>
 
namespace reco {
  class PFCandidateElectronExtra {
  public:
    enum StatusFlag {
      X = 0,                  // undefined
      Selected,               // selected
      ECALDrivenPreselected,  // ECAL-driven electron pre-selected
      MVASelected,            // Passed the internal particle-flow selection (mva selection)
      Rejected                // Rejected
    };
 
    // if you had a variable update NMvaVariables
    enum MvaVariable {
      MVA_FIRST = 0,
      MVA_LnPtGsf = MVA_FIRST,
      MVA_EtaGsf,
      MVA_SigmaPtOverPt,
      MVA_Fbrem,
      MVA_Chi2Gsf,
      MVA_NhitsKf,
      MVA_Chi2Kf,
      MVA_EtotOverPin,
      MVA_EseedOverPout,
      MVA_EbremOverDeltaP,
      MVA_DeltaEtaTrackCluster,
      MVA_LogSigmaEtaEta,
      MVA_HOverHE,
      MVA_LateBrem,
      MVA_FirstBrem,
      MVA_MVA,
      MVA_LAST
    };
 
  public:
    PFCandidateElectronExtra();
    PFCandidateElectronExtra(const GsfTrackRef&);
    ~PFCandidateElectronExtra() { ; }
 
    void setGsfTrackRef(const reco::GsfTrackRef& ref);
 
    void setKfTrackRef(const reco::TrackRef& ref);
 
    reco::GsfTrackRef gsfTrackRef() const { return gsfTrackRef_; }
 
    reco::TrackRef kfTrackRef() const { return kfTrackRef_; }
 
    void setLateBrem(float val);
    void setEarlyBrem(float val);
 
    void setGsfTrackPout(const math::XYZTLorentzVector& pout);
 
    void setClusterEnergies(const std::vector<float>& energies);
 
    void setSigmaEtaEta(float val);
 
    void setDeltaEta(float val);
 
    void setHadEnergy(float val);
 
    void setMVA(float val);
 
    void setStatus(StatusFlag type, bool status = true);
 
    bool electronStatus(StatusFlag) const;
 
    int electronStatus() const { return status_; }
 
    bool mvaStatus(MvaVariable flag) const;
 
    const std::vector<float>& mvaVariables() const { return mvaVariables_; }
 
    float mvaVariable(MvaVariable var) const;
 
    float hadEnergy() const { return hadEnergy_; }
    float sigmaEtaEta() const { return sigmaEtaEta_; }
 
  private:
    void setVariable(MvaVariable type, float var);
 
  private:
    reco::GsfTrackRef gsfTrackRef_;
    reco::TrackRef kfTrackRef_;
 
    std::vector<float> clusterEnergies_;
 
    std::vector<float> mvaVariables_;
 
    int mvaStatus_;
 
    int status_;
 
    math::XYZTLorentzVector pout_;
    float earlyBrem_;
    float lateBrem_;
    float sigmaEtaEta_;
    float hadEnergy_;
    float deltaEta_;
  };
 
  std::ostream& operator<<(std::ostream& out, const PFCandidateElectronExtra& c);
 
}  // namespace reco
#endif