ECLWaveformFit.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
/* ECL headers. */
#include <ecl/dataobjects/ECLDigit.h>
#include <ecl/dataobjects/ECLDsp.h>
#include <ecl/dbobjects/ECLAutoCovariance.h>
#include <ecl/dbobjects/ECLCrystalCalib.h>
#include <ecl/dbobjects/ECLDigitWaveformParameters.h>
#include <ecl/dbobjects/ECLDigitWaveformParametersForMC.h>
 
/* Basf2 headers. */
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
 
class TMinuit;
 
namespace Belle2 {
  struct CovariancePacked {
 
    float m_covMatPacked[31 * (31 + 1) / 2] = {};
 
    float sigma{ -1};
 
    float& operator[](int i) { return m_covMatPacked[i];}
 
    const float& operator[](int i) const { return m_covMatPacked[i];}
  };
 
  struct SignalInterpolation2 {
 
    constexpr static int c_nt = 12;
 
    constexpr static int c_ndt = 5;
 
    constexpr static int c_ntail = 20;
 
    constexpr static double c_dt = 0.5;
 
    constexpr static double c_idt = 1 / c_dt;
 
    constexpr static double c_dtn = c_dt / c_ndt;
 
    constexpr static double c_idtn = c_ndt / c_dt;
 
    double m_FunctionInterpolation[c_nt * c_ndt + c_ntail];
 
    double m_DerivativeInterpolation[c_nt * c_ndt + c_ntail];
 
    double m_r0;
 
    double m_r1;
 
    SignalInterpolation2() {};
 
    explicit SignalInterpolation2(const std::vector<double>&);
 
    void getShape(double t0, double* function, double* derivatives) const;
 
  };
 
  class ECLWaveformFitModule : public Module {
 
  public:
 
    ECLWaveformFitModule();
 
    ~ECLWaveformFitModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    virtual const char* eclDigitArrayName() const
    { return "ECLDigits" ; }
 
    virtual const char* eclDspArrayName() const
    { return "ECLDsps" ; }
 
 
  private:
 
    void loadTemplateParameterArray();
 
    void fitPhotonHadron(
      double& pedestal, double& amplitudePhoton, double& signalTime,
      double& amplitudeHadron, double& chi2);
 
    void fitPhotonHadronBackgroundPhoton(
      double& pedestal, double& amplitudePhoton, double& signalTime,
      double& amplitudeHadron, double& amplitudeBackgroundPhoton,
      double& timeBackgroundPhoton, double& chi2);
 
    double m_EnergyThreshold{0.03};
 
    double m_Chi2Threshold25dof{57.1};
 
    double m_Chi2Threshold27dof{60.0};
 
    bool m_CovarianceMatrix{true};
 
    bool m_TemplatesLoaded{false};
 
    std::vector<double> m_ADCtoEnergy;
 
    TMinuit* m_MinuitPhotonHadron = nullptr;
 
    TMinuit* m_MinuitPhotonHadronBackgroundPhoton = nullptr;
 
    SignalInterpolation2 m_SignalInterpolation[ECLElementNumbers::c_NCrystals][3];
 
    CovariancePacked m_PackedCovariance[ECLElementNumbers::c_NCrystals] = {};
 
    bool m_IsMCFlag{false};
 
    DBObjPtr<ECLCrystalCalib> m_CrystalElectronics{"ECLCrystalElectronics"};
 
    DBObjPtr<ECLCrystalCalib> m_CrystalEnergy{"ECLCrystalEnergy"};
 
    DBObjPtr<ECLDigitWaveformParameters> m_WaveformParameters;
 
    DBObjPtr<ECLDigitWaveformParametersForMC> m_WaveformParametersForMC;
 
    DBObjPtr<ECLAutoCovariance> m_AutoCovariance;
 
    StoreArray<ECLDsp> m_eclDSPs;
 
    StoreArray<ECLDigit> m_eclDigits;
 
  };
 
}