SimpleClusterProbabilitySeedComparitor.cc

Go to the documentation of this file.

#include "RecoTracker/TkSeedingLayers/interface/SeedComparitor.h"
#include "RecoTracker/TkSeedingLayers/interface/SeedComparitorFactory.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include <cmath>

class SimpleClusterProbabilitySeedComparitor : public SeedComparitor {
    public:
        SimpleClusterProbabilitySeedComparitor(const edm::ParameterSet &cfg, edm::ConsumesCollector& iC) ;
        ~SimpleClusterProbabilitySeedComparitor() override ; 
        void init(const edm::Event& ev, const edm::EventSetup& es) override {}
        bool compatible(const SeedingHitSet  &hits) const override { return true; }
        bool compatible(const TrajectoryStateOnSurface &,
                SeedingHitSet::ConstRecHitPointer hit) const override ;
        bool compatible(const SeedingHitSet  &hits, 
                const GlobalTrajectoryParameters &helixStateAtVertex,
                const FastHelix                  &helix) const override { return true; }


    private:
        float probCut_;
};


SimpleClusterProbabilitySeedComparitor::SimpleClusterProbabilitySeedComparitor(const edm::ParameterSet &cfg, edm::ConsumesCollector& iC) :
    probCut_(cfg.getParameter<double>("LogPixelProbabilityCut"))
{
}

SimpleClusterProbabilitySeedComparitor::~SimpleClusterProbabilitySeedComparitor() 
{
}

bool
SimpleClusterProbabilitySeedComparitor::compatible(const TrajectoryStateOnSurface &tsos,
                           SeedingHitSet::ConstRecHitPointer hit) const
{
    return (probCut_ < -15.) || (log10(hit->clusterProbability()) > probCut_);
}

DEFINE_EDM_PLUGIN(SeedComparitorFactory, SimpleClusterProbabilitySeedComparitor, "SimpleClusterProbabilitySeedComparitor");