bscpkgs/garlic/fig/creams/ss.R

library(ggplot2)
library(dplyr)
library(scales)
library(jsonlite)

args=commandArgs(trailingOnly=TRUE)

# Read the timetable from args[1]
input_file = "input.json"
if (length(args)>0) { input_file = args[1] }

# Load the dataset in NDJSON format
dataset = jsonlite::stream_in(file(input_file), verbose=FALSE) %>%
	jsonlite::flatten()

# We only need some colums
df = select(dataset, unit, config.nodes, config.gitBranch, time) %>%
	rename(nodes=config.nodes, gitBranch=config.gitBranch)

df$unit = as.factor(df$unit)
df$nnodes = df$nodes
df$nodes = as.factor(df$nodes)
df$gitBranch = as.factor(df$gitBranch)

# Remove the "garlic/" prefix from the gitBranch
levels(df$gitBranch) <- substring((levels(df$gitBranch)), 8)

# Compute new columns
D=group_by(df, unit) %>%
	mutate(tnorm = time / median(time) - 1) %>%
	mutate(bad = ifelse(max(abs(tnorm)) >= 0.01, 1, 0)) %>%
  mutate(variability = ifelse(bad > 0, "large", "ok")) %>%
	mutate(mtime = median(time)) %>%
	mutate(nmtime = mtime*nnodes) %>%
	mutate(ntime = time*nnodes) %>%
  ungroup() %>%
  mutate(min_nmtime = min(nmtime)) %>%
  mutate(rnmtime = nmtime / min_nmtime) %>%
  mutate(rntime = ntime / min_nmtime) %>%
  mutate(rmeff = 1.0 / rnmtime) %>%
  mutate(reff = 1.0 / rntime) %>%
	group_by(gitBranch) %>%
	mutate(tmax = max(mtime)) %>%
	mutate(speedup=tmax/time) %>%
	mutate(eff=speedup/nnodes) %>%
	mutate(mspeedup=tmax/mtime) %>%
  mutate(meff=mspeedup/nnodes) %>%
  ungroup()

D$bad = as.factor(D$bad > 0)
D$variability = as.factor(D$variability)

ppi=300
h=5
w=5

png("variability.png", width=1.5*w*ppi, height=h*ppi, res=ppi)
p = ggplot(data=D, aes(x=nodes, y=tnorm, color=variability)) +
	theme_bw() +
	theme(plot.subtitle=element_text(size=8)) +
	# Add the maximum allowed error lines
	geom_hline(yintercept=c(-0.01, 0.01),
		linetype="dashed", color="gray") +
	# Draw boxplots
	geom_boxplot(aes(fill=gitBranch)) +
	scale_color_manual(values=c("brown", "black")) +
	# Labels
	labs(x="Nodes", y="Normalized time", title="Creams strong scaling",
    subtitle=input_file)
print(p)
dev.off()

png("time.png", width=w*1.5*ppi, height=h*ppi, res=ppi)
p = ggplot(D, aes(x=nodes, y=mtime, color=gitBranch)) +
	theme_bw() +
	theme(plot.subtitle=element_text(size=8)) +
	geom_line(aes(group=gitBranch)) +
	#geom_point() +
	geom_point(aes(shape=variability), size=3) +
	scale_shape_manual(values=c(21, 19)) +
  #  position=position_dodge(width=0.3)) +
	#scale_x_continuous(trans=log2_trans()) +
	scale_y_continuous(trans=log2_trans()) +
	labs(x="Nodes", y="Time (s)",
    title="Creams strong scaling (lower is better)",
    subtitle=input_file)
print(p)
dev.off()

png("refficiency.png", width=w*1.5*ppi, height=h*ppi, res=ppi)
p = ggplot(D, aes(x=nodes, y=rmeff, color=gitBranch)) +
	theme_bw() +
	theme(plot.subtitle=element_text(size=8)) +
	geom_line(aes(group=gitBranch)) +
	geom_point(aes(shape=variability), size=3) +
	#geom_boxplot(aes(y=reff),
  #  position=position_dodge(width=0.0)) +
	scale_shape_manual(values=c(21, 19)) +
	#geom_point(aes(y=rntime),
  #  position=position_dodge(width=0.3)) +
	#scale_x_continuous(trans=log2_trans()) +
	#scale_y_continuous(trans=log2_trans()) +
	labs(x="Nodes", y="Relative efficiency (to best)",
    title="Creams strong scaling (higher is better)",
    subtitle=input_file)
print(p)
dev.off()