git.pond.sub.org Git - empserver/blob - src/lib/gen/chance.c

   1 /*
   2  *  Empire - A multi-player, client/server Internet based war game.
   3  *  Copyright (C) 1986-2013, Dave Pare, Jeff Bailey, Thomas Ruschak,
   4  *                Ken Stevens, Steve McClure, Markus Armbruster
   5  *
   6  *  Empire is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  *
  19  *  ---
  20  *
  21  *  See files README, COPYING and CREDITS in the root of the source
  22  *  tree for related information and legal notices.  It is expected
  23  *  that future projects/authors will amend these files as needed.
  24  *
  25  *  ---
  26  *
  27  *  chance.c: Roll dice
  28  *
  29  *  Known contributors to this file:
  30  *     Markus Armbruster, 2006-2012
  31  */
  32
  33 #include <config.h>
  34
  35 #include <math.h>
  36 #include <stdlib.h>
  37 #include "chance.h"
  38 #include "mt19937ar.h"
  39
  40 /*
  41  * Return non-zero with probability D.
  42  */
  43 int
  44 chance(double d)
  45 {
  46     return d > genrand_real2();
  47 }
  48
  49 /*
  50  * Return non-zero with probability PCT%.
  51  */
  52 int
  53 pct_chance(int pct)
  54 {
  55     return roll(100) <= pct;
  56 }
  57
  58 static unsigned
  59 round_up_to_pow2(unsigned val)
  60 {
  61     val--;
  62     val |= val >> 1;
  63     val |= val >> 2;
  64     val |= val >> 4;
  65     val |= val >> 8;
  66     val |= val >> 16;
  67     val++;
  68     return val;
  69 }
  70
  71 /*
  72  * Return a random number in [0..N-1].
  73  * N must be in [1..2^31-1].
  74  */
  75 int
  76 roll0(int n)
  77 {
  78     unsigned pow2 = round_up_to_pow2(n);
  79     int r;
  80
  81     do
  82         r = genrand_int32() & (pow2 - 1);
  83     while (r >= n);
  84     return r;
  85 }
  86
  87 /*
  88  * Return a random number in [1..N].
  89  * N must be in [0..2^31-1].
  90  */
  91 int
  92 roll(int n)
  93 {
  94     return 1 + roll0(n);
  95 }
  96
  97 /*
  98  * Round VAL to nearest integer (on the average).
  99  * VAL's fractional part is chance to round up.
 100  */
 101 int
 102 roundavg(double val)
 103 {
 104     double flr = floor(val);
 105     return (int)(flr + chance(val - flr));
 106 }
 107
 108 /*
 109  * Seed the pseudo-random number generator with SEED.
 110  * The sequence of pseudo-random numbers is repeatable by seeding it
 111  * with the same value.
 112  */
 113 void
 114 seed_prng(unsigned seed)
 115 {
 116     init_genrand(seed);
 117 }