R/correlation.b.R

In jamovi/jmvbaseR: Base R analyses made available in jamovi

correlationClass <- if (requireNamespace('jmvcore')) R6::R6Class(
    "correlationClass",
    inherit = correlationBase,
    private = list(
        .init = function() {
            preformatted <- jmvcore::Preformatted$new(self$options, 'pre')
            self$results$add(preformatted)
        },
        .run = function() {

            preformatted <- self$results$get('pre')
            preformatted$content <- ''

            if (is.null(self$options$x) ||
                is.null(self$options$y))
                return()

            x <- jmvcore::toNumeric(self$data[[self$options$x]])
            y <- jmvcore::toNumeric(self$data[[self$options$y]])

            cor <- cor.test(x=x, y=y,
                alternative=self$options$alternative,
                method=self$options$method,
                conf.level=self$options$conf.level)

            preformatted$content <- paste0(capture.output(cor), collapse='\n')
        },
        .sourcifyOption = function(option) {
            if (option$name == 'x')
                return('')
            if (option$name == 'y')
                return('')
            if (option$name == 'data')
                return('')
            if (option$name == 'conf.level') {
                if (isTRUE(all.equal(option$value, 0.95)))
                    return('')
                else
                    return(paste0('conf.level=', option$value))
            }

            super$.sourcifyOption(option)
        }),
    public=list(
        asSource=function() {
            args <- private$.asArgs(incData=FALSE)
            if (args != '')
                args <- paste0(',', args)

            paste0('cor.test(\n    x = data$x,\n    y = data$y', args, '\n)')
        })
)