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();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
    void endRun() override;
 
    void terminate() override;
 
  private:
 
    enum EfficiencyMode {
 
      c_Strip,
 
      c_Plane,
 
    };
 
    void checkScintillatorFEEParameters();
 
    void digitizeRPC();
 
    void digitizeScintillator();
 
    void digitizeAsic();
 
    bool checkActive(KLMChannelNumber channel);
 
    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;
 
    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;
 
  };
 
}