Migrate string capitalize APIs to pylibcudf

python/cudf/cudf/_lib/pylibcudf/libcudf/scalar/scalar_factories.pxd

@@ -0,0 +1,10 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+from libcpp.memory cimport unique_ptr
+from libcpp.string cimport string
+
+from cudf._lib.pylibcudf.libcudf.scalar.scalar cimport scalar
+
+
+cdef extern from "cudf/scalar/scalar_factories.hpp" namespace "cudf" nogil:
+    cdef unique_ptr[scalar] make_string_scalar(const string & _string) except +

python/cudf/cudf/_lib/pylibcudf/libcudf/strings/capitalize.pxd

@@ -3,11 +3,14 @@ from libcpp.memory cimport unique_ptr
 
 from cudf._lib.pylibcudf.libcudf.column.column cimport column
 from cudf._lib.pylibcudf.libcudf.column.column_view cimport column_view
+from cudf._lib.pylibcudf.libcudf.scalar.scalar cimport string_scalar
 
 
 cdef extern from "cudf/strings/capitalize.hpp" namespace "cudf::strings" nogil:
     cdef unique_ptr[column] capitalize(
-        const column_view & strings) except +
+        const column_view & strings,
+        const string_scalar & delimiters
+        ) except +
 
     cdef unique_ptr[column] title(
         const column_view & strings) except +

python/cudf/cudf/_lib/pylibcudf/libcudf/strings/case.pxd

@@ -6,6 +6,12 @@ from cudf._lib.pylibcudf.libcudf.column.column_view cimport column_view
 
 
 cdef extern from "cudf/strings/case.hpp" namespace "cudf::strings" nogil:
+    cdef unique_ptr[column] capitalize(
+        const column_view & input) except +
+
+    cdef unique_ptr[column] is_title(
+        const column_view & input) except +
+
     cdef unique_ptr[column] to_lower(
         const column_view & strings) except +
 

python/cudf/cudf/_lib/pylibcudf/strings/CMakeLists.txt

@@ -12,7 +12,8 @@
 # the License.
 # =============================================================================
 
