get_wave to support decimal wave numbers

Author: nmdefries

facebook/qsf-tools/qsf-utils.R

@@ -46,20 +46,21 @@ get_block_item_map <- function(qsf) {
 
 
 #' Get wave number from qsf filename
+#' 
+#' Wave number as provided in the qsf name should be an integer or a float with
+#' one decimal place.
 #'
 #' @param path_to_qsf
 #'
-#' @return integer wave number
+#' @return (mostly) integer wave number
 get_wave <- function(path_to_qsf) {
-  qsf_name_pattern <- "(.*Wave_)([0-9]*)([.]qsf)$"
+  qsf_name_pattern <- "(.*Wave_)([0-9]*([.][0-9])?)([.]qsf)$"
   if (!grepl(qsf_name_pattern, path_to_qsf)) {
     stop("qsf filename should be of the format 'Survey_of_COVID-Like_Illness_-_Wave_XX.qsf'")
   }
 
-  wave <- as.integer(sub(
-    "(.*Wave_)([0-9]*)([.]qsf)$",
-    "\\2",
-    path_to_qsf)
+  wave <- as.numeric(
+    sub(qsf_name_pattern, "\\2", path_to_qsf)
   ) 
 
   return(wave)