R/tally_tmhs.R

In pureseqtmr: Predict Transmembrane Protein Topology

#' Count the number of transmembrane helices in a topology
#' @inheritParams default_params_doc
#' @return a \link[tibble]{tibble} with the number of
#' TMHs per protein
#' @examples
#' if (is_pureseqtm_installed()) {
#'   tally_tmhs(
#'     predict_topology(
#'       get_example_filename("1bhaA.fasta")
#'     )
#'   )
#' }
#' @export
tally_tmhs <- function(
  topology
) {
  pureseqtmr::check_topology(topology)
  tally <- tibble::tibble(
    name = topology$name,
    n_tmhs = NA
  )
  n_rows <- nrow(tally)
  for (i in seq_len(n_rows)) {
    t <- topology$topology[i]
    tally$n_tmhs[i] <- pureseqtmr::count_n_tmhs(t)
  }
  tally
}