Added print method for ARMAX functions

1 files changed, 49 insertions, 0 deletions

diff --git a/R/poly.R b/R/poly.R
index fd1b373..7db046c 100644
--- a/R/poly.R
+++ b/R/poly.R
@@ -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")
+  }
+}
+