HGCDigiConverter.cc

Go to the documentation of this file.

//STL includes
#include <memory>

//framework includes
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

//other includes
#include "DataFormats/HGCDigi/interface/HGCDigiCollections.h"

class HGCDigiConverter : public edm::global::EDProducer<> {
    public:
        explicit HGCDigiConverter(const edm::ParameterSet&);
        static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
    private:
        void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
        edm::EDGetTokenT<HGCEEDigiCollection> tok_ee_;
        edm::EDGetTokenT<HGCHEDigiCollection> tok_fh_;
        edm::EDGetTokenT<HGCBHDigiCollection> tok_bh_;
};

HGCDigiConverter::HGCDigiConverter(const edm::ParameterSet& iConfig) :
    tok_ee_(consumes<HGCEEDigiCollection>(iConfig.getParameter<edm::InputTag>("eeDigis"))),
    tok_fh_(consumes<HGCHEDigiCollection>(iConfig.getParameter<edm::InputTag>("fhDigis"))),
    tok_bh_(consumes<HGCBHDigiCollection>(iConfig.getParameter<edm::InputTag>("bhDigis")))
{
    produces<HGCalDigiCollection>("EE");
    produces<HGCalDigiCollection>("HEfront");
    produces<HGCalDigiCollection>("HEback");
}

void HGCDigiConverter::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
    edm::Handle<HGCEEDigiCollection> h_ee;
    edm::Handle<HGCHEDigiCollection> h_fh;
    edm::Handle<HGCBHDigiCollection> h_bh;

    iEvent.getByToken(tok_ee_,h_ee);
    iEvent.getByToken(tok_fh_,h_fh);
    iEvent.getByToken(tok_bh_,h_bh);

    auto out_ee = std::make_unique<HGCalDigiCollection>();
    if(h_ee.isValid()){
        for(const auto& df_ee : *h_ee){
            HGCalDataFrame tmp(df_ee.id());
            tmp.setData(df_ee.data());
            out_ee->push_back(tmp);
        }
    }
    iEvent.put(std::move(out_ee),"EE");

    auto out_fh = std::make_unique<HGCalDigiCollection>();
    if(h_fh.isValid()){
        for(const auto& df_fh : *h_fh){
            HGCalDataFrame tmp(df_fh.id());
            tmp.setData(df_fh.data());
            out_fh->emplace_back(tmp);
        }
    }
    iEvent.put(std::move(out_fh),"HEfront");

    auto out_bh = std::make_unique<HGCalDigiCollection>();
    if(h_bh.isValid()){
        for(const auto& df_bh : *h_bh){
            HGCalDataFrame tmp(df_bh.id());
            tmp.setData(df_bh.data());
            out_bh->emplace_back(tmp);
        }
    }
    iEvent.put(std::move(out_bh),"HEback");
}

void HGCDigiConverter::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.add<edm::InputTag>("eeDigis",edm::InputTag("mix:HGCDigisEE"));
    desc.add<edm::InputTag>("fhDigis",edm::InputTag("mix:HGCDigisHEfront"));
    desc.add<edm::InputTag>("bhDigis",edm::InputTag("mix:HGCDigisHEback"));

    descriptions.add("HGCDigiConverter",desc);
}

//define this as a plug-in
DEFINE_FWK_MODULE(HGCDigiConverter);