Method Encode

cpp/src/arrow/compute/row/row_encoder_internal.cc:76–106

}

Status BooleanKeyEncoder::Encode(const ExecValue& data, int64_t batch_length,
                                 uint8_t** encoded_bytes) {
  auto handle_next_valid_value = [&encoded_bytes](bool value) {
    auto& encoded_ptr = *encoded_bytes++;
    *encoded_ptr++ = kValidByte;
    *encoded_ptr++ = value;
  };
  auto handle_next_null_value = [&encoded_bytes]() {
    auto& encoded_ptr = *encoded_bytes++;
    *encoded_ptr++ = kNullByte;
    *encoded_ptr++ = 0;
  };

  if (data.is_array()) {
    VisitArraySpanInline<BooleanType>(data.array, handle_next_valid_value,
                                      handle_next_null_value);
  } else {
    const auto& scalar = data.scalar_as<BooleanScalar>();
    if (!scalar.is_valid) {
      for (int64_t i = 0; i < batch_length; i++) {
        handle_next_null_value();
      }
    } else {
      const bool value = scalar.value;
      for (int64_t i = 0; i < batch_length; i++) {
        handle_next_valid_value(value);
      }
    }
  }
  return Status::OK();
}

void BooleanKeyEncoder::EncodeNull(uint8_t** encoded_bytes) {
  auto& encoded_ptr = *encoded_bytes;