yul-phaser: Switch from uint32_t to size_t in SimulationRNG

- Also pass the appriopriate type internally to the distribution instead of relying on the default (which is uint32_t)
This commit is contained in:
Kamil Śliwak
2020-06-04 17:19:42 +02:00
parent 9b3d1c11ff
commit 8f55ead48d
3 changed files with 21 additions and 12 deletions
+18 -9
View File
@@ -17,12 +17,17 @@
#include <tools/yulPhaser/SimulationRNG.h>
// NOTE: The code would work with std::random but the results for a given seed would not be reproducible
// across different STL implementations. Boost does not guarantee this either but at least it has only one
// implementation. Reproducibility is not a hard requirement for yul-phaser but it's nice to have.
#include <boost/random/bernoulli_distribution.hpp>
#include <boost/random/binomial_distribution.hpp>
#include <boost/random/uniform_int_distribution.hpp>
#include <ctime>
#include <limits>
using namespace std;
using namespace solidity;
using namespace solidity::phaser;
@@ -30,23 +35,27 @@ thread_local boost::random::mt19937 SimulationRNG::s_generator(SimulationRNG::ge
bool SimulationRNG::bernoulliTrial(double _successProbability)
{
boost::random::bernoulli_distribution<> distribution(_successProbability);
boost::random::bernoulli_distribution<double> distribution(_successProbability);
return static_cast<bool>(distribution(s_generator));
}
uint32_t SimulationRNG::uniformInt(uint32_t _min, uint32_t _max)
{
boost::random::uniform_int_distribution<> distribution(_min, _max);
return distribution(s_generator);
}
uint32_t SimulationRNG::binomialInt(uint32_t _numTrials, double _successProbability)
size_t SimulationRNG::uniformInt(size_t _min, size_t _max)
{
boost::random::binomial_distribution<> distribution(_numTrials, _successProbability);
boost::random::uniform_int_distribution<size_t> distribution(_min, _max);
return distribution(s_generator);
}
size_t SimulationRNG::binomialInt(size_t _numTrials, double _successProbability)
{
// NOTE: binomial_distribution<size_t> would not work because it internally tries to use abs()
// and fails to compile due to ambiguous conversion.
assert(_numTrials <= static_cast<size_t>(numeric_limits<long>::max()));
boost::random::binomial_distribution<long> distribution(static_cast<long>(_numTrials), _successProbability);
return static_cast<size_t>(distribution(s_generator));
}
uint32_t SimulationRNG::generateSeed()
{
// This is not a secure way to seed the generator but it's good enough for simulation purposes.