random_generator.hpp 3.18 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
//| This file is a part of the limbo framework.
//| Copyright 2014,  Universite Pierre et Marie Curie (UPMC) / Inria
//|
//| This software is a computer program whose purpose is to facilitate
//| experiments in evolutionary computation and evolutionary robotics.
//|
//| This software is governed by the CeCILL license under French law
//| and abiding by the rules of distribution of free software.  You
//| can use, modify and/ or redistribute the software under the terms
//| of the CeCILL license as circulated by CEA, CNRS and INRIA at the
//| following URL "http://www.cecill.info".
//|
//| As a counterpart to the access to the source code and rights to
//| copy, modify and redistribute granted by the license, users are
//| provided only with a limited warranty and the software's author,
//| the holder of the economic rights, and the successive licensors
//| have only limited liability.
//|
//| In this respect, the user's attention is drawn to the risks
//| associated with loading, using, modifying and/or developing or
//| reproducing the software by the user in light of its specific
//| status of free software, that may mean that it is complicated to
//| manipulate, and that also therefore means that it is reserved for
//| developers and experienced professionals having in-depth computer
//| knowledge. Users are therefore encouraged to load and test the
//| software's suitability as regards their requirements in conditions
//| enabling the security of their systems and/or data to be ensured
//| and, more generally, to use and operate it in the same conditions
//| as regards security.
//|
//| The fact that you are presently reading this means that you have
//| had knowledge of the CeCILL license and that you accept its terms.

#ifndef LIMBO_TOOLS_RANDOM_GENERATOR_HPP
#define LIMBO_TOOLS_RANDOM_GENERATOR_HPP

#include <cstdlib>
#include <cmath>
#include <ctime>
#include <list>
#include <stdlib.h>
#include <random>
#include <utility>
#include <mutex>

namespace limbo {
    namespace tools {
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
48
49
        ///@ingroup tools
        /// random vector in [0, 1]
50
51
        Eigen::VectorXd random_vector(int size)
        {
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
52
53
            // Eigen returns in [-1:1] (??)
            return ((Eigen::VectorXd::Random(size)).array() + 1.0) / 2.0;
54
55
        }

Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
56
57
58
59
60
61
        /// @ingroup tools
        /// a mt19937-based random generator (mutex-protected)
        ///
        /// usage :
        /// - rgen_double_t(0.0, 1.0);
        /// - double r = rgen.rand();
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
62
        template <typename D>
63
64
        class RandomGenerator {
        public:
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
65
66
67
68
69
70
71
72
            using result_type = typename D::result_type;
            RandomGenerator(result_type min, result_type max) : _dist(min, max), _rgen(std::random_device()()) {}
            result_type rand()
            {
                std::lock_guard<std::mutex> lock(_mutex);
                return _dist(_rgen);
            }

73
        private:
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
74
75
76
            D _dist;
            std::mt19937 _rgen;
            std::mutex _mutex;
77
78
79
80
81
82
83
84
85
86
        };
        using rdist_double_t = std::uniform_real_distribution<double>;
        using rdist_int_t = std::uniform_int_distribution<int>;

        using rgen_double_t = RandomGenerator<rdist_double_t>;
        using rgen_int_t = RandomGenerator<rdist_int_t>;
    }
}

#endif