#include <stdio.h>
#include <time.h>
#include <random>
#include "stc/crand.h"
#include "others/pcg_random.hpp"

enum {N = 1000000000};

void test1(void)
{
    clock_t diff, before;
    uint64_t sum;

    std::random_device device;
    std::mt19937 rng(device());
    std::uniform_int_distribution<int> idist(1, 10);
    std::uniform_real_distribution<double> fdist(1, 10);

    before = clock();
    sum = 0;
    c_forrange (N) {
        sum += rng();
    }
    diff = clock() - before;
    printf("std::random:\t\t%.02f, %zu, sz:%zu\n", (float) diff / CLOCKS_PER_SEC, sum, sizeof rng);

    before = clock();
    sum = 0;
    c_forrange (N) {
        sum += idist(rng);
    }
    diff = clock() - before;
    printf("std::uniform:\t\t%.02f, %zu\n\n", (float) diff / CLOCKS_PER_SEC, sum);

    c_forrange (30) printf("%02d ", idist(rng));
    puts("");
    c_forrange (8) printf("%f ", fdist(rng));
    puts("\n");
}

void test2()
{
    clock_t diff, before;
    uint64_t sum;

    // Seed with a real random value, if available
    pcg_extras::seed_seq_from<std::random_device> seed_source;

    // Make a random number engine
    pcg64 rng(seed_source);

    // Choose a random mean between 1 and 10
    std::uniform_int_distribution<int> idist(1, 10);
    std::uniform_real_distribution<double> fdist(1, 10);

   before = clock();
    sum = 0;
    c_forrange (N) {
        sum += rng();
    }
    diff = clock() - before;
    printf("pcg64::random:\t\t%.02f, %zu, sz:%zu\n", (float) diff / CLOCKS_PER_SEC, sum, sizeof rng);

    before = clock();
    sum = 0;
    c_forrange (N) {
        sum += idist(rng);
    }
    diff = clock() - before;
    printf("pcg64::uniform:\t\t%.02f, %zu\n\n", (float) diff / CLOCKS_PER_SEC, sum);

    c_forrange (30) printf("%02d ", idist(rng));
    puts("");
    c_forrange (8) printf("%f ", fdist(rng));
    puts("\n");
}


void test3(void)
{
    clock_t diff, before;
    uint64_t sum;

    stc64_t rng = stc64_init(time(NULL));
    stc64_uniform_t idist = stc64_uniform_init(1, 10);
    stc64_uniformf_t fdist = stc64_uniformf_init(1, 10);

    before = clock();
    sum = 0;
    c_forrange (N) {
        sum += stc64_rand(&rng);
    }
    diff = clock() - before;
    printf("stc64_random:\t\t%.02f, %zu sz:%zu\n", (float) diff / CLOCKS_PER_SEC, sum, sizeof rng);

    before = clock();
    sum = 0;
    c_forrange (N) {
        sum += stc64_uniform(&rng, &idist);
    }
    diff = clock() - before;
    printf("stc64_uniform:\t\t%.02f, %zu\n\n", (float) diff / CLOCKS_PER_SEC, sum);

    c_forrange (30) printf("%02zd ", stc64_uniform(&rng, &idist));
    puts("");
    c_forrange (8) printf("%f ", stc64_uniformf(&rng, &fdist));
    puts("\n");
}

int main()
{
    test1();
    test2();
    test3();
}