feat: clonal prop and plot

NAMESPACE

@@ -2,6 +2,7 @@
 
 export("%>%")
 export(get_study)
+export(plot_clonal_prop_per_sample)
 export(plot_motif_counts)
 exportClasses(Basic)
 exportClasses(Clonality)

R/plot_clonal_prop_per_sample.R

@@ -0,0 +1,47 @@
+#' Plot Clonal Proportion per Sample
+#'
+#' This function plots the clonal proportion per sample.
+#'
+#' @param data A data frame containing the clonal proportion per sample.
+#' @param attr_col Character. Attribute column name.
+#'
+#' @return A ggplot2 object representing the clonal proportion per sample plot.
+#' @export
+#'
+#' @examples
+#' \dontrun{
+#' proj <- reTCR::get_study(id="PRJNA473147", attr_col="cmv_status")
+#' clonal_prop <- proj@clonality@clonal_prop
+#' reTCR::plot_clonal_prop_per_sample(clonal_prop, "cmv_status")
+#' }
+plot_clonal_prop_per_sample <- function(data, attr_col) {
+  ggplot2::ggplot(
+    data,
+    ggplot2::aes(
+      x = sample,
+      y = clonality_portion,
+      fill = !!rlang::sym(attr_col)
+    )
+  ) +
+    ggplot2::geom_bar(stat = "identity", position = ggplot2::position_dodge()) +
+    ggplot2::labs(x = "sample", y = "number of clonotypes") +
+    ggplot2::theme_classic() +
+    ggplot2::theme(
+      axis.text.x = ggplot2::element_text(
+        angle = 90, vjust = 0.5, hjust = 1, size = 10
+      ),
+      axis.text.y = ggplot2::element_text(
+        size = 12,
+        margin = ggplot2::margin(l = 10)
+      ),
+      axis.title.x = ggplot2::element_text(
+        size = 14,
+        margin = ggplot2::margin(t = 10)
+      ),
+      axis.title.y = ggplot2::element_text(size = 14),
+      legend.title = ggplot2::element_text(size = 14),
+      legend.text = ggplot2::element_text(size = 12)
+    ) +
+    ggplot2::scale_y_continuous(expand = c(0, 0)) +
+    ggplot2::scale_fill_brewer(palette = "Set1", name = attr_col)
+}

R/plot_motif_counts.R

@@ -10,7 +10,7 @@
 #'
 #' @examples
 #' \dontrun{
-#' proj <- reTCR::get_study(id = "PRJNA473147")
+#' proj <- reTCR::get_study(id="PRJNA473147", attr_col="cmv_status")
 #' reTCR::plot_motif_counts(proj@motif@aa_motif_count, 20)
 #' }
 plot_motif_counts <- function(data, threshold = 60) {
@@ -29,7 +29,11 @@ plot_motif_counts <- function(data, threshold = 60) {
     )
   ) +
     ggplot2::geom_jitter(width = 0.2, size = 3) +
-    ggplot2::labs(x = "Amino acid motif", y = "Count", color = "CMV Status") +
+    ggplot2::labs(
+      x = "Amino acid motif",
+      y = "Count",
+      color = "CMV Status"
+    ) +
     ggplot2::theme_bw() +
     ggplot2::theme(
       axis.text.x = ggplot2::element_text(

README.md

@@ -106,6 +106,10 @@ print(proj@clonality@pielou)
 # clonal proportion
 print(proj@clonality@clonal_prop)
 
+# plot clonal proportion per sample
+clonal_prop <- proj@clonality@clonal_prop
+reTCR::plot_clonal_prop_per_sample(clonal_prop, "cmv_status")
+
 # relative abundance (in all clonotypes)
 print(proj@clonality@abundance)