Добавление вертикальных линий на графики chart_Series и add_TA - PullRequest
Новая
       12

Добавление вертикальных линий на графики chart_Series и add_TA

6 голосов
/ 10 февраля 2012

После доброго ответа Джошуа в моем предыдущем посте об этой «проблеме» ( Проблема с QuantMod add_TA и chart_Series - строки и текст исчезают после того, как следующая add_TA называется ) Я добавил несколько «расширений» в Quantmod которые позволяют рисовать линии / сегменты и тексты. При этом я столкнулся с проблемой, которую не могу решить: 

require(quantmod)

add_VerticalLine<-function(xCoordinatesOfLines, on=1, ...) {
    lenv <- new.env()
    lenv$add_verticalline <- function(x, xCoordinatesOfLines, ...) {
        xdata <- x$Env$xdata
        xsubset <- x$Env$xsubset
        xcoords <- seq(1:NROW(xdata[xsubset]))[index(xdata[xsubset]) %in% index(xCoordinatesOfLines[xsubset])]
        abline(v=xcoords, ...)
    }
    mapply(function(name, value) {assign(name,value,envir=lenv)}, names(list(xCoordinatesOfLines=xCoordinatesOfLines, ...)), list(xCoordinatesOfLines=xCoordinatesOfLines, ...))
    exp <- parse(text=gsub("list","add_verticalline", as.expression(substitute(list(x=current.chob(),
                                            xCoordinatesOfLines=xCoordinatesOfLines, ...)))), srcfile=NULL)
    plot_object <- current.chob()
    lenv$xdata <- plot_object$Env$xdata
    plot_object$set_frame(sign(on)*abs(on)+1L)
    plot_object$add(exp,env=c(lenv, plot_object$Env),expr=TRUE)
    plot_object
}

datesForLines <- c("2012-02-06", "2012-02-07")

verticalLines.xts <- xts(rep(0, length(datesForLines)), order.by=as.Date(datesForLines))

SPX <- getSymbols("^GSPC", from="2012-01-01", auto.assign=FALSE)
chart_Series(SPX, subset="2012")
add_VerticalLine(verticalLines.xts, on=1, col=c('red', 'blue'), lwd=2)
add_TA(SMA(SPX))
# Everything is fine up to this point, but, when you execute next line (adding vertical line also to second segment of the graph):
add_VerticalLine(verticalLines.xts, on=2, col=c('blue', 'red'), lwd=2)
# You can see that vertical lines are drawn below the SMA and not visible. There seems to be some layering mechanism I do not understand...

Любые идеи о том, как сделать это таким образом, чтобы вертикальные линии были видны только на втором сегменте графика (в части add_TA, второй сегмент), получили высокую оценку.

Я также создал другие функции для добавления текстовых и линейных сегментов таким же образом, чтобы избежать проблем с повторным размещением (проблема описана в связанном вопросе), если используются только графические примитивы. Если кому-то интересно, могу поделиться.

Лучший, Samo

...