Métodos para funciones incorporadas

Parte de la sección Programación orientada a objetos del Journey de R de Coddy. Lección 28 de 57.

setMethod() también añade métodos S4 a funciones base como length() y summary(). La función base sigue funcionando para todo lo demás, y tu clase obtiene su propio comportamiento:

setClass("Playlist", slots = c(name = "character", songs = "character"))
setMethod("length", "Playlist", function(x) length(x@songs))
pl <- new("Playlist", name = "gym", songs = c("a", "b", "c"))
print(length(pl))
print(length(c(1, 2)))

Salida:

[1] 3
[1] 2

Los operadores también son genéricos. Un método para + nombra las clases de ambos lados en una signature(). Los dos argumentos se llaman e1 y e2:

setClass("Money", slots = c(amount = "numeric"))
setMethod("show", "Money", function(object) cat(sprintf("%.2f", object@amount), "\n"))
setMethod("+", signature("Money", "Money"), function(e1, e2) {
  new("Money", amount = e1@amount + e2@amount)
})
new("Money", amount = 2.5) + new("Money", amount = 4)

Salida:

6.50

El genérico de grupo Arith representa +, -, *, /, ^ y el resto a la vez. Dentro de su método, callGeneric() repite el operador que se haya utilizado, sobre los valores que pases:

setClass("Money", slots = c(amount = "numeric"))
setMethod("Arith", signature("Money", "numeric"), function(e1, e2) {
  new("Money", amount = callGeneric(e1@amount, e2))
})
m <- new("Money", amount = 10)
cat((m * 3)@amount, (m / 4)@amount, (m - 1)@amount)

Salida:

30 2.5 9

El genérico de grupo Compare abarca ==, <, > y los demás. Su método devuelve lo que callGeneric() devuelve para los slots, un valor lógico:

setClass("Money", slots = c(amount = "numeric"))
setMethod("Compare", signature("Money", "Money"), function(e1, e2) {
  callGeneric(e1@amount, e2@amount)
})
a <- new("Money", amount = 5)
b <- new("Money", amount = 8)
cat(a < b, a == b, a >= b)

Salida:

TRUE FALSE FALSE

Desafío

Fácil

La clase Money en Money.R tiene los slots amount (numérico) y currency (carácter). Añade cuatro métodos:

show() imprime 12.50 EUR y una nueva línea;
+ para dos objetos Money devuelve su suma, o se detiene con cannot add EUR and USD (usa call. = FALSE) cuando las monedas difieren;
* para Money y un número devuelve la cantidad escalada;
== para dos objetos Money es TRUE cuando tanto la cantidad como la moneda son iguales.

El código proporcionado lee líneas como 10:EUR + 2.5:EUR, 4:USD * 3 o 1:EUR == 1:EUR e imprime el resultado o error: y el mensaje.

Tu código va en Money.R. main.R contiene el código proporcionado de entrada/salida y no se puede editar.

Pruébalo tú mismo

source("Money.R")

# Código de entrada/salida suministrado: déjalo tal como está
input <- suppressWarnings(readLines(file("stdin")))
parse_money <- function(t) {
  p <- strsplit(t, ":")[[1]]
  new("Money", amount = as.numeric(p[1]), currency = p[2])
}
for (line in input) {
  p <- strsplit(line, " ")[[1]]
  a <- parse_money(p[1])
  tryCatch({
    if (p[2] == "+") print(a + parse_money(p[3]))
    if (p[2] == "*") print(a * as.numeric(p[3]))
    if (p[2] == "==") cat(a == parse_money(p[3]), "\n", sep = "")
  }, error = function(e) cat("error: ", conditionMessage(e), "\n", sep = ""))
}