EcalSelectiveReadoutSuppressor.h

Go to the documentation of this file.

#ifndef EcalSelectiveReadoutSuppressor_h
#define EcalSelectiveReadoutSuppressor_h

#include <vector>
#include "boost/multi_array.hpp"
#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloTopology/interface/EcalTrigTowerConstituentsMap.h"
#include "Geometry/EcalMapping/interface/EcalElectronicsMapping.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimCalorimetry/EcalSelectiveReadoutAlgos/interface/EcalSelectiveReadout.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "CondFormats/EcalObjects/interface/EcalSRSettings.h"

#include <memory>

class CaloGeometryRecord;

class EcalSelectiveReadoutSuppressor {
public:
  EcalSelectiveReadoutSuppressor(const edm::ParameterSet& params, edm::ConsumesCollector iC);

  void setSettings(const EcalSRSettings* settings);

  enum { BARREL, ENDCAP };

  static int getFIRTapCount() { return nFIRTaps; }

  void setTriggerMap(const EcalTrigTowerConstituentsMap* map);

  void setElecMap(const EcalElectronicsMapping* map);

  void setGeometry(const CaloGeometry* caloGeometry);

  void run(const edm::EventSetup& eventSetup,
           const EcalTrigPrimDigiCollection& trigPrims,
           EBDigiCollection& barrelDigis,
           EEDigiCollection& endcapDigis);

  void run(const edm::EventSetup& eventSetup,
           const EcalTrigPrimDigiCollection& trigPrims,
           const EBDigiCollection& barrelDigis,
           const EEDigiCollection& endcapDigis,
           EBDigiCollection* selectedBarrelDigis,
           EEDigiCollection* selectedEndcapDigis,
           EBSrFlagCollection* ebSrFlags,
           EESrFlagCollection* eeSrFlags);

  EcalSelectiveReadout* getEcalSelectiveReadout() { return ecalSelectiveReadout.get(); }

  void printTTFlags(std::ostream& os, int iEvent = -1, bool withHeader = true) const;

private:
  bool accept(const edm::DataFrame& frame, int thr);


  void initCellThresholds(double barrelLowInterest,
                          double endcapLowInterest,
                          double barrelHighInterest,
                          double endcapHighInterest);
  int internalThreshold(double thresholdInGeV, int iSubDet) const;

  std::vector<int> getFIRWeigths();

  int iEta2cIndex(int iEta) const { return (iEta < 0) ? iEta + 85 : iEta + 84; }

  int iPhi2cIndex(int iPhi) const { return iPhi - 1; }

  int iTTEta2cIndex(int iEta) const { return (iEta < 0) ? iEta + 28 : iEta + 27; }

  int iTTPhi2cIndex(int iPhi) const { return iPhi - 1; }

  void setTtFlags(const edm::EventSetup& eventSetup, const EBDigiCollection& ebDigis, const EEDigiCollection& eeDigis);

  void setTtFlags(const EcalTrigPrimDigiCollection& trigPrims);

  template <class T>
  double frame2Energy(const T& frame, int timeOffset = 0) const;

  //   /** Help function to get SR flag from ZS threshold using min/max convention
  //    * for SUPPRESS and FULL_READOUT: see zsThreshold.
  //    * @param thr ZS threshold in thrUnit
  //    * @param flag for Zero suppression: EcalSrFlag::SRF_ZS1 or
  //    * EcalSrFlag::SRF_ZS2
  //    * @return the SR flag
  //    */
  //   int thr2Srf(int thr, int zsFlag) const;

  const static size_t nEndcaps = 2;

  const static size_t nEndcapTriggerTowersInEta = 11;

  const static size_t nBarrelTriggerTowersInEta = 34;

  const static size_t nTriggerTowersInEta = 2 * nEndcapTriggerTowersInEta + nBarrelTriggerTowersInEta;

  const static size_t nTriggerTowersInPhi = 72;

  std::unique_ptr<EcalSelectiveReadout> ecalSelectiveReadout;

  const EcalTrigTowerConstituentsMap* theTriggerMap;

  EcalSelectiveReadout::ttFlag_t ttFlags[nTriggerTowersInEta][nTriggerTowersInPhi];

  int firstFIRSample;

  std::vector<int> firWeights;

  static const int nFIRTaps;

  std::vector<float> weights;

  bool symetricZS;

  int zsThreshold[2][8];

  double thrUnit[2];

  bool trigPrimBypass_;

  int trigPrimBypassMode_;

  std::vector<int> actions_;

  bool ttThresOnCompressedEt_;

  bool trigPrimBypassWithPeakFinder_;

  double trigPrimBypassLTH_;

  double trigPrimBypassHTH_;

  int srFlags[2][8];

  EcalSelectiveReadout::ttFlag_t defaultTtf_;

  int ievt_;

  edm::ESGetToken<CaloGeometry, CaloGeometryRecord> geoToken_;
};
#endif