uniformOnSphere.hpp

Go to the documentation of this file.

#pragma once
 
#ifndef NUMCPP_NO_USE_BOOST
 
#include <algorithm>
#include <string>
 
#include "boost/random/uniform_on_sphere.hpp"
 
#include "NumCpp/Core/Internal/StaticAsserts.hpp"
#include "NumCpp/Core/Shape.hpp"
#include "NumCpp/Core/Types.hpp"
#include "NumCpp/NdArray.hpp"
#include "NumCpp/Random/generator.hpp"
 
namespace nc::random
{
    namespace detail
    {
        //============================================================================
        // Method Description:
        template<typename dtype, typename GeneratorType = std::mt19937>
        NdArray<dtype> uniformOnSphere(GeneratorType& generator, uint32 inNumPoints, uint32 inDims = 2)
        {
            STATIC_ASSERT_FLOAT(dtype);
 
            if (inNumPoints == 0)
            {
                return {};
            }
 
            boost::random::uniform_on_sphere<dtype> dist(static_cast<int>(inDims));
 
            NdArray<dtype> returnArray(inNumPoints, inDims);
            for (uint32 row = 0; row < inNumPoints; ++row)
            {
                const auto& point = dist(generator);
                std::copy(point.begin(), point.end(), returnArray.begin(row));
            }
 
            return returnArray;
        }
    } // namespace detail
 
    //============================================================================
    // Method Description:
    template<typename dtype>
    NdArray<dtype> uniformOnSphere(uint32 inNumPoints, uint32 inDims = 2)
    {
        return detail::uniformOnSphere<dtype>(generator_, inNumPoints, inDims);
    }
} // namespace nc::random
 
#endif // #ifndef NUMCPP_NO_USE_BOOST