rolling.hpp

Go to the documentation of this file.

/*
 * Copyright (c) 2019-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/rolling/range_window_bounds.hpp>
#include <cudf/types.hpp>
 
#include <rmm/mr/device/per_device_resource.hpp>
 
#include <memory>
 
namespace cudf {
std::unique_ptr<column> rolling_window(
  column_view const& input,
  size_type preceding_window,
  size_type following_window,
  size_type min_periods,
  rolling_aggregation const& agg,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> rolling_window(
  column_view const& input,
  column_view const& default_outputs,
  size_type preceding_window,
  size_type following_window,
  size_type min_periods,
  rolling_aggregation const& agg,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
struct window_bounds {
 public:
  static window_bounds get(size_type value) { return window_bounds(false, value); }
 
  static window_bounds unbounded()
  {
    return window_bounds(true, std::numeric_limits<cudf::size_type>::max());
  }
 
  // TODO: In the future, add units for bounds.
  //       E.g. {value=1, unit=DAYS, unbounded=false}
  //       For the present, assume units from context:
  //         1. For time-based window functions, assume DAYS as before
  //         2. For all else, assume ROWS as before.
  const bool is_unbounded;  
  const size_type value;    
 
 private:
  explicit window_bounds(bool is_unbounded_, size_type value_ = 0)
    : is_unbounded{is_unbounded_}, value{value_}
  {
  }
};
 
std::unique_ptr<column> grouped_rolling_window(
  table_view const& group_keys,
  column_view const& input,
  size_type preceding_window,
  size_type following_window,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_rolling_window(
  table_view const& group_keys,
  column_view const& input,
  window_bounds preceding_window,
  window_bounds following_window,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_rolling_window(
  table_view const& group_keys,
  column_view const& input,
  column_view const& default_outputs,
  size_type preceding_window,
  size_type following_window,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_rolling_window(
  table_view const& group_keys,
  column_view const& input,
  column_view const& default_outputs,
  window_bounds preceding_window,
  window_bounds following_window,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_time_range_rolling_window(
  table_view const& group_keys,
  column_view const& timestamp_column,
  cudf::order const& timestamp_order,
  column_view const& input,
  size_type preceding_window_in_days,
  size_type following_window_in_days,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_time_range_rolling_window(
  table_view const& group_keys,
  column_view const& timestamp_column,
  cudf::order const& timestamp_order,
  column_view const& input,
  window_bounds preceding_window_in_days,
  window_bounds following_window_in_days,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> grouped_range_rolling_window(
  table_view const& group_keys,
  column_view const& orderby_column,
  cudf::order const& order,
  column_view const& input,
  range_window_bounds const& preceding,
  range_window_bounds const& following,
  size_type min_periods,
  rolling_aggregation const& aggr,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
 
std::unique_ptr<column> rolling_window(
  column_view const& input,
  column_view const& preceding_window,
  column_view const& following_window,
  size_type min_periods,
  rolling_aggregation const& agg,
  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
  // end of group
}  // namespace cudf