fixes and improvements

[talweg.git] / R / Data.R

#' @title Data
#'
#' @description Data encapsulation
#'
#' @field data List of
#' \itemize{
#'   \item time: vector of times
#'   \item serie: centered series
#'   \item level: corresponding levels
#'   \item exo_hat: predicted exogenous variables
#'   \item exo_Dm1: List of measured exogenous variables at day minus 1
#' }
#'
#' @exportClass Data
#' @export Data
Data = setRefClass(
	Class = "Data",

	fields = list(
		data = "list"
	),

	methods = list(
		initialize = function(...)
		{
			"Initialize empty Data object"

			callSuper(...)
		},
		getSize = function()
		{
			length(data)
		},
		append = function(new_time, new_serie, new_level, new_exo_hat, new_exo_Dm1)
		{
			"Acquire a new vector of lists (time, serie, level, exo_hat, exo_Dm1)"

			data[[length(data)+1]] <<- list("time"=new_time,"serie"=new_serie,"level"=new_level,
				"exo_hat"=new_exo_hat,"exo_Dm1"=new_exo_Dm1)
		},
		getTime = function(index)
		{
			"Get time values at specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$time
		},
		getCenteredSerie = function(index)
		{
			"Get serie values (centered) at specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$serie
		},
		getLevel = function(index)
		{
			"Get level for the serie at specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$level
		},
		getSerie = function(index)
		{
			"Get serie values (centered+level) at specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$serie + data[[index]]$level
		},
		getExoHat = function(index)
		{
			"Get exogeous predictions at specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$exo_hat
		},
		getExoDm1 = function(index)
		{
			"Get exogenous measures the day before specified index"

			index = dateIndexToInteger(index, .self)
			data[[index]]$exo_Dm1
		}
	)
)