mirror of
https://github.com/neovim/neovim.git
synced 2025-02-25 18:55:25 -06:00
vim-patch:8.1.2356: rand() does not use the best algorithm
Problem: rand() does not use the best algorithm.
Solution: use xoshiro128** instead of xorshift. (Kaito Udagawa,
closes vim/vim#5279)
f8c1f9200c
This commit is contained in:
parent
061b06a8ae
commit
c97614d98f
@ -5533,7 +5533,7 @@ range({expr} [, {max} [, {stride}]]) *range()*
|
|||||||
GetExpr()->range()
|
GetExpr()->range()
|
||||||
<
|
<
|
||||||
rand([{expr}]) *rand()*
|
rand([{expr}]) *rand()*
|
||||||
Return a pseudo-random Number generated with an xorshift
|
Return a pseudo-random Number generated with an xoshiro128**
|
||||||
algorithm using seed {expr}. The returned number is 32 bits,
|
algorithm using seed {expr}. The returned number is 32 bits,
|
||||||
also on 64 bits systems, for consistency.
|
also on 64 bits systems, for consistency.
|
||||||
{expr} can be initialized by |srand()| and will be updated by
|
{expr} can be initialized by |srand()| and will be updated by
|
||||||
@ -7200,11 +7200,11 @@ sqrt({expr}) *sqrt()*
|
|||||||
srand([{expr}]) *srand()*
|
srand([{expr}]) *srand()*
|
||||||
Initialize seed used by |rand()|:
|
Initialize seed used by |rand()|:
|
||||||
- If {expr} is not given, seed values are initialized by
|
- If {expr} is not given, seed values are initialized by
|
||||||
time(NULL) a.k.a. epoch time. This only has second
|
reading from /dev/urandom, if possible, or using time(NULL)
|
||||||
accuracy.
|
a.k.a. epoch time otherwise; this only has second accuracy.
|
||||||
- If {expr} is given, return seed values which x element is
|
- If {expr} is given it must be a Number. It is used to
|
||||||
{expr}. This is useful for testing or when a predictable
|
initialize the seed values. This is useful for testing or
|
||||||
sequence is expected.
|
when a predictable sequence is intended.
|
||||||
|
|
||||||
Examples: >
|
Examples: >
|
||||||
:let seed = srand()
|
:let seed = srand()
|
||||||
|
@ -6982,76 +6982,79 @@ static void f_py3eval(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
|||||||
/// "rand()" function
|
/// "rand()" function
|
||||||
static void f_rand(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
static void f_rand(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
||||||
{
|
{
|
||||||
uint32_t w;
|
list_T *l = NULL;
|
||||||
#define SHUFFLE_XORSHIFT128 \
|
|
||||||
const uint32_t t = x ^ (x << 11); \
|
|
||||||
x = y; \
|
|
||||||
y = z; \
|
|
||||||
z = w; \
|
|
||||||
w = (w ^ (w >> 19)) ^ (t ^ (t >> 8));
|
|
||||||
|
|
||||||
if (argvars[0].v_type == VAR_UNKNOWN) {
|
if (argvars[0].v_type == VAR_UNKNOWN) {
|
||||||
static bool rand_seed_initialized = false;
|
static list_T *globl = NULL;
|
||||||
static uint32_t xyzw[4] = { 123456789, 362436069, 521288629, 88675123 };
|
|
||||||
|
|
||||||
// When argument is not given, return random number initialized
|
// When no argument is given use the global seed list.
|
||||||
// statically.
|
if (globl == NULL) {
|
||||||
if (!rand_seed_initialized) {
|
// Initialize the global seed list.
|
||||||
xyzw[0] = time(NULL);
|
f_srand(argvars, rettv, fptr);
|
||||||
rand_seed_initialized = true;
|
l = rettv->vval.v_list;
|
||||||
|
if (tv_list_len(l) != 4) {
|
||||||
|
tv_clear(rettv);
|
||||||
|
goto theend;
|
||||||
|
}
|
||||||
|
globl = l;
|
||||||
|
} else {
|
||||||
|
l = globl;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint32_t x = xyzw[0];
|
|
||||||
uint32_t y = xyzw[1];
|
|
||||||
uint32_t z = xyzw[2];
|
|
||||||
w = xyzw[3];
|
|
||||||
SHUFFLE_XORSHIFT128;
|
|
||||||
xyzw[0] = x;
|
|
||||||
xyzw[1] = y;
|
|
||||||
xyzw[2] = z;
|
|
||||||
xyzw[3] = w;
|
|
||||||
} else if (argvars[0].v_type == VAR_LIST) {
|
} else if (argvars[0].v_type == VAR_LIST) {
|
||||||
list_T *const l = argvars[0].vval.v_list;
|
l = argvars[0].vval.v_list;
|
||||||
if (tv_list_len(l) != 4) {
|
if (tv_list_len(l) != 4) {
|
||||||
goto theend;
|
goto theend;
|
||||||
}
|
}
|
||||||
|
|
||||||
typval_T *const tvx = TV_LIST_ITEM_TV(tv_list_find(l, 0L));
|
|
||||||
typval_T *const tvy = TV_LIST_ITEM_TV(tv_list_find(l, 1L));
|
|
||||||
typval_T *const tvz = TV_LIST_ITEM_TV(tv_list_find(l, 2L));
|
|
||||||
typval_T *const tvw = TV_LIST_ITEM_TV(tv_list_find(l, 3L));
|
|
||||||
if (tvx->v_type != VAR_NUMBER) {
|
|
||||||
goto theend;
|
|
||||||
}
|
|
||||||
if (tvy->v_type != VAR_NUMBER) {
|
|
||||||
goto theend;
|
|
||||||
}
|
|
||||||
if (tvz->v_type != VAR_NUMBER) {
|
|
||||||
goto theend;
|
|
||||||
}
|
|
||||||
if (tvw->v_type != VAR_NUMBER) {
|
|
||||||
goto theend;
|
|
||||||
}
|
|
||||||
uint32_t x = tvx->vval.v_number;
|
|
||||||
uint32_t y = tvy->vval.v_number;
|
|
||||||
uint32_t z = tvz->vval.v_number;
|
|
||||||
w = tvw->vval.v_number;
|
|
||||||
SHUFFLE_XORSHIFT128;
|
|
||||||
tvx->vval.v_number = (varnumber_T)x;
|
|
||||||
tvy->vval.v_number = (varnumber_T)y;
|
|
||||||
tvz->vval.v_number = (varnumber_T)z;
|
|
||||||
tvw->vval.v_number = (varnumber_T)w;
|
|
||||||
} else {
|
} else {
|
||||||
goto theend;
|
goto theend;
|
||||||
}
|
}
|
||||||
|
|
||||||
#undef SHUFFLE_XORSHIFT128
|
typval_T *const tvx = TV_LIST_ITEM_TV(tv_list_find(l, 0L));
|
||||||
|
typval_T *const tvy = TV_LIST_ITEM_TV(tv_list_find(l, 1L));
|
||||||
|
typval_T *const tvz = TV_LIST_ITEM_TV(tv_list_find(l, 2L));
|
||||||
|
typval_T *const tvw = TV_LIST_ITEM_TV(tv_list_find(l, 3L));
|
||||||
|
if (tvx->v_type != VAR_NUMBER) {
|
||||||
|
goto theend;
|
||||||
|
}
|
||||||
|
if (tvy->v_type != VAR_NUMBER) {
|
||||||
|
goto theend;
|
||||||
|
}
|
||||||
|
if (tvz->v_type != VAR_NUMBER) {
|
||||||
|
goto theend;
|
||||||
|
}
|
||||||
|
if (tvw->v_type != VAR_NUMBER) {
|
||||||
|
goto theend;
|
||||||
|
}
|
||||||
|
uint32_t x = tvx->vval.v_number;
|
||||||
|
uint32_t y = tvy->vval.v_number;
|
||||||
|
uint32_t z = tvz->vval.v_number;
|
||||||
|
uint32_t w = tvw->vval.v_number;
|
||||||
|
|
||||||
|
// SHUFFLE_XOSHIRO128STARSTAR
|
||||||
|
#define ROTL(x, k) ((x << k) | (x >> (32 - k)))
|
||||||
|
const uint32_t result = ROTL(y * 5, 7) * 9;
|
||||||
|
const uint32_t t = y << 9;
|
||||||
|
z ^= x;
|
||||||
|
w ^= y;
|
||||||
|
y ^= z;
|
||||||
|
x ^= w;
|
||||||
|
z ^= t;
|
||||||
|
w = ROTL(w, 11);
|
||||||
|
#undef ROTL
|
||||||
|
|
||||||
|
tvx->vval.v_number = (varnumber_T)x;
|
||||||
|
tvy->vval.v_number = (varnumber_T)y;
|
||||||
|
tvz->vval.v_number = (varnumber_T)z;
|
||||||
|
tvw->vval.v_number = (varnumber_T)w;
|
||||||
|
|
||||||
rettv->v_type = VAR_NUMBER;
|
rettv->v_type = VAR_NUMBER;
|
||||||
rettv->vval.v_number = (varnumber_T)w;
|
rettv->vval.v_number = (varnumber_T)result;
|
||||||
return;
|
return;
|
||||||
|
|
||||||
theend:
|
theend:
|
||||||
semsg(_(e_invarg2), tv_get_string(&argvars[0]));
|
semsg(_(e_invarg2), tv_get_string(&argvars[0]));
|
||||||
|
rettv->v_type = VAR_NUMBER;
|
||||||
|
rettv->vval.v_number = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// "perleval()" function
|
/// "perleval()" function
|
||||||
@ -10529,6 +10532,7 @@ static void f_stdpath(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
|||||||
static void f_srand(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
static void f_srand(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
||||||
{
|
{
|
||||||
static int dev_urandom_state = -1; // FAIL or OK once tried
|
static int dev_urandom_state = -1; // FAIL or OK once tried
|
||||||
|
uint32_t x = 0;
|
||||||
|
|
||||||
tv_list_alloc_ret(rettv, 4);
|
tv_list_alloc_ret(rettv, 4);
|
||||||
if (argvars[0].v_type == VAR_UNKNOWN) {
|
if (argvars[0].v_type == VAR_UNKNOWN) {
|
||||||
@ -10550,26 +10554,35 @@ static void f_srand(typval_T *argvars, typval_T *rettv, FunPtr fptr)
|
|||||||
dev_urandom_state = FAIL;
|
dev_urandom_state = FAIL;
|
||||||
} else {
|
} else {
|
||||||
dev_urandom_state = OK;
|
dev_urandom_state = OK;
|
||||||
tv_list_append_number(rettv->vval.v_list, (varnumber_T)buf.cont.number);
|
x = buf.cont.number;
|
||||||
}
|
}
|
||||||
os_close(fd);
|
os_close(fd);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (dev_urandom_state != OK) {
|
if (dev_urandom_state != OK) {
|
||||||
// Reading /dev/urandom doesn't work, fall back to time().
|
// Reading /dev/urandom doesn't work, fall back to time().
|
||||||
tv_list_append_number(rettv->vval.v_list, (varnumber_T)time(NULL));
|
x = time(NULL);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
bool error = false;
|
bool error = false;
|
||||||
const uint32_t x = tv_get_number_chk(&argvars[0], &error);
|
x = tv_get_number_chk(&argvars[0], &error);
|
||||||
if (error) {
|
if (error) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
tv_list_append_number(rettv->vval.v_list, (varnumber_T)x);
|
|
||||||
}
|
}
|
||||||
tv_list_append_number(rettv->vval.v_list, 362436069);
|
|
||||||
tv_list_append_number(rettv->vval.v_list, 521288629);
|
uint32_t z;
|
||||||
tv_list_append_number(rettv->vval.v_list, 88675123);
|
#define SPLITMIX32 ( \
|
||||||
|
z = (x += 0x9e3779b9), \
|
||||||
|
z = (z ^ (z >> 16)) * 0x85ebca6b, \
|
||||||
|
z = (z ^ (z >> 13)) * 0xc2b2ae35, \
|
||||||
|
z ^ (z >> 16))
|
||||||
|
|
||||||
|
tv_list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32);
|
||||||
|
tv_list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32);
|
||||||
|
tv_list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32);
|
||||||
|
tv_list_append_number(rettv->vval.v_list, (varnumber_T)SPLITMIX32);
|
||||||
|
#undef SPLITMIX32
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
@ -2,12 +2,12 @@
|
|||||||
|
|
||||||
func Test_Rand()
|
func Test_Rand()
|
||||||
let r = srand(123456789)
|
let r = srand(123456789)
|
||||||
call assert_equal([123456789, 362436069, 521288629, 88675123], r)
|
call assert_equal([1573771921, 319883699, 2742014374, 1324369493], r)
|
||||||
call assert_equal(3701687786, rand(r))
|
call assert_equal(4284103975, rand(r))
|
||||||
call assert_equal(458299110, rand(r))
|
call assert_equal(1001954530, rand(r))
|
||||||
call assert_equal(2500872618, rand(r))
|
call assert_equal(2701803082, rand(r))
|
||||||
call assert_equal(3633119408, rand(r))
|
call assert_equal(2658065534, rand(r))
|
||||||
call assert_equal(516391518, rand(r))
|
call assert_equal(3104308804, rand(r))
|
||||||
|
|
||||||
" Nvim does not support test_settime
|
" Nvim does not support test_settime
|
||||||
" call test_settime(12341234)
|
" call test_settime(12341234)
|
||||||
|
Loading…
Reference in New Issue
Block a user