Skip to content

Commit bdf79ac

Browse files
committed
[Seedpool] refactor construct
1 parent ff5f87b commit bdf79ac

File tree

1 file changed

+12
-15
lines changed

1 file changed

+12
-15
lines changed

src/seedpool.ml

Lines changed: 12 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -343,32 +343,31 @@ let compute_init_cov seeds =
343343
(fun accu seed -> EdgeCoverage.union accu seed.Seed.edge_cov)
344344
EdgeCoverage.empty seeds
345345

346+
let make_pool_and_cov seeds =
347+
let open Coverage in
348+
let pool = create () in
349+
seeds |> List.to_seq |> add_seq pool;
350+
let init_cov = compute_init_cov seeds in
351+
(pool, init_cov)
352+
346353
let evaluate_seeds_and_construct_seedpool raw_seeds node_tbl distmap =
347354
let open AUtil in
348355
let open Coverage in
349-
let pool = create () in
350-
if is_optimuzz_base () then (
356+
if is_optimuzz_base () then
351357
let _, seeds = classify_seeds node_tbl distmap raw_seeds in
352-
let init_cov = compute_init_cov seeds in
353-
seeds |> List.to_seq |> add_seq pool;
354-
(pool, init_cov))
358+
make_pool_and_cov seeds
355359
else
356360
let cover_seeds, noncover_seeds =
357361
classify_seeds node_tbl distmap raw_seeds
358362
in
359363
if cover_seeds = [] then (
360364
L.info "No covering seeds found. Using closest seeds.";
361365
let pool_closest = closest_seeds noncover_seeds in
362-
let init_cov = compute_init_cov pool_closest in
363-
pool_closest |> List.to_seq |> add_seq pool;
364-
(pool, init_cov))
366+
make_pool_and_cov pool_closest)
365367
else (
366-
(* if we have covering seeds, we use covering seeds only. *)
367368
L.info "Covering seeds found. Using them only.";
368369
let sorted_seeds = cover_seeds |> List.sort_uniq compare_score in
369-
let init_cov = compute_init_cov sorted_seeds in
370-
sorted_seeds |> List.to_seq |> add_seq pool;
371-
(pool, init_cov))
370+
make_pool_and_cov sorted_seeds)
372371

373372
let make llctx node_tbl (distmap : float Coverage.DistanceTable.t) =
374373
let open AUtil in
@@ -426,6 +425,4 @@ let make llctx node_tbl (distmap : float Coverage.DistanceTable.t) =
426425
|> List.filter_map preprocess_seed
427426
in
428427

429-
let pool = evaluate_seeds_and_construct_seedpool seeds node_tbl distmap in
430-
431-
pool
428+
evaluate_seeds_and_construct_seedpool seeds node_tbl distmap

0 commit comments

Comments
 (0)