random_fill.hpp

/****************************************************************************
 *  hipipe library
 *  Copyright (c) 2017, Cognexa Solutions s.r.o.
 *  Copyright (c) 2018, Iterait a.s.
 *  Author(s) Filip Matzner
 *
 *  This file is distributed under the MIT License.
 *  See the accompanying file LICENSE.txt for the complete license agreement.
 ****************************************************************************/
 
#pragma once
 
#include <hipipe/core/stream/generate.hpp>
 
namespace hipipe::stream {
 
template<typename FromColumn, typename ToColumn, typename Prng = std::mt19937,
         typename Dist = std::uniform_real_distribution<double>,
         int Dim = utility::ndims<typename ToColumn::data_type>::value
                 - utility::ndims<std::result_of_t<Dist(Prng&)>>::value>
auto random_fill(from_t<FromColumn> size_from,
                 to_t<ToColumn> fill_to,
                 long rnddims = std::numeric_limits<long>::max(),
                 Dist dist = Dist{0, 1},
                 Prng& prng = hipipe::utility::random_generator,
                 dim_t<Dim> d = dim_t<Dim>{})
{
    // a bit of function type erasure to speed up compilation
    using GenT = std::function<
      utility::ndim_type_t<typename ToColumn::data_type, Dim>()>;
    // distribution is always copied to avoid race conditions
    GenT fun = [dist, &prng]() { return std::invoke(Dist{dist}, prng); };
    return stream::generate(size_from, fill_to, std::move(fun), rnddims, d);
}
 
}  // namespace hipipe::stream