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 ) {}

  virtual ~BPHParticleChargeSelect() {}

  virtual bool accept( const reco::Candidate& cand ) const {
    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:

  // private copy and assigment constructors
  BPHParticleChargeSelect           ( const BPHParticleChargeSelect& x );
  BPHParticleChargeSelect& operator=( const BPHParticleChargeSelect& x );

  int charge;

};


#endif