PXDHitRateCounter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric, Benjamin Schwenker                         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <background/modules/BeamBkgHitRateMonitor/HitRateBase.h>
#include <framework/datastore/StoreArray.h>
#include <pxd/dataobjects/PXDDigit.h>
#include <pxd/dataobjects/PXDCluster.h>
#include <vxd/dataobjects/VxdID.h>
#include <pxd/geometry/SensorInfo.h>
#include <vxd/geometry/GeoCache.h>
#include <TTree.h>
#include <map>
 
#include <framework/gearbox/Unit.h>
 
namespace Belle2 {
  namespace Background {
 
    class PXDHitRateCounter: public HitRateBase {
 
    public:
 
      struct TreeStruct {
        float meanOccupancies[40] = {0}; 
        float maxOccupancies[40] = {0}; 
        float doseRates[40] = {0}; 
        float softPhotonFluxes[40] = {0}; 
        float hardPhotonFluxes[40] = {0}; 
        float chargedFluxes[40] = {0}; 
        float segmentDoseRates[240] = {0};  
        float segmentSoftPhotonFluxes[240] = {0}; 
        float segmentHardPhotonFluxes[240] = {0}; 
        float segmentChargedFluxes[240] = {0}; 
        float averageRate = 0; 
        int numEvents = 0; 
        bool valid = false;  
      };
 
      PXDHitRateCounter()
      {}
 
      virtual void initialize(TTree* tree) override;
 
      virtual void clear() override;
 
      virtual void accumulate(unsigned timeStamp) override;
 
      virtual void normalize(unsigned timeStamp) override;
 
    private:
 
      inline const PXD::SensorInfo& getInfo(VxdID sensorID) const;
 
      void setActivePixels();
 
      // class parameters: to be set via constructor or setters
      double m_integrationTime = 20 * Unit::us; 
      bool m_maskDeadPixels = true; 
      int m_nPXDSensors = 0; 
      const double c_densitySi = 2.3290 * Unit::g_cm3; 
      // tree structure
      TreeStruct m_rates; 
      // buffer
      std::map<unsigned, TreeStruct> m_buffer; 
      // collections
      StoreArray<PXDDigit> m_digits;  
      StoreArray<PXDCluster> m_clusters;  
      // other
      double m_activePixels[40] = {0}; 
      double m_activeAreas[40] = {0}; 
      double m_segmentActivePixels[240] = {0}; 
      double m_segmentActiveAreas[240] = {0}; 
    };
 
    inline const PXD::SensorInfo& PXDHitRateCounter::getInfo(VxdID sensorID) const
    {
      return dynamic_cast<const PXD::SensorInfo&>(VXD::GeoCache::get(sensorID));
    }
 
  } // Background namespace
} // Belle2 namespace