#include "strings/strip_shim.h"
#include <cudf/strings/strip.hpp>
#include <cudf/scalar/scalar.hpp>
#include <cudf/scalar/scalar_factories.hpp>
#include <cudf/utilities/default_stream.hpp>
namespace cudf_shims {
std::unique_ptr<OwnedColumn> str_strip(const OwnedColumn& col) {
auto stream = cudf::get_default_stream();
auto mr = cudf::get_current_device_resource_ref();
auto result = cudf::strings::strip(
col.view(), cudf::strings::side_type::BOTH,
cudf::string_scalar("", true, stream), stream, mr);
return std::make_unique<OwnedColumn>(std::move(result));
}
std::unique_ptr<OwnedColumn> str_lstrip(const OwnedColumn& col) {
auto stream = cudf::get_default_stream();
auto mr = cudf::get_current_device_resource_ref();
auto result = cudf::strings::strip(
col.view(), cudf::strings::side_type::LEFT,
cudf::string_scalar("", true, stream), stream, mr);
return std::make_unique<OwnedColumn>(std::move(result));
}
std::unique_ptr<OwnedColumn> str_rstrip(const OwnedColumn& col) {
auto stream = cudf::get_default_stream();
auto mr = cudf::get_current_device_resource_ref();
auto result = cudf::strings::strip(
col.view(), cudf::strings::side_type::RIGHT,
cudf::string_scalar("", true, stream), stream, mr);
return std::make_unique<OwnedColumn>(std::move(result));
}
}