KLMDigitizerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Timofey Uglov, Kirill Chilikin                           *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/dataobjects/bklm/BKLMSimHit.h>
#include <klm/dataobjects/eklm/EKLMSimHit.h>
#include <klm/dataobjects/eklm/EKLMElementNumbers.h>
#include <klm/dataobjects/KLMDigit.h>
#include <klm/dataobjects/KLMElementNumbers.h>
#include <klm/dbobjects/eklm/EKLMChannels.h>
#include <klm/dbobjects/KLMChannelStatus.h>
#include <klm/dbobjects/KLMScintillatorDigitizationParameters.h>
#include <klm/dbobjects/KLMStripEfficiency.h>
#include <klm/dbobjects/KLMTimeConversion.h>
#include <klm/simulation/ScintillatorFirmware.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();
 
    virtual ~KLMDigitizerModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    enum EfficiencyMode {
 
      c_Strip,
 
      c_Plane,
 
    };
 
    void checkChannelParameters();
 
    void digitizeBKLM();
 
    void digitizeEKLM();
 
    bool checkActive(uint16_t channel);
 
    bool efficiencyCorrection(float efficiency);
 
    DBObjPtr<KLMScintillatorDigitizationParameters> m_DigPar;
 
    DBObjPtr<KLMTimeConversion> m_TimeConversion;
 
    DBObjPtr<EKLMChannels> m_Channels;
 
    DBObjPtr<KLMChannelStatus> m_ChannelStatus;
 
    DBObjPtr<KLMStripEfficiency> m_StripEfficiency;
 
    const KLMElementNumbers* m_ElementNumbers;
 
    const EKLMElementNumbers* m_eklmElementNumbers;
 
    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<uint16_t, const BKLMSimHit*> m_bklmSimHitChannelMap;
 
    std::multimap<uint16_t, const BKLMSimHit*> m_bklmSimHitPlaneMap;
 
    std::multimap<uint16_t, const EKLMSimHit*> m_eklmSimHitChannelMap;
 
    std::multimap<uint16_t, const EKLMSimHit*> m_eklmSimHitPlaneMap;
 
    KLM::ScintillatorFirmware* m_Fitter;
 
    StoreArray<BKLMSimHit> m_bklmSimHits;
 
    StoreArray<EKLMSimHit> m_eklmSimHits;
 
    StoreArray<KLMDigit> m_Digits;
 
    StoreArray<KLMScintillatorFirmwareFitResult> m_FPGAFits;
 
  };
 
}