Function DecodeLevels

cpp/src/parquet/column_reader_benchmark.cc:309–337

}

static void DecodeLevels(Encoding::type level_encoding, int16_t max_level, int num_levels,
                         int batch_size, int level_repeat_count,
                         ::benchmark::State& state) {
  std::vector<uint8_t> bytes;
  {
    std::vector<int16_t> input_levels;
    GenerateLevels(/*level_repeats=*/level_repeat_count, /*max_repeat_factor=*/max_level,
                   num_levels, &input_levels);
    EncodeLevels(level_encoding, max_level, num_levels, input_levels.data(), &bytes);
  }

  LevelDecoder decoder;
  std::vector<int16_t> output_levels(batch_size);
  for (auto _ : state) {
    state.PauseTiming();
    decoder.SetData(level_encoding, max_level, num_levels, bytes.data(),
                    static_cast<int>(bytes.size()));
    state.ResumeTiming();
    // Decode multiple times with batch_size
    while (true) {
      int levels_decoded = decoder.Decode(batch_size, output_levels.data());
      if (levels_decoded == 0) {
        break;
      }
    }
  }
  state.SetBytesProcessed(state.iterations() * num_levels * sizeof(int16_t));
  state.SetItemsProcessed(state.iterations() * num_levels);
}

static void ReadLevels_Rle(::benchmark::State& state) {
  int16_t max_level = static_cast<int16_t>(state.range(0));