fix bug in wl_queue_sizes

NHSRwaitinglist.Rproj

@@ -15,7 +15,6 @@ LaTeX: pdfLaTeX
 StripTrailingWhitespace: Yes
 
 BuildType: Package
-PackageUseDevtools: Yes
+PackageCleanBeforeInstall: No
 PackageInstallArgs: --no-multiarch --with-keep.source
-PackageCheckArgs: --as-cran
 PackageRoxygenize: rd,collate,namespace,vignette

R/wl_queue_size.R

@@ -34,15 +34,23 @@ wl_queue_size <- function(waiting_list, start_date = NULL, end_date = NULL) {
   queues[which(queues[, 1] %in% arrival_counts[, 1]), 2] <- arrival_counts[, 2]
   queues$cummul_arrivals <- cumsum(queues[, 2])
 
-  departure_counts <- data.frame(
-    table(wl[which((start_date <= wl[, 2]) & (wl[, 2] <= end_date)), 2])
-  )
-  queues$departures <- rep(0, length(dates))
-  queues[which(queues[, 1] %in% departure_counts[, 1]), 4] <-
-    departure_counts[, 2]
-  queues$cummul_departures <- cumsum(queues[, 4])
-
-  queues$queue_size <- queues$cummul_arrivals - queues$cummul_departures
+  departures <- wl[which((start_date <= wl[, 2]) & (wl[, 2] <= end_date)), 2]
+  if(length(departures>0))
+  {
+    departure_counts <- data.frame(
+      table(wl[which((start_date <= wl[, 2]) & (wl[, 2] <= end_date)), 2])
+    )
+    queues$departures <- rep(0, length(dates))
+    queues[which(queues[, 1] %in% departure_counts[, 1]), 4] <-
+      departure_counts[, 2]
+    queues$cummul_departures <- cumsum(queues[, 4])
+
+    queues$queue_size <- queues$cummul_arrivals - queues$cummul_departures
+  } else {
+    queues$departures <- rep(0, length(dates))
+    queues$cummul_departures <- rep(0, length(dates))
+    queues$queue_size <- queues$cummul_arrivals
+  }
 
   return(queues[, c(1, 6)])
-}
+}

R/wl_simulator.R

@@ -12,6 +12,8 @@
 #'   contains the referral dates
 #'
 #' @return dataframe. A df of simulated referrals and removals
+#'
+#' @import dplyr
 #' @export
 #'
 #' @examples