CalibrationCollectorModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Tadeas Bilka (tadeas.bilka@gmail.com)                    *
 *               David Dossett (david.dossett@unimelb.edu.au)             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <memory>
 
#include <TDirectory.h>
#include <TRandom.h>
 
#include <framework/core/HistoModule.h>
#include <framework/dataobjects/EventMetaData.h>
#include <framework/datastore/StoreObjPtr.h>
 
#include <calibration/dataobjects/RunRange.h>
#include <calibration/CalibObjManager.h>
#include <calibration/Utilities.h>
 
namespace Belle2 {
  class CalibrationCollectorModule: public HistoModule {
 
  public:
    CalibrationCollectorModule();
    virtual ~CalibrationCollectorModule() {}
 
    void initialize() final;
    void event() final;
    void beginRun() final;
    void endRun() final;
    void terminate() final;
    void defineHisto() final;
 
    template <class T>
    void registerObject(std::string name, T* obj)
    {
      std::shared_ptr<T> calObj(obj);
      calObj->SetName(name.c_str());
      m_manager.addObject(name, calObj);
    }
 
    template<class T>
    T* getObjectPtr(std::string name)
    {
      return m_manager.getObject<T>(name, m_expRun);
    }
 
  protected:
    virtual void prepare() {}
    virtual void collect() {}
    virtual void startRun() {}
    virtual void closeRun() {}
    virtual void finish() {}
    virtual void inDefineHisto() {}
 
    TDirectory* m_dir;
 
    CalibObjManager m_manager;
 
    RunRange* m_runRange;
 
    Calibration::ExpRun m_expRun;
 
    StoreObjPtr<EventMetaData> m_emd;
 
  private:
    /****** Module Parameters *******/
    std::string m_granularity;
    int m_maxEventsPerRun;
    float m_preScale;
    /********************************/
 
    StoreObjPtr<EventMetaData> m_evtMetaData; 
    bool m_runCollectOnRun = true;
    std::map<Calibration::ExpRun, int> m_expRunEvents;
    int* m_eventsCollectedInRun;
    bool getPreScaleChoice()
    {
      if (m_preScale == 1.) {
        return true;
      } else if (m_preScale == 0.) {
        return false;
      } else {
        const double randomNumber = gRandom->Uniform();
        return randomNumber < m_preScale;
      }
    }
  };
} // Belle2 namespace