R/GetReplicateAverages.R

In KamilSJaron/ConjunctionStats: The parser of simulations of hybrid zones performed by simulator Conjunction (http://github.com/KamilSJaron/Conjunction)

#' @title GetReplicateAverages
#'
#' @description
#' \code{GetReplicateAverages} compute width averages over replicates.
#'
#' @param GradTable is a data frame filled by \code{FillSettingByHZAR}
#'
#' @param filename is an optional name; if specified a Figure of widths is saved.
#'
#' @param G is an optional parameter; if specified a dataset will be filtered only to row of specifiied generation.
#'
#' @param filter is an optional parameter; if specified a dataset will be filtered only to rows of widths greater than the value of this parameter.
#'
#' @author Kamil Jaron \email{kamiljaron at gmail.com}
#'
#' @export

GetReplicateAverages <- function(GradTable, filename = NA, G = NA, filter = NA){
  # GradTable - filled GradTable
  # filename - name of pdf for boxplot, default, no plot
  # G - subset of GradTable using generation

  if(!is.na(G)){
    GradTable <- GradTable[GradTable$G == G,]
  }

  if(!(is.na(filter))){
    GradTable <- GradTable[GradTable$width > filter,]
  }

  if(!is.na(filename)){
    pdf(filename)
    x <-seq(0,1,by=0.01)

    plot(x,twidth(x,sqrt(0.5)), type = 'l',
         ylim = c(min(GradTable$width, na.rm = T), max(GradTable$width, na.rm = T)),
         xlim = c(min(GradTable$s), max(GradTable$s)),
         xlab = 'selection', ylab = 'width')
  }

  GradTable_means <- data.frame(s = numeric(0),
                                b = numeric(0),
                                width = numeric(0))

  for(sel in unique(GradTable$s)){
    for(beta in unique(GradTable$b)){
      mw <- mean(GradTable$width[GradTable$s == sel & GradTable$b == beta], na.rm = T)
      GradTable_means <- rbind(GradTable_means,
                               data.frame(s = sel,
                                          b = beta,
                                          width = mw))
      if(!is.na(filename)){
        boxplot(GradTable$width[GradTable$s == sel],
                at = sel,
                boxwex = 0.05,
                add = T)
      }
    }
  }

  if(!is.na(filename)){
    dev.off()
  }

  return(GradTable_means)
}