summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source/kit/mersenne_twister_64.c16
-rw-r--r--source/kit/mersenne_twister_64.h4
2 files changed, 16 insertions, 4 deletions
diff --git a/source/kit/mersenne_twister_64.c b/source/kit/mersenne_twister_64.c
index 9ab39d3..c8bbbe2 100644
--- a/source/kit/mersenne_twister_64.c
+++ b/source/kit/mersenne_twister_64.c
@@ -2,9 +2,12 @@
#include "time.h"
-void kit_mt64_init(kit_mt64_state_t *state, uint64_t seed) {
- state->mt[0] = seed;
- for (state->index = 1; state->index < KIT_MT64_N; state->index++)
+void kit_mt64_init_array(kit_mt64_state_t *const state,
+ ptrdiff_t const size,
+ uint64_t const *const seed) {
+ for (ptrdiff_t i = 0; i < size && i < KIT_MT64_N; i++)
+ state->mt[i] = seed[i];
+ for (state->index = size; state->index < KIT_MT64_N; state->index++)
state->mt[state->index] = (6364136223846793005ull *
(state->mt[state->index - 1] ^
(state->mt[state->index - 1] >>
@@ -12,7 +15,12 @@ void kit_mt64_init(kit_mt64_state_t *state, uint64_t seed) {
state->index);
}
-uint64_t kit_mt64_generate(kit_mt64_state_t *state) {
+void kit_mt64_init(kit_mt64_state_t *const state,
+ uint64_t const seed) {
+ kit_mt64_init_array(state, 1, &seed);
+}
+
+uint64_t kit_mt64_generate(kit_mt64_state_t *const state) {
static uint64_t const mag01[2] = { 0ull, 0xB5026F5AA96619E9ull };
int i;
diff --git a/source/kit/mersenne_twister_64.h b/source/kit/mersenne_twister_64.h
index 416ef27..fb509c2 100644
--- a/source/kit/mersenne_twister_64.h
+++ b/source/kit/mersenne_twister_64.h
@@ -17,6 +17,9 @@ typedef struct {
uint64_t index;
} kit_mt64_state_t;
+void kit_mt64_init_array(kit_mt64_state_t *state, ptrdiff_t size,
+ uint64_t const *seed);
+
void kit_mt64_init(kit_mt64_state_t *state, uint64_t seed);
uint64_t kit_mt64_generate(kit_mt64_state_t *state);
@@ -25,6 +28,7 @@ uint64_t kit_mt64_seed();
#ifndef KIT_DISABLE_SHORT_NAMES
# define mt64_state_t kit_mt64_state_t
+# define mt64_init_array kit_mt64_init_array
# define mt64_init kit_mt64_init
# define mt64_generate kit_mt64_generate
# define mt64_seed kit_mt64_seed