From f1e8ee1b0a2fcdf1720da504f80792ec66a15f2d Mon Sep 17 00:00:00 2001 From: Myk Taylor Date: Sun, 29 Jan 2023 16:28:56 -0800 Subject: [PATCH] use new API in autochop --- plugins/autochop.cpp | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/plugins/autochop.cpp b/plugins/autochop.cpp index 34a8979ff..e44b4f804 100644 --- a/plugins/autochop.cpp +++ b/plugins/autochop.cpp @@ -369,13 +369,6 @@ static void bucket_tree(df::plant *plant, bool designate_clearcut, bool *designa *can_chop = true; } -static void get_citizens(vector &vec) { - for (auto &unit : world->units.active) { - if (Units::isCitizen(unit)) - vec.emplace_back(unit); - } -} - static void bucket_watched_burrows(color_ostream & out, map &clearcut_burrows, map &chop_burrows) { @@ -546,7 +539,7 @@ static int32_t do_cycle(color_ostream &out, bool force_designate) { int32_t expected_yield; TreesBySize designatable_trees_by_size; vector citizens; - get_citizens(citizens); + Units::getCitizens(citizens); int32_t newly_marked = scan_trees(out, &expected_yield, &designatable_trees_by_size, true, citizens); @@ -634,7 +627,7 @@ static void autochop_printStatus(color_ostream &out) { int32_t designated_trees, expected_yield, accessible_yield; map tree_counts, designated_tree_counts; vector citizens; - get_citizens(citizens); + Units::getCitizens(citizens); scan_logs(&usable_logs, citizens, &inaccessible_logs); scan_trees(out, &expected_yield, NULL, false, citizens, &accessible_trees, &inaccessible_trees, &designated_trees, &accessible_yield, &tree_counts, &designated_tree_counts); @@ -739,7 +732,7 @@ static int autochop_getLogCounts(lua_State *L) { DEBUG(status,*out).print("entering autochop_getNumLogs\n"); int32_t usable_logs, inaccessible_logs; vector citizens; - get_citizens(citizens); + Units::getCitizens(citizens); scan_logs(&usable_logs, citizens, &inaccessible_logs); Lua::Push(L, usable_logs); Lua::Push(L, inaccessible_logs); @@ -778,7 +771,7 @@ static int autochop_getTreeCountsAndBurrowConfigs(lua_State *L) { int32_t designated_trees, expected_yield, accessible_yield; map tree_counts, designated_tree_counts; vector citizens; - get_citizens(citizens); + Units::getCitizens(citizens); scan_trees(*out, &expected_yield, NULL, false, citizens, &accessible_trees, &inaccessible_trees, &designated_trees, &accessible_yield, &tree_counts, &designated_tree_counts);