R/compare-variables.R

In Crunch-io/rcrunch: Crunch.io Data Tools

compareVariables <- function(A, B) {
    ## Match variable catalogs on alias
    aa <- aliases(A)
    ab <- aliases(B)
    allaliases <- c(aa, setdiff(ab, aa))
    match.a <- match(allaliases, aa)
    match.b <- match(allaliases, ab)

    out <- structure(data.frame(
        name.A = names(A)[match.a],
        type.A = types(A)[match.a],
        alias = allaliases,
        type.B = types(B)[match.b],
        name.B = names(B)[match.b],
        stringsAsFactors = FALSE
    ),
    class = c("compareVariables", "data.frame")
    )
    return(out)
}

summarizeCompareVariables <- function(compdf) {
    ## Check for mismatch of types where aliases match
    mismatched.type <- as.character(na.omit(compdf$alias[compdf$type.A != compdf$type.B]))

    return(structure(list(
        variables = compdf,
        problems = list(
            mismatched.type = mismatched.type,
            mismatched.name = findMismatches(compdf, "alias", "name")
        )
    ),
    class = "compareVariablesSummary"
    ))
}

#' @export
summary.compareVariables <- function(object, ...) summarizeCompareVariables(object)

#' @export
print.compareVariablesSummary <- function(x, ...) {
    cat("Total variables:", nrow(x$variables), "\n")
    mismatched.type <- x$problems$mismatched.type
    if (length(mismatched.type)) {
        cat("\n")
        cat("Type mismatch:", length(mismatched.type), "\n")
        print(x$variables[x$variables$alias %in% mismatched.type, , drop = FALSE],
            row.names = FALSE
        )
    }
    mismatched.name <- x$problems$mismatched.name
    if (length(mismatched.name)) {
        cat("\n")
        cat("Name mismatch:", length(mismatched.name), "\n")
        bad <- x$variables$alias %in% mismatched.name
        print(x$variables[bad, c("name.A", "alias", "name.B"), drop = FALSE],
            row.names = FALSE
        )
    }
    if (length(mismatched.type) + length(mismatched.name) == 0) {
        cat("No type or name mismatches.\n")
    }
    invisible(x)
}