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/bklm/BKLMSimHit.h>
#include <klm/dataobjects/eklm/EKLMSimHit.h>
#include <klm/dataobjects/KLMDigit.h>
#include <klm/dataobjects/KLMElementNumbers.h>
#include <klm/dbobjects/KLMChannelStatus.h>
#include <klm/dbobjects/KLMScintillatorDigitizationParameters.h>
#include <klm/dbobjects/KLMScintillatorFEEParameters.h>
#include <klm/dbobjects/KLMStripEfficiency.h>
#include <klm/simulation/ScintillatorFirmware.h>
#include <klm/time/KLMTime.h>
 
/* Belle 2 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 digitizeBKLM();
 
    void digitizeEKLM();
 
    bool checkActive(KLMChannelNumber channel);
 
    bool efficiencyCorrection(float efficiency);
 
    DBObjPtr<KLMChannelStatus> m_ChannelStatus;
 
    DBObjPtr<KLMScintillatorDigitizationParameters> m_DigPar;
 
    DBObjPtr<KLMScintillatorFEEParameters> m_FEEPar;
 
    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_Debug;
 
    std::multimap<KLMChannelNumber, const BKLMSimHit*> m_bklmSimHitChannelMap;
 
    std::multimap<KLMPlaneNumber, const BKLMSimHit*> m_bklmSimHitPlaneMap;
 
    std::multimap<KLMChannelNumber, const EKLMSimHit*> m_eklmSimHitChannelMap;
 
    std::multimap<KLMPlaneNumber, const EKLMSimHit*> m_eklmSimHitPlaneMap;
 
    KLM::ScintillatorFirmware* m_Fitter;
 
    StoreArray<BKLMSimHit> m_bklmSimHits;
 
    StoreArray<EKLMSimHit> m_eklmSimHits;
 
    StoreArray<KLMDigit> m_Digits;
 
    StoreArray<KLMScintillatorFirmwareFitResult> m_FPGAFits;
 
  };
 
}