R/detect.DR.state.R

In covid19-dashboard-us/cdcar: COVID-19 Data Checking and Repairing (CDCAR)

#' COVID-19 Data Checking and Reparing (CDCAR)
#'
#' This function detects the order-dependency violation of state level covid-19 data (or similar type of data including the count time series and epidemic data).
#'
detect.DR.state <- function(dat = list(), state.show = NULL, log.test = FALSE, plot.show = FALSE, test = "wo"){
	# Extract subset of the data based on state.show and county.show
	if(length(state.show) > 0){
		dat.sub <- dat %>% filter(State %in% state.show) # dim(dat.sub)
	}else{
		dat.sub <- dat
	}

	states = dat.sub$State
	dates0 = names(dat.sub)[-c(1)]
	dates = as.Date(dates0, "X%Y.%m.%d")

	dr.res = dat.sub %>% mutate(DR = 0) %>% dplyr::select("State", "DR")

	for(i in 1:nrow( dat.sub )){
		print(paste0( states[i]))
		x = as.vector(unlist(dat.sub[i, -c(1)]))

		if(sum(x>0) > 14 & sum(!is.na(x)) > 0 ){
			myts = ts (x,frequency =7)
			logts = ts (log(x[which(x>0)]),frequency =7)
			if(log.test == TRUE){
				d = logts
			}else{
				d = myts
			}
			dr.res[i,2]= try(isSeasonal(d, test = test, freq = 7))

			if(dr.res[i,2] == TRUE & plot.show == TRUE ){
				ts = d %>%
					  stl(s.window = "periodic") %>%
					  autoplot()
				print(ts)
			}
		}else{
			warning(paste0(states[i], "'s positive sequence shorter than two weeks."))
			dr.res[i,2]= 0
		}
	}
	list(dat.sub = dat.sub, dr.res = dr.res)
}