CmsLikeJetEfficiency.cc

#include "CmsLikeJetEfficiency.h"
#include "generator/pythia6.h"

#include "PythiaJetGun.hh"

#define CODE_KLONG 130

namespace stab {

    CmsLikeJetEfficiency::CmsLikeJetEfficiency(
        const double i_minTrackPt,
        const double i_minNeutralHadronPt,
        const double i_minPhotonPt ,
        const double i_trackingEfficiencyInJets)
        : m_minTrackPt(i_minTrackPt), 
          m_trackRap(5.5),
          m_minNeutralHadronPt(i_minNeutralHadronPt),
          m_hadRap(5.5),
          m_minPhotonPt(i_minPhotonPt),
          m_photoRap(5.5),
          m_trackingEfficiencyInJets(i_trackingEfficiencyInJets),
      m_friendInfo(NULL)
    {
        m_klong_mass = pymass(CODE_KLONG);
    }

    void CmsLikeJetEfficiency::processPhoton(
        const PseudoJet& in, const StabilityUserInfo &info, ParticleCollection* out
    ) const
    {
        if ( in.pt() >= m_minPhotonPt && fabs(in.eta()) < m_photoRap ) {
            out->push_back(in);
        m_friendInfo->push_back( info );
    }
    }

    void CmsLikeJetEfficiency::processTrack(
        const PseudoJet& in, const StabilityUserInfo &info, ParticleCollection* out
    ) const
    {
        if ( in.pt() >= m_minTrackPt && fabs(in.eta()) < m_trackRap )
        {
            // Emulate tracking inefficiency
            const double rnd = PythiaJetGun::pyrandom();
            if (rnd > m_trackingEfficiencyInJets) {
                // Replace the input particle by K_L
        PseudoJet part(in.px(),in.py(),in.pz(),in.E());
            //StabilityUserInfo *info = new StabilityUserInfo(CODE_KLONG,m_klong_mass);
        //part.set_user_info( info );
                out->push_back( part );
        m_friendInfo->push_back( StabilityUserInfo(CODE_KLONG,m_klong_mass) );
        }
            else
                out->push_back(in);
        m_friendInfo->push_back( info );
        }
    }

    void CmsLikeJetEfficiency::processNeutralHadron(
        const PseudoJet& in, const StabilityUserInfo &info, ParticleCollection* out
    ) const
    {
        if ( in.pt() >= m_minNeutralHadronPt && fabs(in.eta()) < m_hadRap )
            out->push_back(in);
        m_friendInfo->push_back(info);
    }


} // end stab namespace