add integer argument to gs_power_ahr

Author: LittleBeannie

R/gs_power_ahr.R

@@ -53,9 +53,12 @@
 #' @param tol Tolerance parameter for boundary convergence (on Z-scale).
 #' @param interval An interval that is presumed to include the time at which
 #'   expected event count is equal to targeted event.
+#' @param integer Logical value integer whether it is an integer design
+#' (i.e., integer sample size and events) or not. This argument is commonly
+#' used when creating integer design via [to_integer()].
 #'
-#' @return A tibble with columns `Analysis`, `Bound`, `Z`, `Probability`,
-#'   `theta`, `Time`, `AHR`, `Events`.
+#' @return A tibble with columns `analysis`, `bound`, `z`, `probability`,
+#'   `theta`, `time`, `ahr`, `event`.
 #'   Contains a row for each analysis and each bound.
 #'
 #' @details
@@ -159,7 +162,8 @@ gs_power_ahr <- function(
     info_scale = c("h0_h1_info", "h0_info", "h1_info"),
     r = 18,
     tol = 1e-6,
-    interval = c(.01, 1000)) {
+    interval = c(.01, 1000),
+    integer = FALSE) {
   # Get the number of analysis
   n_analysis <- max(length(event), length(analysis_time), na.rm = TRUE)
 
@@ -198,8 +202,7 @@ gs_power_ahr <- function(
   )
 
   # if both events and sample size are integers, then elaborate the info and info0
-  sample_size <- sum(enroll_rate$duration * enroll_rate$rate)
-  if (abs(sample_size - round(sample_size)) < 1e-5 && all(abs(event - round(event)) < 1e-5)) {
+  if (integer) {
 
     # elaborate info0
     q_e <- ratio / (1 + ratio)

R/to_integer.R

@@ -350,7 +350,8 @@ to_integer.gs_design <- function(x, sample_size = TRUE, ...) {
       test_lower = x$input$test_lower,
       binding = x$input$binding,
       info_scale = x$input$info_scale, r = x$input$r, tol = x$input$tol,
-      interval = c(0.01, max(x$analysis$time) + 100)
+      interval = c(0.01, max(x$analysis$time) + 100),
+      integer = TRUE
     )
     if (is_wlr) power_args[c("weight", "approx")] <- x$input[c("weight", "approx")]
     x_new <- do.call(if (is_wlr) gs_power_wlr else gs_power_ahr, power_args)