From 95be046985dcbf340baf13c6532bdc8b2dd253ab Mon Sep 17 00:00:00 2001
From: Jens Sauer <jsauer65@gmail.com>
Date: Tue, 17 Nov 2020 14:33:56 +0100
Subject: [PATCH] survival: Add plots per diagnosis

This adds survival curves per diagnosis. They still look a bit
overloaded.
---
 survival.R | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/survival.R b/survival.R
index 8c09792..f6ef2b2 100644
--- a/survival.R
+++ b/survival.R
@@ -57,6 +57,24 @@ sma_plot_surv_asct_sex <- function() {
 		   pval = TRUE)
 }
 
+# plot survival after diagnosis per diagnosis
+sma_plot_surv_dx_dx <- function() {
+	ggsurvplot(survfit(surv_dx ~ diagnosis, data = secmal), data = secmal,
+		   xscale = "d_y", title = "Survival after diagnosis",
+		   break.time.by = sma_break.time.by,
+		   risk.table = "abs_pct",
+		   ggtheme = theme_bw())
+}
+
+# plot survival after diagnosis per diagnosis
+sma_plot_surv_asct_dx <- function() {
+	ggsurvplot(survfit(surv_asct ~ diagnosis, data = secmal), data = secmal,
+		   xscale = "d_y", title = "Survival after transplantation",
+		   break.time.by = sma_break.time.by,
+		   risk.table = "abs_pct",
+		   ggtheme = theme_bw())
+}
+
 #
 # Write survival plots to files
 #
@@ -65,4 +83,6 @@ sma_plot_file_surv <- function() {
 	sma_plot_file("survival_dx_sex.png", png, sma_plot_surv_dx_sex)
 	sma_plot_file("survival_asct.png", png, sma_plot_surv_asct)
 	sma_plot_file("survival_asct_sex.png", png, sma_plot_surv_asct_sex)
+	sma_plot_file("survival_dx_dx.png", png, sma_plot_surv_dx_dx)
+	sma_plot_file("survival_asct_dx.png", png, sma_plot_surv_asct_dx)
 }