miga-base 1.2.17.0 → 1.2.17.1

data/utils/enveomics/enveomics.R/R/recplot.R

@@ -0,0 +1,354 @@
+#' Enveomics: Recruitment Plots
+#' 
+#' @description 
+#' Produces recruitment plots provided that BlastTab.catsbj.pl has
+#' been previously executed. Requires the \pkg{gplots} library.
+#' 
+#' @param prefix
+#' Path to the prefix of the BlastTab.catsbj.pl output files. At
+#' least the files \strong{.rec} and \strong{.lim} must exist with this prefix.
+#' @param id.min
+#' Minimum identity to be considered. By default, the minimum detected
+#' identity. This value is a percentage.
+#' @param id.max
+#' Maximum identity to be considered. By default, 100\%.
+#' @param id.binsize
+#' Size of the identity bins (vertical histograms). By default, 0.1 for
+#' identity metrics and 5 for bit score.
+#' @param id.splines
+#' Smoothing parameter for the splines in the identity histogram. Zero (0) for no
+#' splines. A generally good value is 1/2. If non-zero, requires the \pkg{stats} package.
+#' @param id.metric
+#' Metric of identity to be used (Y-axis). 
+#' It can be any unambiguous prefix of:
+#' \itemize{
+#' \item "identity"
+#' \item "corrected identity" 
+#' \item "bit score"}
+#' @param id.summary
+#' Method used to build the identity histogram (Horizontal axis of the right panel).
+#' It can be any unambiguous prefix of: 
+#' \itemize{
+#' \item "sum"
+#' \item "average" 
+#' \item "median"
+#' \item "90\% lower bound"
+#' \item "90\% upper bound"
+#' \item "95\% lower bound"
+#' \item "95\% upper bound" }
+#' The last four options
+#' correspond to the upper and lower boundaries of the 90\% and 95\% empirical confidence
+#' intervals.
+#' @param pos.min
+#' Minimum (leftmost) position in the reference (concatenated) genome (in bp).
+#' @param pos.max
+#' Maximum (rightmost) position in the reference (concatenated) genome (in bp).
+#' By default: Length of the genome.
+#' @param pos.binsize
+#' Size of the position bins (horizontal histograms) in bp.
+#' @param pos.splines
+#' Smoothing parameter for the splines in the position histogram. Zero (0) for no splines.
+#' If non-zero, requires the stats package.
+#' @param rec.col1
+#' Lightest color in the recruitment plot.
+#' @param rec.col2
+#' Darkest color in the recruitment plot.
+#' @param main
+#' Title of the plot.
+#' @param contig.col
+#' Color of the Contig boundaries. Set to \code{NA} to ignore Contig boundaries.
+#' @param ret.recplot
+#' Indicates if the matrix of the recruitment plot is to be returned.
+#' @param ret.hist
+#' Ignored, for backwards compatibility.
+#' @param ret.mode
+#' Indicates if the mode of the identity is to be computed. It requires the 
+#' \pkg{modeest} package.
+#' @param id.cutoff
+#' Minimum identity to consider an alignment as "top". By default, it is 0.95 for the
+#' identity metrics and 95\% of the best scoring alignment for bit score.
+#' @param verbose
+#' Indicates if the function should report the advance.
+#' @param ...
+#' Any additional graphic parameters to be passed to plot for all panels except the
+#' recruitment plot (lower-left).
+#' 
+#' @return 
+#' 
+#' Returns a list with the following elements:
+#' 
+#' \describe{
+#' \item{\code{pos.marks}}{Midpoints of the position histogram.}
+#' \item{\code{id.matrix}}{Midpoints of the identity histogram.}
+#' \item{\code{recplot}}{Matrix containing the recruitment plot values 
+#'      (if \code{ret.recplot=TRUE}).}
+#' \item{\code{id.mean}}{Mean identity.}
+#' \item{\code{id.median}}{Median identity.}
+#' \item{\code{id.mode}}{Mode of the identity (if \code{ret.mode=TRUE}). Deprecated.}
+#' \item{\code{id.hist}}{Values of the identity histogram (if \code{ret.hist=TRUE}).}
+#' \item{\code{pos.hist.low}}{Values of the position histogram (depth) with "low"
+#'      identity (i.e., below id.cutoff) (if \code{ret.hist=TRUE}).}
+#' \item{\code{pos.hist.top}}{Values of the position histogram (depth) with "top"
+#'      identity (i.e., above id.cutoff) (if \code{ret.hist=TRUE}).}
+#' \item{\code{id.max}}{Value of \code{id.max}. This is returned because 
+#'      \code{id.max=NULL} may vary.}
+#' \item{\code{id.cutoff}}{Value of \code{id.cutoff}. 
+#'      This is returned because \code{id.cutoff=NULL} may vary.}
+#' \item{\code{seqdepth.mean.top}}{Average sequencing depth with identity above 
+#'      \code{id.cutoff}.}
+#' \item{\code{seqdepth.mean.low}}{Average sequencing depth with identity below 
+#'      \code{id.cutoff}.} 
+#' \item{\code{seqdepth.mean.all}}{Average sequencing depth without identity filtering.}
+#' \item{\code{seqdepth.median.top}}{Median sequencing depth with identity above 
+#'      \code{id.cutoff}.}
+#' \item{\code{seqdepth.median.low}}{Median sequencing depth with identity below 
+#'      \code{id.cutoff}.}
+#' \item{\code{seqdepth.median.all}}{Median sequencing depth without identity filtering.}
+#' \item{\code{id.metric}}{Full name of the used identity metric.}
+#' \item{\code{id.summary}}{Full name of the summary method used to build the identity plot.}}
+#'
+#' @author Luis M. Rodriguez-R [aut, cre]
+#'
+#' @export
+
+enve.recplot <- structure(function(
+  prefix,
+  
+  # Id. hist.
+  id.min=NULL,
+  id.max=NULL,
+  id.binsize=NULL,
+  id.splines=0,
+  id.metric='id',
+  id.summary='sum',
+  
+  # Pos. hist.
+  pos.min=1,
+  pos.max=NULL,
+  pos.binsize=1e3,
+  pos.splines=0,
+  
+  # Rec. plot
+  rec.col1='white',
+  rec.col2='black',
+  
+  # General
+  main=NULL,
+  contig.col=grey(0.85),
+  
+  # Return
+  ret.recplot=FALSE,
+  ret.hist=FALSE,
+  ret.mode=FALSE,
+  
+  # General
+  id.cutoff=NULL,
+  verbose=TRUE,
+  ...
+){
+  
+  # Settings
+  METRICS <- c('identity', 'corrected identity', 'bit score');
+  SUMMARY <- c('sum', 'average', 'median', '');
+  if(is.null(prefix)) stop('Parameter prefix is mandatory.');
+  if(!requireNamespace("gplots", quietly=TRUE)) stop('Unavailable gplots library.');
+  
+  # Read files
+  if(verbose) cat("Reading files.\n")
+  rec <- read.table(paste(prefix, '.rec', sep=''), sep="\t", comment.char='', quote='');
+  lim <- read.table(paste(prefix, '.lim', sep=''), sep="\t", comment.char='', quote='');
+  
+  # Configure ID summary
+  id.summary <- pmatch(id.summary, SUMMARY);
+  if(is.na(id.summary)) stop('Invalid identity summary.');
+  if(id.summary == -1) stop('Ambiguous identity summary.');
+  if(id.summary==1){
+    id.summary.func <- function(x) colSums(x);
+    id.summary.name <- 'sum'
+  }else if(id.summary==2){
+    id.summary.func <- function(x) colMeans(x);
+    id.summary.name <- 'mean'
+  }else if(id.summary==3){
+    id.summary.func <- function(x) apply(x,2,median);
+    id.summary.name <- 'median'
+  }else if(id.summary==4){
+    id.summary.func <- function(x) apply(x,2,quantile,probs=0.05,names=FALSE);
+    id.summary.name <- '90% LB'
+  }else if(id.summary==5){
+    id.summary.func <- function(x) apply(x,2,quantile,probs=0.95,names=FALSE);
+    id.summary.name <- '90% UB'
+  }else if(id.summary==6){
+    id.summary.func <- function(x) apply(x,2,quantile,probs=0.025,names=FALSE);
+    id.summary.name <- '95% LB'
+  }else if(id.summary==7){
+    id.summary.func <- function(x) apply(x,2,quantile,probs=0.975,names=FALSE);
+    id.summary.name <- '95% UB'
+  }
+  
+  # Configure metrics
+  id.metric <- pmatch(id.metric, METRICS);
+  if(is.na(id.metric)) stop('Invalid identity metric.');
+  if(id.metric == -1) stop('Ambiguous identity metric.');
+  if(id.metric==1){
+    id.reccol <- 3
+    id.shortname <- 'Id.'
+    id.fullname  <- 'Identity'
+    id.units     <- '%'
+    id.hallmarks <- seq(0, 100, by=5)
+    if(is.null(id.max)) id.max <- 100
+    if(is.null(id.cutoff)) id.cutoff <- 95
+    if(is.null(id.binsize)) id.binsize <- 0.1
+  }else if(id.metric==2){
+    if(ncol(rec)<6) stop("Requesting corrected identity, but .rec file doesn't have 6th column")
+    id.reccol <- 6
+    id.shortname <- 'cId.'
+    id.fullname  <- 'Corrected identity'
+    id.units     <- '%'
+    id.hallmarks <- seq(0, 100, by=5)
+    if(is.null(id.max)) id.max <- 100
+    if(is.null(id.cutoff)) id.cutoff <- 95
+    if(is.null(id.binsize)) id.binsize <- 0.1
+  }else if(id.metric==3){
+    id.reccol <- 4
+    id.shortname <- 'BSc.'
+    id.fullname  <- 'Bit score'
+    id.units     <- 'bits'
+    max.bs <- max(rec[, id.reccol])
+    id.hallmarks <- seq(0, max.bs*1.2, by=50)
+    if(is.null(id.max)) id.max <- max.bs
+    if(is.null(id.cutoff)) id.cutoff <- 0.95 * max.bs
+    if(is.null(id.binsize)) id.binsize <- 5
+  }
+  if(is.null(id.min)) id.min <- min(rec[, id.reccol]);
+  if(is.null(pos.max)) pos.max <- max(lim[, 3]);
+  id.lim <- c(id.min, id.max);
+  pos.lim <- c(pos.min, pos.max)/1e6;
+  id.breaks <- round((id.max-id.min)/id.binsize);
+  pos.breaks <- round((pos.max-pos.min)/pos.binsize);
+  if(is.null(main)) main <- paste('Recruitment plot of ', prefix, sep='');
+  pos.marks=seq(pos.min, pos.max, length.out=pos.breaks+1)/1e6;
+  id.marks=seq(id.min, id.max, length.out=id.breaks+1);
+  id.topclasses <- 0;
+  for(i in length(id.marks):1) if(id.marks[i]>id.cutoff) id.topclasses <- id.topclasses + 1;
+  
+  # Set-up image
+  layout(matrix(c(3,4,1,2), nrow=2, byrow=TRUE), widths=c(2,1), heights=c(1,2));
+  out <- list();
+  
+  # Recruitment plot
+  if(verbose) cat("Rec. plot.\n")
+  par(mar=c(5,4,0,0)+0.1);
+  rec.hist <- matrix(0, nrow=pos.breaks, ncol=id.breaks);
+  for(i in 1:nrow(rec)){
+    id.class <- ceiling((id.breaks)*((rec[i, id.reccol]-id.min)/(id.max-id.min)));
+    if(id.class<=id.breaks & id.class>0){
+      for(pos in rec[i, 1]:rec[i, 2]){
+        pos.class <- ceiling((pos.breaks)*((pos-pos.min)/(pos.max-pos.min)));
+        if(pos.class<=pos.breaks & pos.class>0) rec.hist[pos.class, id.class] <- rec.hist[pos.class, id.class]+1;
+      }
+    }
+  }
+  id.top <- c((1-id.topclasses):0) + id.breaks;
+  rec.col=gplots::colorpanel(256, rec.col1, rec.col2);
+  image(x=pos.marks, y=id.marks, z=log10(rec.hist),
+        breaks=seq(0, log10(max(rec.hist)), length.out=1+length(rec.col)), col=rec.col,
+        xlim=pos.lim, ylim=id.lim, xlab='Position in genome (Mbp)',
+        ylab=paste(id.fullname, ' (',id.units,')', sep=''), xaxs='i', yaxs='r');
+  if(!is.na(contig.col)) abline(v=c(lim$V2, lim$V3)/1e6, lty=1, col=contig.col);
+  abline(h=id.hallmarks, lty=2, col=grey(0.7));
+  abline(h=id.marks[id.top[1]], lty=3, col=grey(0.5))
+  legend('bottomleft', 'Rec. plot', bg=rgb(1,1,1,2/3));
+  out <- c(out, list(pos.marks=pos.marks, id.marks=id.marks));
+  if(ret.recplot) out <- c(out, list(recplot=rec.hist));
+  
+  # Identity histogram
+  if(verbose) cat(id.shortname, " hist.\n", sep='')
+  par(mar=c(5,0,0,2)+0.1);
+  id.hist <- id.summary.func(rec.hist);
+  plot(1, t='n', xlim=c(1, max(id.hist)), ylim=id.lim, ylab='', yaxt='n', xlab=paste('Sequences (bp),', id.summary.name), log='x', ...);
+  id.x <- rep(id.marks, each=2)[2:(id.breaks*2+1)]
+  id.f <- rep(id.hist, each=2)[1:(id.breaks*2)]
+  if(sum(id.f)>0){
+    lines(id.f, id.x, lwd=ifelse(id.splines>0, 1/2, 2), type='o', pch='.');
+    if(id.splines>0){
+      id.spline <- smooth.spline(id.x[id.f>0], log(id.f[id.f>0]), spar=id.splines)
+      lines(exp(id.spline$y), id.spline$x, lwd=2)
+    }
+  }
+  
+  abline(h=id.hallmarks, lty=2, col=grey(0.7));
+  abline(h=id.marks[id.top[1]], lty=3, col=grey(0.5))
+  legend('bottomright', paste(id.shortname, 'histogram'), bg=rgb(1,1,1,2/3));
+  out <- c(out, list(id.mean=mean(rec[, id.reccol])));
+  out <- c(out, list(id.median=median(rec[, id.reccol])));
+  if(ret.hist)  out <- c(out, list(id.hist=id.hist));
+  
+  # Position histogram
+  if(verbose) cat("Pos. hist.\n")
+  par(mar=c(0,4,4,0)+0.1);
+  h1<-rep(0,nrow(rec.hist)) ;
+  h2<-rep(0,nrow(rec.hist)) ;
+  pos.winsize <- (pos.max-pos.min+1)/pos.breaks;
+  if(sum(rec.hist[, id.top])>0) h1 <- rowSums(matrix(rec.hist[, id.top], nrow=nrow(rec.hist)))/pos.winsize;
+  if(sum(rec.hist[,-id.top])>0) h2 <- rowSums(matrix(rec.hist[,-id.top], nrow=nrow(rec.hist)))/pos.winsize;
+  
+  ymin <- min(1, h1[h1>0], h2[h2>0]);
+  ymax <- max(10, h1, h2);
+  if(is.na(ymin) || ymin<=0) ymin <- 1e-10;
+  if(is.na(ymax) || ymax<=0) ymax <- 1;
+  plot(1, t='n', xlab='', xaxt='n', ylab='Sequencing depth (X)', log='y', xlim=pos.lim,
+       ylim=c(ymin, ymax), xaxs='i', main=main, ...);
+  if(!is.na(contig.col)) abline(v=c(lim[,2], lim[,3])/1e6, lty=1, col=contig.col);
+  abline(h=10^c(0:5), lty=2, col=grey(0.7));
+  if(sum(h2)>0){
+    h2.x <- rep(pos.marks, each=2)[2:(pos.breaks*2+1)]
+    h2.y <- rep(h2, each=2)[1:(pos.breaks*2)]
+    lines(h2.x, h2.y, lwd=ifelse(pos.splines>0, 1/2, 2), col=grey(0.5));
+    if(pos.splines>0){
+      h2.spline <- smooth.spline(h2.x[h2.y>0], log(h2.y[h2.y>0]), spar=pos.splines)
+      lines(h2.spline$x, exp(h2.spline$y), lwd=2, col=grey(0.5))
+    }
+    if(ret.hist) out <- c(out, list(pos.hist.low=h2.y));
+  }
+  if(sum(h1)>0){
+    h1.x <- rep(pos.marks, each=2)[2:(pos.breaks*2+1)]
+    h1.y <- rep(h1, each=2)[1:(pos.breaks*2)]
+    lines(h1.x, h1.y, lwd=ifelse(pos.splines>0, 1/2, 2), col=grey(0));
+    if(pos.splines>0){
+      h1.spline <- smooth.spline(h1.x[h1.y>0], log(h1.y[h1.y>0]), spar=pos.splines)
+      lines(h1.spline$x, exp(h1.spline$y), lwd=2, col=grey(0))
+    }
+    if(ret.hist) out <- c(out, list(pos.hist.top=h1.y));
+  }
+  legend('topleft', 'Pos. histogram', bg=rgb(1,1,1,2/3));
+  out <- c(out, list(id.max=id.max, id.cutoff=id.marks[id.top[1]]));
+  out <- c(out, list(seqdepth.mean.top=mean(h1)));
+  out <- c(out, list(seqdepth.mean.low=mean(h2)));
+  out <- c(out, list(seqdepth.mean=mean(h1+h2)));
+  out <- c(out, list(seqdepth.median.top=median(h1)));
+  out <- c(out, list(seqdepth.median.low=median(h2)));
+  out <- c(out, list(seqdepth.median=median(h1+h2)));
+  out <- c(out, list(id.metric=id.fullname));
+  out <- c(out, list(id.summary=id.summary.name));
+  
+  # Legend
+  par(mar=c(0,0,4,2)+0.1);
+  plot(1, t='n', xlab='', xaxt='n', ylab='', yaxt='n', xlim=c(0,1), ylim=c(0,1), xaxs='r', yaxs='i', ...);
+  text(1/2, 5/6, labels=paste('Reads per ', signif((pos.max-pos.min)/pos.breaks, 2), ' bp (rec. plot)', sep=''), pos=3);
+  leg.col <- gplots::colorpanel(100, rec.col1, rec.col2);
+  leg.lab <- signif(10^seq(0, log10(max(rec.hist)), length.out=10), 2);
+  for(i in 1:10){
+    for(j in 1:10){
+      k <- (i-1)*10 + j;
+      polygon(c(k-1, k, k, k-1)/100, c(2/3, 2/3, 5/6, 5/6), border=leg.col[k], col=leg.col[k]);
+    }
+    text((i-0.5)/10, 2/3, labels=paste(leg.lab[i], ''), srt=90, pos=2, offset=0, cex=3/4);
+  }
+  legend('bottom',
+         legend=c('Contig boundary', 'Hallmark', paste(id.fullname, 'cutoff'),
+                  paste('Pos. hist.: ',id.shortname,' > ',signif(id.marks[id.top[1]],2),id.units,sep=''),
+                  paste('Pos. hist.: ',id.shortname,' < ',signif(id.marks[id.top[1]],2),id.units,sep='')), ncol=2,
+         col=grey(c(0.85, 0.7, 0.5, 0, 0.5)), lty=c(1,2,3,1,1), lwd=c(1,1,1,2,2), bty='n', inset=0.05, cex=5/6);
+  return(out);
+});
+