Function FillWithRle

be/src/benchmarks/rle-benchmark.cc:61–81

RLE encodes NUM_OUT_VALUES number of bytes into the buffer. The length of runs are pseudo random between 1 and max_run_length.

/// RLE encodes NUM_OUT_VALUES number of bytes into the buffer.
/// The length of runs are pseudo random between 1 and max_run_length.
int FillWithRle(vector<uint8_t>* buffer, int bit_width, int max_run_length) {
  RleEncoder encoder(buffer->data(), buffer->size(), bit_width);

  uniform_int_distribution<int> uniform_dist(1, max_run_length);
  minstd_rand rand_engine;
  uint8_t val = 0;
  int run_length = 0;
  for (int i = 0; i < NUM_OUT_VALUES; ++i) {
    if (!encoder.Put(val)) {
      LOG(ERROR) << Substitute(
          "Error during RLE encoding. bit_widths: $0 max_run_length: $1",
          bit_width, max_run_length);
    }
    if (run_length == 0) {
      run_length = uniform_dist(rand_engine);
      val = (val + 1) % (1 << bit_width);
    }
    --run_length;
  }
  return encoder.Flush();
}

struct BenchmarkParams {
  int bit_width;