R/egg.plot.R

In eggs: Distances and areas on the surface of an egg

#' Plot the fitted egg object
#'
#' Adds the fitted egg object to the current plot window
#'
#' @param egg an object of class \code{egg.fit}
#' @param ... additional parameters passed to the \code{\link[graphics]{lines}} function
#'
#' @return None
#'
#' @examples
#' stop()
#'
#' @export
egg.plot = function(egg, ...) {
  Rinv = solve(egg$R)
  n = 100
  qq = matrix(0, nrow=2*n, ncol=2)
  qq[,2] = c(seq(-egg$length/2, egg$length/2, length.out=n), -seq(-egg$length/2, egg$length/2, length.out=n))
  z.hat = normalise(qq[,2], -egg$length/2, egg$length/2, 0, 1)
  for(i in 1:(2*n)) {
    qq[i,1] = get.d(z.hat[i], egg)
    if(i > n) {
      qq[i,1] = -qq[i,1]
    }
    qq[i,] = t(Rinv %*% cbind(c(qq[i,1], qq[i,2]))) + egg$mid.point
  }
  lines(qq[,1], qq[,2], ...)
}