added predict routine for idpoly objects

2 files changed, 19 insertions, 7 deletions

diff --git a/R/estpoly.R b/R/estpoly.R
index 2fa4351..25a65d2 100644
--- a/R/estpoly.R
+++ b/R/estpoly.R
@@ -67,16 +67,15 @@ predict.estpoly <- function(x,newdata=NULL,nahead=1){
   if(is.null(newdata)&&nahead=1){
     return(fitted(x))
   } else{
-    mod <- x$sys
+    model <- x$sys
     if(is.null(newdata)){
-      y <- fitted(mod)+resid(mod)
-      u <- mod$input
+      y <- fitted(x)+resid(x)
+      u <- x$input
+      z <- idframe(y,u,Ts = deltat(y),start=start(y))
     } else{
-      y <- outputData(newdata); u <- inputData(newdata)
+      z <- newdata
     }
-   
-    
-    return(ypred)
+    predict(model,z,nahead)
   } 
 }
 
diff --git a/R/poly.R b/R/poly.R
index 15591e3..a4e777b 100644
--- a/R/poly.R
+++ b/R/poly.R
@@ -164,4 +164,17 @@ print.idpoly <- function(mod,se=NULL,dig=3){
     }
   }
   cat("\n")
+}
+
+#' @export
+predict.idpoly <- function(x,data,nahead=1){
+  y <- outputData(z); u<- inputData(z)
+  G <- signal::Arma(b=c(rep(0,x$ioDelay),x$B),
+                    a= as.numeric(polynom::polynomial(x$A)*
+                                    polynom::polynomial(x$F1)))
+  
+  Hden <- as.numeric(polynom::polynomial(x$A)*polynom::polynomial(x$D))
+  Hinv <- signal::Arma(b=Hden,a=x$C)
+  
+  return(ypred)
 }
\ No newline at end of file