70 lines
2.0 KiB
C++
70 lines
2.0 KiB
C++
/*
|
|
* Copyright (C) 2017 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef CHRE_WIFI_OFFLOAD_TEST_RANDOM_GENERATOR_H_
|
|
#define CHRE_WIFI_OFFLOAD_TEST_RANDOM_GENERATOR_H_
|
|
|
|
#include <cinttypes>
|
|
#include <random>
|
|
|
|
namespace wifi_offload_test {
|
|
|
|
/**
|
|
* @class RandomGenerator A class to generate random values for any uint type:
|
|
* (uint8_t, ..., uint64_t). Also supports resetting to its initial
|
|
* state to be able to reproduce the same random sequence.
|
|
*/
|
|
class RandomGenerator {
|
|
public:
|
|
/**
|
|
* Default constructs a RandomGenerator object
|
|
*/
|
|
RandomGenerator();
|
|
|
|
/**
|
|
* Resets the object to its initial state. Useful if we want to reproduce the
|
|
* exact same sequenct again.
|
|
*/
|
|
void Reset();
|
|
|
|
/**
|
|
* Generates a random number of type IntType
|
|
*
|
|
* @return A random number of type IntType
|
|
*/
|
|
template <typename IntType>
|
|
IntType get() {
|
|
return static_cast<IntType>(uniform_distribution_(random_engine_));
|
|
}
|
|
|
|
private:
|
|
/* The initial seed kept to use again when need to reset */
|
|
std::random_device::result_type initial_seed_;
|
|
|
|
/* Standard mersenne_twister_engine */
|
|
std::mt19937 random_engine_;
|
|
|
|
/**
|
|
* Use uniform_distribution_ to transform the random unsigned int generated by
|
|
* random_engine_ into an uint64_t
|
|
*/
|
|
std::uniform_int_distribution<uint64_t> uniform_distribution_;
|
|
};
|
|
|
|
} // namespace wifi_offload_test
|
|
|
|
#endif // CHRE_WIFI_OFFLOAD_TEST_RANDOM_GENERATOR_H_
|