54 lines
1.1 KiB
C
54 lines
1.1 KiB
C
|
|
|||
|
|
#pragma once
|
|||
|
|
/**
|
|||
|
|
* @file RandT.h
|
|||
|
|
* @brief 随机数
|
|||
|
|
* @author your name (you@domain.com)
|
|||
|
|
* @version 0.1
|
|||
|
|
* @date 2023-12-22
|
|||
|
|
*
|
|||
|
|
* Copyright: Baosight Co. Ltd.
|
|||
|
|
* DO NOT COPY/USE WITHOUT PERMISSION
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
#include <time.h>
|
|||
|
|
#include <cmath>
|
|||
|
|
namespace RandT {
|
|||
|
|
/**
|
|||
|
|
* @brief 生成指定范围内的随机数
|
|||
|
|
* @tparam T
|
|||
|
|
* @param _min min
|
|||
|
|
* @param _max max
|
|||
|
|
* @return T
|
|||
|
|
*/
|
|||
|
|
template <typename T>
|
|||
|
|
T RandT(T _min, T _max) {
|
|||
|
|
T temp;
|
|||
|
|
if (_min > _max) {
|
|||
|
|
temp = _max;
|
|||
|
|
_max = _min;
|
|||
|
|
_min = temp;
|
|||
|
|
}
|
|||
|
|
return rand() / (double)RAND_MAX * (_max - _min) + _min;
|
|||
|
|
}
|
|||
|
|
/**
|
|||
|
|
* @brief 生成高斯分布的随机数
|
|||
|
|
* @param mean 均值
|
|||
|
|
* @param sigma 标准差
|
|||
|
|
* @return double
|
|||
|
|
*/
|
|||
|
|
double GuassRand(double mean = 0, double sigma = 0) {
|
|||
|
|
double U1 = rand() * 1.0f / RAND_MAX; // 0~1均匀分布
|
|||
|
|
double U2 = rand() * 1.0f / RAND_MAX; // 0~1均匀分布
|
|||
|
|
double Z =
|
|||
|
|
sqrt(-2 * log(U1)) * cos(2 * M_PI * U2); // 均值为0，方差为1的正态分布
|
|||
|
|
double Y;
|
|||
|
|
if (sigma >= 0) {
|
|||
|
|
Y = mean + sqrt(sigma) * Z;
|
|||
|
|
} else {
|
|||
|
|
Y = mean + Z;
|
|||
|
|
}
|
|||
|
|
return Y;
|
|||
|
|
}
|
|||
|
|
} // namespace RandT
|