brass-venn.R

do_more <- function(classStr, feats, profiler=profiler){
    # need to generalize this ...
    strOrigins <- list(
        orphan       = classStr[grepl("^[^O]+$",     classStr)]
      , Arabidopsis  = classStr[grepl("^O[^O][^O]$", classStr, perl=TRUE)]
      , Camelineae   = classStr[grepl("^.O[^O]$",    classStr, perl=TRUE)]
      , Brassicaceae = classStr[grepl("^..O$",       classStr, perl=TRUE)]
    )
    feats <- lapply(feats, function(d){
        d[, c("seqid",
              "gen", "trn", "orf", # ORFic
              "nuc", "cds", "exo", # non-ORFic
              "rna", "tec", "una", "ind", "nst", "scr" # Unknown
              )]
    })
    strSums <- lapply(strOrigins, function(x) summary(factor(x)))

    group_profile <- function(group, origins, feats, profiler){
        mats <- lapply(names(origins[[group]]), profiler, feats)
        names(mats) <- names(origins[[group]])
        mats
    }

    profiles <- lapply(names(strOrigins)[-1], group_profile, origins=strOrigins, feats=feats, profiler=profiler)
    names(profiles) <- names(strOrigins)[-1]
    list(
        strOrigins = strOrigins
      , strSums    = strSums
      , profiles   = profiles
    )
}

get_brass_uno_comp <- function(m, strata){

  gene_profiler <- function(seqid, feats){
      y <- lapply(feats, function(f) f[f$seqid == seqid, -1]) %>% do.call(what=rbind)
      rownames(y) <- sub("feature_table/query/", "", rownames(y))
      y
  }

  classStr <- rmonad::get_value(m, tag='query_origins')[[1]]$classStr
  feats <- rmonad::get_value(m, tag='feature_table/query')
  x <- do_more(classStr, feats, profiler=gene_profiler)

  .labels <- rmonad::get_value(m, tag="query_labels")[[1]]
  strict_origins <- fagin:::.determine_origins(.labels, con, get_classifier=by_strict)

  y <- do_more(strict_origins$classStr, feats, profiler=gene_profiler)

  list(
    comparisons = list(
       orphan       = merge_named_vectors(x=x$strSums[[1]], y=y$strSums[[1]], .fill=0)
     , Arabidopsis  = merge_named_vectors(x=x$strSums[[2]], y=y$strSums[[2]], .fill=0)
     , Camelineae   = merge_named_vectors(x=x$strSums[[3]], y=y$strSums[[3]], .fill=0)
     , Brassicaceae = merge_named_vectors(x=x$strSums[[4]], y=y$strSums[[4]], .fill=0)
    ),
    venn = make_venn_function(strata, x, y)
  )
}