BPHChi2Select.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHChi2Select_h
#define HeavyFlavorAnalysis_SpecificDecay_BPHChi2Select_h

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

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHDecayVertex.h"
#include "TMath.h"

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


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

class BPHChi2Select: public BPHVertexSelect {

 public:

  BPHChi2Select( double prob ): probMin( prob ) {}

  virtual ~BPHChi2Select() {}

  virtual bool accept( const BPHDecayVertex& cand ) const {
    const reco::Vertex& v = cand.vertex();
    if ( v.isFake() ) return false;
    if ( !v.isValid() ) return false;
    return ( TMath::Prob( v.chi2(), lround( v.ndof() ) ) > probMin );
  }

  void setProbMin( double p ) { probMin = p; return; }

  double getProbMin() const { return probMin; }

 private:

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

  double probMin;

};


#endif