R/ExportBaselineSurveyResponses.R

In legenepi/exceed-pipelines: Reusable pipelines for the EXCEED study

#' ExportBaselineSurveyResponses
#'
#' Prepare data export of baseline survey responses.
#'
#' @docType class
#' @format An R6 class object.
#' @description ExportBaselineSurveyResponses class
#' @importFrom R6 R6Class
#' @export
ExportBaselineSurveyResponses <- R6::R6Class(
  "ExportBaselineSurveyResponses",
  inherit = LoadBaselineSurveyResponses,

  public = list(
    transform = function(.data, .collect, ...) {
      .data <- super$transform(.data, .collect, ...) %>%
        .collect()

      pipeline <- self$client$pipeline() %>%
        add_step(
          ConvertToKilograms,
          unit = "mea1",
          kilograms = "mea1a",
          stones = "mea1b",
          pounds = "mea1c"
        ) %>%
        add_step(
          ConvertToCentimeters,
          unit = "mea2",
          centimeters = "mea2a",
          inches = "mea2b"
        ) %>%
        add_step(
          ConvertToCentimeters,
          unit = "mea3",
          centimeters = "mea3a",
          feet = "mea3b",
          inches = "mea3c"
        )

      pipeline %>%
        .collect(.data) %>%
        mutate(mea1a = ifelse(mea1a >= 30 & mea1a <= 250, mea1a, NA)) %>%
        mutate(mea2a = ifelse(mea2a >= 10 & mea2a <= 200, mea2a, NA)) %>%
        mutate(mea3a = ifelse(mea3a >= 120 & mea3a <= 220, mea3a, NA))
    }
  )
)