BPHParticleChargeSelect.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHParticleChargeSelect_h
#define HeavyFlavorAnalysis_SpecificDecay_BPHParticleChargeSelect_h

//----------------------
// Base Class Headers --
//----------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoSelect.h"

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"

//---------------
// C++ Headers --
//---------------

//              ---------------------
//              -- Class Interface --
//              ---------------------

class BPHParticleChargeSelect : public BPHRecoSelect {
public:
  BPHParticleChargeSelect(int c) : charge(c ? (c > 0 ? 1 : -1) : 0) {}

  // deleted copy constructor and assignment operator
  BPHParticleChargeSelect(const BPHParticleChargeSelect& x) = delete;
  BPHParticleChargeSelect& operator=(const BPHParticleChargeSelect& x) = delete;

  ~BPHParticleChargeSelect() override = default;

  bool accept(const reco::Candidate& cand) const override {
    switch (charge) {
      default:
      case 0:
        return (cand.charge() != 0);
      case 1:
        return (cand.charge() > 0);
      case -1:
        return (cand.charge() < 0);
    }
    return true;
  };

  void setCharge(int c) {
    charge = (c ? (c > 0 ? 1 : -1) : 0);
    return;
  }

  double getCharge() const { return charge; }

private:
  int charge;
};

#endif