R/benchmark_taxize_spectra.R

In taxonomicallyinformedannotation/tima-r: Taxonomically Informed Metabolite Annotation

#' @title Taxize spectra benchmark
#'
#' @description This function adds taxa to the benchmark
#'
#' @details Because they are still quite dirty
#'
#' @param input Initial features
#' @param keys SOP keys
#' @param org_tax_ott Taxonomy
#' @param output Prepared features
#'
#' @return NULL
#'
#' @export
#'
#' @examples NULL
benchmark_taxize_spectra <-
  function(input,
           keys,
           org_tax_ott,
           output) {
    features <- input |>
      tidytable::fread(
        na.strings = c("", "NA"),
        colClasses = "character"
      )
    sop <- keys |>
      tidytable::fread(
        na.strings = c("", "NA"),
        colClasses = "character"
      ) |>
      tidytable::mutate(inchikey_no_stereo = stringi::stri_sub(
        str = structure_inchikey,
        from = 1,
        to = 14
      ))
    taxo <- org_tax_ott |>
      tidytable::fread(
        na.strings = c("", "NA"),
        colClasses = "character"
      )

    features_pretaxed <- features |>
      tidytable::left_join(sop |>
        tidytable::distinct(
          organism_name,
          inchikey_no_stereo
        ))
    rm(features, sop)
    set.seed(42)
    features_sampled <- features_pretaxed |>
      tidytable::filter(!is.na(organism_name)) |>
      tidytable::slice_sample(n = 1, .by = feature_id) |>
      tidytable::bind_rows(features_pretaxed |> tidytable::filter(is.na(organism_name)))
    rm(features_pretaxed)

    features_taxed <- features_sampled |>
      tidytable::left_join(taxo |>
        tidytable::distinct(organism_name, .keep_all = TRUE)) |>
      tidytable::select(
        feature_id,
        sample_organism_01_domain = organism_taxonomy_01domain,
        sample_organism_02_kingdom = organism_taxonomy_02kingdom,
        sample_organism_03_phylum = organism_taxonomy_03phylum,
        sample_organism_04_class = organism_taxonomy_04class,
        sample_organism_05_order = organism_taxonomy_05order,
        sample_organism_06_family = organism_taxonomy_06family,
        sample_organism_07_tribe = organism_taxonomy_07tribe,
        sample_organism_08_genus = organism_taxonomy_08genus,
        sample_organism_09_species = organism_taxonomy_09species,
        sample_organism_10_varietas = organism_taxonomy_10varietas
      )
    rm(features_sampled, taxo)

    export_output(x = features_taxed, file = output)
    rm(features_taxed)

    return(c(output))
  }