Наследование и callSuper

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

contains = "Account" заставляет Reference Class наследовать каждое поле и метод класса Account. Подкласс добавляет собственные поля и методы:

Account <- suppressMessages(setRefClass("Account",
  fields = list(owner = "character", balance = "numeric"),
  methods = list(deposit = function(x) balance <<- balance + x)
))
Savings <- suppressMessages(setRefClass("Savings", contains = "Account",
  fields = list(rate = "numeric"),
  methods = list(add_interest = function() deposit(balance * rate))
))
s <- Savings$new(owner = "Ada", balance = 100, rate = 0.05)
s$deposit(100)
s$add_interest()
cat(s$balance, is(s, "Account"))

Вывод:

210 TRUE

Подкласс переопределяет метод, определяя метод с тем же именем. Внутри него callSuper(...) вызывает версию родительского класса, поэтому дочерний класс добавляет только то, что является новым:

Account <- suppressMessages(setRefClass("Account",
  fields = list(balance = "numeric"),
  methods = list(withdraw = function(x) {
    if (x > balance) stop("insufficient funds", call. = FALSE)
    balance <<- balance - x
  })
))
Savings <- suppressMessages(setRefClass("Savings", contains = "Account",
  methods = list(withdraw = function(x) {
    if (x > 500) stop("limit is 500", call. = FALSE)
    callSuper(x)
  })
))
s <- Savings$new(balance = 300)
s$withdraw(200)
for (x in c(600, 400)) cat(tryCatch({ s$withdraw(x); "ok" }, error = function(e) conditionMessage(e)), "\n")
cat(s$balance, "\n")

Вывод:

limit is 500 
insufficient funds 
100

initialize наследуется и переопределяется так же, как любой метод. initialize дочернего класса вызывает callSuper(...), чтобы родительский класс задал свои поля, а затем устанавливает собственные значения по умолчанию:

Account <- suppressMessages(setRefClass("Account",
  fields = list(owner = "character", balance = "numeric"),
  methods = list(initialize = function(...) {
    initFields(...)
    if (length(balance) == 0) balance <<- 0
    invisible(.self)
  })
))
Savings <- suppressMessages(setRefClass("Savings", contains = "Account",
  fields = list(rate = "numeric"),
  methods = list(initialize = function(...) {
    callSuper(...)
    if (length(rate) == 0) rate <<- 0.01
    invisible(.self)
  })
))
s <- Savings$new(owner = "Bo")
cat(s$owner, s$balance, s$rate)

Вывод:

Bo 0 0.01

Объекты разных подклассов отвечают на один и тот же вызов метода по-своему, поэтому циклу по смешанному списку не нужны проверки класса:

Account <- suppressMessages(setRefClass("Account",
  fields = list(balance = "numeric"),
  methods = list(month_end = function() balance <<- balance - 2)
))
Savings <- suppressMessages(setRefClass("Savings", contains = "Account",
  methods = list(month_end = function() balance <<- balance * 1.01)
))
bank <- list(Account$new(balance = 100), Savings$new(balance = 100))
for (a in bank) a$month_end()
cat(vapply(bank, function(a) a$balance, numeric(1)))

Результат:

98 101

Задание

Легко

Счета и сберегательные счета являются Reference Classes в двух файлах. Реализуйте методы:

Account в Account.R (поля owner, balance): initialize устанавливает баланс в 0, если он не задан; deposit(x); withdraw(x) завершает выполнение с сообщением insufficient funds; month_end() взимает комиссию в размере 1; describe() возвращает ada: 99.
Savings в Savings.R (содержит Account, поле rate): withdraw(x) завершается с сообщением limit is 500 при значении свыше 500, а в противном случае вызывает callSuper(x); month_end() добавляет проценты, balance * rate, вместо комиссии; describe() добавляет (savings) с помощью callSuper().

Используйте call. = FALSE. Оба определения обёрнуты в suppressMessages(). Предоставленный код считывает savings ada 0.1 или account bo, затем dep 100, wd 30 или month и после каждой операции выводит describe() либо refused: и сообщение.

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

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

source("Account.R")
source("Savings.R")

# Предоставленный код ввода/вывода: оставьте его как есть
input <- suppressWarnings(readLines(file("stdin")))
p <- strsplit(input[1], " ")[[1]]
acc <- if (p[1] == "savings") Savings$new(owner = p[2], rate = as.numeric(p[3])) else Account$new(owner = p[2])
for (line in input[-1]) {
  q <- strsplit(line, " ")[[1]]
  msg <- tryCatch({
    if (q[1] == "dep") acc$deposit(as.numeric(q[2]))
    if (q[1] == "wd") acc$withdraw(as.numeric(q[2]))
    if (q[1] == "month") acc$month_end()
    acc$describe()
  }, error = function(e) paste("refused:", conditionMessage(e)))
  cat(line, ": ", msg, "\n", sep = "")
}