BeamBkgHitRateMonitorModule.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/Module.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/dataobjects/EventMetaData.h>
#include <framework/dataobjects/FileMetaData.h>
#include <mdst/dataobjects/TRGSummary.h>
#include <background/modules/BeamBkgHitRateMonitor/HitRateBase.h>
 
#include <string>
#include <vector>
#include <map>
#include <TFile.h>
#include <TTree.h>
#include <TH1F.h>
 
namespace Belle2 {
  class BeamBkgHitRateMonitorModule : public Module {
 
  public:
 
    BeamBkgHitRateMonitorModule();
 
    virtual ~BeamBkgHitRateMonitorModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    bool isEventSelected();
 
    void collectFileMetaData();
 
    void setFileMetaData();
 
    // steering parameters
    std::string m_outputFileName; 
    std::string m_treeName;       
    std::vector<int> m_trgTypes;  
    bool m_writeEmptyTimeStamps;  
    double m_topTimeOffset; 
    double m_topTimeWindow; 
    std::string m_svdShaperDigitsName; 
    double m_svdThrCharge; 
    bool m_svdIgnoreHotStripsPayload; 
    bool m_svdIgnoreMaskedStripsPayload; 
    std::map<std::string, std::string> m_additionalDataDescription; 
    int m_cdcTimeWindowLowerEdgeSmallCell;  
    int m_cdcTimeWindowUpperEdgeSmallCell;  
    int m_cdcTimeWindowLowerEdgeNormalCell; 
    int m_cdcTimeWindowUpperEdgeNormalCell; 
    bool m_cdcEnableBadWireTreatment;       
    bool m_cdcEnableBackgroundHitFilter;    
    bool m_cdcEnableMarkBackgroundHit;      
    // collections
    StoreObjPtr<EventMetaData> m_eventMetaData; 
    StoreObjPtr<TRGSummary> m_trgSummary;  
    StoreObjPtr<FileMetaData> m_fileMetaData{"", DataStore::c_Persistent}; 
    // output file and tree
    TFile* m_file = 0; 
    TTree* m_tree = 0; 
    // tree variables
    int m_run = 0;  
    int m_numEvents = 0;  
    unsigned m_timeStamp = 0;  
    int m_time = 0;  
    // control histograms
    TH1F* m_trgAll = 0; 
    TH1F* m_trgSel = 0; 
    // buffers
    std::map<unsigned, int> m_eventCounts; 
    // rate monitoring classes: these provide additional buffers and tree branches
    std::vector<Background::HitRateBase*> m_monitors; 
    // other
    unsigned m_numEventsSelected = 0;  
    std::map<TRGSummary::ETimingType, unsigned> m_trgTypesCount; 
    unsigned m_utimeFirst = 0; 
    unsigned m_utimeMin = 0; 
    unsigned m_utimeMax = 0; 
    // output file meta data
    TTree* m_persistent = 0; 
    FileMetaData m_outputFileMetaData; 
    std::vector<std::string> m_parentLfns; 
    unsigned long m_experimentLow = 1;  
    unsigned long m_runLow = 0;         
    unsigned long m_eventLow = 0;       
    unsigned long m_experimentHigh = 0; 
    unsigned long m_runHigh = 0;        
    unsigned long m_eventHigh = 0;      
    unsigned m_allEventsSelected = 0;  
  };
 
} // Belle2 namespace