PATTriggerObjectStandAloneSlimmer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PatAlgos
// Class:      pat::PATTriggerObjectStandAloneSlimmer
//
//
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/PatCandidates/interface/TriggerObjectStandAlone.h"
#include <set>

namespace pat {
  
  class PATTriggerObjectStandAloneSlimmer : public edm::global::EDProducer<> {
    
  public:
    
    explicit PATTriggerObjectStandAloneSlimmer( const edm::ParameterSet & iConfig );
    ~PATTriggerObjectStandAloneSlimmer() {};
    
  private:
    
    virtual void produce(edm::StreamID, edm::Event & iEvent, const edm::EventSetup& iSetup) const override;
    
    const edm::EDGetTokenT<TriggerObjectStandAloneCollection> srcToken_;
    const edm::EDGetTokenT< edm::TriggerResults > triggerResultsToken_;
 
    bool packFilterLabels_, packP4_;
 
  };
  
}


using namespace pat;


PATTriggerObjectStandAloneSlimmer::PATTriggerObjectStandAloneSlimmer( const edm::ParameterSet & iConfig ) : 
    srcToken_( consumes<TriggerObjectStandAloneCollection>( iConfig.getParameter<edm::InputTag>( "src" ) ) ),
    triggerResultsToken_( consumes< edm::TriggerResults >( iConfig.getParameter< edm::InputTag >( "triggerResults" ) ) ),
    packFilterLabels_( iConfig.getParameter<bool>("packFilterLabels") ),
    packP4_( iConfig.getParameter<bool>("packP4") )
{
    produces<TriggerObjectStandAloneCollection>();
    if (packFilterLabels_) {
        produces<std::vector<std::string>>("filterLabels");
    }
}

void PATTriggerObjectStandAloneSlimmer::produce( edm::StreamID, edm::Event & iEvent, const edm::EventSetup& iSetup) const
{
    edm::Handle<TriggerObjectStandAloneCollection> src;
    iEvent.getByToken( srcToken_, src );
    edm::Handle< edm::TriggerResults > triggerResults;
    iEvent.getByToken( triggerResultsToken_, triggerResults );

    auto slimmed = std::make_unique<TriggerObjectStandAloneCollection>(*src);

    if (packFilterLabels_) {
        std::set<std::string> allLabels;
        for (auto & obj : *slimmed) {
            obj.packFilterLabels(iEvent,*triggerResults);
        }

    }
    if (packP4_) {
        for (TriggerObjectStandAlone & obj : *slimmed) {
            obj.packP4();
        }
    }
    iEvent.put(std::move(slimmed) );
}


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