add cjetreconstruction backend to benchmark.jl

Author: m-fila

src/benchmark.jl

@@ -16,7 +16,7 @@ using LorentzVectorHEP
 using JetReconstruction
 
 # Backends for the jet reconstruction
-@enumx T=Backend Backends Julia FastJet
+@enumx T=Backend Backends Julia FastJet CJetReconstruction
 const AllBackends = [String(Symbol(x)) for x in instances(Backends.Backend)]
 
 # Parsing for Enum types
@@ -118,13 +118,14 @@ function julia_jet_process_avg_time(events::Vector{Vector{PseudoJet}};
     lowest_time
 end
 
-function fastjet_jet_process_avg_time(input_file::AbstractString;
-                                      ptmin::Float64 = 5.0,
-                                      distance::Float64 = 0.4,
-                                      p::Union{Real, Nothing} = nothing,
-                                      algorithm::JetAlgorithm.Algorithm = JetAlgorithm.AntiKt,
-                                      strategy::RecoStrategy.Strategy,
-                                      nsamples::Integer = 1)
+function external_benchmark_avg_time(input_file::AbstractString;
+                                     ptmin::Float64 = 5.0,
+                                     distance::Float64 = 0.4,
+                                     p::Union{Real, Nothing} = nothing,
+                                     algorithm::JetAlgorithm.Algorithm = JetAlgorithm.AntiKt,
+                                     strategy::RecoStrategy.Strategy,
+                                     nsamples::Integer = 1,
+                                     backend::Backends.Backend = Backends.FastJet)
 
     # FastJet reader cannot handle gzipped files
     if endswith(input_file, ".gz")
@@ -136,17 +137,24 @@ function fastjet_jet_process_avg_time(input_file::AbstractString;
                                                                        algorithm = algorithm)
 
     # @warn "FastJet timing not implemented yet"
-    fj_bin = joinpath(@__DIR__, "..", "fastjet", "build", "fastjet-finder")
-    fj_args = String[]
-    push!(fj_args, "-p", string(p))
-    push!(fj_args, "-s", string(strategy))
-    push!(fj_args, "-R", string(distance))
-    push!(fj_args, "--ptmin", string(ptmin))
-
-    push!(fj_args, "-n", string(nsamples))
-    @info "Fastjet command: $fj_bin $fj_args $input_file"
-    fj_output = read(`$fj_bin $fj_args $input_file`, String)
-    tryparse(Float64, match(r"Lowest time per event ([\d\.]+) us", fj_output)[1])
+    if backend == Backends.FastJet
+        bench_dir = "fastjet"
+        bench_name = "fastjet-finder"
+    elseif  backend == Backends.CJetReconstruction
+        bench_dir = "cjetreconstruction"
+        bench_name = "cjetreconstruction-finder"
+    end
+    bench_bin = joinpath(@__DIR__, "..", bench_dir, "build", bench_name)
+    bench_args = String[]
+    push!(bench_args, "-p", string(p))
+    push!(bench_args, "-s", string(strategy))
+    push!(bench_args, "-R", string(distance))
+    push!(bench_args, "--ptmin", string(ptmin))
+
+    push!(bench_args, "-n", string(nsamples))
+    @info "Benchmark command: $bench_bin $bench_args $input_file"
+    bench_output = read(`$bench_bin $bench_args $input_file`, String)
+    tryparse(Float64, match(r"Lowest time per event ([\d\.]+) us", bench_output)[1])
 end
 
 function parse_command_line(args)
@@ -269,13 +277,14 @@ function main()
                                                         p = args[:power],
                                                         strategy = args[:strategy],
                                                         nsamples = samples, repeats = args[:repeats])
-        elseif args[:backend] == Backends.FastJet
-            time_per_event = fastjet_jet_process_avg_time(event_file, ptmin = args[:ptmin],
-                                                          distance = args[:distance],
-                                                          algorithm = args[:algorithm],
-                                                          p = args[:power],
-                                                          strategy = args[:strategy],
-                                                          nsamples = samples)
+        elseif args[:backend] ∈ (Backends.FastJet, Backends.CJetReconstruction)
+            time_per_event = external_benchmark_avg_time(event_file, ptmin = args[:ptmin],
+                                                         distance = args[:distance],
+                                                         algorithm = args[:algorithm],
+                                                         p = args[:power],
+                                                         strategy = args[:strategy],
+                                                         nsamples = samples,
+                                                         backend = args[:backend])
         end
 
         push!(event_timing, time_per_event)