GRLNeuroTrainerModule.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.                  *
 **************************************************************************/
 
#ifndef GRLNEUROTRAINERMODULE_H
#define GRLNEUROTRAINERMODULE_H
 
#include <framework/core/Module.h>
 
#include <trg/grl/GRLNeuro.h>
#include <trg/grl/dataobjects/GRLMLPData.h>
 
#include <TH1D.h>
 
namespace Belle2 {
  class GRLNeuroTrainerModule : public Module {
  public:
    GRLNeuroTrainerModule();
 
    virtual ~GRLNeuroTrainerModule() {}
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
    void updateRelevantID(unsigned isector);
 
    void train(unsigned isector);
 
    void saveTraindata(const std::string& filename, const std::string& arrayname = "trainSets");
 
    bool loadTraindata(const std::string& filename, const std::string& arrayname = "trainSets");
 
  protected:
    //module parameters
    std::string m_TrgECLClusterName;
    std::string m_2DfinderCollectionName;
    std::string m_GRLCollectionName;
    std::string m_filename;
    std::string m_trainFilename;
    std::string m_logFilename;
    std::string m_arrayname;
    std::string m_trainArrayname;
    bool m_saveDebug;
    bool m_load;
    GRLNeuro::Parameters m_parameters;
    double m_nTrainMin;
    double m_nTrainMax;
    bool m_multiplyNTrain;
    int m_nValid;
    int m_nTest;
    double m_wMax;
    int m_nThreads;
    int m_checkInterval;
    int m_maxEpochs;
    int m_repeatTrain;
 
    GRLNeuro m_GRLNeuro;
    std::vector<GRLMLPData> m_trainSets;
 
    //..ECL look up tables
    std::vector<int> TCThetaID;
    std::vector<float> TCPhiLab;
    std::vector<float> TCcotThetaLab;
    std::vector<float> TCPhiCOM;
    std::vector<float> TCThetaCOM;
    std::vector<float> TC1GeV;
    double radtodeg = 0;
 
    int n_cdc_sector = 0;
    int n_ecl_sector = 0;
    int n_sector = 0;
 
    std::vector<TH1D*> h_cdc2d_phi_sig;
    std::vector<TH1D*> h_cdc2d_pt_sig;
    std::vector<TH1D*> h_selE_sig;
    std::vector<TH1D*> h_selPhi_sig;
    std::vector<TH1D*> h_selTheta_sig;
    std::vector<TH1D*> h_result_sig;
    std::vector<TH1D*> h_cdc2d_phi_bg;
    std::vector<TH1D*> h_cdc2d_pt_bg;
    std::vector<TH1D*> h_selE_bg;
    std::vector<TH1D*> h_selPhi_bg;
    std::vector<TH1D*> h_selTheta_bg;
    std::vector<TH1D*> h_result_bg;
    std::vector<TH1D*> h_ncdc_sig;
    std::vector<TH1D*> h_ncdcf_sig;
    std::vector<TH1D*> h_ncdcs_sig;
    std::vector<TH1D*> h_ncdci_sig;
    std::vector<TH1D*> h_necl_sig;
    std::vector<TH1D*> h_ncdc_bg;
    std::vector<TH1D*> h_ncdcf_bg;
    std::vector<TH1D*> h_ncdcs_bg;
    std::vector<TH1D*> h_ncdci_bg;
    std::vector<TH1D*> h_necl_bg;
 
    std::vector<int> scale_bg;
 
  };
}
 
#endif