Blame src/utils/random_utils.h

Packit 9c6abc
// Copyright 2013 Google Inc. All Rights Reserved.
Packit 9c6abc
//
Packit 9c6abc
// Use of this source code is governed by a BSD-style license
Packit 9c6abc
// that can be found in the COPYING file in the root of the source
Packit 9c6abc
// tree. An additional intellectual property rights grant can be found
Packit 9c6abc
// in the file PATENTS. All contributing project authors may
Packit 9c6abc
// be found in the AUTHORS file in the root of the source tree.
Packit 9c6abc
// -----------------------------------------------------------------------------
Packit 9c6abc
//
Packit 9c6abc
// Pseudo-random utilities
Packit 9c6abc
//
Packit 9c6abc
// Author: Skal (pascal.massimino@gmail.com)
Packit 9c6abc
Packit 9c6abc
#ifndef WEBP_UTILS_RANDOM_UTILS_H_
Packit 9c6abc
#define WEBP_UTILS_RANDOM_UTILS_H_
Packit 9c6abc
Packit 9c6abc
#include <assert.h>
Packit 9c6abc
#include "src/webp/types.h"
Packit 9c6abc
Packit 9c6abc
#ifdef __cplusplus
Packit 9c6abc
extern "C" {
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
#define VP8_RANDOM_DITHER_FIX 8   // fixed-point precision for dithering
Packit 9c6abc
#define VP8_RANDOM_TABLE_SIZE 55
Packit 9c6abc
Packit 9c6abc
typedef struct {
Packit 9c6abc
  int index1_, index2_;
Packit 9c6abc
  uint32_t tab_[VP8_RANDOM_TABLE_SIZE];
Packit 9c6abc
  int amp_;
Packit 9c6abc
} VP8Random;
Packit 9c6abc
Packit 9c6abc
// Initializes random generator with an amplitude 'dithering' in range [0..1].
Packit 9c6abc
void VP8InitRandom(VP8Random* const rg, float dithering);
Packit 9c6abc
Packit 9c6abc
// Returns a centered pseudo-random number with 'num_bits' amplitude.
Packit 9c6abc
// (uses D.Knuth's Difference-based random generator).
Packit 9c6abc
// 'amp' is in VP8_RANDOM_DITHER_FIX fixed-point precision.
Packit 9c6abc
static WEBP_INLINE int VP8RandomBits2(VP8Random* const rg, int num_bits,
Packit 9c6abc
                                      int amp) {
Packit 9c6abc
  int diff;
Packit 9c6abc
  assert(num_bits + VP8_RANDOM_DITHER_FIX <= 31);
Packit 9c6abc
  diff = rg->tab_[rg->index1_] - rg->tab_[rg->index2_];
Packit 9c6abc
  if (diff < 0) diff += (1u << 31);
Packit 9c6abc
  rg->tab_[rg->index1_] = diff;
Packit 9c6abc
  if (++rg->index1_ == VP8_RANDOM_TABLE_SIZE) rg->index1_ = 0;
Packit 9c6abc
  if (++rg->index2_ == VP8_RANDOM_TABLE_SIZE) rg->index2_ = 0;
Packit 9c6abc
  // sign-extend, 0-center
Packit 9c6abc
  diff = (int)((uint32_t)diff << 1) >> (32 - num_bits);
Packit 9c6abc
  diff = (diff * amp) >> VP8_RANDOM_DITHER_FIX;  // restrict range
Packit 9c6abc
  diff += 1 << (num_bits - 1);                   // shift back to 0.5-center
Packit 9c6abc
  return diff;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
static WEBP_INLINE int VP8RandomBits(VP8Random* const rg, int num_bits) {
Packit 9c6abc
  return VP8RandomBits2(rg, num_bits, rg->amp_);
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#ifdef __cplusplus
Packit 9c6abc
}    // extern "C"
Packit 9c6abc
#endif
Packit 9c6abc
Packit 9c6abc
#endif  /* WEBP_UTILS_RANDOM_UTILS_H_ */