Function rk_random_uint16

numpy/random/src/mt19937/randomkit.c:333–366

* Fills an array with cnt random npy_uint16 between off and off + rng * inclusive. The numbers wrap if rng is sufficiently large. */

 * inclusive. The numbers wrap if rng is sufficiently large.
 */
void rk_random_uint16(npy_uint16 off, npy_uint16 rng, npy_intp cnt,
                      npy_uint16 *out, rk_state *state) {
  npy_uint16 val, mask = rng;
  npy_intp i;
  npy_uint32 buf;
  int bcnt = 0;

  if (rng == 0) {
    for (i = 0; i < cnt; i++) {
      out[i] = off;
    }
    return;
  }

  /* Smallest bit mask >= max */
  mask |= mask >> 1;
  mask |= mask >> 2;
  mask |= mask >> 4;
  mask |= mask >> 8;

  for (i = 0; i < cnt; i++) {
    do {
      if (!bcnt) {
        buf = rk_uint32(state);
        bcnt = 1;
      } else {
        buf >>= 16;
        bcnt--;
      }
      val = (npy_uint16)buf & mask;
    } while (val > rng);
    out[i] = off + val;
  }
}

/*
 * Fills an array with cnt random npy_uint8 between off and off + rng