SVDROIDQMModule.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 <framework/core/HistoModule.h>
#include <framework/datastore/StoreArray.h>
 
#include <vxd/geometry/GeoCache.h>
 
#include <TH1.h>
#include <TH1F.h>
 
#include <unordered_map>
#include <utility>
#include <functional>
 
namespace Belle2 {
  class ROIid;
  class SVDIntercept;
  class SVDShaperDigit;
  class SVDRecoDigit;
  class SVDCluster;

 
  class SVDROIDQMModule : public HistoModule {
 
  public:

    SVDROIDQMModule();
 
    void initialize() override; 
 
    void event() override;
 
    void endRun() override; 
 
  private:
 
    std::string m_SVDShaperDigitsName; 
    std::string m_SVDRecoDigitsName; 
    std::string m_SVDClustersName; 
 
    StoreArray<ROIid> m_ROIs; 
    StoreArray<SVDIntercept> m_Intercepts; 
    StoreArray<SVDShaperDigit> m_SVDShaperDigits; 
    StoreArray<SVDRecoDigit> m_SVDRecoDigits; 
    StoreArray<SVDCluster> m_SVDClusters; 
 
    VXD::GeoCache& m_geoCache = VXD::GeoCache::getInstance(); 
 
    std::string m_ROIsName; 
    std::string m_InterceptsName; 
 
    TDirectory* m_InterDir; 
    TDirectory* m_ROIDir; 

    typedef std::pair< TH1*, std::function< void(TH1*, const SVDIntercept*) > > InterHistoAndFill;
    std::unordered_multimap<Belle2::VxdID, InterHistoAndFill, std::function<size_t (const Belle2::VxdID&)> > hInterDictionary;

    typedef std::pair< TH1*, std::function< void(TH1*, const ROIid*) > > ROIHistoAndFill;
    std::unordered_multimap<Belle2::VxdID, ROIHistoAndFill, std::function<size_t (const Belle2::VxdID&)> > hROIDictionary;

    struct ROIHistoAccumulateAndFill {
      TH1* hPtr;  
      std::function< void (const ROIid*, double&) > accumulate;   
      std::function< void (TH1*, double&) > fill;   
      double value; 
    };

    std::unordered_multimap<Belle2::VxdID, ROIHistoAccumulateAndFill&, std::function<size_t (const Belle2::VxdID&) > >
    hROIDictionaryEvt;
 
    void createHistosDictionaries(); 
    void fillSensorROIHistos(const ROIid* roi); 
    void fillSensorInterHistos(const SVDIntercept* inter); 
 
    int m_numModules; 
    int m_specificLayer =
      -1; 
    bool m_plotRecoDigits = false; 
 
    TH1F* hnROIs; 
    TH1F* hnInter; 
    TH1F* harea; 
    TH1F* hredFactor; 
    TH1F* hCellU; 
    TH1F* hCellV; 
    int n_events; 
 
    void defineHisto() override; 
 
  };//end class declaration
 

} // end namespace Belle2