RPDDataAnalyzer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ZDCANALYSIS_RPDDataAnalyzer_h
#define ZDCANALYSIS_RPDDataAnalyzer_h
 
#include <string>
#include <vector>
#include <functional>
#include <bitset>
 
#include "ZdcAnalysis/ZDCMsg.h"
 
#include "CxxUtils/checker_macros.h"
ATLAS_NO_CHECK_FILE_THREAD_SAFETY;
 
struct RPDConfig {
  unsigned int nRows;
  unsigned int nColumns;
  unsigned int nSamples;
  unsigned int nBaselineSamples;
  unsigned int endSignalSample;
  float pulse2ndDerivThresh;
  float postPulseFracThresh;
  unsigned int goodPulseSampleStart;
  unsigned int goodPulseSampleStop;
  unsigned int nominalBaseline;
  float pileupBaselineSumThresh;
  float pileupBaselineStdDevThresh;
  unsigned int nNegativesAllowed;
  unsigned int AdcOverflow;
};
 
class RPDDataAnalyzer
{
 public:
  enum {
    ValidBit                        =  0, // analysis and output are valid
    OutOfTimePileupBit              =  1, // OOT detected, pileup subtraction attempted
    OverflowBit                     =  2, // overflow detected => invalid
    PrePulseBit                     =  3, // pulse detected before expected range => invalid
    PostPulseBit                    =  4, // pulse detected after expected range => invalid
    NoPulseBit                      =  5, // no pulse detected => invalid
    BadAvgBaselineSubtrBit          =  6, // subtraction of avg. of baseline samples yielded too many negatives => invalid
    InsufficientPileupFitPointsBit  =  7, // baseline samples indicate pileup, but there are not enough points to perform fit -> nominal baseline used without pileup subtraction
    PileupStretchedExpFitFailBit    =  8, // fit to stretched exponential failed -> fallback to exponential fit
    PileupStretchedExpGrowthBit     =  9, // fit to stretched exponential does not decay -> fallback to exponential fit
    PileupBadStretchedExpSubtrBit   = 10, // subtraction of stretched exponential fit yielded too many negatives -> fallback to exponential fit
    PileupExpFitFailBit             = 11, // fit to exponential failed => invalid IF stretched exponential fit is also bad
    PileupExpGrowthBit              = 12, // fit to exponential does not decay => invalid IF stretched exponential fit is also bad
    PileupBadExpSubtrBit            = 13, // subtraction of stretched exponential yielded too many negatives => invalid IF stretched exponential fit is also bad
    PileupStretchedExpPulseLikeBit  = 14, // fit to stretched exponential probably looks more like a pulse than pileup
  };
 
  RPDDataAnalyzer(ZDCMsg::MessageFunctionPtr messageFunc_p, const std::string& tag, const RPDConfig& config, std::vector<float> const& calibFactors);
 
  virtual ~RPDDataAnalyzer() = default;
 
  void loadChannelData(unsigned int channel, const std::vector<uint16_t>& FadcData);
  void analyzeData();
 
  unsigned int getChMaxSample(unsigned int channel) const;
  float getChSumAdc(unsigned int channel) const;
  float getChSumAdcCalib(unsigned int channel) const;
  float getChMaxAdc(unsigned int channel) const;
  float getChMaxAdcCalib(unsigned int channel) const;
  float getChPileupFrac(unsigned int channel) const;
  float getChBaseline(unsigned int channel) const;
  const std::vector<float>& getChPileupExpFitParams(unsigned int channel) const;
  const std::vector<float>& getChPileupStretchedExpFitParams(unsigned int channel) const;
  const std::vector<float>& getChPileupExpFitParamErrs(unsigned int channel) const;
  const std::vector<float>& getChPileupStretchedExpFitParamErrs(unsigned int channel) const;
  float getChPileupExpFitMSE(unsigned int channel) const;
  float getChPileupStretchedExpFitMSE(unsigned int channel) const;
 
  unsigned int getChStatus(unsigned int channel) const;
  unsigned int getSideStatus() const;
 
  void reset();
 
 private:
  bool checkOverflow(unsigned int channel);
  bool checkPulses(unsigned int channel);
  unsigned int countSignalRangeNegatives(std::vector<float> const& values) const;
  bool doBaselinePileupSubtraction(unsigned int channel);
  void calculateMaxSampleMaxAdc(unsigned int channel);
  void calculateSumAdc(unsigned int channel);
 
  void setSideStatusBits();
 
  bool doPileupExpFit(unsigned int channel, std::vector<std::pair<unsigned int, float>> const& pileupFitPoints);
  bool doPileupStretchedExpFit(unsigned int channel, std::vector<std::pair<unsigned int, float>> const& pileupFitPoints);
  float calculateBaselineSamplesMSE(unsigned int channel, std::function<float(unsigned int)> const& fit) const;
 
  ZDCMsg::MessageFunctionPtr m_msgFunc_p;
  std::string m_tag;
  unsigned int m_nRows;
  unsigned int m_nColumns;
  unsigned int m_nChannels;
  unsigned int m_nChannelsLoaded = 0;
  unsigned int m_nSamples;
  unsigned int m_nBaselineSamples; 
  unsigned int m_endSignalSample; 
  float m_pulse2ndDerivThresh; 
  float m_postPulseFracThresh; 
  unsigned int m_goodPulseSampleStart; 
  unsigned int m_goodPulseSampleStop; 
  unsigned int m_nominalBaseline; 
  float m_pileupBaselineSumThresh; 
  float m_pileupBaselineStdDevThresh; 
  unsigned int m_nNegativesAllowed; 
  unsigned int m_AdcOverflow; 
  std::vector<float> m_outputCalibFactors; 
  std::vector<std::vector<float>> m_chFadcData; 
  std::vector<std::vector<float>> m_chCorrectedFadcData; 
  std::vector<float> m_chMaxSample; 
  std::vector<float> m_chSumAdc; 
  std::vector<float> m_chSumAdcCalib; 
  std::vector<float> m_chMaxAdc; 
  std::vector<float> m_chMaxAdcCalib; 
  std::vector<float> m_chPileupFrac; 
  std::vector<float> m_chBaseline; 
  std::vector<std::vector<float>> m_chPileupExpFitParams; 
  std::vector<std::vector<float>> m_chPileupStretchedExpFitParams; 
  std::vector<std::vector<float>> m_chPileupExpFitParamErrs; 
  std::vector<std::vector<float>> m_chPileupStretchedExpFitParamErrs; 
  std::vector<bool> m_chPileupFuncType; 
  std::vector<std::function<float(unsigned int)>> m_chExpPileupFuncs; 
  std::vector<std::function<float(unsigned int)>> m_ch2ndOrderStretchedExpPileupFuncs; 
  std::vector<float> m_chExpPileupMSE; 
  std::vector<float> m_ch2ndOrderStretchedExpPileupMSE; 
  std::vector<std::bitset<32>> m_chStatus; 
  std::bitset<32> m_sideStatus; 
  static unsigned int constexpr s_minPileupFitPoints = 3;
};
#endif