Brandon Potter
231 строка
8.2 KiB
R
Исполняемый файл
231 строка
8.2 KiB
R
Исполняемый файл
#!/usr/bin/env Rscript
|
|
|
|
# load the required libraries:
|
|
library(tidyverse)
|
|
library(RColorBrewer)
|
|
library(optparse)
|
|
|
|
# declare some helper functions
|
|
ggpreview <- function (..., device = "png") {
|
|
fname <- tempfile(fileext = paste0(".", device))
|
|
ggplot2::ggsave(filename = fname, device = device, ...)
|
|
system2("open", fname)
|
|
invisible(NULL)
|
|
}
|
|
|
|
set_right_order <- function(df) {
|
|
# reverse the order of the rows so that oldest commit is first
|
|
df <- df %>% map_df(rev)
|
|
# ensure that ggplot plots the x-axis in the right order
|
|
df$Commit <- factor(df$Commit, levels = unique(df$Commit))
|
|
return(df)
|
|
}
|
|
|
|
plot_and_save <- function(df, xval, yval, title, subtitle, xlabel, filename) {
|
|
p <- ggplot(df, aes_string(x=xval, y=yval, group=1)) +
|
|
geom_line(size = 0.5, color=mycolors[1]) +
|
|
geom_point(size = 1.5, alpha = 1, color=mycolors[2]) +
|
|
theme_minimal() +
|
|
expand_limits(y=0) +
|
|
xlab(xlabel) +
|
|
ggtitle(title, subtitle = subtitle) +
|
|
theme(
|
|
axis.text.x = element_text(angle=90,hjust=1),
|
|
axis.title.y = element_blank()
|
|
) +
|
|
scale_fill_manual(values = mycolors)
|
|
#ggpreview(width=7.5, height=5, units="in", dpi=500)
|
|
ggsave(filename, p, device=pdf, dpi=500)
|
|
}
|
|
|
|
## Set up options ##
|
|
|
|
option_list = list(
|
|
make_option(c("-o", "--output"), type="character", default=NULL, action="store",
|
|
help="path (without trailing /) to a folder that will
|
|
contain the plots", metavar="folder-path"),
|
|
make_option(c("-a", "--changeset_a"), type="character", default=NULL, action="store",
|
|
help="beginning (inclusive) changeset of slice", metavar="changeset"),
|
|
make_option(c("-b", "--changeset_b"), type="character", default=NULL, action="store",
|
|
help="ending (inclusive) changeset of slice", metavar="changeset"),
|
|
make_option(c("-c", "--one_changeset"), type="character", default=NULL, action="store",
|
|
help="if set, will prepare plots for one changeset; if not, plots for a changeset slice")
|
|
|
|
)
|
|
|
|
## SCRIPT START ##
|
|
|
|
# parse the options
|
|
opt_parser <- OptionParser(option_list=option_list)
|
|
opts <- parse_args(opt_parser)
|
|
if (is.null(opts$output)) {
|
|
print_help(opt_parser)
|
|
stop("Please set the --output flag.", call.=FALSE)
|
|
}
|
|
slice_opt = 0
|
|
single_opt = 0
|
|
if (!is.null(opts$changeset_a) && !is.null(opts$changeset_b)) {
|
|
slice_opt = 1
|
|
}
|
|
if (!is.null(opts$one_changeset)) {
|
|
single_opt = 1
|
|
}
|
|
|
|
if ( (slice_opt && single_opt) || (!slice_opt && !single_opt) ) {
|
|
stop("Please supply a slice or a single changeset, not both.", call.= FALSE)
|
|
}
|
|
|
|
# choose color palette
|
|
mycolors <- brewer.pal(5, "Set2")
|
|
|
|
if (length(opts$one_changeset) > 0) {
|
|
## Plotting data for a single changeset ##
|
|
|
|
# read the files
|
|
non_amo <- read.csv("non_amo_one_changeset.csv", header=TRUE)
|
|
amo <- read.csv("amo_one_changeset.csv", header=TRUE)
|
|
ping_pong <- read.csv("ping_pong_one_changeset.csv", header=TRUE)
|
|
|
|
# ensure that ggplot plots the x-axis in the right order
|
|
non_amo$size <- factor(non_amo$size, levels = unique(non_amo$size))
|
|
amo$op <- factor(amo$op, levels = unique(amo$op))
|
|
|
|
# plot
|
|
non_amo_ops <- list("put","put_nbi","get","get_nbi")
|
|
for (op in non_amo_ops) {
|
|
plot_and_save(df=non_amo,
|
|
xval="size",
|
|
yval=op,
|
|
title=op,
|
|
subtitle="Latency (us)",
|
|
xlabel="Message size (bytes)",
|
|
filename=paste(opts$output,"/",op,"_changeset_",opts$one_changeset,".pdf", sep="")
|
|
)
|
|
}
|
|
|
|
# prepare data for plots with fixed message size and ops as x axis
|
|
non_amo$bsize <- paste("b",non_amo$size,sep="") # (so that the columns in non_amo_t start with a character)
|
|
non_amo_t <- setNames(data.frame(t(non_amo[,2:5])), non_amo[,6]) # transpose + set column names
|
|
non_amo_t$op <- colnames(non_amo[,2:5]) # make a column with operation names
|
|
|
|
sizes <- colnames(non_amo_t[,-(length(colnames(non_amo_t)))])
|
|
for (size in sizes) {
|
|
plot_and_save(df=non_amo_t,
|
|
xval="op",
|
|
yval=size,
|
|
title=paste(sub('.', '', size),"byte"),
|
|
subtitle="Latency (us)",
|
|
xlabel="Operation",
|
|
filename=paste(opts$output,"/",size,"_changeset_",opts$one_changeset,".pdf", sep="")
|
|
)
|
|
}
|
|
|
|
plot_and_save(df=amo,
|
|
xval="op",
|
|
yval="latency",
|
|
title="Atomics",
|
|
subtitle="Latency (us)",
|
|
xlabel="Operation",
|
|
filename=paste(opts$output,"/atomic_changeset_",opts$one_changeset,".pdf", sep="")
|
|
)
|
|
|
|
ping_pong$type <- c("ping_pong")
|
|
p<-ggplot(ping_pong, aes(x=type, y=latency, fill=type)) +
|
|
geom_bar(stat="identity", width=0.5) +
|
|
theme_minimal() +
|
|
ggtitle("Ping pong", subtitle = "Latency (us)") +
|
|
theme(
|
|
axis.title.y = element_blank(),
|
|
axis.text.y = element_blank(),
|
|
axis.title.x = element_blank(),
|
|
legend.position = "none"
|
|
) +
|
|
coord_flip() +
|
|
scale_fill_manual(values = mycolors)
|
|
#ggpreview(width=7.5, height=5, units="in", dpi=500)
|
|
ggsave(paste(opts$output,"/ping_pong_changeset_",opts$one_changeset,".pdf", sep=""), p, device=pdf, dpi=500)
|
|
|
|
} else {
|
|
## Plotting across a changeset slice ##
|
|
|
|
# read the files
|
|
put <- read.csv("put.csv", header=TRUE)
|
|
put_nbi <- read.csv("put_nbi.csv", header=TRUE)
|
|
get <- read.csv("get.csv", header=TRUE)
|
|
get_nbi <- read.csv("get_nbi.csv", header=TRUE)
|
|
amo <- read.csv("amo.csv", header=TRUE)
|
|
ping_pong <- read.csv("ping_pong.csv", header=TRUE)
|
|
|
|
# slice out the commits
|
|
start <- match(c(opts$changeset_a), put$Commit)
|
|
end <- match(c(opts$changeset_b), put$Commit)
|
|
# (start and end should be the same for all the frames) #
|
|
put <- put[start:end,]
|
|
put_nbi <- put_nbi[start:end,]
|
|
get <- get[start:end,]
|
|
get_nbi <- get_nbi[start:end,]
|
|
amo <- amo[start:end,]
|
|
ping_pong <- ping_pong[start:end,]
|
|
|
|
put <- set_right_order(put)
|
|
put_nbi <- set_right_order(put_nbi)
|
|
get <- set_right_order(get)
|
|
get_nbi <- set_right_order(get_nbi)
|
|
amo <- set_right_order(amo)
|
|
ping_pong <- set_right_order(ping_pong)
|
|
|
|
# plot
|
|
non_amo_ops <- list("put","put_nbi","get","get_nbi")
|
|
sizes_to_subtitle_map <- list("b1"="1 byte",
|
|
"b2"="2 bytes",
|
|
"b4"="4 bytes",
|
|
"b8"="8 bytes",
|
|
"b16"="16 bytes",
|
|
"b32"="32 bytes",
|
|
"b64"="64 bytes",
|
|
"b128"="128 bytes",
|
|
"b256"="256 bytes",
|
|
"b512"="512 bytes",
|
|
"b1024"="1024 bytes",
|
|
"b2048"="2048 bytes",
|
|
"b4096"="4096 bytes",
|
|
"b8192"="8192 bytes",
|
|
"b16384"="16384 bytes",
|
|
"b32768"="32768 bytes")
|
|
for (op in non_amo_ops) {
|
|
for (size in names(sizes_to_subtitle_map)) {
|
|
plot_and_save(df=eval(parse(text=op)),
|
|
xval="Commit",
|
|
yval=size,
|
|
title=op,
|
|
subtitle=paste("Latency (us) for ",sizes_to_subtitle_map[[size]],sep=""),
|
|
xlabel="Commit (older to newer)",
|
|
filename=paste(opts$output,"/",op,"_",size,".pdf", sep="")
|
|
)
|
|
}
|
|
}
|
|
|
|
amo_ops <- list("add","cswap","fadd","fcswap","fetch","finc","inc")
|
|
for (op in amo_ops) {
|
|
plot_and_save(df=amo,
|
|
xval="Commit",
|
|
yval=op,
|
|
title=op,
|
|
subtitle="Latency (us)",
|
|
xlabel="Commit (older to newer)",
|
|
filename=paste(opts$output,"/",op,".pdf", sep="")
|
|
)
|
|
}
|
|
|
|
plot_and_save(df=ping_pong,
|
|
xval="Commit",
|
|
yval="latency",
|
|
title="ping_pong",
|
|
subtitle="Latency (us)",
|
|
xlabel="Commit (older to newer)",
|
|
filename=paste(opts$output,"/","ping_pong.pdf", sep="")
|
|
)
|
|
}
|
|
|
|
## SCRIPT END ##
|