RemovePileUpDominatedEventsGen.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    RemovePileUpDominatedEventsGen/RemovePileUpDominatedEventsGen
// Class:      RemovePileUpDominatedEventsGen
// 
//
// Original Author:  Silvio DONATO
//         Created:  Fri, 12 Dec 2014 12:48:57 GMT
//
//

#include <memory>

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include <iostream>
#include <SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h>
#include <SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h>

class RemovePileUpDominatedEventsGen : public edm::stream::EDFilter <>{
   public:
      explicit RemovePileUpDominatedEventsGen(const edm::ParameterSet&);
      ~RemovePileUpDominatedEventsGen() override;
      static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

   private:
      bool filter(edm::Event&, const edm::EventSetup&) override;

      const edm::EDGetTokenT<std::vector<PileupSummaryInfo> > pileupSummaryInfos_;
      const edm::EDGetTokenT<GenEventInfoProduct > generatorInfo_;
      unsigned int bunchCrossing;
};

RemovePileUpDominatedEventsGen::RemovePileUpDominatedEventsGen(const edm::ParameterSet& iConfig):
pileupSummaryInfos_(    consumes<std::vector<PileupSummaryInfo> >(iConfig.getParameter<edm::InputTag>("pileupSummaryInfos"))    ),
generatorInfo_(         consumes<GenEventInfoProduct>(iConfig.getParameter<edm::InputTag>("generatorInfo"))                     )
{
bunchCrossing=0;
produces<float>();
}

RemovePileUpDominatedEventsGen::~RemovePileUpDominatedEventsGen() = default;

bool RemovePileUpDominatedEventsGen::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
   using namespace edm;
   using namespace std;
   
   edm::Handle<GenEventInfoProduct > generatorInfo;
   iEvent.getByToken(generatorInfo_,generatorInfo);

   edm::Handle<std::vector<PileupSummaryInfo> > pileupSummaryInfos;
   iEvent.getByToken(pileupSummaryInfos_,pileupSummaryInfos);
   
   //find in-time pile-up
   if(bunchCrossing>=pileupSummaryInfos.product()->size() || pileupSummaryInfos.product()->at(bunchCrossing).getBunchCrossing()!=0)
    {
       bool found=false;
       for(bunchCrossing=0; bunchCrossing<pileupSummaryInfos.product()->size() && !found; ++bunchCrossing)
       {
            if( pileupSummaryInfos.product()->at(bunchCrossing).getBunchCrossing() == 0 ){
                found=true;
                bunchCrossing--;
            }
       }
       if(!found){
            edm::LogInfo("RemovePileUpDominatedEventsGen") << "In-time pile-up not found!" << endl;
            return true;
       }
   }
   
//   cout << "Using "<<bunchCrossing<<endl;
//   cout << "pileupSummaryInfos.product()->at(bunchCrossing).getBunchCrossing() "<<pileupSummaryInfos.product()->at(bunchCrossing).getBunchCrossing()<<endl;

   //get the PU pt-hat max
   float signal_pT_hat = -1;
   float pu_pT_hat_max = -1;
   
   PileupSummaryInfo puSummary_onTime = pileupSummaryInfos.product()->at(bunchCrossing);
   for(const auto& pu_pT_hat : puSummary_onTime.getPU_pT_hats()) if (pu_pT_hat>pu_pT_hat_max) pu_pT_hat_max = pu_pT_hat;

   //get the signal pt-hat
   signal_pT_hat = generatorInfo->qScale();

   //save PU - signal pt-hat
   std::unique_ptr<float> pOut(new float());
   *pOut=signal_pT_hat-pu_pT_hat_max;   
   iEvent.put(std::move(pOut));

   //filter the event
   if (signal_pT_hat>pu_pT_hat_max) return true;
   return false;
}


void RemovePileUpDominatedEventsGen::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag> ("pileupSummaryInfos",edm::InputTag("addPileupInfo"));
  desc.add<edm::InputTag> ("generatorInfo",edm::InputTag("generator"));
  descriptions.addDefault(desc);
}

DEFINE_FWK_MODULE(RemovePileUpDominatedEventsGen);