NNWaveFitter.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
#ifndef _SVD_RECONSTRUCTION_NNwAVEFITTER_H
#define _SVD_RECONSTRUCTION_NNWAVEFITTER_H
 
#include <cmath>
#include <string>
#include <vector>
#include <map>
#include <memory>
#include <functional>
#include <svd/reconstruction/NNWaveFitTool.h>
#include <svd/simulation/SVDSimulationTools.h>
#include <Eigen/Dense>
 
namespace Belle2 {
  namespace SVD {
 
// ==============================================================================
// The NNWaveFitter class - neural network fitter of APV25 waveforms
// ------------------------------------------------------------------------------
    class NNWaveFitter {
 
    public:
 
 
      typedef std::pair<double, double> nnBoundsType;
 
      NNWaveFitter(std::string xmlData = "");
 
      void setNetwrok(const std::string& xmlData);
 
      std::shared_ptr<nnFitterBinData> getFit(const apvSamples& samples, double tau);
 
      const NNWaveFitTool& getFitTool() const { return *m_fitTool; }
 
      const nnFitterBinData& getBinCenters() const { return m_binCenters; }
 
      const nnFitterBins& getBins() const { return m_bins; }
 
      nnBoundsType getWaveWidthBounds() const { return m_waveWidthBounds; }
 
      nnBoundsType getAmplitudeBounds() const { return m_amplitudeBounds; }
 
      nnBoundsType getTimeShiftBounds() const { return m_timeShiftBounds; }
 
      bool isValid() const { return m_isValid; }
 
      bool checkCoefficients(const std::string& dumpname, double tol = 1.0e-10);
 
    private:
 
      typedef std::function<double(double)> activationType;
      activationType relu =
        [](double x) -> double { return std::max(double(0.0), x); };
      activationType sigmoid =
        [](double x) -> double { double e = std::exp(x); return e / (1.0 + e); };
 
      Eigen::VectorXd softmax(const Eigen::VectorXd& input)
      {
        auto output = input.array().unaryExpr(
                        [](double x)->double { return ((x > 0.0) ? exp(x) : 0.0); }
                      );
        double norm = output.sum();
        return output / norm;
      }
 
      int readNetworkData(const std::string& xmlFileName);
 
      typedef std::map< size_t, std::pair< Eigen::MatrixXd, Eigen::VectorXd > >
      networkWeightsType;
 
      typedef std::vector<Eigen::VectorXd> layerStatesType;
 
      // Data members
      bool m_isValid; 
      std::size_t m_nLayers; 
      nnFitterBinData m_binCenters;  
      nnFitterBins m_bins; 
      layerStatesType m_layerStates; 
      std::vector<std::size_t> m_layerSizes; 
      networkWeightsType m_networkCoefs; 
      activationType m_activation; 
      nnBoundsType m_amplitudeBounds; 
      nnBoundsType m_waveWidthBounds; 
      nnBoundsType m_timeShiftBounds; 
      TauEncoder m_tauCoder; 
      WaveformShape m_wave; 
      std::shared_ptr<NNWaveFitTool> m_fitTool; 
    }; // class NNWaveFitter
 
  } // namespace SVD
} // namespace Belle2
 
#endif