R/class_timespan_compat_stats.R

In parttime: Partial Datetime Handling

#' start S3 generic
#'
#' A generic method to retrieve the start of an object
#'
#' @param x An object to retrieve the start from
#' @param ... Additional arguments passed to methods
#'
#' @return The starting `partial_time` of a `partial_timespan` object.
#'
start <- function(x, ...) {
  UseMethod("start")
}

#' @importFrom utils head
#' @rawNamespace S3method(stats::start,timespan)
#' @export
start.timespan <- function(x, ...) {
  cols <- utils::head(dimnames(vctrs::field(x, "tmspn_arr"))[[2]], -1L)
  dns <- dimnames(vctrs::field(x, "tmspn_arr"))
  as.parttime(array(
    vctrs::field(x, "tmspn_arr")[, cols, "lb"],
    dim = c(length(x), length(cols)),
    dimnames = list(dns[[1L]], cols)
  ))
}

#' end S3 generic
#'
#' A generic method to retrieve the end of an object
#'
#' @param x An object to retrieve the end from
#' @param ... Additional arguments passed to methods
#'
#' @return The ending `partial_time` of a `partial_timespan` object.
#'
end <- function(x, ...) {
  UseMethod("end")
}

#' @importFrom utils head
#' @rawNamespace S3method(stats::end,timespan)
#' @export
end.timespan <- function(x, ...) {
  cols <- utils::head(dimnames(vctrs::field(x, "tmspn_arr"))[[2]], -1L)
  dns <- dimnames(vctrs::field(x, "tmspn_arr"))
  as.parttime(array(
    vctrs::field(x, "tmspn_arr")[, cols, "ub"],
    dim = c(length(x), length(cols)),
    dimnames = list(dns[[1L]], cols)
  ))
}