tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clang-Formated utils/random/generator.h

Browse Source
This commit is contained in:
sale 2016-11-22 16:46:31 +00:00
parent 1c3cf9c401
commit 8916b77418
1 changed files with 8 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
include/utils/random/generator.h
View File

@ -18,7 +18,6 @@ class RandomGenerator {
public:
RandomGenerator(Distribution dist) : gen_(device_()), dist_(dist) {}
};
class StringGenerator
@ -30,45 +29,27 @@ class StringGenerator
std::string next(int size) {
std::string random_string;
random_string.reserve(size);
for (int i = 0; i < size; i++) random_string += (dist_(gen_) + '\0');
return random_string;
}
};
template <class Distribution, class Generator, class DistributionRangeType>
class IntegerGenerator : public RandomGenerator<Distribution, Generator> {
class NumberGenerator : public RandomGenerator<Distribution, Generator> {
public:
IntegerGenerator(DistributionRangeType start, DistributionRangeType end) :
RandomGenerator<Distribution,Generator>(Distribution(start, end)){}
int next() {
return this->dist_(this->gen_);
}
};
template <class Distribution, class Generator, class DistributionRangeType>
class DecimalGenerator : public RandomGenerator<Distribution, Generator> {
public:
DecimalGenerator(DistributionRangeType start, DistributionRangeType end) :
RandomGenerator<Distribution, Generator>(Distribution(start, end)) {}
auto next() {
return this->dist_(this->gen_);
}
NumberGenerator(DistributionRangeType start, DistributionRangeType end)
: RandomGenerator<Distribution, Generator>(Distribution(start, end)) {}
auto next() { return this->dist_(this->gen_); }
};
template <class RandomGenerator>
auto generate_vector(RandomGenerator& gen, int size) {
std::vector<decltype (gen.next())> elements(size);
for (int i = 0; i < size; i++)
elements[i] = gen.next();
std::vector<decltype(gen.next())> elements(size);
for (int i = 0; i < size; i++) elements[i] = gen.next();
return elements;
}