defaultParameterGenerator.hpp

Go to the documentation of this file.

#ifndef DEFAULTPARAMETERGENERATOR_HPP
#define DEFAULTPARAMETERGENERATOR_HPP

#include <random>
#include <array>
#include <algorithm>

namespace canopy
{

template <unsigned TNumParams>
class defaultParameterGenerator
{
    public:
        defaultParameterGenerator(const int limit)
        {
            // Setup RNG
            std::random_device rd{};
            rand_engine.seed(rd());

            // Set all limits to be the same value
            std::fill(param_limits.begin(),param_limits.end(),limit);
        }

        defaultParameterGenerator(const std::array<int,TNumParams>& limits)
        {
            // Setup RNG
            std::random_device rd{};
            rand_engine.seed(rd());

            // Store limit values
            std::copy(limits.cbegin(),limits.cend(),param_limits.begin());
        }

        void operator() (std::array<int,TNumParams>& params)
        {
            // Generate a valid combination of parameters and store in params
            for(unsigned p = 0; p < TNumParams; ++p)
            {
                params[p] = uni_int_dist(rand_engine,std::uniform_int_distribution<int>::param_type{0,param_limits[p]});
            }
        }

    private:
        std::array<int,TNumParams> param_limits; 
        std::default_random_engine rand_engine; 
        std::uniform_int_distribution<int> uni_int_dist; 
};

} // end of namespace

#endif // DEFAULTPARAMETERGENERATOR_HPP