Method Plain

cpp/src/parquet/encoding_test.cc:922–955

  static std::shared_ptr<::arrow::DataType> arrow_type();

  void Plain(int seed, int rounds = 1, int offset = 0) {
    auto random_array = GetValues(seed)->Slice(offset);
    auto encoder = MakeTypedEncoder<ParquetType>(
        Encoding::PLAIN, /*use_dictionary=*/false, column_descr());
    auto decoder = MakeTypedDecoder<ParquetType>(Encoding::PLAIN, column_descr());

    for (int i = 0; i < rounds; ++i) {
      ASSERT_NO_THROW(encoder->Put(*random_array));
    }
    std::shared_ptr<::arrow::Array> values;
    if (rounds == 1) {
      values = random_array;
    } else {
      ::arrow::ArrayVector arrays(rounds, random_array);
      EXPECT_OK_AND_ASSIGN(values,
                           ::arrow::Concatenate(arrays, ::arrow::default_memory_pool()));
    }
    auto buf = encoder->FlushValues();

    decoder->SetData(static_cast<int>(values->length()), buf->data(),
                     static_cast<int>(buf->size()));

    BuilderType acc(arrow_type(), ::arrow::default_memory_pool());
    ASSERT_EQ(static_cast<int>(values->length() - values->null_count()),
              decoder->DecodeArrow(static_cast<int>(values->length()),
                                   static_cast<int>(values->null_count()),
                                   values->null_bitmap_data(), values->offset(), &acc));

    std::shared_ptr<::arrow::Array> result;
    ASSERT_OK(acc.Finish(&result));
    ASSERT_OK(result->ValidateFull());
    ASSERT_EQ(values->length(), result->length());
    ::arrow::AssertArraysEqual(*values, *result, /*verbose=*/true);
  }

  void ByteStreamSplit(int seed) {
    if constexpr (!std::is_same_v<ParquetType, FloatType> &&