random_generator.hpp 3.67 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
//| 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>
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
45
#include <limbo/tools/rand_utils.hpp>
46
47
48

namespace limbo {
    namespace tools {
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
49
50
51
52
        /// @ingroup tools
        /// a mt19937-based random generator (mutex-protected)
        ///
        /// usage :
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
53
        /// - RandomGenerator<double>(0.0, 1.0);
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
54
        /// - double r = rgen.rand();
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
55
        template <typename D>
56
57
        class RandomGenerator {
        public:
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
58
            using result_type = typename D::result_type;
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
59
            RandomGenerator(result_type min, result_type max) : _dist(min, max), _rgen(randutils::auto_seed_128{}.base()) {}
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
60
61
62
63
64
65
            result_type rand()
            {
                std::lock_guard<std::mutex> lock(_mutex);
                return _dist(_rgen);
            }

66
        private:
Konstantinos Chatzilygeroudis's avatar
Konstantinos Chatzilygeroudis committed
67
68
69
            D _dist;
            std::mt19937 _rgen;
            std::mutex _mutex;
70
        };
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
71
72

        /// @ingroup tools
73
        using rdist_double_t = std::uniform_real_distribution<double>;
Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
74
        /// @ingroup tools
75
76
        using rdist_int_t = std::uniform_int_distribution<int>;

Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
77
78
        /// @ingroup tools
        /// Double random number generator
79
        using rgen_double_t = RandomGenerator<rdist_double_t>;
80

Jean-Baptiste Mouret's avatar
Jean-Baptiste Mouret committed
81
82
        ///@ingroup tools
        ///integer random number generator
83
        using rgen_int_t = RandomGenerator<rdist_int_t>;
84
85
86
87
88
89
90
91
92
93
94

        /// @ingroup tools
        /// random vector in [0, 1]
        ///
        /// - this function is thread safe because the random number generator we use is thread-safe
        /// - we use a C++11 random number generator
        Eigen::VectorXd random_vector(int size)
        {
            static rgen_double_t rgen(0.0, 1.0);
            Eigen::VectorXd res(size);
            for (int i = 0; i < size; ++i)
95
                res[i] = rgen.rand();
96
97
            return res;
        }
98
99
100
101
    }
}

#endif