random.hpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//
 
#pragma once

 
#include "shambase/constants.hpp"
#include "shamalgs/primitives/mock_value.hpp"
#include "shambackends/DeviceBuffer.hpp"
#include "shambackends/DeviceScheduler.hpp"
#include "shambackends/sycl.hpp"
#include "shambackends/typeAliasVec.hpp"
#include "shambackends/vec.hpp"
#include <shambackends/sycl.hpp>
#include <random>

namespace shamalgs::random {
 
    template<class T, class Engine = std::mt19937>
    T mock_gaussian(Engine &eng) {
        using namespace shambase::constants;
 
        constexpr T _2pi = pi<T> * 2;
        T r_3            = shamalgs::primitives::mock_value<T>(eng, 0, 1);
        T r_4            = shamalgs::primitives::mock_value<T>(eng, 0, 1);
        return sycl::sqrt(-2 * sycl::log(r_3)) * sycl::cos(_2pi * r_4);
    }
 
    template<class T, class Engine = std::mt19937>
    T mock_gaussian_multidim(Engine &eng) {
        T ret;
        constexpr int n = shambase::VectorProperties<T>::dimension;
        using Tscal     = shambase::VecComponent<T>;
#pragma unroll
        for (int i = 0; i < n; i++) {
            ret[i] = mock_gaussian<Tscal>(eng);
        }
        return ret;
    }
 
    template<class T, class Engine = std::mt19937>
    T mock_unit_vector(Engine &eng) {
        T ret    = mock_gaussian_multidim<T>(eng);
        auto len = sycl::length(ret);
 
        auto default_unit_vec = []() {
            T ret  = {};
            ret[0] = 1;
            return ret;
        };
 
        return (len > 0) ? (ret / len) : default_unit_vec();
    }
 
    template<class T>
    sycl::buffer<T> mock_buffer(u64 seed, u32 len, T min_bound, T max_bound);
 
    template<class T>
    sham::DeviceBuffer<T> mock_buffer_usm(
        const sham::DeviceScheduler_ptr &sched, u64 seed, u32 len, T min_bound, T max_bound);
 
    template<class T>
    inline sham::DeviceBuffer<T> mock_buffer_usm(
        const sham::DeviceScheduler_ptr &sched, u64 seed, u32 len) {
        using Prop = shambase::VectorProperties<T>;
        return mock_buffer_usm(sched, seed, len, Prop::get_min(), Prop::get_max());
    }
 
    template<class T>
    inline sycl::buffer<T> mock_buffer(u64 seed, u32 len) {
        using Prop = shambase::VectorProperties<T>;
        return mock_buffer(seed, len, Prop::get_min(), Prop::get_max());
    }
 
    template<class T>
    inline std::unique_ptr<sycl::buffer<T>> mock_buffer_ptr(
        u64 seed, u32 len, T min_bound, T max_bound) {
        return std::make_unique<sycl::buffer<T>>(mock_buffer(seed, len, min_bound, max_bound));
    }
    template<class T>
    inline std::unique_ptr<sycl::buffer<T>> mock_buffer_ptr(u64 seed, u32 len) {
        using Prop = shambase::VectorProperties<T>;
        return mock_buffer_ptr(seed, len, Prop::get_min(), Prop::get_max());
    }
 
} // namespace shamalgs::random