R/plot.pgd.R

In quesadagranja/pgd: Proper Generalized Decomposition

#' Surface plot of the \code{pgd} class
#' @param o The \code{pgd} class.
#' @param nmodes The number of modes to be plotted.
#' @param phi Angle defining the viewing direction: azimuthal direction.
#' @param theta Angle defining the viewing direction: colatitude.
#' @return Surface plot of the approximated result.
#' @export

plot.pgd <- function(o, nmodes = dim(o$f$x)[2], phi = 45, theta = 45) {
  # Compute surface
  z <- o$f$x[,1:nmodes] %*% t(o$f$y[,1:nmodes])
  # Plot surface
  persp(
    x = o$coor$x,
    y = o$coor$y,
    z = z,
    phi = phi,
    theta = theta,
    xlab = "x",
    ylab = "y"
  )
}