EcalSimRawData.h

Go to the documentation of this file.

/*
 */

#include "DataFormats/EcalDigi/interface/EcalDigiCollections.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloTopology/interface/EcalTrigTowerConstituentsMap.h"
#include "SimCalorimetry/EcalSelectiveReadoutAlgos/interface/EcalSelectiveReadout.h"
#include <fstream>
#include <string>

class EcalSimRawData : public edm::one::EDAnalyzer<> {
public:
  explicit EcalSimRawData(const edm::ParameterSet &pset);

  ~EcalSimRawData() override = default;

  void analyze(const edm::Event &, const edm::EventSetup &) override;

private:
  enum tokenType { tcp = 0, tp = 1 };

  int iEvent;

  static const int nEbEta = 170;

  static const int nEbPhi = 360;

  static const int nEeX = 100;

  static const int nEeY = 100;

  static const int nEndcaps = 2;

  static const int scEdge = 5;

  static const int nScX = 20;

  static const int nScY = 20;

  /* Edge size of a barrel TT in num. of crystals
   */
  static const int ttEdge = 5;

  static const int nTtSmPhi = 4;

  static const int nTtSmEta = 17;

  static const int nTtPhi = nEbPhi / ttEdge;  // 72

  static const int nEbTtEta = nEbEta / ttEdge;  // 34

  static const int nEeTtEta = 11;

  static const int nTtEta = nEbTtEta + 2 * nEeTtEta;  // 56

  static const int nDccInPhi = 18;

  static const int nDccEndcap = 9;

  static const int ebTccPhiEdge = 20;

  static const int nTccInPhi = 18;

  static const int nTccEndcap = 36;

  static const int ebDccPhiEdge = 20;

  static const int nTtPhisPerEbDcc = 4;

  static const int nTtPhisPerEbTcc = 4;

  static const int nTtTypes = 2;

  static const int ttType[nEbTtEta];

  static const int stripCh2Phi[nTtTypes][ttEdge][ttEdge];

  static const int strip2Eta[nTtTypes][ttEdge];

  enum writeMode_t { littleEndian, bigEndian, ascii };

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

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

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

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

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

  int iPhi2cIndex(int iPhi) const {
    int iPhi0 = iPhi - 11;
    if (iPhi0 < 0)
      iPhi0 += nEbPhi;
    return iPhi0;
  }

  int iTtEta2cIndex(int iTtEta) const { return (iTtEta < 0) ? (iTtEta + 28) : (iTtEta + 27); }

  int cIndex2iTtEta(int iTtEta0) const { return (iTtEta0 < 28) ? (iTtEta0 - 28) : (iTtEta0 - 27); }

  int cIndex2TtPhi(int iTtPhi0) const { return iTtPhi0 + 1; }

  int iTtPhi2cIndex(int iTtPhi) const { return iTtPhi - 1; }

  /*
    int iXY2cIndex(int iX) const{
    return iX-1;
  }
  */

  void elec2GeomNum(int ittEta0, int ittPhi0, int strip1, int ch1, int &iEta0, int &iPhi0) const;

  /* Set horizontal parity of a 16-bit word of FE data
   * @param a the word whose parity must be set.
   */
  void setHParity(uint16_t &a) const;

  void genFeData(std::string &basename, int iEvent, const std::vector<uint16_t> adcCount[nEbEta][nEbPhi]) const;

  void genTccIn(std::string &basename, int iEvent, const int tps[nTtEta][nTtPhi]) const;

  void genTccOut(std::string &basename, int iEvent, const int tps[nTtEta][nTtPhi]) const;

  void getEbDigi(const edm::Event &event, std::vector<uint16_t> adc[nEbEta][nEbPhi]) const;

  void getTp(const edm::Event &event, tokenType type, int tp[nTtEta][nTtPhi]) const;

  std::string getExt() const;

  void fwrite(std::ofstream &f, uint16_t data, int &iword, bool hpar = true) const;

  void getSrfs(const edm::Event &event, int ebSrf[nTtEta][nTtPhi], int eeSrf[nEndcaps][nScX][nScY]) const;

  void genSrData(std::string &basename, int iEvent, int ttf[nEbTtEta][nTtPhi]) const;

private:
  std::string digiProducer_;

  std::string ebDigiCollection_;

  std::string eeDigiCollection_;

  std::string srDigiProducer_;

  std::string ebSrFlagCollection_;

  std::string eeSrFlagCollection_;

  std::string tpDigiCollection_;

  std::string tcpDigiCollection_;

  const CaloGeometry *theGeometry;

  std::string tpProducer_;

  writeMode_t writeMode_;

  bool xtalVerbose_;

  bool tpVerbose_;

  bool srp2dcc_;

  bool tcc2dcc_;

  bool fe2dcc_;

  bool fe2tcc_;

  const EcalTrigTowerConstituentsMap *theTriggerTowerMap;

  std::unique_ptr<EcalSelectiveReadout> esr_;

  std::ofstream ttfFile;

  std::ofstream srfFile;

  int tccNum_;

  int dccNum_;

  int tccInDefaultVal_;

  std::string basename_;

  edm::EDGetTokenT<EESrFlagCollection> eeSrFlagToken_;
  edm::EDGetTokenT<EBSrFlagCollection> ebSrFlagToken_;
  edm::EDGetTokenT<EBDigiCollection> ebDigisToken_;
  edm::EDGetTokenT<EcalTrigPrimDigiCollection> trigPrimDigisToken_[2];
};