-- Pseudo random number generation module for Lunatic

local ffi = require("ffi")
local ffiC = ffi.C

local rawset = rawset

local type = type
local decl = decl  -- comes from above (defs.ilua)

local print = print  -- for commented out debug block in new() below

module(...)


-- PRNG state struct
ffi.cdef[[
typedef struct {
    uint32_t x, y, z, c;
} rng_jkiss_t;

typedef union { unsigned char u[16]; double d[2]; } uchar_double_u_t;
]]

-- PRNG functions
decl[[
uint32_t rand_jkiss_u32(rng_jkiss_t *s);
double rand_jkiss_dbl(rng_jkiss_t *s);

uint32_t crc32once(uint8_t *blk, uint32_t len);
]]

local function get_rand_u32(tin)
    tin.d[0] = ffiC.gethitickms() % 1
    tin.d[1] = ffiC.gethitickms() % 1
    return ffiC.crc32once(tin.u, 16)
end

local mt = {
    __tostring = function(s)
        return "rand.new("..s.x..","..s.y..","..s.z..","..s.c..")"
    end,

    __index = {
        getu32 = ffiC.rand_jkiss_u32,
        getdbl = ffiC.rand_jkiss_dbl,

        -- Initialize the JKISS PRNG using the MD4 of the lower bits of the
        -- profiling timer.
        init_time_md4 = function(s)
            local tin = ffi.new("uchar_double_u_t")
            local tout = ffi.new("uint32_t [4]")

            repeat
                s.y = get_rand_u32()
            until (s.y ~= 0)  -- y must not be zero!

            s.x = get_rand_u32()
            s.z = get_rand_u32()
            s.c = get_rand_u32() % 698769068 + 1  -- Should be less than 698769069
        end,
    },
}
local jkiss = ffi.metatype("rng_jkiss_t", mt)

function new(x,y,z,c)
    local s
    if (x == nil or type(x)=="boolean") then
        -- initialize with arbitrary but fixed state
        s = jkiss(123456789, 987654321, 43219876, 6543217)
        if (x) then
            s:init_time_md4()
        end
    else
        s = jkiss(x,y,z,c)
    end
--[[
    print("TEST")
    local r=ffi.new("rng_jkiss_t")
    r.x = 123456789; r.y = 987654321; r.z = 43219876; r.c = 6543217;

    t=gv.gethitickms()
    for i=1,4*2*1e6 do
        ffiC.rand_jkiss_dbl(r)
    end
    print("TIME: "..gv.gethitickms()-t)  -- x86_64: approx. 100 ms
--]]

    return s
end