SVDClusterQualityEstimatorCalibrationModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: James Webb                                               *
 *                                                                        *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
 
#include <tracking/dataobjects/RecoTrack.h>
 
#include <string>
#include <map>
 
#include <TH2.h>
 
namespace Belle2 {
  class SVDClusterQualityEstimatorCalibrationModule : public Module {
 
  public:
    SVDClusterQualityEstimatorCalibrationModule();
 
 
 
    virtual void initialize() override;
 
 
    virtual void event() override;
 
 
    virtual void terminate() override;
 
 
 
  protected:
 
    void calculateProb(TH2F* signal, TH2F* background, TH2F* probability); 
    void calculateError(TH2F* signal, TH2F* background, TH2F* error); 
    // Data members
    std::string m_svdClustersName; 
    std::string m_recoTracksName; 
    StoreArray<SVDCluster> m_svdClusters; 
    StoreArray<RecoTrack> m_recoTracks; 
    int m_binSizeCharge; 
    int m_binSizeTime; 
    int m_maxClusterSize; 
    bool m_useLegacyNaming; 
    std::string m_outputFileName; 
    std::string
    m_nameOfInstance; 
    //for setting up histograms
 
    std::map<std::string, TH2F* > signalHistMap; 
    std::map<std::string, TH2F*> backgroundHistMap; 
  };
} // end namespace Belle2