GH-50027: [Format][C++][Python] Add fixed and variable closedness range canonical extension types

cpp/src/arrow/CMakeLists.txt

@@ -1116,6 +1116,7 @@ if(ARROW_JSON)
   arrow_add_object_library(ARROW_JSON
                            extension/fixed_shape_tensor.cc
                            extension/opaque.cc
+                           extension/range.cc
                            extension/tensor_internal.cc
                            extension/variable_shape_tensor.cc
                            json/options.cc

cpp/src/arrow/extension/CMakeLists.txt

@@ -18,7 +18,11 @@
 set(CANONICAL_EXTENSION_TESTS bool8_test.cc json_test.cc uuid_test.cc)
 
 if(ARROW_JSON)
-  list(APPEND CANONICAL_EXTENSION_TESTS tensor_extension_array_test.cc opaque_test.cc)
+  list(APPEND
+       CANONICAL_EXTENSION_TESTS
+       tensor_extension_array_test.cc
+       opaque_test.cc
+       range_test.cc)
 endif()
 
 add_arrow_test(test

cpp/src/arrow/extension/meson.build

@@ -21,6 +21,7 @@ if needs_json
     canonical_extension_tests += [
         'tensor_extension_array_test.cc',
         'opaque_test.cc',
+        'range_test.cc',
     ]
 endif
 
@@ -38,6 +39,7 @@ install_headers(
         'json.h',
         'opaque.h',
         'parquet_variant.h',
+        'range.h',
         'uuid.h',
         'variable_shape_tensor.h',
     ],

cpp/src/arrow/extension/range.cc

@@ -0,0 +1,344 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+#include "arrow/extension/range.h"
+
+#include <sstream>
+#include <string_view>
+
+#include "arrow/util/logging_internal.h"
+#include "arrow/util/simdjson_internal.h"
+
+#include <simdjson.h>
+
+using ::arrow::internal::JsonWriter;
+
+namespace arrow::extension {
+
+namespace {
+
+/// Map RangeClosed -> the JSON string value used in serialization.
+std::string_view ClosedToString(RangeClosed closed) {
+  switch (closed) {
+    case RangeClosed::Left:
+      return "left";
+    case RangeClosed::Right:
+      return "right";
+    case RangeClosed::Both:
+      return "both";
+    case RangeClosed::Neither:
+      return "neither";
+  }
+  // unreachable
+  return "right";
+}
+
+/// Parse the JSON "closed" string into a RangeClosed enum.
+/// Returns an error if the string is not one of the four valid values.
+Result<RangeClosed> ClosedFromString(std::string_view s) {
+  if (s == "left") return RangeClosed::Left;
+  if (s == "right") return RangeClosed::Right;
+  if (s == "both") return RangeClosed::Both;
+  if (s == "neither") return RangeClosed::Neither;
+  return Status::Invalid(
+      "Invalid value for FixedClosednessRangeType \"closed\" parameter: \"", s,
+      "\". Expected one of: \"left\", \"right\", \"both\", \"neither\".");
+}
+
+/// Build the storage Struct type for a given value subtype.
+std::shared_ptr<DataType> MakeFixedClosednessStorageType(
+    const std::shared_ptr<DataType>& value_type, bool allow_unbounded) {
+  // Nullable bounds can represent an unbounded (infinite) endpoint; non-nullable
+  // bounds are always finite.
+  return struct_({field("lower", value_type, allow_unbounded),
+                  field("upper", value_type, allow_unbounded)});
+}
+
+}  // namespace
+
+// ---------------------------------------------------------------------------
+// FixedClosednessRangeType
+
+std::shared_ptr<DataType> FixedClosednessRangeType::value_type() const {
+  // storage_type() is a struct with two fields; both share the same type.
+  return internal::checked_cast<const StructType&>(*storage_type()).field(0)->type();
+}
+
+std::string FixedClosednessRangeType::ToString(bool show_metadata) const {
+  std::stringstream ss;
+  ss << "extension<" << this->extension_name()
+     << "[value_type=" << value_type()->ToString(show_metadata)
+     << ", closed=" << ClosedToString(closed_) << "]>";
+  return ss.str();
+}
+
+bool FixedClosednessRangeType::ExtensionEquals(const ExtensionType& other) const {
+  if (extension_name() != other.extension_name()) {
+    return false;
+  }
+  const auto& other_range =
+      internal::checked_cast<const FixedClosednessRangeType&>(other);
+  return storage_type()->Equals(*other_range.storage_type()) &&
+         closed_ == other_range.closed_;
+}
+
+std::string FixedClosednessRangeType::Serialize() const {
+  JsonWriter writer;
+
+  writer.StartObject();
+  writer.StringField("closed", ClosedToString(closed_));
+  writer.EndObject();
+
+  Result<std::string_view> json = writer.GetString();
+  // can only fail in OutOfMemory scenarios
+  ARROW_CHECK_OK(json.status());
+  return std::string(*json);
+}
+
+Result<std::shared_ptr<DataType>> FixedClosednessRangeType::Deserialize(
+    std::shared_ptr<DataType> storage_type, const std::string& serialized_data) const {
+  // Validate storage type structure.
+  if (storage_type->id() != Type::STRUCT) {
+    return Status::Invalid("FixedClosednessRangeType storage type must be a Struct, got ",
+                           storage_type->ToString());
+  }
+  const auto& struct_type = internal::checked_cast<const StructType&>(*storage_type);
+  if (struct_type.num_fields() != 2) {
+    return Status::Invalid(
+        "FixedClosednessRangeType storage Struct must have exactly 2 fields, got ",
+        struct_type.num_fields());
+  }
+  const auto& lower_field = struct_type.field(0);
+  const auto& upper_field = struct_type.field(1);
+  if (lower_field->name() != "lower") {
+    return Status::Invalid(
+        "FixedClosednessRangeType storage Struct field 0 must be named \"lower\", got \"",
+        lower_field->name(), "\"");
+  }
+  if (upper_field->name() != "upper") {
+    return Status::Invalid(
+        "FixedClosednessRangeType storage Struct field 1 must be named \"upper\", got \"",
+        upper_field->name(), "\"");
+  }
+  if (!lower_field->type()->Equals(*upper_field->type())) {
+    return Status::Invalid(
+        "FixedClosednessRangeType storage Struct fields \"lower\" and \"upper\" must "
+        "have the same "
+        "type, got \"",
+        lower_field->type()->ToString(), "\" and \"", upper_field->type()->ToString(),
+        "\"");
+  }
+
+  // Parse the required "closed" parameter from JSON metadata. The closedness
+  // is not defaulted on the wire: empty metadata or a missing key is invalid.
+  if (serialized_data.empty()) {
+    return Status::Invalid(
+        "FixedClosednessRangeType metadata must be a JSON object with a required "
+        "\"closed\" key, "
+        "got an empty string");
+  }
+  simdjson::dom::parser parser;
+  ARROW_ASSIGN_OR_RAISE(auto object, internal::ParseJsonObject(parser, serialized_data));
+
+  ARROW_ASSIGN_OR_RAISE(auto closed_value,
+                        internal::GetOptionalJsonField(object, "closed"));
+  if (!closed_value.has_value()) {
+    return Status::Invalid(
+        "FixedClosednessRangeType metadata is missing the required \"closed\" key: ",
+        serialized_data);
+  }
+  std::string_view closed_str;
+  if (closed_value->get_string().get(closed_str) != simdjson::SUCCESS) {
+    return Status::Invalid(
+        "Invalid serialized JSON data for FixedClosednessRangeType: \"closed\" is not a "
+        "string");
+  }
+  ARROW_ASSIGN_OR_RAISE(RangeClosed closed, ClosedFromString(closed_str));
+
+  return std::make_shared<FixedClosednessRangeType>(std::move(storage_type), closed);
+}
+
+std::shared_ptr<Array> FixedClosednessRangeType::MakeArray(
+    std::shared_ptr<ArrayData> data) const {
+  DCHECK_EQ(data->type->id(), Type::EXTENSION);
+  DCHECK_EQ("arrow.fixed_closedness_range",
+            internal::checked_cast<const ExtensionType&>(*data->type).extension_name());
+  return std::make_shared<FixedClosednessRangeArray>(data);
+}
+
+Result<std::shared_ptr<DataType>> FixedClosednessRangeType::Make(
+    std::shared_ptr<DataType> value_type, RangeClosed closed, bool allow_unbounded) {
+  auto storage = MakeFixedClosednessStorageType(value_type, allow_unbounded);
+  return std::make_shared<FixedClosednessRangeType>(std::move(storage), closed);
+}
+
+// ---------------------------------------------------------------------------
+// Free factory function
+
+std::shared_ptr<DataType> fixed_closedness_range(std::shared_ptr<DataType> value_type,
+                                                 RangeClosed closed,
+                                                 bool allow_unbounded) {
+  auto result =
+      FixedClosednessRangeType::Make(std::move(value_type), closed, allow_unbounded);
+  ARROW_CHECK_OK(result.status());
+  return std::move(result).ValueOrDie();
+}
+
+// ---------------------------------------------------------------------------
+// VariableClosednessRangeType
+
+namespace {
+
+/// Build the storage Struct type for a per-value-inclusivity range. In addition
+/// to the "lower"/"upper" bounds (nullable iff unbounded endpoints are allowed),
+/// it carries two non-nullable boolean fields recording each bound's inclusivity.
+std::shared_ptr<DataType> MakeVariableClosednessStorageType(
+    const std::shared_ptr<DataType>& value_type, bool allow_unbounded) {
+  return struct_({field("lower", value_type, allow_unbounded),
+                  field("upper", value_type, allow_unbounded),
+                  field("lower_inc", boolean(), /*nullable=*/false),
+                  field("upper_inc", boolean(), /*nullable=*/false)});
+}
+
+}  // namespace
+
+std::shared_ptr<DataType> VariableClosednessRangeType::value_type() const {
+  // storage_type() is a struct whose "lower"/"upper" fields share the same type.
+  return internal::checked_cast<const StructType&>(*storage_type()).field(0)->type();
+}
+
+std::string VariableClosednessRangeType::ToString(bool show_metadata) const {
+  std::stringstream ss;
+  ss << "extension<" << this->extension_name()
+     << "[value_type=" << value_type()->ToString(show_metadata) << "]>";
+  return ss.str();
+}
+
+bool VariableClosednessRangeType::ExtensionEquals(const ExtensionType& other) const {
+  if (extension_name() != other.extension_name()) {
+    return false;
+  }
+  // All parameters (value type, bound nullability, the boolean flag fields) are
+  // part of the storage type, so a storage comparison is sufficient.
+  return storage_type()->Equals(*other.storage_type());
+}
+
+std::string VariableClosednessRangeType::Serialize() const {
+  // Inclusivity is stored per value, so there is no type-level parameter to
+  // serialize. Emit an empty JSON object for explicitness and forward-compat.
+  return "{}";
+}
+
+Result<std::shared_ptr<DataType>> VariableClosednessRangeType::Deserialize(
+    std::shared_ptr<DataType> storage_type, const std::string& serialized_data) const {
+  // Validate storage type structure.
+  if (storage_type->id() != Type::STRUCT) {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage type must be a Struct, got ",
+        storage_type->ToString());
+  }
+  const auto& struct_type = internal::checked_cast<const StructType&>(*storage_type);
+  if (struct_type.num_fields() != 4) {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct must have exactly 4 fields, got ",
+        struct_type.num_fields());
+  }
+  const auto& lower_field = struct_type.field(0);
+  const auto& upper_field = struct_type.field(1);
+  const auto& lower_inc_field = struct_type.field(2);
+  const auto& upper_inc_field = struct_type.field(3);
+  if (lower_field->name() != "lower") {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct field 0 must be named \"lower\", got "
+        "\"",
+        lower_field->name(), "\"");
+  }
+  if (upper_field->name() != "upper") {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct field 1 must be named \"upper\", got "
+        "\"",
+        upper_field->name(), "\"");
+  }
+  if (lower_inc_field->name() != "lower_inc") {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct field 2 must be named \"lower_inc\", "
+        "got \"",
+        lower_inc_field->name(), "\"");
+  }
+  if (upper_inc_field->name() != "upper_inc") {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct field 3 must be named \"upper_inc\", "
+        "got \"",
+        upper_inc_field->name(), "\"");
+  }
+  if (!lower_field->type()->Equals(*upper_field->type())) {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct fields \"lower\" and \"upper\" must "
+        "have the "
+        "same type, got \"",
+        lower_field->type()->ToString(), "\" and \"", upper_field->type()->ToString(),
+        "\"");
+  }
+  if (lower_inc_field->type()->id() != Type::BOOL ||
+      upper_inc_field->type()->id() != Type::BOOL) {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct fields \"lower_inc\" and "
+        "\"upper_inc\" must be "
+        "boolean, got \"",
+        lower_inc_field->type()->ToString(), "\" and \"",
+        upper_inc_field->type()->ToString(), "\"");
+  }
+  if (lower_inc_field->nullable() || upper_inc_field->nullable()) {
+    return Status::Invalid(
+        "VariableClosednessRangeType storage Struct fields \"lower_inc\" and "
+        "\"upper_inc\" must be "
+        "non-nullable");
+  }
+
+  // Unlike FixedClosednessRangeType, the metadata carries no parameters: inclusivity
+  // lives in the storage fields. Accept an empty string or any JSON object (ignoring
+  // unknown keys for forward compatibility).
+  if (!serialized_data.empty()) {
+    simdjson::dom::parser parser;
+    RETURN_NOT_OK(internal::ParseJsonObject(parser, serialized_data).status());
+  }
+
+  return std::make_shared<VariableClosednessRangeType>(std::move(storage_type));
+}
+
+std::shared_ptr<Array> VariableClosednessRangeType::MakeArray(
+    std::shared_ptr<ArrayData> data) const {
+  DCHECK_EQ(data->type->id(), Type::EXTENSION);
+  DCHECK_EQ("arrow.variable_closedness_range",
+            internal::checked_cast<const ExtensionType&>(*data->type).extension_name());
+  return std::make_shared<VariableClosednessRangeArray>(data);
+}
+
+Result<std::shared_ptr<DataType>> VariableClosednessRangeType::Make(
+    std::shared_ptr<DataType> value_type, bool allow_unbounded) {
+  auto storage = MakeVariableClosednessStorageType(value_type, allow_unbounded);
+  return std::make_shared<VariableClosednessRangeType>(std::move(storage));
+}
+
+std::shared_ptr<DataType> variable_closedness_range(std::shared_ptr<DataType> value_type,
+                                                    bool allow_unbounded) {
+  auto result = VariableClosednessRangeType::Make(std::move(value_type), allow_unbounded);
+  ARROW_CHECK_OK(result.status());
+  return std::move(result).ValueOrDie();
+}
+
+}  // namespace arrow::extension