HLTPMDocaFilter.cc

Go to the documentation of this file.

#include "HLTPMDocaFilter.h"

#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/EgammaCandidates/interface/Electron.h"
#include "DataFormats/EgammaCandidates/interface/ElectronFwd.h"

//
// constructors and destructor
//
HLTPMDocaFilter::HLTPMDocaFilter(const edm::ParameterSet& iConfig) : HLTFilter(iConfig)
{
  candTag_             = iConfig.getParameter< edm::InputTag > ("candTag");
  docaDiffPerpCutHigh_ = iConfig.getParameter<double> ("docaDiffPerpCutHigh");
  docaDiffPerpCutLow_  = iConfig.getParameter<double> ("docaDiffPerpCutLow");
  nZcandcut_           = iConfig.getParameter<int> ("nZcandcut");
  candToken_ = consumes<trigger::TriggerFilterObjectWithRefs>(candTag_);
}

HLTPMDocaFilter::~HLTPMDocaFilter()= default;

void
HLTPMDocaFilter::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  makeHLTFilterDescription(desc);
  desc.add<edm::InputTag>("candTag",edm::InputTag("HltZeePMMassFilter"));
  desc.add<double>("docaDiffPerpCutHigh",0.055691);
  desc.add<double>("docaDiffPerpCutLow",0.0);
  desc.add<int>("nZcandcut",1);
  descriptions.add("hltPMDocaFilter",desc);
}

// ------------ method called to produce the data  ------------
bool
HLTPMDocaFilter::hltFilter(edm::Event& iEvent, const edm::EventSetup& iSetup, trigger::TriggerFilterObjectWithRefs & filterproduct) const
{
  using namespace std;
  using namespace edm;
  using namespace reco;
  using namespace trigger;

  // Ref to Candidate object to be recorded in filter object
  edm::Ref<reco::ElectronCollection> ref;

  edm::Handle<trigger::TriggerFilterObjectWithRefs> PrevFilterOutput;
  iEvent.getByToken (candToken_,PrevFilterOutput);

  std::vector<edm::Ref<reco::ElectronCollection> > electrons;
  PrevFilterOutput->getObjects(TriggerElectron, electrons);


  int n = 0;

  unsigned int size = electrons.size();
  std::vector<double> vx(size);
  std::vector<double> vy(size);

  for (unsigned int i=0; i< size; i++) {
    ref = electrons[i];
    vx[i]=ref->vx();
    vy[i]=ref->vy();
  }

  for(unsigned int jj=0;jj<size;jj++){
     for(unsigned int ii=jj+1;ii<size;ii++){
     double docaDiffPerp = sqrt( (vx[jj]-vx[ii])*(vx[jj]-vx[ii])+(vy[jj]-vy[ii])*(vy[jj]-vy[ii]));
     // std::cout<<"docaDiffPerp= "<<docaDiffPerp<<std::endl;
     if((docaDiffPerp>=docaDiffPerpCutLow_) && (docaDiffPerp<= docaDiffPerpCutHigh_)){
       n++;
       ref = electrons[ii];
       filterproduct.addObject(TriggerElectron, ref);
       ref = electrons[jj];
       filterproduct.addObject(TriggerElectron, ref);

     }
     }
  }


  // filter decision
  bool accept(n>=nZcandcut_);

  return accept;
}


// declare this class as a framework plugin
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(HLTPMDocaFilter);