///////////////////////////////////////////////////////////////////////////////////
// RAT::Classifiers::PositionANN
//
// Position method that uses an ANN to classify radius using calibrated hit time
// and angle between PMTs and an estimated event orientation as inputs to find
// a radius.
//
// author: Matt Mottram <m.mottram@qmul.ac.uk>
//
// REVISION HISTORY:
// - 2016-06-20: M Mottram - new file
//
///////////////////////////////////////////////////////////////////////////////////

#ifndef __RAT_PositionANN__
#define __RAT_PositionANN__

#include <RAT/DS/FitResult.hh>
#include <RAT/Method.hh>

namespace RAT {

  namespace DS {
    class Entry;
    class EV;
  }

  namespace Methods {
    class PositionANN : public Method {

    public:
      virtual std::string GetName() const {return PositionANN::Name();}
      static std::string Name() {return std::string("positionANN");}
      void Initialise(const std::string&) {};
      void BeginOfRun(DS::Run&);
      void EndOfRun(DS::Run&) {};
      virtual DS::FitResult GetBestFit();

    private:
      std::vector<double> FeedForward(const std::vector<double>& inputs,
                                      const std::vector< std::vector<double> >& weights,
                                      const std::vector<double>& biases);
      std::vector< std::vector< std::vector<double> > > fWeights; // Deep!
      std::vector< std::vector<double> > fBiases;
      size_t fNLayers;
      int fNDots;
      int fNTimes;
      double fDotLow;
      double fDotHigh;
      double fDotWidth;
      double fTimeLow;
      double fTimeHigh;
      double fTimeWidth;
      double fRadiusScale;

    };
  }

}

#endif // __RAT_PositionANN__