diff options
| -rw-r--r-- | R/poly.R | 49 |
1 files changed, 49 insertions, 0 deletions
@@ -27,6 +27,8 @@ checkUnity <- function(x){ print.idpoly <- function(x){ if(x$type=="arx"){ print_arx(x) + } else if(x$type=="armax"){ + print_armax(x) } } @@ -64,3 +66,50 @@ print_arx <- function(obj){ } } +print_armax <- function(obj){ + cat("Discrete-time ARX model: A(q^{-1})y[k] = B(q^{-1})u[k] + C(q^{-1})e[k] \n\n") + cat("A(q^{-1}) = ") + for(i in seq_along(obj$A)){ + if(i-1==0){ + cat(obj$A[i]) + } else{ + if(obj$A[i]>0) cat(" + ") else cat("- ") + + if(!(abs(obj$A[i])==1)) cat(abs(obj$A[i])) + cat("q^{-",i-1,"}",sep="") + } + cat("\t") + } + cat("\n") + cat("B(q^{-1}) = ") + for(i in seq_along(obj$B)){ + if(i+obj$ioDelay-1==0){ + cat(obj$B[i]) + } else{ + + if(!((obj$ioDelay!=0) && (i==1))){ + if(obj$B[i]>0) cat(" + ") else cat("- ") + } else{ + if(obj$B[i]<0) cat("-") + } + + if(!(abs(obj$B[i])==1)) cat(abs(obj$B[i])) + cat("q^{-",i+obj$ioDelay-1,"}",sep="") + } + cat("\t") + } + cat("\n") + cat("C(q^{-1}) = ") + for(i in seq_along(obj$C)){ + if(i-1==0){ + cat(obj$C[i]) + } else{ + if(obj$C[i]>0) cat(" + ") else cat("- ") + + if(!(abs(obj$C[i])==1)) cat(abs(obj$C[i])) + cat("q^{-",i-1,"}",sep="") + } + cat("\t") + } +} + |