oct-rand.h

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2003-2024 The Octave Project Developers
//
// See the file COPYRIGHT.md in the top-level directory of this
// distribution or <https://octave.org/copyright/>.
//
// This file is part of Octave.
//
// Octave is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Octave is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Octave; see the file COPYING.  If not, see
// <https://www.gnu.org/licenses/>.
//
////////////////////////////////////////////////////////////////////////
 
#if ! defined (octave_oct_rand_h)
#define octave_oct_rand_h 1
 
#include "octave-config.h"
 
#include <map>
#include <string>
 
#include "Array.h"
#include "dNDArray.h"
#include "fNDArray.h"
#include "lo-ieee.h"
#include "uint32NDArray.h"
 
//class dim_vector;
 
OCTAVE_BEGIN_NAMESPACE(octave)
 
class OCTAVE_API rand
{
protected:
 
  OCTAVE_API rand ();
 
public:
 
  OCTAVE_DISABLE_COPY_MOVE (rand)
 
  ~rand () = default;
 
  static bool instance_ok ();
 
  // Return the current seed.
  static double seed ()
  {
    return (instance_ok ()
            ? s_instance->do_seed () : numeric_limits<double>::NaN ());
  }
 
  // Set the seed.
  static void seed (double s)
  {
    if (instance_ok ())
      s_instance->do_seed (s);
  }
 
  // Reset the seed.
  static void reset ()
  {
    if (instance_ok ())
      s_instance->do_reset ();
  }
 
  // Return the current state.
  static uint32NDArray state (const std::string& d = "")
  {
    return instance_ok () ? s_instance->do_state (d) : uint32NDArray ();
  }
 
  // Set the current state/
  static void state (const uint32NDArray& s,
                     const std::string& d = "")
  {
    if (instance_ok ())
      s_instance->do_state (s, d);
  }
 
  // Reset the current state/
  static void reset (const std::string& d)
  {
    if (instance_ok ())
      s_instance->do_reset (d);
  }
 
  // Return the current distribution.
  static std::string distribution ()
  {
    return instance_ok () ? s_instance->do_distribution () : "";
  }
 
  // Set the current distribution.  May be either "uniform" (the
  // default), "normal", "exponential", "poisson", or "gamma".
  static void distribution (const std::string& d)
  {
    if (instance_ok ())
      s_instance->do_distribution (d);
  }
 
  static void uniform_distribution ()
  {
    if (instance_ok ())
      s_instance->do_uniform_distribution ();
  }
 
  static void normal_distribution ()
  {
    if (instance_ok ())
      s_instance->do_normal_distribution ();
  }
 
  static void exponential_distribution ()
  {
    if (instance_ok ())
      s_instance->do_exponential_distribution ();
  }
 
  static void poisson_distribution ()
  {
    if (instance_ok ())
      s_instance->do_poisson_distribution ();
  }
 
  static void gamma_distribution ()
  {
    if (instance_ok ())
      s_instance->do_gamma_distribution ();
  }
 
  // Return the next number from the sequence.
  static double scalar (double a = 1.0)
  {
    return (instance_ok ()
            ? s_instance->do_scalar (a) : numeric_limits<double>::NaN ());
  }
 
  // Return the next number from the sequence.
  static float float_scalar (float a = 1.0)
  {
    return (instance_ok ()
            ? s_instance->do_scalar (a) : numeric_limits<float>::NaN ());
  }
 
  // Return an array of numbers from the sequence.
  static Array<double> vector (octave_idx_type n, double a = 1.0)
  {
    return instance_ok () ? s_instance->do_vector (n, a) : Array<double> ();
  }
 
  // Return an array of numbers from the sequence.
  static Array<float> float_vector (octave_idx_type n, float a = 1.0)
  {
    return instance_ok () ? s_instance->do_vector (n, a) : Array<float> ();
  }
 
