HcalSimpleRecAlgo.h

Go to the documentation of this file.

#ifndef HCALSIMPLERECALGO_H
#define HCALSIMPLERECALGO_H 1

#include <memory>

#include "DataFormats/HcalDigi/interface/QIE10DataFrame.h"
#include "DataFormats/HcalDigi/interface/HFDataFrame.h"
#include "DataFormats/HcalDigi/interface/HODataFrame.h"
#include "DataFormats/HcalDigi/interface/ZDCDataFrame.h"
#include "DataFormats/HcalDigi/interface/HcalCalibDataFrame.h"
#include "DataFormats/HcalRecHit/interface/HFRecHit.h"
#include "DataFormats/HcalRecHit/interface/HORecHit.h"
#include "DataFormats/HcalRecHit/interface/ZDCRecHit.h"
#include "DataFormats/HcalRecHit/interface/HcalCalibRecHit.h"
#include "CalibFormats/HcalObjects/interface/HcalCoder.h"
#include "CalibFormats/HcalObjects/interface/HcalCalibrations.h"
#include "CalibCalorimetry/HcalAlgos/interface/HcalPulseContainmentManager.h"
#include "CondFormats/HcalObjects/interface/AbsOOTPileupCorrection.h"

#include "CalibCalorimetry/HcalAlgos/interface/HcalTimeSlew.h"

class HcalSimpleRecAlgo {
public:
  HcalSimpleRecAlgo(bool correctForTimeslew, bool correctForContainment, float fixedPhaseNs);

  void beginRun(edm::EventSetup const& es);
  void endRun();

  void initPulseCorr(int toadd);

  // set RecoParams channel-by-channel.
  void setRecoParams(
      bool correctForTimeslew, bool correctForPulse, bool setLeakCorrection, int pileupCleaningID, float phaseNS);

  // usage of leak correction
  void setLeakCorrection();

  // set OOT pileup corrections
  void setHFPileupCorrection(std::shared_ptr<AbsOOTPileupCorrection> corr);
  void setHOPileupCorrection(std::shared_ptr<AbsOOTPileupCorrection> corr);

  // Set bunch crossing information.
  // This object will not manage the pointer.
  void setBXInfo(const BunchXParameter* info, unsigned lenInfo);

  HFRecHit reconstruct(
      const HFDataFrame& digi, int first, int toadd, const HcalCoder& coder, const HcalCalibrations& calibs) const;
  HFRecHit reconstructQIE10(
      const QIE10DataFrame& digi, int first, int toadd, const HcalCoder& coder, const HcalCalibrations& calibs) const;

  HORecHit reconstruct(
      const HODataFrame& digi, int first, int toadd, const HcalCoder& coder, const HcalCalibrations& calibs) const;
  HcalCalibRecHit reconstruct(const HcalCalibDataFrame& digi,
                              int first,
                              int toadd,
                              const HcalCoder& coder,
                              const HcalCalibrations& calibs) const;

private:
  bool correctForTimeslew_;
  bool correctForPulse_;
  float phaseNS_;
  std::unique_ptr<HcalPulseContainmentManager> pulseCorr_;
  int runnum_;  // data run numer
  bool setLeakCorrection_;
  int pileupCleaningID_;
  const BunchXParameter* bunchCrossingInfo_;
  unsigned lenBunchCrossingInfo_;
  std::shared_ptr<AbsOOTPileupCorrection> hbhePileupCorr_;
  std::shared_ptr<AbsOOTPileupCorrection> hfPileupCorr_;
  std::shared_ptr<AbsOOTPileupCorrection> hoPileupCorr_;

  HcalPulseShapes theHcalPulseShapes_;

  int puCorrMethod_;

  const HcalTimeSlew* hcalTimeSlew_delay_;
};

#endif