HLTTrackWithHits.h

Go to the documentation of this file.

#ifndef HLTrigger_HLTTrackWithHits_H

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDFilter.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/MessageService/interface/MessageLogger.h"

#include "HLTrigger/HLTcore/interface/HLTFilter.h"
#include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

class HLTTrackWithHits : public HLTFilter {
public:
  explicit HLTTrackWithHits(const edm::ParameterSet& iConfig) : HLTFilter(iConfig),
    src_(iConfig.getParameter<edm::InputTag>("src")),
    minN_(iConfig.getParameter<int>("MinN")),
    maxN_(iConfig.getParameter<int>("MaxN")),
    MinBPX_(iConfig.getParameter<int>("MinBPX")),
    MinFPX_(iConfig.getParameter<int>("MinFPX")),
    MinPXL_(iConfig.getParameter<int>("MinPXL")),
    MinPT_(iConfig.getParameter<double>("MinPT"))
  {
    srcToken_ = consumes<reco::TrackCollection>(src_);
  }

  ~HLTTrackWithHits() { }

  static void fillDescriptions(edm::ConfigurationDescriptions & descriptions)
    {
      edm::ParameterSetDescription desc;
      makeHLTFilterDescription(desc);
      desc.add<edm::InputTag>("src",edm::InputTag(""));
      desc.add<int>("MinN",0);
      desc.add<int>("MaxN",99999);
      desc.add<int>("MinBPX",0);
      desc.add<int>("MinFPX",0);
      desc.add<int>("MinPXL",0);
      desc.add<double>("MinPT",0.);
      descriptions.add("hltTrackWithHits",desc);
    }

private:
  virtual bool hltFilter(edm::Event& iEvent, const edm::EventSetup&, trigger::TriggerFilterObjectWithRefs & filterproduct) const override
  {
    edm::Handle<reco::TrackCollection> oHandle;
    iEvent.getByToken(srcToken_, oHandle);
    int s=oHandle->size();
    int count=0;
    for (int i=0;i!=s;++i){
      const reco::Track & track = (*oHandle)[i];
      if (track.pt() < MinPT_) continue;
      const reco::HitPattern & hits = track.hitPattern();
      if ( MinBPX_>0 && hits.numberOfValidPixelBarrelHits() >= MinBPX_ ) { ++count; continue; }
      if ( MinFPX_>0 && hits.numberOfValidPixelEndcapHits() >= MinFPX_ ) { ++count; continue; }
      if ( MinPXL_>0 && hits.numberOfValidPixelHits() >= MinPXL_ )       { ++count; continue; }
    }

    bool answer=(count>=minN_ && count<=maxN_);
    LogDebug("HLTTrackWithHits")<<module(iEvent)<<" sees: "<<s<<" objects. Only: "<<count<<" satisfy the hit requirement. Filter answer is: "<<(answer?"true":"false")<<std::endl;
    return answer;
  }

  edm::InputTag src_;
  edm::EDGetTokenT<reco::TrackCollection> srcToken_;
  int minN_,maxN_,MinBPX_,MinFPX_,MinPXL_;
  double MinPT_;
};


#endif