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/ParameterSet/interface/ParameterSet.h"
#include "SimCalorimetry/EcalSelectiveReadoutAlgos/src/EcalSelectiveReadout.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "CondFormats/EcalObjects/interface/EcalSRSettings.h"

#include <memory>

class EcalSelectiveReadoutSuppressor{
public:
  EcalSelectiveReadoutSuppressor(const edm::ParameterSet & params, 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_;
};
#endif