EcalSelectiveReadoutProducer.h

Go to the documentation of this file.

#ifndef ECALZEROSUPPRESSIONPRODUCER_H
#define ECALZEROSUPPRESSIONPRODUCER_H

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "SimCalorimetry/EcalSelectiveReadoutAlgos/interface/EcalSelectiveReadoutSuppressor.h"
#include "DataFormats/Provenance/interface/ProductID.h"
#include "CondFormats/EcalObjects/interface/EcalSRSettings.h"
#include "CondFormats/DataRecord/interface/EcalSRSettingsRcd.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "Geometry/EcalMapping/interface/EcalMappingRcd.h"

#include <memory>
#include <vector>

class EcalSelectiveReadoutProducer : public edm::one::EDProducer<> {
public:
  explicit EcalSelectiveReadoutProducer(const edm::ParameterSet& params);

  ~EcalSelectiveReadoutProducer() override;

  void produce(edm::Event& event, const edm::EventSetup& eventSetup) override;

  static void printSrFlags(std::ostream& os,
                           const EBSrFlagCollection& ebSrFlags,
                           const EESrFlagCollection& eeSrFlags,
                           int iEvent = -1,
                           bool withHeader = true);

private:
  void checkWeights(const edm::Event& evt, const edm::ProductID& noZSDigiId) const;

  bool getBinOfMax(const edm::Event& evt, const edm::ProductID& noZsDigiId, int& binOfMax) const;

  const EBDigiCollection* getEBDigis(edm::Event& event);

  const EEDigiCollection* getEEDigis(edm::Event& event);

  const EcalTrigPrimDigiCollection* getTrigPrims(edm::Event& event) const;

  void checkGeometry(const edm::EventSetup& eventSetup);
  void checkTriggerMap(const edm::EventSetup& eventSetup);
  void checkElecMap(const edm::EventSetup& eventSetup);


  static void checkValidity(const EcalSRSettings& settings);

  void printTTFlags(const EcalTrigPrimDigiCollection& tp, std::ostream& os) const;

private:
  EcalSelectiveReadoutSuppressor suppressor_;
  std::string digiProducer_;         // name of module/plugin/producer making digis
  std::string ebdigiCollection_;     // secondary name given to collection of input digis
  std::string eedigiCollection_;     // secondary name given to collection of input digis
  std::string ebSRPdigiCollection_;  // secondary name given to collection of suppressed digis
  std::string eeSRPdigiCollection_;  // secondary name given to collection of suppressed digis
  std::string ebSrFlagCollection_;   // secondary name given to collection of SR flag digis
  std::string eeSrFlagCollection_;   // secondary name given to collection of SR flag digis
  std::string trigPrimProducer_;     // name of module/plugin/producer making triggere primitives
  std::string trigPrimCollection_;   // name of module/plugin/producer making triggere primitives

  // store the pointer, so we don't have to update it every event
  const CaloGeometry* theGeometry;
  const EcalTrigTowerConstituentsMap* theTriggerTowerMap;
  const EcalElectronicsMapping* theElecMap;

  bool suppressorSettingsSet_ = false;

  bool trigPrimBypass_;

  int trigPrimBypassMode_;

  int dumpFlags_;

  bool writeSrFlags_;

  bool produceDigis_;

  const EcalSRSettings* settings_;

  bool useCondDb_;

  bool useFullReadout_;

  bool firstCallEB_;
  bool firstCallEE_;
  int iEvent_;

  std::unique_ptr<EcalSRSettings> settingsFromFile_;

  // Tokens for consumes collection:

  edm::EDGetTokenT<EBDigiCollection> EB_token;
  edm::EDGetTokenT<EEDigiCollection> EE_token;
  edm::EDGetTokenT<EcalTrigPrimDigiCollection> EcTP_token;
  edm::ESGetToken<EcalSRSettings, EcalSRSettingsRcd> hSr_token_;
  edm::ESGetToken<CaloGeometry, CaloGeometryRecord> geom_token_;
  edm::ESGetToken<EcalTrigTowerConstituentsMap, IdealGeometryRecord> eTTmap_token_;
  edm::ESGetToken<EcalElectronicsMapping, EcalMappingRcd> eElecmap_token_;
};

#endif