  // Return an N-dimensional array of numbers from the sequence,
  // filled in column major order.
  static NDArray nd_array (const dim_vector& dims, double a = 1.0)
  {
    return instance_ok () ? s_instance->do_nd_array (dims, a) : NDArray ();
  }
 
  // Return an N-dimensional array of numbers from the sequence,
  // filled in column major order.
  static FloatNDArray float_nd_array (const dim_vector& dims, float a = 1.0)
  {
    return (instance_ok ()
            ? s_instance->do_float_nd_array (dims, a) : FloatNDArray ());
  }
 
private:
 
  static rand *s_instance;
 
  static void cleanup_instance ()
  { delete s_instance; s_instance = nullptr; }
 
  enum
  {
    unknown_dist,
    uniform_dist,
    normal_dist,
    expon_dist,
    poisson_dist,
    gamma_dist
  };
 
  // Current distribution of random numbers.
  int m_current_distribution;
 
  // If TRUE, use old RANLIB generators.  Otherwise, use Mersenne
  // Twister generator.
  bool m_use_old_generators;
 
  // Saved MT states.
  std::map<int, uint32NDArray> m_rand_states;
 
  // Return the current seed.
  OCTAVE_API double do_seed ();
 
  // Set the seed.
  OCTAVE_API void do_seed (double s);
 
  // Reset the seed.
  OCTAVE_API void do_reset ();
 
  // Return the current state.
  OCTAVE_API uint32NDArray do_state (const std::string& d);
 
  // Set the current state/
  OCTAVE_API void do_state (const uint32NDArray& s, const std::string& d);
 
  // Reset the current state/
  OCTAVE_API void do_reset (const std::string& d);
 
  // Return the current distribution.
  OCTAVE_API std::string do_distribution ();
 
  // Set the current distribution.  May be either "uniform" (the
  // default), "normal", "exponential", "poisson", or "gamma".
  OCTAVE_API void do_distribution (const std::string& d);
 
  OCTAVE_API void do_uniform_distribution ();
 
  OCTAVE_API void do_normal_distribution ();
 
  OCTAVE_API void do_exponential_distribution ();
 
  OCTAVE_API void do_poisson_distribution ();
 
  OCTAVE_API void do_gamma_distribution ();
 
  // The following templates only make sense for double and float
  // types.
 
  template <typename T> OCTAVE_API T uniform ();
 
  template <typename T> OCTAVE_API T normal ();
 
  template <typename T> OCTAVE_API T exponential ();
 
  template <typename T> OCTAVE_API T poisson (T a);
 
  template <typename T> OCTAVE_API T gamma (T a);
 
  // Return the next number from the sequence.
  template <typename T> OCTAVE_API T do_scalar (T a = 1);
 
  // Return an array of numbers from the sequence.
  template <typename T> OCTAVE_API Array<T>
  do_vector (octave_idx_type n, T a = 1);
 
  // Return an N-dimensional array of numbers from the sequence,
  // filled in column major order.
  OCTAVE_API NDArray do_nd_array (const dim_vector& dims, double a = 1.);
 
  // Return an N-dimensional array of numbers from the sequence,
  // filled in column major order.
  OCTAVE_API FloatNDArray
  do_float_nd_array (const dim_vector& dims, float a = 1.);
 
  // Some helper functions.
 
  OCTAVE_API void initialize_ranlib_generators ();
 
  OCTAVE_API void initialize_mersenne_twister ();
 
  OCTAVE_API uint32NDArray get_internal_state ();
 
  OCTAVE_API void save_state ();
 
  OCTAVE_API int get_dist_id (const std::string& d);
 
  OCTAVE_API void set_internal_state (const uint32NDArray& s);
 
  OCTAVE_API void switch_to_generator (int dist);
 
  OCTAVE_API void fill (octave_idx_type len, double *v, double a);
 
  OCTAVE_API void fill (octave_idx_type len, float *v, float a);
};
 
OCTAVE_END_NAMESPACE(octave)
 
#endif