after merg

[talweg.git] / pkg / R / F_Persistence.R

diff --git a/pkg/R/F_Persistence.R b/pkg/R/F_Persistence.R
index dde8612..7035231 100644 (file)
--- a/pkg/R/F_Persistence.R
+++ b/pkg/R/F_Persistence.R
@@ -18,23 +18,21 @@ PersistenceForecaster = R6::R6Class("PersistenceForecaster",
        inherit = Forecaster,
 
        public = list(
-               predictShape = function(data, today, memory, horizon, ...)
+               predictShape = function(data, today, memory, predict_from, horizon, ...)
                {
                        # Return centered last (similar) day curve, avoiding NAs until memory is run
                        first_day = max(1, today-memory)
                        same_day = ifelse(hasArg("same_day"), list(...)$same_day, TRUE)
-                       # If 'same_day', get the last known future of similar day: -7 + 1 == -6
-                       index = today - ifelse(same_day,6,0)
+                       index <- today
                        repeat
                        {
-                               {
-                                       last_serie = data$getCenteredSerie(index)[1:horizon]
-                                       index = index - ifelse(same_day,7,1)
-                               };
-                               if (!any(is.na(last_serie)))
-                                       return (last_serie);
+                               # If 'same_day', get the last known future of similar day
+                               index = index - ifelse(same_day,7,1)
                                if (index < first_day)
                                        return (NA)
+                               last_serie = data$getCenteredSerie(index)[predict_from:horizon]
+                               if (!any(is.na(last_serie)))
+                                       return (last_serie)
                        }
                }
        )