ElectronConversionRejectionVars.cc

Go to the documentation of this file.

//

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Common/interface/View.h"

#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "RecoEgamma/EgammaTools/interface/ConversionFinder.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/Scalers/interface/DcsStatus.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"

class ElectronConversionRejectionVars : public edm::EDProducer {
public:
  explicit ElectronConversionRejectionVars(const edm::ParameterSet& iConfig);
  ~ElectronConversionRejectionVars() override;

  void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;

private:
  edm::EDGetTokenT<edm::View<reco::Candidate>> probesToken_;
  edm::EDGetTokenT<reco::TrackCollection> tracksToken_;
  edm::EDGetTokenT<reco::GsfElectronCollection> gsfElecsToken_;
};

ElectronConversionRejectionVars::ElectronConversionRejectionVars(const edm::ParameterSet& iConfig)
    : probesToken_(consumes<edm::View<reco::Candidate>>(iConfig.getParameter<edm::InputTag>("probes"))),
      tracksToken_(consumes<reco::TrackCollection>(edm::InputTag("generalTracks"))),
      gsfElecsToken_(consumes<reco::GsfElectronCollection>(edm::InputTag("gsfElectrons"))) {
  produces<edm::ValueMap<float>>("dist");
  produces<edm::ValueMap<float>>("dcot");
  produces<edm::ValueMap<float>>("convradius");
  produces<edm::ValueMap<float>>("passConvRej");
}

ElectronConversionRejectionVars::~ElectronConversionRejectionVars() {}

void ElectronConversionRejectionVars::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;

  // read input
  Handle<View<reco::Candidate>> probes;
  edm::Handle<reco::TrackCollection> tracks_h;
  edm::Handle<reco::GsfElectronCollection> elHandle;

  iEvent.getByToken(probesToken_, probes);
  iEvent.getByToken(tracksToken_, tracks_h);
  iEvent.getByToken(gsfElecsToken_, elHandle);

  float evt_bField = 3.8;

  // prepare vector for output
  std::vector<float> values;
  std::vector<float> values2;
  std::vector<float> values3;
  std::vector<float> values4;

  // fill: use brute force
  double dist = 0.0;
  double dcot = 0.0;
  double convradius = 0.0;
  double passConvRej = 0.0;

  View<reco::Candidate>::const_iterator probe, endprobes = probes->end();
  const reco::GsfElectronCollection* electronCollection = elHandle.product();
  reco::GsfElectronCollection::const_iterator eleIt = electronCollection->begin();

  for (probe = probes->begin(); probe != endprobes; ++probe) {
    for (eleIt = electronCollection->begin(); eleIt != electronCollection->end(); eleIt++) {
      if (fabs(eleIt->et() - probe->et()) < 0.05 && fabs(eleIt->eta() - probe->eta()) < 0.01 &&
          fabs(eleIt->phi() - probe->phi()) < 0.01) {
        //we have a match
        ConversionInfo convInfo = egammaTools::getConversionInfo(*eleIt, tracks_h, evt_bField);
        dist = convInfo.dist;
        dcot = convInfo.dcot;
        convradius = convInfo.radiusOfConversion;
        if (fabs(dist) > 0.02 && fabs(dcot) > 0.02)
          passConvRej = 1.0;
        break;  //got our guy, so break
      }
    }
    values.push_back(dist);
    values2.push_back(dcot);
    values3.push_back(convradius);
    values4.push_back(passConvRej);
  }

  // convert into ValueMap and store
  auto valMap = std::make_unique<ValueMap<float>>();
  ValueMap<float>::Filler filler(*valMap);
  filler.insert(probes, values.begin(), values.end());
  filler.fill();
  iEvent.put(std::move(valMap), "dist");

  // ---> same for dcot
  auto valMap2 = std::make_unique<ValueMap<float>>();
  ValueMap<float>::Filler filler2(*valMap2);
  filler2.insert(probes, values2.begin(), values2.end());
  filler2.fill();
  iEvent.put(std::move(valMap2), "dcot");

  // ---> same for convradius
  auto valMap3 = std::make_unique<ValueMap<float>>();
  ValueMap<float>::Filler filler3(*valMap3);
  filler3.insert(probes, values3.begin(), values3.end());
  filler3.fill();
  iEvent.put(std::move(valMap3), "convradius");

  // ---> same for passConvRej
  auto valMap4 = std::make_unique<ValueMap<float>>();
  ValueMap<float>::Filler filler4(*valMap4);
  filler4.insert(probes, values4.begin(), values4.end());
  filler4.fill();
  iEvent.put(std::move(valMap4), "passConvRej");
}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ElectronConversionRejectionVars);