PXDClusterChargeCollectorModule.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
 
#include <calibration/CalibrationCollectorModule.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
 
#include <pxd/dbobjects/PXDClusterChargeMapPar.h>
#include <pxd/dbobjects/PXDGainMapPar.h>
 
#include <string>
 
namespace Belle2 {
  class PXDCluster;
  class MCParticle;
  class Track;
  class RecoTrack;
  class EventMetaData;

  class PXDClusterChargeCollectorModule : public CalibrationCollectorModule {
 
  public:

    PXDClusterChargeCollectorModule();
    void prepare() override final;
    void collect() override final;
    void startRun() override final;
 
  private:
    StoreArray<PXDCluster> m_pxdClusters;
    StoreArray<MCParticle> m_mcParticles;
    StoreArray<Track> m_tracks;
    StoreArray<RecoTrack> m_recoTracks;
    StoreObjPtr<EventMetaData> m_evtMetaData;

    std::string m_storeClustersName;
    std::string m_storeTracksName;
    std::string m_storeRecoTracksName;
    std::string m_storeMCParticlesName;

    int m_minClusterCharge;
    int m_minClusterSize;
    int m_maxClusterSize;
    int m_nBinsU;
    int m_nBinsV;
    std::string m_chargeName;
    std::string m_gainName;
    bool m_mcSamples;
    bool m_relationCheck;
    bool m_fillChargeHistogram;
    int m_matchTrack;

    int m_signal;
    int m_run;
    int m_exp;
    PXDClusterChargeMapPar m_chargeMap;
    PXDGainMapPar m_gainMap;

    DBObjPtr<PXDClusterChargeMapPar> m_DBChargeMapPar;
    DBObjPtr<PXDGainMapPar> m_DBGainMapPar;
  };

}