avoid droping color levels in the risk table

Mouksassi · Mouksassi · commit ffd0bc3a800e · 2025-08-27T16:03:53.000-07:00
diff --git a/R/ggkmrisktable.R b/R/ggkmrisktable.R
@@ -547,6 +547,13 @@ ggkmrisktable <- function(data = lung_long, # long format filter to Endpoint of
       risktabledata[[variable]] <- .get_variable_value(variable,risktabledata$strata, surv_object, data.long)
     }
   }
+
+  if  (!is.null( levels(data.long[[colorinputvar]]))){
+    collevels <- levels(data.long[[colorinputvar]])
+    risktabledata[[colorinputvar]] <- factor(risktabledata[[colorinputvar]],
+                                             levels = collevels)
+    }
+
   if(nrisk_filterout0){
     risktabledata <- risktabledata |> 
       dplyr::filter(n.risk > 0)

Original file line number	Diff line number	Diff line change
`@@ -547,6 +547,13 @@ ggkmrisktable <- function(data = lung_long, # long format filter to Endpoint of`
`547`	`547`	`risktabledata[[variable]] <- .get_variable_value(variable,risktabledata$strata, surv_object, data.long)`
`548`	`548`	`}`
`549`	`549`	`}`
	`550`	`+`
	`551`	`+ if (!is.null( levels(data.long[[colorinputvar]]))){`
	`552`	`+ collevels <- levels(data.long[[colorinputvar]])`
	`553`	`+ risktabledata[[colorinputvar]] <- factor(risktabledata[[colorinputvar]],`
	`554`	`+ levels = collevels)`
	`555`	`+ }`
	`556`	`+`
`550`	`557`	`if(nrisk_filterout0){`
`551`	`558`	`risktabledata <- risktabledata \|>`
`552`	`559`	`dplyr::filter(n.risk > 0)`