R/plot_n_tmhs_per_protein.R

In richelbilderbeek/ncbi_results: Creates Results

#' The number of TMHs per protein
#' @inheritParams default_params_doc
#' @export
plot_n_tmhs_per_protein <- function(folder_name = folder_name) {

  topo_filenames <- list.files(
    path = folder_name,
    pattern = ".*\\.topo$",
    full.names = TRUE
  )
  testthat::expect_true(length(topo_filenames) > 0)

  tibbles <- list()
  for (i in seq_along(topo_filenames)) {
    t <- dplyr::distinct(
      pureseqtmr::load_fasta_file_as_tibble_cpp(topo_filenames[i])
    )
    tibbles[[i]] <- pureseqtmr::count_n_tmhs(t$sequence)
  }

  n_tmhs <- unlist(tibbles)
  n_tmhs

  n_nonzero_tmhs <- n_tmhs[n_tmhs > 0]
  # n_snps <- NULL; rm(n_snps) # nolint, fixes warning: no visible binding for global variable

  ggplot2::ggplot() +
    ggplot2::aes(n_nonzero_tmhs) +
    ggplot2::geom_histogram(fill = "#BBBBBB", bins = 50) +
    ggplot2::scale_x_continuous(name = "Number of TMHs") +
    ggplot2::scale_y_continuous(name = "Number of proteins") +
    bbbq::get_bbbq_theme()
}