R/fix_uncontinue.R

In kongdd/RHtestsHelper: What the Package Does (One Line, Title Case)

#' fix un-continuous observations for the INPUT data
#'
#' fix un-continuous observations for the INPUT data and only keep the complete
#' year.
#'
#' @param df A data.frame with the columns of `site`, `date` and others
#' @param complete_year Boolean. Note that `complete_year` has to be true for RHtests.
#'
#' @export
fix_uncontinue <- function(df, complete_year = TRUE) {
  # df %>% group_by(site) %>% group_modify(fix_uncontinue_site)
  df[, fix_uncontinue_site(.SD, complete_year = complete_year), .(site)] # about 2x faster
  # ddply(df, .(site), fix_uncontinue_site, .progress = "text")
}

fix_uncontinue_site <- function(d, ..., complete_year = TRUE) {
  n <- nrow(d)
  # if (n <= 365*4) return(NULL)
  date_begin <- d$date %>% min()
  date_end <- d$date %>% max()
  
  if (complete_year) {
    if (format(date_begin, "%m-%d") != "01-01") {
      date_begin <- make_date(year(date_begin), 1, 1)
    }
    if (format(date_end, "%m-%d") != "12-31") {
      date_end <- make_date(year(date_end), 12, 31)
    }
  }
  # print2(date_begin, date_end)
  temp <- data.table(date = seq.Date(date_begin, date_end, by = "day"))
  merge(d, temp, c("date"), all.y = TRUE) #%>% select(-site)
}