PixelFEDChannelCollectionProducer

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
// -*- C++ -*-
//
// Package:    CalibTracker/PixelFEDChannelCollectionProducer
// Class:      PixelFEDChannelCollectionProducer
//
/**\class PixelFEDChannelCollectionProducer

 Description: [one line class summary]

 Implementation:
     [Notes on implementation]
*/
//
// Original Author:  Marco Musich
//         Created:  Thu, 13 Dec 2018 08:48:22 GMT
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"

#include "CondFormats/DataRecord/interface/SiPixelStatusScenariosRcd.h"
#include "CondFormats/SiPixelObjects/interface/SiPixelFEDChannelContainer.h"
#include "CalibTracker/Records/interface/SiPixelFEDChannelContainerESProducerRcd.h"

// Need to add #include statements for definitions of
// the data type and record type here

//
// class declaration
//

class PixelFEDChannelCollectionProducer : public edm::ESProducer {
public:
  PixelFEDChannelCollectionProducer(const edm::ParameterSet&);
  ~PixelFEDChannelCollectionProducer() override;

  typedef std::unordered_map<std::string, PixelFEDChannelCollection> PixelFEDChannelCollectionMap;
  using ReturnType = std::unique_ptr<PixelFEDChannelCollectionMap>;

  ReturnType produce(const SiPixelFEDChannelContainerESProducerRcd&);

private:
  // ----------member data ---------------------------
  const edm::ESGetToken<SiPixelFEDChannelContainer, SiPixelStatusScenariosRcd> qualityToken_;
};

PixelFEDChannelCollectionProducer::PixelFEDChannelCollectionProducer(const edm::ParameterSet& iConfig)
    : qualityToken_(setWhatProduced(this).consumes()) {}

PixelFEDChannelCollectionProducer::~PixelFEDChannelCollectionProducer() {
  // do anything here that needs to be done at destruction time
  // (e.g. close files, deallocate resources etc.)
}

//
// member functions
//

// ------------ method called to produce the data  ------------
PixelFEDChannelCollectionProducer::ReturnType PixelFEDChannelCollectionProducer::produce(
    const SiPixelFEDChannelContainerESProducerRcd& iRecord) {
  const auto& qualityCollection = iRecord.get(qualityToken_);

  auto out = std::make_unique<PixelFEDChannelCollectionMap>();

  for (const auto& it : qualityCollection.getScenarioMap()) {
    const std::string& scenario = it.first;
    // getScenarioMap() is an unordered_map<string, ...>, so each scenario appears exactly once
    PixelFEDChannelCollection& disabled_channelcollection = (*out)[scenario];

    const auto& SiPixelBadFedChannels = it.second;
    for (const auto& entry : SiPixelBadFedChannels) {
      disabled_channelcollection.insert(entry.first, entry.second.data(), entry.second.size());
    }
  }

  return out;
}

//define this as a plug-in
DEFINE_FWK_EVENTSETUP_MODULE(PixelFEDChannelCollectionProducer);