R/ggfda.R

In YunlongNie/sFuSIM: R package for Functional Single Index Model with compact support

#' This function computes the coeffcients of the index function 
#' @param fdobj fda object 
#' @export fdagg 
#' @import fda
#' @import dplyr
#' @import ggplot2
#' @import reshape2
#' @examples
#' \dontrun{
#' library(fda)
#' library(cFuSIM)
#' spline.basis=create.bspline.basis(rangeval=c(1,24),nbasis=10,norder)
#' beta_fd = fd(rnorm(10), spline.basis)
#' plot(beta_fd)
#' }
fdagg = function(fdobj, ylab='Value', xlab="Time", addzero = FALSE)
{
range= fdobj$basis$rangeval

t = seq(range[1], range[2], len=1e3)

names_y = eval.fd(t, fdobj)%>%colnames

data=  data.frame(Time=t, eval.fd(t, fdobj))
names(data) = c('Time',names_y)

plotdata = melt(data, id=1)

plot = ggplot(plotdata,aes(x=Time, y=value,group=factor(variable), color=factor(variable)))+geom_line()+theme_bw()+ylab(ylab)+xlab(xlab)
if (addzero) plot = plot+geom_hline(yintercept=0, linetype=2, col=2)
plot = plot+theme(legend.position = "none")
return(plot)
}