-set(cython_sources case.pyx find.pyx)
+set(cython_sources capitalize.pyx case.pyx find.pyx)
+
 set(linked_libraries cudf::cudf)
 rapids_cython_create_modules(
   CXX

python/cudf/cudf/_lib/pylibcudf/strings/__init__.pxd

@@ -1,3 +1,3 @@
 # Copyright (c) 2024, NVIDIA CORPORATION.
 
-from . cimport case, find
+from . cimport capitalize, case, find

python/cudf/cudf/_lib/pylibcudf/strings/__init__.py

@@ -1,3 +1,3 @@
 # Copyright (c) 2024, NVIDIA CORPORATION.
 
-from . import case, find
+from . import capitalize, case, find

python/cudf/cudf/_lib/pylibcudf/strings/capitalize.pxd

@@ -0,0 +1,9 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+from cudf._lib.pylibcudf.column cimport Column
+from cudf._lib.pylibcudf.scalar cimport Scalar
+
+
+cpdef Column capitalize(Column input, Scalar delimiters=*)
+cpdef Column title(Column input)
+cpdef Column is_title(Column input)

python/cudf/cudf/_lib/pylibcudf/strings/capitalize.pyx

@@ -0,0 +1,58 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+from libcpp.memory cimport unique_ptr
+from libcpp.utility cimport move
+
+from cudf._lib.pylibcudf.column cimport Column
+from cudf._lib.pylibcudf.libcudf.column.column cimport column
+from cudf._lib.pylibcudf.libcudf.scalar.scalar cimport string_scalar
+from cudf._lib.pylibcudf.libcudf.scalar.scalar_factories cimport (
+    make_string_scalar as cpp_make_string_scalar,
+)
+from cudf._lib.pylibcudf.libcudf.strings cimport capitalize as cpp_capitalize
+from cudf._lib.pylibcudf.scalar cimport Scalar
+
+from cython.operator import dereference
+
+
+cpdef Column capitalize(
+    Column input,
+    Scalar delimiters=None
+    # TODO: default scalar values
+    # https://github.com/rapidsai/cudf/issues/15505
+):
+
+    cdef unique_ptr[column] c_result
+
+    if delimiters is None:
+        delimiters = Scalar.from_libcudf(
+            cpp_make_string_scalar("".encode())
+        )
+
+    cdef const string_scalar* cpp_delimiters = <const string_scalar*>(
+        delimiters.c_obj.get()
+    )
+
+    with nogil:
+        c_result = cpp_capitalize.capitalize(
+            input.view(),
+            dereference(cpp_delimiters)
+        )
+
+    return Column.from_libcudf(move(c_result))
+
+
+cpdef Column title(Column input):
+    cdef unique_ptr[column] c_result
+    with nogil:
+        c_result = cpp_capitalize.title(input.view())
+
+    return Column.from_libcudf(move(c_result))
+
+
+cpdef Column is_title(Column input):
+    cdef unique_ptr[column] c_result
+    with nogil:
+        c_result = cpp_capitalize.is_title(input.view())
+
+    return Column.from_libcudf(move(c_result))

python/cudf/cudf/_lib/strings/capitalize.pyx

@@ -2,47 +2,33 @@
 
 from cudf.core.buffer import acquire_spill_lock
 
-from libcpp.memory cimport unique_ptr
-from libcpp.utility cimport move
-
 from cudf._lib.column cimport Column
-from cudf._lib.pylibcudf.libcudf.column.column cimport column
-from cudf._lib.pylibcudf.libcudf.column.column_view cimport column_view
-from cudf._lib.pylibcudf.libcudf.strings.capitalize cimport (
-    capitalize as cpp_capitalize,
-    is_title as cpp_is_title,
-    title as cpp_title,
-)
+
+import cudf._lib.pylibcudf as plc
 
 
 @acquire_spill_lock()
 def capitalize(Column source_strings):
-    cdef unique_ptr[column] c_result
-    cdef column_view source_view = source_strings.view()
-
-    with nogil:
-        c_result = move(cpp_capitalize(source_view))
-
-    return Column.from_unique_ptr(move(c_result))
+    return Column.from_pylibcudf(
+        plc.strings.capitalize.capitalize(
+            source_strings.to_pylibcudf(mode="read")
+        )
+    )
 
 
 @acquire_spill_lock()
 def title(Column source_strings):
-    cdef unique_ptr[column] c_result
-    cdef column_view source_view = source_strings.view()
-
-    with nogil:
-        c_result = move(cpp_title(source_view))
-
-    return Column.from_unique_ptr(move(c_result))
+    return Column.from_pylibcudf(
+        plc.strings.capitalize.title(
+            source_strings.to_pylibcudf(mode="read")
+        )
+    )
 
 
 @acquire_spill_lock()
 def is_title(Column source_strings):
-    cdef unique_ptr[column] c_result
-    cdef column_view source_view = source_strings.view()
-
-    with nogil:
-        c_result = move(cpp_is_title(source_view))
-
-    return Column.from_unique_ptr(move(c_result))
+    return Column.from_pylibcudf(
+        plc.strings.capitalize.is_title(
+            source_strings.to_pylibcudf(mode="read")
+        )
+    )

python/cudf/cudf/pylibcudf_tests/common/utils.py

@@ -35,7 +35,6 @@ def assert_column_eq(plc_column: plc.Column, pa_array: pa.Array) -> None:
         plc_pa = plc_pa.combine_chunks()
     if isinstance(pa_array, pa.ChunkedArray):
         pa_array = pa_array.combine_chunks()
-
     assert plc_pa.equals(pa_array)
 
 

python/cudf/cudf/pylibcudf_tests/test_string_capitalize.py

@@ -0,0 +1,52 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+import pyarrow as pa
+import pytest
+from utils import assert_column_eq
+
+import cudf._lib.pylibcudf as plc
+
+
+@pytest.fixture(scope="module")
+def pa_data():
+    data = [
+        "leopard",
+        "Golden Eagle",
+        "SNAKE",
+        "",
+        "!A",
+        "hello World",
+        "A B C",
+        "#",
+        "AƻB",
+        "Ⓑⓖ",
+        "Art of War",
+        "The quick bRoWn fox juMps over the laze DOG",
+        '123nr98nv9rev!$#INF4390v03n1243<>?}{:-"',
+        "accénted",
+        None,
+    ]
+    return pa.array(data)
+
+
+@pytest.fixture(scope="module")
+def plc_data(pa_data):
+    return plc.interop.from_arrow(pa_data)
+
+
+def test_capitalize(plc_data, pa_data):
+    got = plc.strings.capitalize.capitalize(plc_data)
+    expected = pa.compute.utf8_capitalize(pa_data)
+    assert_column_eq(got, expected)
+
+
+def test_title(plc_data, pa_data):
+    got = plc.strings.capitalize.title(plc_data)
+    expected = pa.compute.utf8_title(pa_data)
+    assert_column_eq(got, expected)
+
+
+def test_is_title(plc_data, pa_data):
+    got = plc.strings.capitalize.is_title(plc_data)
+    expected = pa.compute.utf8_is_title(pa_data)
+    assert_column_eq(got, expected)