X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=pkg%2FR%2FData.R;h=551cfaf6f5dbed4b9e2251cd5275b96b4492d35c;hb=72b9c50162bcdcf6c99fbb8b2ec6ea9ba98379cb;hp=d970f7b825ad6af4e5e57b16f84a1cb2024ce5b3;hpb=25b75559e2d9bf84e2de35b851d93fefdae36e17;p=talweg.git

diff --git a/pkg/R/Data.R b/pkg/R/Data.R
index d970f7b..551cfaf 100644
--- a/pkg/R/Data.R
+++ b/pkg/R/Data.R
@@ -2,130 +2,116 @@
 #'
 #' Data encapsulation
 #'
-#' @field data List of
-#' \itemize{
+#' @docType class
+#' @importFrom R6 R6Class
+#'
+#' @field .data List of \itemize{
 #'   \item time: vector of times
 #'   \item serie: centered series
 #'   \item level: corresponding levels
 #'   \item exo: exogenous variables
-#'   \item exo_hat: predicted exogenous variables
-#' }
+#'   \item exo_hat: predicted exogenous variables}
 #'
-#' @docType class
-#' @importFrom R6 R6Class
+#' @section Methods:
+#' \describe{
+#' \item{\code{getSize()}}{
+#'   Return number of series in dataset.}
+#' \item{\code{getStdHorizon()}}{
+#'   Return number of time steps from serie[1] until midnight}
+#' \item{\code{appendHat(time, hat_serie, hat_exo)}}{
+#'   New estimated data + time.}
+#' \item{\code{append(serie, exo)}}{
+#'   New measured data; call *after* \code{appendHat()}}
+#' \item{\code{getTime(index)}}{
+#'   Get times at specified index.}
+#' \item{\code{getCenteredSerie(index, hat=FALSE)}}{
+#'   Get (measured or predicted) centered serie at specified index.}
+#' \item{\code{getCenteredSeries(indices, hat=FALSE)}}{
+#'   Get centered series at specified indices (in columns).}
+#' \item{\code{getLevel(index, hat=FALSE)}}{
+#'   Get level at specified index.}
+#' \item{\code{getSerie(index, hat=FALSE)}}{
+#'   Get serie (centered+level) at specified index.}
+#' \item{\code{getSeries(indices, hat=FALSE)}}{
+#'   Get series at specified indices (in columns).}
+#' \item{\code{getExo(index, hat=FALSE)}}{
+#'   Get exogenous variables at specified index.}
+#' }
 #'
