TOPBackSplashTimingModule.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/StoreArray.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <mdst/dataobjects/Track.h>
#include <top/dataobjects/TOPBackSplashFitResult.h>
#include <top/dataobjects/TOPDigit.h>
#include <top/dataobjects/TOPRecBunch.h>
 
#include <RooAbsPdf.h>
#include <RooRealVar.h>
#include <RooDataSet.h>
#include <RooFitResult.h>
#include <RooWorkspace.h>
 
namespace Belle2 {
  class TOPBackSplashTimingModule : public Module {
 
  public:
    TOPBackSplashTimingModule();

    void initialize() override;

    void event() override;
 
  private:
    void prepareFitModels();

    static int getModuleFromPhi(double phi);

    TOPBackSplashFitResult* fitTimingDigits(int moduleID, const std::vector<const TOPDigit*>& digitsPerSlot,
                                            double clusterE, double clusterCosTheta, int nTracksPerSlot);

    void makePlot(double cosTheta, double clusterE, int moduleID, RooAbsPdf* model,
                  RooRealVar* x, RooDataSet data, RooFitResult* res, int nTracksPerSlot);
 

    bool m_saveFits = false;

    double m_minClusterE = 0.5;

    int m_minNphotons = 50;

    double m_minClusterNHits = 10.0;

    bool m_includeSlotsWithTracks = false;

    bool m_saveMoreFitParams = false;
 

    StoreArray<ECLCluster> m_eclClusters;

    StoreArray<TOPBackSplashFitResult> m_fitresult;

    StoreArray<TOPDigit> m_digits;

    StoreArray<Track> m_tracks;
 
    StoreObjPtr<TOPRecBunch> m_recBunch; 

    std::array<std::array<double, 11>, 15> m_fitparams;

    std::vector<RooWorkspace> m_wss;
  };

} //end namespace Belle2