random.hh (4045:43eb54e807d1) | random.hh (5190:fc46e0d647b6) |
---|---|
1/* 2 * Copyright (c) 2003-2005 The Regents of The University of Michigan 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer; --- 18 unchanged lines hidden (view full) --- 27 * 28 * Authors: Nathan Binkert 29 * Ali Saidi 30 */ 31 32#ifndef __BASE_RANDOM_HH__ 33#define __BASE_RANDOM_HH__ 34 | 1/* 2 * Copyright (c) 2003-2005 The Regents of The University of Michigan 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer; --- 18 unchanged lines hidden (view full) --- 27 * 28 * Authors: Nathan Binkert 29 * Ali Saidi 30 */ 31 32#ifndef __BASE_RANDOM_HH__ 33#define __BASE_RANDOM_HH__ 34 |
35#include <ios> 36#include <string> 37 38#include "base/range.hh" |
|
35#include "sim/host.hh" 36 | 39#include "sim/host.hh" 40 |
37uint32_t getUInt32(); 38double getDouble(); 39double m5random(double r); 40uint64_t getUniformPos(uint64_t min, uint64_t max); 41int64_t getUniform(int64_t min, int64_t max); | 41class Checkpoint; |
42 | 42 |
43template <typename T> 44struct Random; 45 46template<> struct Random<int8_t> | 43class Random |
47{ | 44{ |
48 static int8_t get() 49 { return getUInt32() & (int8_t)-1; } | 45 protected: 46 static const int N = 624; 47 static const int M = 397; 48 static const uint32_t MATRIX_A = (uint32_t)0x9908b0df; 49 static const uint32_t UPPER_MASK = (uint32_t)0x80000000; 50 static const uint32_t LOWER_MASK = (uint32_t)0x7fffffff; |
50 | 51 |
51 static int8_t uniform(int8_t min, int8_t max) 52 { return getUniform(min, max); } 53}; | 52 uint32_t mt[N]; 53 int mti; |
54 | 54 |
55template<> struct Random<uint8_t> 56{ 57 static uint8_t get() 58 { return getUInt32() & (uint8_t)-1; } | 55 uint32_t genrand(); 56 uint32_t genrand(uint32_t max); 57 uint64_t genrand(uint64_t max); |
59 | 58 |
60 static uint8_t uniform(uint8_t min, uint8_t max) 61 { return getUniformPos(min, max); } 62}; | 59 void 60 _random(int8_t &value) 61 { 62 value = genrand() & (int8_t)-1; 63 } |
63 | 64 |
64template<> struct Random<int16_t> 65{ 66 static int16_t get() 67 { return getUInt32() & (int16_t)-1; } | 65 void 66 _random(int16_t &value) 67 { 68 value = genrand() & (int16_t)-1; 69 } |
68 | 70 |
69 static int16_t uniform(int16_t min, int16_t max) 70 { return getUniform(min, max); } 71}; | 71 void 72 _random(int32_t &value) 73 { 74 value = (int32_t)genrand(); 75 } |
72 | 76 |
73template<> struct Random<uint16_t> 74{ 75 static uint16_t get() 76 { return getUInt32() & (uint16_t)-1; } | 77 void 78 _random(int64_t &value) 79 { 80 value = (int64_t)genrand() << 32 | (int64_t)genrand(); 81 } |
77 | 82 |
78 static uint16_t uniform(uint16_t min, uint16_t max) 79 { return getUniformPos(min, max); } 80}; | 83 void 84 _random(uint8_t &value) 85 { 86 value = genrand() & (uint8_t)-1; 87 } |
81 | 88 |
82template<> struct Random<int32_t> 83{ 84 static int32_t get() 85 { return (int32_t)getUInt32(); } | 89 void 90 _random(uint16_t &value) 91 { 92 value = genrand() & (uint16_t)-1; 93 } |
86 | 94 |
87 static int32_t uniform(int32_t min, int32_t max) 88 { return getUniform(min, max); } 89}; | 95 void 96 _random(uint32_t &value) 97 { 98 value = genrand(); 99 } |
90 | 100 |
91template<> struct Random<uint32_t> 92{ 93 static uint32_t get() 94 { return (uint32_t)getUInt32(); } | 101 void 102 _random(uint64_t &value) 103 { 104 value = (uint64_t)genrand() << 32 | (uint64_t)genrand(); 105 } |
95 | 106 |
96 static uint32_t uniform(uint32_t min, uint32_t max) 97 { return getUniformPos(min, max); } 98}; | 107 // [0,1] 108 void 109 _random(float &value) 110 { 111 // ieee floats have 23 bits of mantissa 112 value = (genrand() >> 9) / 8388608.0; 113 } |
99 | 114 |
100template<> struct Random<int64_t> 101{ 102 static int64_t get() 103 { return (int64_t)getUInt32() << 32 || (uint64_t)getUInt32(); } | 115 // [0,1] 116 void 117 _random(double &value) 118 { 119 double number = genrand() * 2097152.0 + (genrand() >> 11); 120 value = number / 9007199254740992.0; 121 } |
104 | 122 |
105 static int64_t uniform(int64_t min, int64_t max) 106 { return getUniform(min, max); } 107}; | |
108 | 123 |
109template<> struct Random<uint64_t> 110{ 111 static uint64_t get() 112 { return (uint64_t)getUInt32() << 32 || (uint64_t)getUInt32(); } | 124 // Range based versions of the random number generator 125 int8_t 126 _random(int8_t min, int8_t max) 127 { 128 uint32_t diff = max - min; 129 return static_cast<int8_t>(min + genrand(diff)); 130 } |
113 | 131 |
114 static uint64_t uniform(uint64_t min, uint64_t max) 115 { return getUniformPos(min, max); } 116}; | 132 int16_t 133 _random(int16_t min, int16_t max) 134 { 135 uint32_t diff = max - min; 136 return static_cast<int16_t>(min + genrand(diff)); 137 } |
117 | 138 |
118template<> struct Random<float> 119{ 120 static float get() 121 { return getDouble(); } 122}; | 139 int32_t 140 _random(int32_t min, int32_t max) 141 { 142 uint32_t diff = max - min; 143 return static_cast<int32_t>(min + genrand(diff)); 144 } |
123 | 145 |
124template<> struct Random<double> 125{ 126 static double get() 127 { return getDouble(); } | 146 int64_t 147 _random(int64_t min, int64_t max) 148 { 149 uint64_t diff = max - min; 150 return static_cast<int64_t>(min + genrand(diff)); 151 } 152 153 uint8_t 154 _random(uint8_t min, uint8_t max) 155 { 156 uint32_t diff = max - min; 157 return static_cast<uint8_t>(min + genrand(diff)); 158 } 159 160 uint16_t 161 _random(uint16_t min, uint16_t max) 162 { 163 uint32_t diff = max - min; 164 return static_cast<uint16_t>(min + genrand(diff)); 165 } 166 167 uint32_t 168 _random(uint32_t min, uint32_t max) 169 { 170 uint32_t diff = max - min; 171 return static_cast<uint32_t>(min + genrand(diff)); 172 } 173 174 uint64_t 175 _random(uint64_t min, uint64_t max) 176 { 177 uint64_t diff = max - min; 178 return static_cast<uint64_t>(min + genrand(diff)); 179 } 180 181 public: 182 Random(); 183 Random(uint32_t s); 184 Random(uint32_t init_key[], int key_length); 185 ~Random(); 186 187 void init(uint32_t s); 188 void init(uint32_t init_key[], int key_length); 189 190 template <typename T> 191 T 192 random() 193 { 194 T value; 195 _random(value); 196 return value; 197 } 198 199 template <typename T> 200 T 201 random(T min, T max) 202 { 203 return _random(min, max); 204 } 205 206 template <typename T> 207 T 208 random(const Range<T> &range) 209 { 210 return _random(range.start, range.end); 211 } 212 213 // [0,1] 214 double 215 gen_real1() 216 { 217 return genrand() / 4294967296.0; 218 } 219 220 // [0,1) 221 double 222 gen_real2() 223 { 224 return genrand() / 4294967295.0; 225 } 226 227 // (0,1) 228 double 229 gen_real3() 230 { 231 return ((double)genrand() + 0.5) / 4294967296.0; 232 } 233 234 public: 235 void serialize(const std::string &base, std::ostream &os); 236 void unserialize(const std::string &base, Checkpoint *cp, 237 const std::string §ion); |
128}; 129 | 238}; 239 |
240extern Random random_mt; 241 |
|
130#endif // __BASE_RANDOM_HH__ | 242#endif // __BASE_RANDOM_HH__ |