multibyte_split.hpp

/*
 * Copyright (c) 2021-2022, NVIDIA CORPORATION.
 *
 * Licensed 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.
 */
 
#pragma once
 
#include <cudf/column/column.hpp>
#include <cudf/io/text/byte_range_info.hpp>
#include <cudf/io/text/data_chunk_source.hpp>
 
#include <rmm/mr/device/device_memory_resource.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
 
#include <memory>
#include <optional>
 
namespace cudf {
namespace io {
namespace text {
 
struct parse_options {
  byte_range_info byte_range = create_byte_range_info_max();
  bool strip_delimiters = false;
};
 
std::unique_ptr<cudf::column> multibyte_split(
  data_chunk_source const& source,
  std::string const& delimiter,
  parse_options options               = {},
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<cudf::column> multibyte_split(
  data_chunk_source const& source,
  std::string const& delimiter,
  std::optional<byte_range_info> byte_range,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<cudf::column> multibyte_split(data_chunk_source const& source,
                                              std::string const& delimiter,
                                              rmm::mr::device_memory_resource* mr);
 
}  // namespace text
}  // namespace io
}  // namespace cudf