File: Rand64.h

package info (click to toggle)
contextfree 3.4%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 3,260 kB
  • sloc: cpp: 37,992; lex: 414; makefile: 123; sh: 43; python: 34
file content (121 lines) | stat: -rw-r--r-- 3,855 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// Rand64.h
// this file is part of Context Free
// ---------------------
// Copyright (C) 2008-2015 John Horigan - john@glyphic.com
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
// 
// John Horigan can be contacted at john@glyphic.com or at
// John Horigan, 1209 Villa St., Mountain View, CA 94041-1123, USA
//
//


#ifndef INCLUDE_RAND64_H
#define INCLUDE_RAND64_H

#include "xorshift64star.h"
#include <cmath>
#include <cstdint>
#include <string>

class Rand64 {
public:
    using result_type = XORshift64star::result_type;
    Rand64(result_type seed = XORshift64star::RAND64_SEED) : mSeed(seed) { }
    std::string serialize() const;
    bool operator==(const Rand64& o) const { return mSeed.mSeed == o.mSeed.mSeed; }
    bool operator!=(const Rand64& o) const { return mSeed.mSeed != o.mSeed.mSeed; }

    // Return double in [0,1)
    double getDouble()
    {
        auto next = mSeed();
        return  std::ldexp(static_cast<double>(next & 0xfffffffffffffULL), -52);
    }
    
    // Return int in [l,u]
    int64_t getInt(std::int64_t l, std::int64_t u);
    
    // Return int in [0,trials]
    int64_t getBinomial(std::int64_t trials, double prob);
    
    // Return int in [0,trials]
    int64_t getNegativeBinomial(std::int64_t trials, double prob);
    
    bool getBernoulli(double prob);
    
    int64_t getGeometric(double p);
    
    int64_t getPoisson(double mean);
    
    double getExponential(double lambda);
    
    double getGamma(double alpha, double beta);
    
    double getWeibull(double a, double b);
    
    double getExtremeValue(double a, double b);
    
    double getNormal(double mean, double stddev);
    
    double getLogNormal(double scale, double shape);
    
    double getChiSquared(double freedom);
    
    double getCauchy(double location, double scale);
    
    double getFisherF(double mfree, double nfree);
    
    double getStudentT(double freedom);
    
    int64_t getDiscrete(unsigned count, const double* weights);
    
    Rand64& operator^=(const Rand64& r)
    {
        mSeed.mSeed ^= r.mSeed.mSeed;
        return *this;
    };
    
    void seed(result_type _s = XORshift64star::RAND64_SEED)
    { mSeed.seed(_s); }
    result_type operator()() { return mSeed(); }

    
    void xorString(const char* t, int& i);
    void xorChar(unsigned char c, unsigned i);
    static double GetDouble()
    { return Common.getDouble(); }
    static std::int64_t GetInt(std::int64_t l, std::int64_t u)
    { return Common.getInt(l,u); }
    static void Seed(result_type s = XORshift64star::RAND64_SEED)
    { Common.seed(s); }
    static void XorString(const char* t, int& i)
    { Common.xorString(t, i); }
    static void XorChar(unsigned char c, unsigned i)
    { Common.xorChar(c, i); }

private:
    XORshift64star  mSeed;
    static Rand64   Common;
    static double prob(double p) { return p < 0.0 ? 0.0 : (p > 1.0 ? 1.0 : p); };
    static double pos(double p) { return p > 0.0 ? p : std::numeric_limits<double>::epsilon(); }
    static double degree(double n) { return n >= 1.0 ? std::floor(n) : 1.0; }
    static std::int64_t nat(std::int64_t i) { return i < 1 ? 1 : i; }
};

#endif  // INCLUDE_RAND64_H