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 "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "SimCalorimetry/EcalSelectiveReadoutAlgos/interface/EcalSelectiveReadoutSuppressor.h"
#include "DataFormats/Provenance/interface/ProductID.h"
#include "CondFormats/EcalObjects/interface/EcalSRSettings.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:
  std::unique_ptr<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;
  edm::ParameterSet params_;

  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;

};

#endif