GRLNeuroModule.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 GRLNEUROMODULE_H
#define GRLNEUROMODULE_H
 
#include <framework/core/Module.h>
 
#include <trg/grl/GRLNeuro.h>
#include <trg/grl/dataobjects/TRGGRLInfo.h>
#include <trg/grl/dataobjects/GRLMLPData.h>
#include <framework/database/DBObjPtr.h>
#include <trg/grl/dbobjects/TRGGRLConfig.h>
 
#include <TH1D.h>
 
namespace Belle2 {
  class GRLNeuroModule : public Module {
  public:
    GRLNeuroModule();

    virtual ~GRLNeuroModule() {}

    virtual void initialize() override;

    virtual void beginRun() override;

    virtual void event() override;
 
    virtual void terminate() override;

    bool loadTraindata(const std::string& filename, const std::string& arrayname = "trainSets");
 
  protected:
    //module parameters
    GRLNeuro::Parameters m_parameters;
    std::vector<std::string> m_weightFileNames;
    std::vector<std::string> m_biasFileNames;
    std::string m_arrayname;
    std::string m_GRLCollectionName;
    GRLNeuro m_GRLNeuro;
    bool m_use_cdc;
    bool m_use_ecl;
    bool m_use_klm;
    int trainMode;
    std::string m_TrgECLClusterName;
    std::vector<float> m_nNodes;
    std::vector<float> m_nn_thres;
    bool m_saveHist;
    std::string m_HistFileName;
    std::string m_TrgGrlInformationName;

    DBObjPtr<TRGGRLConfig> m_db_trggrlconfig;

    std::vector<int> TCThetaID;
    std::vector<double> TCThetaLab;
    std::vector<double> TCPhiLab;

    std::vector<TH1D*> h_target;

    bool m_useDB;
 
  };

}
 
#endif