Dateien
rocm-systems/scripts/omnitrace-merge.jl
T
2021-11-24 04:59:59 -06:00

44 Zeilen
1.2 KiB
Julia
Ausführbare Datei

#!/usr/bin/env julia
using Base: String
using JSON, DataFrames, Dates, CSV
function check_input_args()
message = "Needs only one json file and at least one trace data file(s)."
@assert length(ARGS) >= 2 && occursin(".json", join(ARGS)) message
json_file, trace_files = nothing, String[]
for arg in ARGS
@assert Base.Filesystem.ispath(arg) "'$(arg)' not found!"
if endswith(arg, ".json")
json_file = arg
else
push!(trace_files, arg)
end
end
json_file, trace_files
end
json_file, trace_files = check_input_args()
out = JSON.parsefile(json_file)
for trace in trace_files
@info "processing $(trace) ..."
f = tempname() * ".json"
run(`trace_to_text json $trace $f`)
todo = JSON.parsefile(f)
out["traceEvents"] = vcat(
out["traceEvents"],
get(todo, "traceEvents", [])
)
out["systemTraceEvents"] = (*)(
get(out, "systemTraceEvents", ""),
get(todo, "systemTraceEvents", "")
)
end
output_file = replace("$(string(now())).json", ":" => "-")
open(output_file, "w") do f
write(f, JSON.json(out))
@info "output file: $(output_file)"
end