Add the ListViewScalar class

apache · Apr 26, 2023 · 71cad47 · 71cad47
1 parent 855dddf
commit 71cad47
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 0 deletions.
diff --git a/cpp/src/arrow/scalar.cc b/cpp/src/arrow/scalar.cc
@@ -621,6 +621,17 @@ RunEndEncodedScalar::RunEndEncodedScalar(const std::shared_ptr<DataType>& type)
 
 RunEndEncodedScalar::~RunEndEncodedScalar() = default;
 
+ListViewScalar::ListViewScalar(std::shared_ptr<Array> value,
+                                 std::shared_ptr<DataType> type)
+    : Scalar{std::move(type), /*is_valid=*/value != NULLPTR}, value(std::move(value)) {
+  ARROW_CHECK(this->type->Equals(this->value->type()));
+}
+
+ListViewScalar::ListViewScalar(const std::shared_ptr<DataType>& type)
+    : ListViewScalar(NULLPTR, type) {}
+
+ListViewScalar::~ListViewScalar() = default;
+
 DictionaryScalar::DictionaryScalar(std::shared_ptr<DataType> type)
     : internal::PrimitiveScalarBase(std::move(type)),
       value{MakeNullScalar(checked_cast<const DictionaryType&>(*this->type).index_type()),

diff --git a/cpp/src/arrow/scalar.h b/cpp/src/arrow/scalar.h
@@ -591,6 +591,29 @@ struct ARROW_EXPORT RunEndEncodedScalar : public Scalar {
   const TypeClass& ree_type() const { return internal::checked_cast<TypeClass&>(*type); }
 };
 
+struct ARROW_EXPORT ListViewScalar : public Scalar {
+  using TypeClass = ListViewType;
+  using ValueType = std::shared_ptr<Array>;
+
+  ValueType value;
+
+  ListViewScalar(std::shared_ptr<Array> value, std::shared_ptr<DataType> type);
+
+  /// \brief Constructs a NULL ListViewScalar
+  explicit ListViewScalar(const std::shared_ptr<DataType>& type);
+
+  ~ListViewScalar() override;
+
+  const std::shared_ptr<DataType>& value_type() const {
+    return array_view_type().value_type();
+  }
+
+ private:
+  const TypeClass& array_view_type() const {
+    return internal::checked_cast<TypeClass&>(*type);
+  }
+};
+
 /// \brief A Scalar value for DictionaryType
 ///
 /// `is_valid` denotes the validity of the `index`, regardless of

diff --git a/cpp/src/arrow/type_fwd.h b/cpp/src/arrow/type_fwd.h
@@ -186,6 +186,7 @@ struct RunEndEncodedScalar;
 
 class ListViewType;
 class ListViewArray;
+struct ListViewScalar;
 
 template <typename TypeClass>
 class NumericArray;

diff --git a/cpp/src/arrow/type_traits.h b/cpp/src/arrow/type_traits.h
@@ -394,6 +394,7 @@ template <>
 struct TypeTraits<ListViewType> {
   using ArrayType = ListViewArray;
   // TODO(felipecrv): Add BuilderType
+  using ScalarType = ListViewScalar;
 
   constexpr static bool is_parameter_free = false;
 };