R/profile.uiols.R

In ui: Uncertainty Intervals and Sensitivity Analysis for Missing Data

#' Plot of UI and CI
#'
#' Plot function for objects returned from \code{\link{ui.ols}}.
#' Plots confidence intervals for different values of rho and the uncertainty interval.
#' @param fitted An object of class uiols
#' @param plot.all If TRUE, plots all covariates.
#' @param which Specify which variables should be plotted by either sending in their names in a vector or a vector with their numbers (1 intercept, 2 for the first covariate etc.).
#' @param intercept If TRUE, also plots the intercept.
#' @param xlab Title for x-axis, default is \code{expression(rho)}.
#' @param ylab Title for y-axis, default is the variable names.
#' @param ... Additional arguments, for instance margins.
#'
#'
#' @importFrom graphics lines par plot polygon
#' @export


profile.uiols<-function(fitted, plot.all=TRUE, which=NA, intercept=FALSE, xlab=NULL,ylab=NULL, ...){
p<-dim(fitted$coef)[1]-1
nrho<-dim(fitted$coef)[2]
if(is.null(ylab)){NamnX<-rownames(fitted$coef)
	}else{if(length(as.vector(ylab))==1){
	NamnX<-rep(ylab,length(rownames(fitted$coef)))	
	}else{
		if(length(as.vector(ylab))==length(rownames(fitted$coef))){
	NamnX==ylab	}else{
		stop('Wrong dimensions on ylab.')
	}}}
if(is.null(xlab)){xlab=expression(rho)}
	
if(sum(is.na(which)>0)){
	if(plot.all==FALSE) warning('Need to specify which variable in order to not plot all.')
	plot.all=TRUE
}else{
	plot.all=FALSE
	if(mode(which)=='character'){
		which<-which(NamnX %in% which)
	}
}
if(plot.all){
	if(intercept){which<- 1:(p+1)}else{which<- 2:(p+1)}
	}
	
ui<-fitted$ui
ci<-fitted$ci
n<-length(which)


dim.par<-c(floor(sqrt(n)),ceiling(sqrt(n)))

if(dim.par[1]*dim.par[2]<n){
	dim.par[1]<-dim.par[1]+1
}


par(mfrow=dim.par)

for(i in which){
plot(fitted$gridrho,fitted$coef[i,],type='l',ylim=c(min(c(0, ui[i,])),max(c(0, ui[i,]))),xlab=xlab,ylab=NamnX[i])
polygon(c(fitted$gridrho, rev(fitted$gridrho)), c(ci[i,,2],rev(ci[i,,1])), col = "grey90", border = NA)
lines(fitted$gridrho,fitted$coef[i,])
lines(fitted$gridrho,ci[i,,1],lty=2)
lines(fitted$gridrho,ci[i,,2],lty=2)
lines(c(-1,1),c(0,0))
}	
}