ECLWaveformFit.h

/********************************************************************
 * BASF2 (Belle Analysis Framework 2)                               *
 * Copyright(C) 2018 - Belle II Collaboration                       *
 *                                                                  *
 * Author: The Belle II Collaboration                               *
 * Contributors: Savino Longo (longos@uvic.ca)                      *
 *               Alexei Sibidanov (sibid@uvic.ca)                   *
 *                                                                  *
 * This software is provided "as is" without any warranty.          *
 ********************************************************************/
 
#pragma once
 
// FRAMEWORK
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
 
// ECL
#include <ecl/dataobjects/ECLDigit.h>
#include <ecl/dataobjects/ECLDsp.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 val_der_t {
    double f0;
    double f1;
    double f2;
  };
 
  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;
    std::pair<double, double> m_F[c_nt * c_ndt + c_ntail];
    double m_r0;
    double m_r1;
 
    SignalInterpolation2() {};
    explicit SignalInterpolation2(const std::vector<double>&);
    void getshape(double, val_der_t*) 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:
    StoreArray<ECLDsp> m_eclDSPs;  
    StoreArray<ECLDigit> m_eclDigits;   
    double m_EnergyThreshold{0.03};  
    double m_chi2Threshold25dof{57.1};  
    double m_chi2Threshold27dof{60.0};  
    bool m_TemplatesLoaded{false};  
    void loadTemplateParameterArray();  
    std::vector<double> m_ADCtoEnergy;  
    TMinuit* m_Minit2h{nullptr};   
    TMinuit* m_Minit2h2{nullptr};   
    void Fit2h(double& b, double& a0, double& t0, double& a1, double& chi2);  
    void Fit2hExtraPhoton(double& b, double& a0, double& t0, double& a1, double& A2, double& T2,
                          double& chi2);  
    SignalInterpolation2 m_si[8736][3];  
    CovariancePacked m_c[8736] = {};  
    bool m_CovarianceMatrix{true};  
    bool m_IsMCFlag{false};  
  };
} // end Belle2 namespace