HopfieldNetwork.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
 
//basf2
#include <tracking/trackFindingVXD/trackSetEvaluator/OverlapResolverNodeInfo.h>
 
//ROOT
#include <TRandom.h>
 
 
namespace Belle2 {
  class HopfieldNetwork {
  public:
    HopfieldNetwork(float omega = 0.5, float T = 3.1, float Tmin = 0.1, float cmax = 0.01):
      m_omega(omega), m_T(T), m_Tmin(Tmin), m_cmax(cmax)
    {}
 
    unsigned short doHopfield(std::vector<OverlapResolverNodeInfo>& overlapResolverNodeInfos, unsigned short nIterations = 20);
 
  private:
    float m_omega; 
    float m_T; 
    float m_Tmin; 
    float m_cmax; 
    //--- Structs to help simplify the process ------------------------------------------------------------------------
    struct TRandomWrapper {
      typedef unsigned int result_type;
 
      static constexpr result_type min() { return 0; }
 
      static constexpr result_type max() { return std::numeric_limits<result_type>::max(); }
 
      result_type operator()() { return gRandom->Integer(max()); }
    };
  };
}