You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
130 lines
3.2 KiB
130 lines
3.2 KiB
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <assert.h>
|
|
#include <stdio.h>
|
|
|
|
#include "crand.h"
|
|
|
|
typedef unsigned short uint48_t[3];
|
|
|
|
|
|
struct jr_state
|
|
{
|
|
union {
|
|
long result;
|
|
double fresult;
|
|
};
|
|
struct drand48_data data;
|
|
union {
|
|
unsigned short xsubi[3];
|
|
struct {
|
|
uint64_t xsubh : 48;
|
|
uint16_t _xsub : 16;
|
|
};
|
|
uint64_t xsubl;
|
|
} st;
|
|
};
|
|
_Static_assert( sizeof(uint48_t) == (sizeof(uint16_t) * 3), "bad uint48 (ushort[3])");
|
|
_Static_assert( sizeof(((struct jr_state*)NULL)->st) == sizeof(uint64_t), "bad uint64 (union st)");
|
|
|
|
static unsigned short* _impl__jr_st_resolv__low(struct jr_state* restrict state)
|
|
{
|
|
return state->st.xsubi;
|
|
}
|
|
|
|
static unsigned short* _impl__jr_st_resolv__high(struct jr_state* restrict state)
|
|
{
|
|
return state->st.xsubi+1;
|
|
}
|
|
|
|
static unsigned short* (*_ifun__jr_st_resolv (void)) (struct jr_state* restrict state)
|
|
{
|
|
struct jr_state chk = {0};
|
|
chk.st.xsubh = JR_MAX;
|
|
return chk.st._xsub
|
|
? &_impl__jr_st_resolv__high
|
|
: &_impl__jr_st_resolv__low;
|
|
}
|
|
|
|
static unsigned short* _jr_st_resolv(struct jr_state* restrict state)
|
|
__attribute__((ifunc("_ifun__jr_st_resolv")));
|
|
|
|
void _jr_seed(struct jr_state* restrict state, unsigned long with)
|
|
{
|
|
state->st.xsubh = with;
|
|
seed48_r(_jr_st_resolv(state), &state->data);
|
|
}
|
|
|
|
long _jr_proc(struct jr_state* restrict state)
|
|
{
|
|
jrand48_r(_jr_st_resolv(state), &state->data, &state->result);
|
|
return state->result;
|
|
}
|
|
|
|
//TODO: a version of _jr_proc that uses use `erand48_r()` to return between 0..1.0 for _sample()
|
|
|
|
double _jr_procf(struct jr_state* restrict state)
|
|
{
|
|
erand48_r(_jr_st_resolv(state), &state->data, &state->fresult);
|
|
return state->fresult;
|
|
}
|
|
|
|
struct jr_state* _jr_alloc()
|
|
{
|
|
struct jr_state* bx = aligned_alloc(_Alignof(struct jr_state), sizeof(struct jr_state));
|
|
memset(bx, 0, sizeof(struct jr_state));
|
|
|
|
return bx;
|
|
}
|
|
|
|
struct jr_state* _jr_new(unsigned long with)
|
|
{
|
|
struct jr_state* state = _jr_alloc();
|
|
_jr_seed(state, with);
|
|
return state;
|
|
}
|
|
|
|
void _jr_free(struct jr_state* restrict state)
|
|
{
|
|
free(state);
|
|
}
|
|
|
|
//TODO: Use test macros from tracemac, run with attr(ctor) if `-DTEST` is provided.
|
|
void __TEST__jr_test()
|
|
{
|
|
struct jr_state* st = _jr_alloc();
|
|
assert(!st->st._xsub);
|
|
_jr_seed(st, ~0UL);
|
|
const volatile unsigned short* res_state = _jr_st_resolv(st);
|
|
printf("seeded: %lu (full %lu, spill %u). xsubi = [%04x, %04x, %04x) %04x], resolv = [%04x, %04x, %04x) %04x]\n", (uint64_t)st->st.xsubh, st->st.xsubl, st->st._xsub,
|
|
st->st.xsubi[0],
|
|
st->st.xsubi[1],
|
|
st->st.xsubi[2],
|
|
st->st.xsubi[3],
|
|
|
|
res_state[0],
|
|
res_state[1],
|
|
res_state[2],
|
|
res_state[3]);
|
|
assert(!st->st._xsub);
|
|
for(int i=0;i<10;i++)
|
|
{
|
|
printf("res: %ld\n", _jr_proc(st));
|
|
printf("state: %lu (full %lu, spill %u). xsubi = %p, resolv = %p\n", (uint64_t)st->st.xsubh, st->st.xsubl, st->st._xsub,
|
|
(const void*)st->st.xsubi, (const void*)_jr_st_resolv(st));
|
|
}
|
|
printf("ended: %lu (full %lu, spill %u). xsubi = [%04x, %04x, %04x) %04x], resolv = [%04x, %04x, %04x) %04x]\n", (uint64_t)st->st.xsubh, st->st.xsubl, st->st._xsub,
|
|
st->st.xsubi[0],
|
|
st->st.xsubi[1],
|
|
st->st.xsubi[2],
|
|
st->st.xsubi[3],
|
|
|
|
res_state[0],
|
|
res_state[1],
|
|
res_state[2],
|
|
res_state[3]);
|
|
assert(!st->st._xsub);
|
|
_jr_free(st);
|
|
}
|