index_by

NAMESPACE

@@ -26,6 +26,7 @@ S3method(fill_gaps,temporal_cubble_df)
 S3method(filter,spatial_cubble_df)
 S3method(group_by,spatial_cubble_df)
 S3method(group_by,temporal_cubble_df)
+S3method(index_by,temporal_cubble_df)
 S3method(key,cubble_df)
 S3method(key_data,cubble_df)
 S3method(key_vars,cubble_df)
@@ -123,6 +124,7 @@ importFrom(tidyr,unnest)
 importFrom(tsibble,as_tsibble)
 importFrom(tsibble,fill_gaps)
 importFrom(tsibble,index)
+importFrom(tsibble,index_by)
 importFrom(tsibble,key)
 importFrom(tsibble,key_data)
 importFrom(tsibble,key_vars)

R/cubble-package.R

@@ -9,6 +9,7 @@
 #' @importFrom rlang .data quo_is_missing sym as_name
 #' @importFrom cli cli_abort cli_inform
 #' @importFrom tsibble index as_tsibble
+#' @importFrom tsibble fill_gaps scan_gaps index_by
 #' @aliases cubble-package
 #' @keywords internal
 "_PACKAGE"

R/dplyr.R

@@ -227,6 +227,7 @@ summarise.spatial_cubble_df <- function(.data, ..., .by = NULL,
 #' @export
 summarise.temporal_cubble_df <- function(.data, ..., .by = key_vars(.data),
  .groups = NULL){
+ #browser()
  vars <- enquos(..., .named = TRUE)
  index <- index_var(.data)
  key <- key_vars(.data)
@@ -247,10 +248,22 @@ summarise.temporal_cubble_df <- function(.data, ..., .by = key_vars(.data),
  index <- setdiff(potential_index, key)
  }
 
- if (!index %in% colnames(out) || !key %in% colnames(out)){
+ if (is.null(attr(.data, "index2"))) {
+ idx_in <- index %in% colnames(out)
+ } else{
+ idx_in <- index %in% colnames(out) && attr(.data, "index2") %in% colnames(out)
+ }
+
+ if (!idx_in || !key %in% colnames(out)){
  return(as_tibble(out))
  }
 
+ index <- if (index %in% colnames(out)){
+ index <- index
+ } else{
+ index <- attr(.data, "index2")
+ }
+
  new_temporal_cubble(
  out, key = key, index = index, coords = coords, spatial = spatial)
 }

R/tsibble.R

@@ -20,7 +20,6 @@ make_temporal_tsibble <- function(x){
 
 #' Gap-filling on the temporal component of a cubble object
 #' @inheritParams tsibble::fill_gaps
-#' @importFrom tsibble fill_gaps scan_gaps
 #' @return a cubble object
 #' @rdname tsibble
 #' @export
@@ -62,3 +61,22 @@ scan_gaps.temporal_cubble_df <- function(.data, ...){
 
 
 }
+
+#' @rdname tsibble
+#' @export
+index_by.temporal_cubble_df <- function(.data, ...){
+ #browser()
+ stopifnot(is_cubble_temporal(.data))
+ key <- key_vars(.data)
+ index <- index_var(.data)
+ coords <- coords(.data)
+ spatial <- spatial(.data)
+ .data <- as_tsibble(.data, key = key , index = index)
+ res <- index_by(.data, ...)
+
+ new_temporal_cubble(
+ res, key = key, index = index, coords = coords(.data), spatial = spatial)
+
+
+}
+

R/zzz.R

@@ -18,18 +18,10 @@
  s3_register("dplyr::rowwise", "spatial_cubble_df")
  s3_register("dplyr::rowwise", "temporal_cubble_df")
 
- #s3_register("tsibble::index_by", "cubble_df")
- #s3_register("dplyr::ungroup", "cubble_df")
-
- # s3_register("dplyr::slice_head", "cubble_df")
- # s3_register("dplyr::slice_tail", "cubble_df")
- # s3_register("dplyr::slice_min", "cubble_df")
- # s3_register("dplyr::slice_max", "cubble_df")
- # s3_register("dplyr::slice_sample", "cubble_df")
-
  s3_register("base::print", "cubble_df")
  s3_register("tsibble::fill_gaps", "temporal_cubble_df")
  s3_register("tsibble::scan_gaps", "temporal_cubble_df")
+ s3_register("tsibble::index_by", "temporal_cubble_df")
  s3_register("tsibble::key", "cubble_df")
  s3_register("tsibble::key_data", "cubble_df")
  s3_register("tsibble::key_vars", "cubble_df")