datetime.hpp

前往此文件的文档。

/*
 * 版权所有 (c) 2019-2025, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非符合许可证，否则您不得使用此文件。
 * 您可以在以下位置获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件
 * 是基于“原样”提供的，
 * 不附带任何明示或暗示的保证或条件。
 * 有关特定语言的管理权限和
 * 限制，请参阅许可证。
 */

#pragma once

#include <cudf/types.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <rmm/cuda_stream_view.hpp>

#include <memory>

namespace CUDF_EXPORT cudf {
namespace datetime {
enum class datetime_component : uint8_t {
 YEAR,
 MONTH,
 DAY,
 WEEKDAY,
 HOUR,
 MINUTE,
 SECOND,
 MILLISECOND,
 MICROSECOND,
 NANOSECOND
};

std::unique_ptr<cudf::column> extract_datetime_component(
 cudf::column_view const& column,
 datetime_component component,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 // 组结束
std::unique_ptr<cudf::column> last_day_of_month(
 cudf::column_view const& column,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> day_of_year(
 cudf::column_view const& column,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> add_calendrical_months(
 cudf::column_view const& timestamps,
 cudf::column_view const& months,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> add_calendrical_months(
 cudf::column_view const& timestamps,
 cudf::scalar const& months,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> is_leap_year(
 cudf::column_view const& column,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> days_in_month(
 cudf::column_view const& column,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> extract_quarter(
 cudf::column_view const& column,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

enum class rounding_frequency : int32_t {
 DAY,
 HOUR,
 MINUTE,
 SECOND,
 MILLISECOND,
 MICROSECOND,
 NANOSECOND
};

std::unique_ptr<cudf::column> ceil_datetimes(
 cudf::column_view const& column,
 rounding_frequency freq,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> floor_datetimes(
 cudf::column_view const& column,
 rounding_frequency freq,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<cudf::column> round_datetimes(
 cudf::column_view const& column,
 rounding_frequency freq,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 // 组结束

} // 命名空间 datetime
} // 命名空间 CUDF_EXPORT cudf