KLMDigitizerModule.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
 
/* KLM headers. */
#include <klm/dataobjects/KLMDigit.h>
#include <klm/dataobjects/KLMElementNumbers.h>
#include <klm/dataobjects/KLMSimHit.h>
#include <klm/dbobjects/KLMChannelStatus.h>
#include <klm/dbobjects/KLMElectronicsMap.h>
#include <klm/dbobjects/KLMScintillatorDigitizationParameters.h>
#include <klm/dbobjects/KLMScintillatorFEEParameters.h>
#include <klm/dbobjects/KLMScintillatorFirmware.h>
#include <klm/dbobjects/KLMStripEfficiency.h>
#include <klm/rawdata/RawData.h>
#include <klm/simulation/ScintillatorFirmware.h>
#include <klm/time/KLMTime.h>
 
/* Basf2 headers. */
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
 
namespace Belle2 {

  class KLMDigitizerModule : public Module {
 
  public:

    KLMDigitizerModule();

    ~KLMDigitizerModule() override;

    void initialize() override;

    void beginRun() override;

    void event() override;

    void endRun() override;
 
  private:

    enum EfficiencyMode {

      c_Strip,

      c_Plane,
 
    };

    void checkScintillatorFEEParameters();

    void digitizeRPC();

    void digitizeScintillator();

    void digitizeAsic();

    bool checkActive(KLMChannelNumber channel);

    static bool efficiencyCorrection(float efficiency);

    DBObjPtr<KLMChannelStatus> m_ChannelStatus;

    DBObjPtr<KLMElectronicsMap> m_ElectronicsMap;

    DBObjPtr<KLMScintillatorDigitizationParameters> m_DigPar;

    DBObjPtr<KLMScintillatorFEEParameters> m_FEEPar;

    DBObjPtr<KLMScintillatorFirmware> m_ScintillatorFirmware;

    DBObjPtr<KLMStripEfficiency> m_StripEfficiency;

    KLMTime* m_Time;

    const KLMElementNumbers* m_ElementNumbers;

    std::string m_SimulationMode;

    bool m_ChannelSpecificSimulation;

    int m_DigitizationInitialTime;

    bool m_SaveFPGAFit;

    std::string m_Efficiency;

    EfficiencyMode m_EfficiencyMode;

    bool m_CreateMultiStripDigits;

    bool m_CreateMultiStripDigitsByRun = false;

    bool m_Debug;

    std::multimap<KLMPlaneNumber, const KLMSimHit*> m_MapPlaneSimHit;

    std::multimap<KLMChannelNumber, const KLMSimHit*> m_MapChannelSimHit;

    std::multimap<KLMElectronicsChannel, const KLMSimHit*> m_MapAsicSimHit;

    KLMDigit* m_AsicDigits[KLM::c_NChannelsAsic] {};

    std::multimap<KLMChannelNumber, const KLMSimHit*>::iterator
    m_AsicDigitSimHitsLowerBound[KLM::c_NChannelsAsic] {};

    std::multimap<KLMChannelNumber, const KLMSimHit*>::iterator
    m_AsicDigitSimHitsUpperBound[KLM::c_NChannelsAsic] {};

    KLM::ScintillatorFirmware* m_Fitter;

    StoreArray<KLMSimHit> m_SimHits;

    StoreArray<KLMDigit> m_Digits;

    StoreArray<KLMScintillatorFirmwareFitResult> m_FPGAFits;
 
  };

}