philox.h

Go to the documentation of this file.

//============================================================================
// vSMC/include/vsmc/rngc/philox.h
//----------------------------------------------------------------------------
//                         vSMC: Scalable Monte Carlo
//----------------------------------------------------------------------------
// Copyright (c) 2013-2016, Yan Zhou
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
//   Redistributions of source code must retain the above copyright notice,
//   this list of conditions and the following disclaimer.
//
//   Redistributions in binary form must reproduce the above copyright notice,
//   this list of conditions and the following disclaimer in the documentation
//   and/or other materials provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS AS IS
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//============================================================================

#ifndef VSMC_RNGC_PHILOX_H
#define VSMC_RNGC_PHILOX_H

#include <vsmc/internal/config.h>

typedef struct {
    uint32_t v[2];
} vsmc_philox2x32_ctr_t;

typedef struct {
    uint32_t v[4];
} vsmc_philox4x32_ctr_t;

typedef struct {
    uint32_t v[1];
} vsmc_philox2x32_key_t;

typedef struct {
    uint32_t v[2];
} vsmc_philox4x32_key_t;

typedef struct {
    uint32_t v[1];
} vsmc_philox2x32_par_t;

typedef struct {
    uint32_t v[2];
} vsmc_philox4x32_par_t;

typedef struct {
    vsmc_philox2x32_ctr_t state;
    vsmc_philox2x32_ctr_t ctr;
    vsmc_philox2x32_key_t key;
    uint32_t index;
} vsmc_philox2x32;

typedef struct {
    vsmc_philox4x32_ctr_t state;
    vsmc_philox4x32_ctr_t ctr;
    vsmc_philox4x32_key_t key;
    uint32_t index;
} vsmc_philox4x32;

static inline void vsmc_philox2x32_inc(vsmc_philox2x32_ctr_t *ctr)
{
    if (++ctr->v[0] != 0)
        return;
    if (++ctr->v[1] != 0)
        return;
}

static inline void vsmc_philox4x32_inc(vsmc_philox4x32_ctr_t *ctr)
{
    if (++ctr->v[0] != 0)
        return;
    if (++ctr->v[1] != 0)
        return;
    if (++ctr->v[2] != 0)
        return;
    if (++ctr->v[3] != 0)
        return;
}

static inline void vsmc_philox2x32_initpar(
    const vsmc_philox2x32_key_t *key, vsmc_philox2x32_par_t *par)
{
    par->v[0] = key->v[0];
}

static inline void vsmc_philox4x32_initpar(
    const vsmc_philox4x32_key_t *key, vsmc_philox4x32_par_t *par)
{
    par->v[0] = key->v[0];
    par->v[1] = key->v[1];
}

static inline void vsmc_philox2x32_bumpkey(vsmc_philox2x32_par_t *par)
{
    par->v[0] += UINT32_C(0x9E3779B9);
}

static inline void vsmc_philox4x32_bumpkey(vsmc_philox4x32_par_t *par)
{
    par->v[0] += UINT32_C(0x9E3779B9);
    par->v[1] += UINT32_C(0xBB67AE85);
}

static inline void vsmc_philox2x32_round(
    vsmc_philox2x32_ctr_t *state, const vsmc_philox2x32_par_t *par)
{
#ifdef __cplusplus
    uint64_t p = static_cast<uint64_t>(state->v[0]) * UINT64_C(0xD256D193);
    uint32_t hi = static_cast<uint32_t>(p >> 32);
    uint32_t lo = static_cast<uint32_t>(p);
#else
    uint64_t p = ((uint64_t)(state->v[0])) * UINT64_C(0xD256D193);
    uint32_t hi = (uint32_t)(p >> 32);
    uint32_t lo = (uint32_t)(p);
#endif
    hi ^= par->v[0];
    state->v[0] = hi ^ (state->v[1]);
    state->v[1] = lo;
}

static inline void vsmc_philox4x32_round(
    vsmc_philox4x32_ctr_t *state, const vsmc_philox4x32_par_t *par)
{
#ifdef __cplusplus
    uint64_t p0 = static_cast<uint64_t>(state->v[0]) * UINT64_C(0xD2511F53);
    uint64_t p2 = static_cast<uint64_t>(state->v[2]) * UINT64_C(0xCD9E8D57);
    uint32_t hi0 = static_cast<uint32_t>(p2 >> 32);
    uint32_t lo1 = static_cast<uint32_t>(p2);
    uint32_t hi2 = static_cast<uint32_t>(p0 >> 32);
    uint32_t lo3 = static_cast<uint32_t>(p0);
#else
    uint64_t p0 = ((uint64_t)(state->v[0])) * UINT64_C(0xD2511F53);
    uint64_t p2 = ((uint64_t)(state->v[2])) * UINT64_C(0xCD9E8D57);
    uint32_t hi0 = (uint32_t)(p2 >> 32);
    uint32_t lo1 = (uint32_t)(p2);
    uint32_t hi2 = (uint32_t)(p0 >> 32);
    uint32_t lo3 = (uint32_t)(p0);
#endif
    hi0 ^= par->v[0];
    hi2 ^= par->v[1];
    state->v[0] = hi0 ^ (state->v[1]);
    state->v[1] = lo1;
    state->v[2] = hi2 ^ (state->v[3]);
    state->v[3] = lo3;
}

