SVDDQMDoseModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2021 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Ludovico Massaccesi                                      *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/HistoModule.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <TH1.h>
#include <TH2.h>
#include <string>
#include <vector>
#include <functional>
 
namespace Belle2 {
  // Forward declarations to avoid the need of #includes here in the header
  class RawFTSW;
  class SVDShaperDigit;
  class TRGSummary;
  class VxdID;
  // class SVDFADCMaskedStrips;
 
  namespace SVD {
    class SVDDQMDoseModule : public HistoModule {
    public:
      enum EEventType {
        c_HERInjection = 1,
        c_LERInjection = 2,
        c_NoInjection = 4
      };
 
      SVDDQMDoseModule(); 
      void initialize() override final; 
      void beginRun() override final; 
      void event() override final; 
      void defineHisto() override final; 
    private:
      typedef struct SensorGroup {
        const char* nameSuffix;
        const char* titleSuffix;
        int nBins;
        double xMin;
        double xMax;
        std::function<bool(const VxdID&)> contains;
        mutable int nStripsU = 0;
      } SensorGroup;
 
      // Steerable data members (parameters)
      unsigned int m_eventFilter; 
      std::string m_histogramDirectoryName; 
      std::string m_SVDShaperDigitsName; 
      double m_noInjectionTime; 
      std::vector<int> m_trgTypes; 
      // Inputs
      StoreArray<RawFTSW> m_rawTTD; 
      StoreArray<SVDShaperDigit> m_digits; 
      StoreObjPtr<TRGSummary> m_trgSummary; 
      // Outputs (histograms)
      TH2F* h_nEvtsVsTime = nullptr;
      std::vector<TH2F*> m_groupNHitsU;
      std::vector<TH1F*> m_groupOccupanciesU;
      TH1F* h_nEvtsVsTime1 = nullptr;
      std::vector<TH1F*> m_groupNHits1U;
      TH2F* h_bunchNumVsNHits = nullptr;
 
      // Other stuff
      static constexpr double c_revolutionTime = 5120.0 / 508.0;
      static constexpr double c_globalClock = 127.0;
 
      static constexpr double c_defaultOccuMin = 0.0;
      static const int c_defaultNBins = 90;
      static constexpr double c_defaultOccuMax = 100.0 / 1536.0 * c_defaultNBins;
 
      static const std::vector<SensorGroup> c_sensorGroups;
    };
  }
}