R/gadget_areafile.R

In sCervino/mfdb: MareFrame DB querying library

gadget_areafile <- function (size, temperature, area = attr(size, 'area')) {
    # Replace areas in temp output with matching integers
    temperature$area <- as.factor(temperature$area)
    levels(temperature$area) <- vapply(
        levels(temperature$area),
        function (n) which(names(area) == n),
        0)

    # Make sure sizes coming in match the aggregation file
    mapped_sizes <- vapply(names(area), function(n) {
        out <- size[size$area == n, 'size']
        if (length(out) == 0) 0 else out
    }, 0)

    structure(list(
        labels = names(area),
        areas = seq_len(length(area)),
        size = mapped_sizes,
        temperature = temperature
        ), class = "gadget_areafile")
}

gadget_dir_write.gadget_areafile <- function(gd, obj) {

    # Write out areafile
    gadget_dir_write(gd, gadget_file("Modelfiles/area", components = list(structure(list(
        areas = obj$areas,
        size = obj$size,
        temperature = ""  # NB: Hack in a line with just "temperature"
    ), preamble = paste(obj$labels, collapse = "\t"))), data = obj$temperature))

    # Add to mainfile
    gadget_mainfile_update(gd, areafile = "Modelfiles/area")
}