BPHMassSymSelect.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHMassSymSelect_h
#define HeavyFlavorAnalysis_SpecificDecay_BPHMassSymSelect_h

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

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHDecayMomentum.h"
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHMassSelect.h"

//---------------
// C++ Headers --
//---------------
#include <string>

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

class BPHMassSymSelect: public BPHMomentumSelect {

 public:

  BPHMassSymSelect( const std::string& np, const std::string& nn,
                    const BPHMassSelect* ms ): nPos( np ), nNeg( nn ),
                                               mSel( ms ) {}

  virtual ~BPHMassSymSelect() {}

  virtual bool accept( const BPHDecayMomentum& cand ) const {

    if ( mSel->accept( cand ) ) return true;

    const
    reco::Candidate* pp = cand.getDaug( nPos );
    const
    reco::Candidate* np = cand.getDaug( nNeg );

    reco::Candidate* pc = cand.originalReco( pp )->clone();
    reco::Candidate* nc = cand.originalReco( np )->clone();

    pc->setMass( np->p4().mass() );
    nc->setMass( pp->p4().mass() );
    const reco::Candidate::LorentzVector  s4 = pc->p4() + nc->p4();
    double mass = s4.mass();

    delete pc;
    delete nc;
    return ( ( mass > mSel->getMassMin() ) &&
             ( mass < mSel->getMassMax() ) );

  }

 private:

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

  std::string nPos;
  std::string nNeg;
  const BPHMassSelect* mSel;

};


#endif