123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365 |
- #ifndef BOOST_STATS_NORMAL_HPP
- #define BOOST_STATS_NORMAL_HPP
- #include <boost/math/distributions/fwd.hpp>
- #include <boost/math/special_functions/erf.hpp>
- #include <boost/math/distributions/complement.hpp>
- #include <boost/math/distributions/detail/common_error_handling.hpp>
- #include <utility>
- #include <type_traits>
- namespace boost{ namespace math{
- template <class RealType = double, class Policy = policies::policy<> >
- class normal_distribution
- {
- public:
- using value_type = RealType;
- using policy_type = Policy;
- explicit normal_distribution(RealType l_mean = 0, RealType sd = 1)
- : m_mean(l_mean), m_sd(sd)
- {
- static const char* function = "boost::math::normal_distribution<%1%>::normal_distribution";
- RealType result;
- detail::check_scale(function, sd, &result, Policy());
- detail::check_location(function, l_mean, &result, Policy());
- }
- RealType mean()const
- {
- return m_mean;
- }
- RealType standard_deviation()const
- {
- return m_sd;
- }
-
- RealType location()const
- {
- return m_mean;
- }
- RealType scale()const
- {
- return m_sd;
- }
- private:
-
-
-
- RealType m_mean;
- RealType m_sd;
- };
- using normal = normal_distribution<double>;
- #ifdef __cpp_deduction_guides
- template <class RealType>
- normal_distribution(RealType, RealType)->normal_distribution<typename boost::math::tools::promote_args<RealType>::type>;
- template <class RealType>
- normal_distribution(RealType)->normal_distribution<typename boost::math::tools::promote_args<RealType>::type>;
- #endif
- #ifdef _MSC_VER
- #pragma warning(push)
- #pragma warning(disable:4127)
- #endif
- template <class RealType, class Policy>
- inline std::pair<RealType, RealType> range(const normal_distribution<RealType, Policy>& )
- {
- if (std::numeric_limits<RealType>::has_infinity)
- {
- return std::pair<RealType, RealType>(-std::numeric_limits<RealType>::infinity(), std::numeric_limits<RealType>::infinity());
- }
- else
- {
- using boost::math::tools::max_value;
- return std::pair<RealType, RealType>(-max_value<RealType>(), max_value<RealType>());
- }
- }
- template <class RealType, class Policy>
- inline std::pair<RealType, RealType> support(const normal_distribution<RealType, Policy>& )
- {
- if (std::numeric_limits<RealType>::has_infinity)
- {
- return std::pair<RealType, RealType>(-std::numeric_limits<RealType>::infinity(), std::numeric_limits<RealType>::infinity());
- }
- else
- {
- using boost::math::tools::max_value;
- return std::pair<RealType, RealType>(-max_value<RealType>(), max_value<RealType>());
- }
- }
- #ifdef _MSC_VER
- #pragma warning(pop)
- #endif
- template <class RealType, class Policy>
- inline RealType pdf(const normal_distribution<RealType, Policy>& dist, const RealType& x)
- {
- BOOST_MATH_STD_USING
- RealType sd = dist.standard_deviation();
- RealType mean = dist.mean();
- static const char* function = "boost::math::pdf(const normal_distribution<%1%>&, %1%)";
- RealType result = 0;
- if(false == detail::check_scale(function, sd, &result, Policy()))
- {
- return result;
- }
- if(false == detail::check_location(function, mean, &result, Policy()))
- {
- return result;
- }
- if((boost::math::isinf)(x))
- {
- return 0;
- }
- if(false == detail::check_x(function, x, &result, Policy()))
- {
- return result;
- }
- RealType exponent = x - mean;
- exponent *= -exponent;
- exponent /= 2 * sd * sd;
- result = exp(exponent);
- result /= sd * sqrt(2 * constants::pi<RealType>());
- return result;
- }
- template <class RealType, class Policy>
- inline RealType logpdf(const normal_distribution<RealType, Policy>& dist, const RealType& x)
- {
- BOOST_MATH_STD_USING
- const RealType sd = dist.standard_deviation();
- const RealType mean = dist.mean();
- static const char* function = "boost::math::logpdf(const normal_distribution<%1%>&, %1%)";
- RealType result = -std::numeric_limits<RealType>::infinity();
- if(false == detail::check_scale(function, sd, &result, Policy()))
- {
- return result;
- }
- if(false == detail::check_location(function, mean, &result, Policy()))
- {
- return result;
- }
- if((boost::math::isinf)(x))
- {
- return result;
- }
- if(false == detail::check_x(function, x, &result, Policy()))
- {
- return result;
- }
- const RealType pi = boost::math::constants::pi<RealType>();
- const RealType half = boost::math::constants::half<RealType>();
- result = -log(sd) - half*log(2*pi) - (x-mean)*(x-mean)/(2*sd*sd);
- return result;
- }
- template <class RealType, class Policy>
- inline RealType cdf(const normal_distribution<RealType, Policy>& dist, const RealType& x)
- {
- BOOST_MATH_STD_USING
- RealType sd = dist.standard_deviation();
- RealType mean = dist.mean();
- static const char* function = "boost::math::cdf(const normal_distribution<%1%>&, %1%)";
- RealType result = 0;
- if(false == detail::check_scale(function, sd, &result, Policy()))
- {
- return result;
- }
- if(false == detail::check_location(function, mean, &result, Policy()))
- {
- return result;
- }
- if((boost::math::isinf)(x))
- {
- if(x < 0) return 0;
- return 1;
- }
- if(false == detail::check_x(function, x, &result, Policy()))
- {
- return result;
- }
- RealType diff = (x - mean) / (sd * constants::root_two<RealType>());
- result = boost::math::erfc(-diff, Policy()) / 2;
- return result;
- }
- template <class RealType, class Policy>
- inline RealType quantile(const normal_distribution<RealType, Policy>& dist, const RealType& p)
- {
- BOOST_MATH_STD_USING
- RealType sd = dist.standard_deviation();
- RealType mean = dist.mean();
- static const char* function = "boost::math::quantile(const normal_distribution<%1%>&, %1%)";
- RealType result = 0;
- if(false == detail::check_scale(function, sd, &result, Policy()))
- return result;
- if(false == detail::check_location(function, mean, &result, Policy()))
- return result;
- if(false == detail::check_probability(function, p, &result, Policy()))
- return result;
- result= boost::math::erfc_inv(2 * p, Policy());
- result = -result;
- result *= sd * constants::root_two<RealType>();
- result += mean;
- return result;
- }
- template <class RealType, class Policy>
- inline RealType cdf(const complemented2_type<normal_distribution<RealType, Policy>, RealType>& c)
- {
- BOOST_MATH_STD_USING
- RealType sd = c.dist.standard_deviation();
- RealType mean = c.dist.mean();
- RealType x = c.param;
- static const char* function = "boost::math::cdf(const complement(normal_distribution<%1%>&), %1%)";
- RealType result = 0;
- if(false == detail::check_scale(function, sd, &result, Policy()))
- return result;
- if(false == detail::check_location(function, mean, &result, Policy()))
- return result;
- if((boost::math::isinf)(x))
- {
- if(x < 0) return 1;
- return 0;
- }
- if(false == detail::check_x(function, x, &result, Policy()))
- return result;
- RealType diff = (x - mean) / (sd * constants::root_two<RealType>());
- result = boost::math::erfc(diff, Policy()) / 2;
- return result;
- }
- template <class RealType, class Policy>
- inline RealType quantile(const complemented2_type<normal_distribution<RealType, Policy>, RealType>& c)
- {
- BOOST_MATH_STD_USING
- RealType sd = c.dist.standard_deviation();
- RealType mean = c.dist.mean();
- static const char* function = "boost::math::quantile(const complement(normal_distribution<%1%>&), %1%)";
- RealType result = 0;
- if(false == detail::check_scale(function, sd, &result, Policy()))
- return result;
- if(false == detail::check_location(function, mean, &result, Policy()))
- return result;
- RealType q = c.param;
- if(false == detail::check_probability(function, q, &result, Policy()))
- return result;
- result = boost::math::erfc_inv(2 * q, Policy());
- result *= sd * constants::root_two<RealType>();
- result += mean;
- return result;
- }
- template <class RealType, class Policy>
- inline RealType mean(const normal_distribution<RealType, Policy>& dist)
- {
- return dist.mean();
- }
- template <class RealType, class Policy>
- inline RealType standard_deviation(const normal_distribution<RealType, Policy>& dist)
- {
- return dist.standard_deviation();
- }
- template <class RealType, class Policy>
- inline RealType mode(const normal_distribution<RealType, Policy>& dist)
- {
- return dist.mean();
- }
- template <class RealType, class Policy>
- inline RealType median(const normal_distribution<RealType, Policy>& dist)
- {
- return dist.mean();
- }
- template <class RealType, class Policy>
- inline RealType skewness(const normal_distribution<RealType, Policy>& )
- {
- return 0;
- }
- template <class RealType, class Policy>
- inline RealType kurtosis(const normal_distribution<RealType, Policy>& )
- {
- return 3;
- }
- template <class RealType, class Policy>
- inline RealType kurtosis_excess(const normal_distribution<RealType, Policy>& )
- {
- return 0;
- }
- template <class RealType, class Policy>
- inline RealType entropy(const normal_distribution<RealType, Policy> & dist)
- {
- using std::log;
- RealType arg = constants::two_pi<RealType>()*constants::e<RealType>()*dist.standard_deviation()*dist.standard_deviation();
- return log(arg)/2;
- }
- }
- }
- #include <boost/math/distributions/detail/derived_accessors.hpp>
- #endif
|