Методы для встроенных объектов

Часть раздела Объектно-ориентированное программирование путешествия по R на Coddy. Урок 28 из 57.

setMethod() также добавляет методы S4 к базовым функциям, таким как length() и summary(). Базовая функция продолжает работать для всего остального, а ваш класс получает собственное поведение:

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)))

Результат:

[1] 3
[1] 2

Операторы тоже являются обобщёнными функциями. Метод для + указывает классы обеих сторон в signature(). Два аргумента называются e1 и 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)

Результат:

6.50

Групповой generic Arith обозначает сразу +, -, *, /, ^ и остальные операторы. Внутри его метода callGeneric() повторяет использованный оператор над переданными вами значениями:

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)

Вывод:

30 2.5 9

Групповой generic Compare охватывает ==, <, > и другие операторы. Его метод возвращает результат callGeneric() для слотов — логическое значение:

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)

Результат:

TRUE FALSE FALSE

Задание

Легко

Класс Money в Money.R имеет слоты amount (числовой) и currency (строковый). Добавьте четыре метода:

show() выводит 12.50 EUR и перевод строки;
+ для двух объектов Money возвращает их сумму или завершается с ошибкой cannot add EUR and USD (используйте call. = FALSE), если валюты различаются;
* для Money и числа возвращает умноженную на это число сумму;
== для двух объектов Money имеет значение TRUE, если и сумма, и валюта совпадают.

Предоставленный код считывает строки 10:EUR + 2.5:EUR, 4:USD * 3 или 1:EUR == 1:EUR и выводит результат либо error: и сообщение.

Ваш код должен находиться в Money.R. В main.R содержится предоставленный код ввода/вывода, и его нельзя редактировать.

Попробуйте сами

source("Money.R")

# Предоставленный код ввода/вывода: оставьте его как есть
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 = ""))
}