368faacac7
This commit reverts 5aaefa51 (and also partly 7f285f48e77 and b6d75682f9, which were related to the original commit). As landed, 5aaefa51 had unintended consequences on some downstream bots and didn't have proper coverage upstream due to a few subtle things. Implementing this is something we should do in libc++, however we'll first need to address a few issues listed in https://reviews.llvm.org/D106124#3349710. Differential Revision: https://reviews.llvm.org/D120683
300 lines
9.2 KiB
C++
300 lines
9.2 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef _LIBCPP___RANDOM_LOGNORMAL_DISTRIBUTION_H
|
|
#define _LIBCPP___RANDOM_LOGNORMAL_DISTRIBUTION_H
|
|
|
|
#include <__config>
|
|
#include <__random/normal_distribution.h>
|
|
#include <cmath>
|
|
#include <iosfwd>
|
|
#include <limits>
|
|
|
|
#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
|
|
# pragma GCC system_header
|
|
#endif
|
|
|
|
_LIBCPP_PUSH_MACROS
|
|
#include <__undef_macros>
|
|
|
|
_LIBCPP_BEGIN_NAMESPACE_STD
|
|
|
|
#ifdef _LIBCPP_ABI_OLD_LOGNORMAL_DISTRIBUTION
|
|
|
|
template<class _RealType = double>
|
|
class _LIBCPP_TEMPLATE_VIS lognormal_distribution
|
|
{
|
|
public:
|
|
// types
|
|
typedef _RealType result_type;
|
|
|
|
class _LIBCPP_TEMPLATE_VIS param_type
|
|
{
|
|
normal_distribution<result_type> __nd_;
|
|
public:
|
|
typedef lognormal_distribution distribution_type;
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit param_type(result_type __m = 0, result_type __s = 1)
|
|
: __nd_(__m, __s) {}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type m() const {return __nd_.mean();}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type s() const {return __nd_.stddev();}
|
|
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator==(const param_type& __x, const param_type& __y)
|
|
{return __x.__nd_ == __y.__nd_;}
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator!=(const param_type& __x, const param_type& __y)
|
|
{return !(__x == __y);}
|
|
friend class lognormal_distribution;
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_ostream<_CharT, _Traits>&
|
|
operator<<(basic_ostream<_CharT, _Traits>& __os,
|
|
const lognormal_distribution<_RT>& __x);
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_istream<_CharT, _Traits>&
|
|
operator>>(basic_istream<_CharT, _Traits>& __is,
|
|
lognormal_distribution<_RT>& __x);
|
|
};
|
|
|
|
private:
|
|
param_type __p_;
|
|
|
|
public:
|
|
// constructor and reset functions
|
|
#ifndef _LIBCPP_CXX03_LANG
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
lognormal_distribution() : lognormal_distribution(0) {}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(result_type __m, result_type __s = 1)
|
|
: __p_(param_type(__m, __s)) {}
|
|
#else
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(result_type __m = 0,
|
|
result_type __s = 1)
|
|
: __p_(param_type(__m, __s)) {}
|
|
#endif
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(const param_type& __p)
|
|
: __p_(__p) {}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void reset() {__p_.__nd_.reset();}
|
|
|
|
// generating functions
|
|
template<class _URNG>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type operator()(_URNG& __g)
|
|
{return (*this)(__g, __p_);}
|
|
template<class _URNG>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type operator()(_URNG& __g, const param_type& __p)
|
|
{return _VSTD::exp(const_cast<normal_distribution<result_type>&>(__p.__nd_)(__g));}
|
|
|
|
// property functions
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type m() const {return __p_.m();}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type s() const {return __p_.s();}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
param_type param() const {return __p_;}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void param(const param_type& __p) {__p_ = __p;}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type min() const {return 0;}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type max() const {return numeric_limits<result_type>::infinity();}
|
|
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator==(const lognormal_distribution& __x,
|
|
const lognormal_distribution& __y)
|
|
{return __x.__p_ == __y.__p_;}
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator!=(const lognormal_distribution& __x,
|
|
const lognormal_distribution& __y)
|
|
{return !(__x == __y);}
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_ostream<_CharT, _Traits>&
|
|
operator<<(basic_ostream<_CharT, _Traits>& __os,
|
|
const lognormal_distribution<_RT>& __x);
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_istream<_CharT, _Traits>&
|
|
operator>>(basic_istream<_CharT, _Traits>& __is,
|
|
lognormal_distribution<_RT>& __x);
|
|
};
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
basic_ostream<_CharT, _Traits>&
|
|
operator<<(basic_ostream<_CharT, _Traits>& __os,
|
|
const lognormal_distribution<_RT>& __x)
|
|
{
|
|
return __os << __x.__p_.__nd_;
|
|
}
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
basic_istream<_CharT, _Traits>&
|
|
operator>>(basic_istream<_CharT, _Traits>& __is,
|
|
lognormal_distribution<_RT>& __x)
|
|
{
|
|
return __is >> __x.__p_.__nd_;
|
|
}
|
|
|
|
#else // _LIBCPP_ABI_OLD_LOGNORMAL_DISTRIBUTION
|
|
|
|
template<class _RealType = double>
|
|
class _LIBCPP_TEMPLATE_VIS lognormal_distribution
|
|
{
|
|
public:
|
|
// types
|
|
typedef _RealType result_type;
|
|
|
|
class _LIBCPP_TEMPLATE_VIS param_type
|
|
{
|
|
result_type __m_;
|
|
result_type __s_;
|
|
public:
|
|
typedef lognormal_distribution distribution_type;
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit param_type(result_type __m = 0, result_type __s = 1)
|
|
: __m_(__m), __s_(__s) {}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type m() const {return __m_;}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type s() const {return __s_;}
|
|
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator==(const param_type& __x, const param_type& __y)
|
|
{return __x.__m_ == __y.__m_ && __x.__s_ == __y.__s_;}
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator!=(const param_type& __x, const param_type& __y)
|
|
{return !(__x == __y);}
|
|
};
|
|
|
|
private:
|
|
normal_distribution<result_type> __nd_;
|
|
|
|
public:
|
|
// constructor and reset functions
|
|
#ifndef _LIBCPP_CXX03_LANG
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
lognormal_distribution() : lognormal_distribution(0) {}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(result_type __m, result_type __s = 1)
|
|
: __nd_(__m, __s) {}
|
|
#else
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(result_type __m = 0,
|
|
result_type __s = 1)
|
|
: __nd_(__m, __s) {}
|
|
#endif
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
explicit lognormal_distribution(const param_type& __p)
|
|
: __nd_(__p.m(), __p.s()) {}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void reset() {__nd_.reset();}
|
|
|
|
// generating functions
|
|
template<class _URNG>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type operator()(_URNG& __g)
|
|
{
|
|
return _VSTD::exp(__nd_(__g));
|
|
}
|
|
|
|
template<class _URNG>
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type operator()(_URNG& __g, const param_type& __p)
|
|
{
|
|
typename normal_distribution<result_type>::param_type __pn(__p.m(), __p.s());
|
|
return _VSTD::exp(__nd_(__g, __pn));
|
|
}
|
|
|
|
// property functions
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type m() const {return __nd_.mean();}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type s() const {return __nd_.stddev();}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
param_type param() const {return param_type(__nd_.mean(), __nd_.stddev());}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
void param(const param_type& __p)
|
|
{
|
|
typename normal_distribution<result_type>::param_type __pn(__p.m(), __p.s());
|
|
__nd_.param(__pn);
|
|
}
|
|
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type min() const {return 0;}
|
|
_LIBCPP_INLINE_VISIBILITY
|
|
result_type max() const {return numeric_limits<result_type>::infinity();}
|
|
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator==(const lognormal_distribution& __x,
|
|
const lognormal_distribution& __y)
|
|
{return __x.__nd_ == __y.__nd_;}
|
|
friend _LIBCPP_INLINE_VISIBILITY
|
|
bool operator!=(const lognormal_distribution& __x,
|
|
const lognormal_distribution& __y)
|
|
{return !(__x == __y);}
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_ostream<_CharT, _Traits>&
|
|
operator<<(basic_ostream<_CharT, _Traits>& __os,
|
|
const lognormal_distribution<_RT>& __x);
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
friend
|
|
basic_istream<_CharT, _Traits>&
|
|
operator>>(basic_istream<_CharT, _Traits>& __is,
|
|
lognormal_distribution<_RT>& __x);
|
|
};
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
basic_ostream<_CharT, _Traits>&
|
|
operator<<(basic_ostream<_CharT, _Traits>& __os,
|
|
const lognormal_distribution<_RT>& __x)
|
|
{
|
|
return __os << __x.__nd_;
|
|
}
|
|
|
|
template <class _CharT, class _Traits, class _RT>
|
|
inline _LIBCPP_INLINE_VISIBILITY
|
|
basic_istream<_CharT, _Traits>&
|
|
operator>>(basic_istream<_CharT, _Traits>& __is,
|
|
lognormal_distribution<_RT>& __x)
|
|
{
|
|
return __is >> __x.__nd_;
|
|
}
|
|
|
|
#endif // _LIBCPP_ABI_OLD_LOGNORMAL_DISTRIBUTION
|
|
|
|
_LIBCPP_END_NAMESPACE_STD
|
|
|
|
_LIBCPP_POP_MACROS
|
|
|
|
#endif // _LIBCPP___RANDOM_LOGNORMAL_DISTRIBUTION_H
|