ECLDigitizerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Poyuan Chen                                              *
 *               Guglielmo De Nardo (denardo@na.infn.it)                  *
 *               Alex Bobrov                                              *
 *               Alexei Sibidanov                                         *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
//STL
#include <vector>
 
//Framework
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <ecl/dbobjects/ECLDigitWaveformParametersForMC.h>
#include <framework/database/DBObjPtr.h>
 
//ECL
#include <ecl/digitization/EclConfiguration.h>
 
namespace Belle2 {
  class ECLWaveformData;
  class ECLNoiseData;
  class ECLWFAlgoParams;
  class ECLHit;
  class ECLSimHit;
  class ECLDigit;
  class ECLDsp;
  class ECLDspWithExtraMCInfo;
  class ECLTrig;
  class ECLWaveforms;
 
  class ECLDigitizerModule : public Module {
  public:
 
    ECLDigitizerModule();
 
    ~ECLDigitizerModule();
 
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
    using algoparams_t   = ECL::EclConfiguration::algoparams_t;   
    using fitparams_t    = ECL::EclConfiguration::fitparams_t;    
    using signalsample_t = ECL::EclConfiguration::signalsample_t; 
    using adccounts_t    = ECL::EclConfiguration::adccounts_t;    
    using int_array_192x16_t = fitparams_t::int_array_192x16_t; 
    using int_array_24x16_t  =
      fitparams_t::int_array_24x16_t;  
    using uint_pair_t        = std::pair<unsigned int, unsigned int>; 
    struct crystallinks_t {
      short unsigned int idn;
      short unsigned int inoise;
      short unsigned int ifunc;
      short unsigned int iss;
    };
 
    std::vector<crystallinks_t> m_tbl;
 
    std::vector<algoparams_t> m_idn; 
    std::vector<fitparams_t> m_fitparams; 
    std::vector<ECLNoiseData> m_noise; 
    std::vector<signalsample_t> m_ss; 
    std::vector<signalsample_t> m_ss_HadronShapeSimulations; 
    std::vector<adccounts_t> m_adc;  
    struct calibration_t {
      float ascale; 
      float tshift; 
    };
    std::vector<calibration_t> m_calib;
 
    std::vector<double> m_Awave;
 
    void shapeFitterWrapper(const int j, const int* FitA, const int m_ttrig,
                            int& m_lar, int& m_ltr, int& m_lq, int& m_chi) const ;
 
    DBObjPtr<ECLDigitWaveformParametersForMC> m_waveformParametersMC;
 
    void callbackHadronSignalShapes();
 
    void readDSPDB();
    int shapeSignals();
    void makeWaveforms();
    void repack(const ECLWFAlgoParams&, algoparams_t&);
    void getfitparams(const ECLWaveformData&, const ECLWFAlgoParams&, fitparams_t&);
    void makeElectronicNoiseAndPedestal(int j, int* FitA);
 
    StoreArray<ECLHit>    m_eclHits;  
    StoreArray<ECLHit>    m_eclDiodeHits; 
    StoreArray<ECLSimHit> m_eclSimHits; 
    StoreObjPtr<ECLWaveforms> m_eclWaveforms; 
    bool m_HadronPulseShape; 
    StoreArray<ECLDigit>  m_eclDigits;
    StoreArray<ECLDsp>    m_eclDsps;
    StoreArray<ECLDspWithExtraMCInfo>    m_eclDspsWithExtraMCInfo;
    StoreArray<ECLTrig>   m_eclTrigs;
    bool m_background;  
    bool m_calibration;  
    bool m_inter; 
    bool m_waveformMaker; 
    bool m_storeDspWithExtraMCInfo;  
    unsigned int m_compAlgo; 
    int m_ADCThreshold; 
    double m_WaveformThresholdOverride; 
    double m_DspWithExtraMCInfoThreshold;  
    std::string m_eclWaveformsName;   
    bool m_dspDataTest; 
  };
}//Belle2