HopfieldNetwork.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2011 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Heck                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#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()); }
    };
  };
}