-#' @export
 Data = R6::R6Class("Data",
 	private = list(
 		.data = list()
 	),
 	public = list(
 		getSize = function()
-			getSizeData(private)
+			length(private$.data)
 		,
 		getStdHorizon = function()
-			getStdHorizonData(private)
-		,
-		append = function(new_time, new_centered_serie, new_level, new_exo, new_exo_hat)
-			appendData(private, new_time, new_centered_serie, new_level, new_exo, new_exo_hat)
+			24 - as.POSIXlt( private$.data[[1]]$time[1] )$hour + 1
 		,
+		appendHat = function(time, hat_serie, hat_exo)
+		{
+			hat_level = mean(hat_serie, na.rm=TRUE)
+			hat_centered_serie = hat_serie - hat_level
+			private$.data[[length(private$.data)+1]] <- list(
+				"time"=time, "hat_centered_serie"=hat_centered_serie,
+				"hat_level"=hat_level, "hat_exo"=hat_exo )
+		},
+		append = function(serie, exo)
+		{
+			level = mean(serie, na.rm=TRUE)
+			centered_serie = serie - level
+			private$.data[[length(private$.data)]]$centered_serie <- centered_serie,
+			private$.data[[length(private$.data)]]$level <- level,
+			private$.data[[length(private$.data)]]$exo <- exo,
+		},
 		getTime = function(index)
-			getTimeData(self, private, index)
-		,
-		getCenteredSerie = function(index)
-			getCenteredSerieData(self, private, index)
-		,
-		getLevel = function(index)
-			getLevelData(self, private, index)
+		{
+			index = dateIndexToInteger(index, self)
+			private$.data[[index]]$time
+		},
+		getCenteredSerie = function(index, hat=FALSE)
+		{
+			index = dateIndexToInteger(index, self)
+			if (hat)
+				private$.data[[index]]$hat_centered_serie
+			else
+				private$.data[[index]]$centered_serie
+		},
+		getCenteredSeries = function(indices, hat=FALSE)
+			sapply(indices, function(i) self$getCenteredSerie(i, hat))
 		,
-		getSerie = function(index)
-			getSerieData(self, private, index)
+		getLevel = function(index, hat=FALSE)
+		{
+			index = dateIndexToInteger(index, self)
+			if (hat)
+				private$.data[[index]]$hat_level
+			else
+				private$.data[[index]]$level
+		},
+		getSerie = function(index, hat=FALSE)
+		{
+			index = dateIndexToInteger(index, self)
+			if (hat)
+				private$.data[[index]]$hat_centered_serie + private$.data[[index]]$hat_level
+			else
+				private$.data[[index]]$centered_serie + private$.data[[index]]$level
+		},
+		getSeries = function(indices, hat=FALSE)
+				sapply(indices, function(i) self$getSerie(i, hat))
 		,
-		getExo = function(index)
-			getExoData(self, private, index)
+		getExo = function(index, hat=FALSE)
+		{
+			index = dateIndexToInteger(index, self)
+			if (hat)
+				private$.data[[index]]$hat_exo
+			else
+				private$.data[[index]]$exo
+		},
+		removeFirst = function()
+			private$.data <- private$.data[2:length(private$.data)]
 		,
-		getExoHat = function(index)
-			getExoHatData(self, private, index)
+		removeLast = function()
+			private$.data <- private$.data[1:(length(private$.data)-1)]
 	)
 )
-
-#' Number of series in the dataset
-#'
-#' @param private List of private members in current object
-getSizeData = function(private)
-	length(private$.data)
-
-#' 'Standard' horizon, from t+1 to midnight
-#'
-#' @inheritParams getSizeData
-getStdHorizonData = function(private)
-	24 - as.POSIXlt( private$.data[[1]]$time[1] )$hour + 1
-
-#' Acquire a new vector of lists (time, centered_serie, level, exo, exo_hat)
-#'
-#' @inheritParams getSizeData
-#' @param new_time Time
-#' @param new_centered_serie Centered serie
-#' @param new_level Level
-#' @param new_exo Exogneous variables
-#' @param new_exo_hat Predicted exogenous variables
-appendData = function(private, new_time, new_centered_serie, new_level, new_exo, new_exo_hat)
-{
-	private$.data[[length(private$.data)+1]] <- list("time"=new_time,
-		"centered_serie"=new_centered_serie,"level"=new_level,"exo"=new_exo,"exo_hat"=new_exo_hat)
-}
-
-#' Time values at specified index
-#'
-#' @inheritParams getSizeData
-#' @param index Return value at this index
-getTimeData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$time
-}
-
-#' Centered serie values at specified index
-#'
-#' @inheritParams getTimeData
-getCenteredSerieData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$centered_serie
-}
-
-#' Level of the serie at specified index
-#'
-#' @inheritParams getTimeData
-getLevelData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$level
-}
-
-#' Serie values (centered+level) at specified index
-#'
-#' @inheritParams getTimeData
-getSerieData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$centered_serie + data[[index]]$level
-}
-
-#' Exogenous measures at specified index
-#'
-#' @inheritParams getTimeData
-getExoData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$exo
-}
-
-#' Exogeous predictions at specified index
-#'
-#' @inheritParams getTimeData
-getExoHatData = function(o, private, index)
-{
-	index = dateIndexToInteger(index, o)
-	private$.data[[index]]$exo_hat
-}