1 | #!/usr/bin/env Rscript
|
2 | #
|
3 | # wedge.R -- Show how long it takes
|
4 | #
|
5 | # Usage:
|
6 | # wedge.R ACTION IN_DIR OUT_DIR
|
7 |
|
8 | library(dplyr)
|
9 | library(ggplot2)
|
10 |
|
11 | source('benchmarks/common.R')
|
12 |
|
13 | options(stringsAsFactors = F,
|
14 | # Make the report wide. tibble.width doesn't appear to do this?
|
15 | width=200,
|
16 | #tibble.print_max=Inf,
|
17 |
|
18 | # for printing of timestamps
|
19 | digits=11
|
20 | )
|
21 |
|
22 | WritePlot <- function(p, filename, width = 800, height = 600) {
|
23 | png(filename, width=width, height=height)
|
24 | plot(p)
|
25 | dev.off()
|
26 | Log('Wrote %s', filename)
|
27 | }
|
28 |
|
29 | PlotElapsed <- function(ctx) {
|
30 | g <- ggplot(ctx$tasks, aes(x = wedge, y = elapsed_secs))
|
31 |
|
32 | # NOTE: stat = "identity" required for x and y, geom_bar makes a histogram by
|
33 | # default
|
34 | b <- geom_bar(stat = "identity")
|
35 | t <- ggtitle('Elapsed Time')
|
36 |
|
37 | g + b + t #+ scale_fill_manual(values=palette)
|
38 | }
|
39 |
|
40 | PlotXargs <- function(ctx) {
|
41 | tasks = ctx$tasks
|
42 |
|
43 | left = min(tasks$start_time)
|
44 | Log("Starting at %f", left)
|
45 |
|
46 | tasks$start_time = tasks$start_time - left
|
47 | tasks$end_time = tasks$end_time - left
|
48 | #print(tasks)
|
49 |
|
50 | g = ggplot(tasks, aes(x = start_time, xend = end_time, y = xargs_slot, yend = xargs_slot))
|
51 | line = geom_segment(linewidth = 10, aes(color = wedge))
|
52 | labs = labs(title = "Parallel Wedge Builds", x = "time (seconds)", y = "xargs slot")
|
53 |
|
54 | g + line + labs
|
55 |
|
56 | # Doesn't work
|
57 | #+ scale_x_time()
|
58 | }
|
59 |
|
60 | # Something like this
|
61 | # https://stackoverflow.com/questions/70767351/plotting-date-intervals-in-ggplot2
|
62 |
|
63 | Report = function(ctx, out_dir) {
|
64 | p = PlotElapsed(ctx)
|
65 |
|
66 | p = PlotElapsed(ctx)
|
67 | WritePlot(p, file.path(out_dir, 'elapsed.png'))
|
68 |
|
69 | p = PlotXargs(ctx)
|
70 | WritePlot(p, file.path(out_dir, 'xargs.png'))
|
71 | }
|
72 |
|
73 | Load = function(in_dir) {
|
74 | list(tasks = read.table(file.path(in_dir, 'tasks.tsv'), header=T))
|
75 | }
|
76 |
|
77 | main = function(argv) {
|
78 | action = argv[[1]]
|
79 |
|
80 | if (action == 'xargs-report') {
|
81 | in_dir = argv[[2]]
|
82 | out_dir = in_dir # same for now
|
83 |
|
84 | ctx = Load(in_dir)
|
85 | Report(ctx, out_dir)
|
86 |
|
87 | } else {
|
88 | Log("Invalid action '%s'", action)
|
89 | quit(status = 1)
|
90 | }
|
91 | }
|
92 |
|
93 | if (length(sys.frames()) == 0) {
|
94 | # increase ggplot font size globally
|
95 | theme_set(theme_grey(base_size = 20))
|
96 | main(commandArgs(TRUE))
|
97 | }
|