static inline void vsmc_philox2x32_gen(const vsmc_philox2x32_ctr_t *ctr,
    const vsmc_philox2x32_key_t *key, vsmc_philox2x32_ctr_t *state)
{
    *state = *ctr;
    vsmc_philox2x32_par_t par;
    vsmc_philox2x32_initpar(key, &par);

    vsmc_philox2x32_round(state, &par); // N = 1
    vsmc_philox2x32_bumpkey(&par);      // N = 2
    vsmc_philox2x32_round(state, &par); // N = 2
    vsmc_philox2x32_bumpkey(&par);      // N = 3
    vsmc_philox2x32_round(state, &par); // N = 3
    vsmc_philox2x32_bumpkey(&par);      // N = 4
    vsmc_philox2x32_round(state, &par); // N = 4
    vsmc_philox2x32_bumpkey(&par);      // N = 5
    vsmc_philox2x32_round(state, &par); // N = 5
    vsmc_philox2x32_bumpkey(&par);      // N = 6
    vsmc_philox2x32_round(state, &par); // N = 6
    vsmc_philox2x32_bumpkey(&par);      // N = 7
    vsmc_philox2x32_round(state, &par); // N = 7
    vsmc_philox2x32_bumpkey(&par);      // N = 8
    vsmc_philox2x32_round(state, &par); // N = 8
    vsmc_philox2x32_bumpkey(&par);      // N = 9
    vsmc_philox2x32_round(state, &par); // N = 9
    vsmc_philox2x32_bumpkey(&par);      // N = 10
    vsmc_philox2x32_round(state, &par); // N = 10
}

static inline void vsmc_philox4x32_gen(const vsmc_philox4x32_ctr_t *ctr,
    const vsmc_philox4x32_key_t *key, vsmc_philox4x32_ctr_t *state)
{
    *state = *ctr;
    vsmc_philox4x32_par_t par;
    vsmc_philox4x32_initpar(key, &par);

    vsmc_philox4x32_round(state, &par); // N = 1
    vsmc_philox4x32_bumpkey(&par);      // N = 2
    vsmc_philox4x32_round(state, &par); // N = 2
    vsmc_philox4x32_bumpkey(&par);      // N = 3
    vsmc_philox4x32_round(state, &par); // N = 3
    vsmc_philox4x32_bumpkey(&par);      // N = 4
    vsmc_philox4x32_round(state, &par); // N = 4
    vsmc_philox4x32_bumpkey(&par);      // N = 5
    vsmc_philox4x32_round(state, &par); // N = 5
    vsmc_philox4x32_bumpkey(&par);      // N = 6
    vsmc_philox4x32_round(state, &par); // N = 6
    vsmc_philox4x32_bumpkey(&par);      // N = 7
    vsmc_philox4x32_round(state, &par); // N = 7
    vsmc_philox4x32_bumpkey(&par);      // N = 8
    vsmc_philox4x32_round(state, &par); // N = 8
    vsmc_philox4x32_bumpkey(&par);      // N = 9
    vsmc_philox4x32_round(state, &par); // N = 9
    vsmc_philox4x32_bumpkey(&par);      // N = 10
    vsmc_philox4x32_round(state, &par); // N = 10
}

static inline void vsmc_philox2x32_init(vsmc_philox2x32 *rng, uint32_t seed)
{
    rng->ctr.v[0] = 0;
    rng->ctr.v[1] = 0;
    rng->key.v[0] = seed;
    rng->index = 2;
}

static inline void vsmc_philox4x32_init(vsmc_philox4x32 *rng, uint32_t seed)
{
    rng->ctr.v[0] = 0;
    rng->ctr.v[1] = 0;
    rng->ctr.v[2] = 0;
    rng->ctr.v[3] = 0;
    rng->key.v[0] = seed;
    rng->key.v[1] = 0;
    rng->index = 4;
}

static inline uint32_t vsmc_philox2x32_rand(vsmc_philox2x32 *rng)
{
    if (rng->index == 2) {
        vsmc_philox2x32_inc(&rng->ctr);
        vsmc_philox2x32_gen(&rng->ctr, &rng->key, &rng->state);
        rng->index = 0;
    }

    return rng->state.v[rng->index++];
}

static inline uint32_t vsmc_philox4x32_rand(vsmc_philox4x32 *rng)
{
    if (rng->index == 4) {
        vsmc_philox4x32_inc(&rng->ctr);
        vsmc_philox4x32_gen(&rng->ctr, &rng->key, &rng->state);
        rng->index = 0;
    }

    return rng->state.v[rng->index++];
}

#endif // VSMC_RNGC_PHILOX_H