33 #include "formula/callable_objects.hpp"
41 #define ERR_CF LOG_STREAM(err, log_config)
42 #define WRN_CF LOG_STREAM(warn, log_config)
43 #define DBG_CF LOG_STREAM(debug, log_config)
46 #define ERR_WML LOG_STREAM(err, log_wml)
53 , use_flat_tod_(false)
69 std::vector<const unit *> ret;
74 if(max_matches == 0) {
87 if (
matches(*u, u->get_location())) {
88 return u.get_shared_ptr();
98 unit_filter_xy(
const std::string& x,
const std::string& y) : x_(x), y_(y) {}
105 if(x.empty() && y.empty()) {
108 else if(x ==
"recall" && y ==
"recall") {
116 const std::string x_;
117 const std::string y_;
122 unit_filter_adjacent(
const vconfig& cfg)
135 std::vector<map_location::direction> dirs;
136 if (i_adjacent.
empty()) {
143 if (unit_itor == units.
end() || !child_.matches(
unit_filter_args{*unit_itor, unit_itor->get_location(), &args.u, args.fc, args.use_flat_tod} )) {
146 auto is_enemy = cfg_[
"is_enemy"];
166 unit_filter_child_literal(
const vconfig& v,
const F&
f) : v_(v) , f_(
f) {}
175 template<
typename T,
typename F>
178 unit_filter_attribute_parsed(T&& v, F&&
f) : v_(std::move(v)), f_(std::move(
f)) {}
187 template<
typename C,
typename F>
190 unit_filter_attribute_literal(std::string&& v, C&&
c, F&&
f) : v_(std::move(v)), c_(std::move(
c)), f_(std::move(
f)) {}
195 return f_(c_(v), args);
202 class contains_dollar_visitor
203 #ifdef USING_BOOST_VARIANT
204 :
public boost::static_visitor<bool>
208 contains_dollar_visitor() {}
212 bool operator()(
const T&)
const {
return false; }
214 bool operator()(
const t_string&)
const {
return true; }
216 bool operator()(
const std::string&
s)
const
218 return s.find(
'$') != std::string::npos;
225 unit_filter_compound::unit_filter_compound(
vconfig cfg)
252 switch (filter.first) {
253 case conditional_type::type::filter_and:
254 res = res && filter.second.matches(args);
256 case conditional_type::type::filter_or:
257 res = res || filter.second.matches(args);
259 case conditional_type::type::filter_not:
260 res = res && !filter.second.matches(args);
270 if (!filter->matches(args)) {
280 children_.emplace_back(
new unit_filter_child_literal<F>(
c, func));
283 template<
typename C,
typename F>
289 children_.emplace_back(
new unit_filter_attribute_literal<C, F>(std::move(v.
str()), std::move(conv), std::move(func)));
292 children_.emplace_back(
new unit_filter_attribute_parsed<decltype(conv(v)), F>(std::move(conv(v)), std::move(func)));
300 std::string tag_name;
304 void get_ability_children_id(std::vector<ability_match>& id_result,
305 const config& parent,
const std::string&
id) {
308 if(cfg[
"id"] ==
id) {
309 ability_match special = { key, &cfg };
310 id_result.push_back(special);
321 if(literal.
empty()) {
return; }
332 return std::find(id_list.begin(), id_list.end(), args.
u.
id()) != id_list.end();
340 return std::find(types.begin(), types.end(), args.
u.
type_id()) != types.end();
348 std::set<std::string> types_expanded;
349 for(
const std::string&
type : types) {
350 if(types_expanded.count(
type)) {
354 const auto& tree = ut->advancement_tree();
355 types_expanded.insert(tree.begin(), tree.end());
356 types_expanded.insert(
type);
359 return types_expanded.find(args.
u.
type_id()) != types_expanded.end();
367 return std::find(types.begin(), types.end(), args.
u.
variation()) != types.end();
379 for(
const std::string& variation_id : types) {
380 if (
type->has_variation(variation_id)) {
392 for(
const std::string& ability_id : abilities) {
405 for(
const std::string& ability : abilities) {
419 for(
const std::string& ability : abilities) {
420 std::vector<ability_match> ability_id_matches_self;
421 get_ability_children_id(ability_id_matches_self, args.
u.
abilities(), ability);
422 for(
const ability_match& entry : ability_id_matches_self) {
429 for(
unsigned i = 0;
i < adjacent.size(); ++
i) {
431 if (it == units.
end() || it->incapacitated())
433 if (&*it == (args.
u.shared_from_this()).get())
436 std::vector<ability_match> ability_id_matches_adj;
437 get_ability_children_id(ability_id_matches_adj, it->abilities(), ability);
438 for(
const ability_match& entry : ability_id_matches_adj) {
453 for(
const std::string& ability : abilities) {
466 std::sort(res.begin(), res.end());
470 [](
const std::vector<std::string>& check_traits,
const unit_filter_args& args)
474 std::vector<std::string> isect;
475 std::sort(have_traits.begin(), have_traits.end());
476 std::set_intersection(check_traits.begin(), check_traits.end(), have_traits.begin(), have_traits.end(), std::back_inserter(isect));
477 return !isect.empty();
485 return std::find(races.begin(), races.end(), args.
u.
race()->
id()) != races.end();
493 return gender == args.
u.
gender();
502 }
catch(std::invalid_argument&) {
515 std::vector<int> res;
518 res.push_back(std::stoi(s));
519 }
catch(std::invalid_argument&) {
520 WRN_CF <<
"ignored invalid side='" << s <<
"' in filter";
527 return std::find(sides.begin(), sides.end(), args.
u.
side()) != sides.end();
531 create_attribute(literal[
"status"],
535 for(
const std::string& status : statuses) {
544 create_attribute(literal[
"has_weapon"],
550 if(a.id() == weapon) {
558 create_attribute(literal[
"role"],
566 create_attribute(literal[
"alignment"],
574 create_attribute(literal[
"ai_special"],
582 create_attribute(literal[
"usage"],
586 for(
const std::string& usage : usages) {
587 if(args.
u.
usage() == usage) {
595 create_attribute(literal[
"canrecruit"],
603 create_attribute(literal[
"recall_cost"],
605 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
611 create_attribute(literal[
"level"],
613 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
619 create_attribute(literal[
"defense"],
621 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
624 return in_ranges(actual_defense, ranges);
628 create_attribute(literal[
"movement_cost"],
630 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
637 create_attribute(literal[
"vision_cost"],
639 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
646 create_attribute(literal[
"jamming_cost"],
648 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
655 create_attribute(literal[
"lua_function"],
666 create_attribute(literal[
"formula"],
678 auto secondary = std::make_shared<wfl::unit_callable>(*args.
u2);
687 lg::log_to_chat() <<
"Formula error in unit filter: " <<
e.type <<
" at " <<
e.filename <<
':' <<
e.line <<
")\n";
688 ERR_WML <<
"Formula error in unit filter: " <<
e.type <<
" at " <<
e.filename <<
':' <<
e.line <<
")";
695 create_attribute(literal[
"find_in"],
703 for (
const config&
c : gd->get_variable_access_read(find_in).as_array())
705 if(
c[
"id"] == args.
u.
id()) {
720 if (!literal[
"x"].
blank() || !literal[
"y"].blank()) {
721 children_.emplace_back(
new unit_filter_xy(literal[
"x"], literal[
"y"]));
724 for(
auto child : cfg.all_ordered()) {
727 cond_children_.emplace_back(std::piecewise_construct_t(), std::tuple(*cond), std::tuple(child.second));
729 else if (child.first ==
"filter_wml") {
731 config fwml = c.get_parsed_config();
735 config::all_children_itors ci = fwml.all_children_range();
736 if (fwml.all_children_count() == 1 && fwml.attribute_count() == 1 && ci.front().key ==
"variables") {
737 return args.u.variables().matches(ci.front().cfg);
741 return ucfg.matches(fwml);
745 else if (child.first ==
"filter_vision") {
747 std::set<int> viewers;
748 side_filter ssf(c, args.fc);
749 std::vector<int> sides = ssf.get_teams();
750 viewers.insert(sides.begin(), sides.end());
752 for (const int viewer : viewers) {
753 bool fogged = args.context().get_disp_context().get_team(viewer).fogged(args.loc);
755 bool hiding = args.context().get_disp_context().get_team(viewer).is_enemy(args.u.side()) && args.u.invisible(args.loc);
756 bool unit_hidden = fogged || hiding;
757 if (c[
"visible"].to_bool(true) != unit_hidden) {
764 else if (child.first ==
"filter_adjacent") {
765 children_.emplace_back(
new unit_filter_adjacent(child.second));
767 else if (child.first ==
"filter_location") {
769 return terrain_filter(c, args.fc, args.use_flat_tod).match(args.loc);
772 else if (child.first ==
"filter_side") {
774 return side_filter(c, args.fc).match(args.u.side());
777 else if ((child.first ==
"filter_ability") || (child.first ==
"experimental_filter_ability")) {
778 if(child.first ==
"experimental_filter_ability"){
782 if(!(c.get_parsed_config())[
"active"].to_bool()){
783 for(const auto [key, cfg] : args.u.abilities().all_children_view()) {
784 if(args.u.ability_matches_filter(cfg, key, c.get_parsed_config())) {
789 const unit_map& units = args.context().get_disp_context().units();
790 for(const auto [key, cfg] : args.u.abilities().all_children_view()) {
791 if(args.u.ability_matches_filter(cfg, key, c.get_parsed_config())) {
792 if (args.u.get_self_ability_bool(cfg, key, args.loc)) {
799 for(
unsigned i = 0;
i < adjacent.size(); ++
i) {
800 const unit_map::const_iterator it = units.find(adjacent[i]);
801 if (it == units.end() || it->incapacitated())
803 if (&*it == (args.u.shared_from_this()).get())
806 for(const auto [key, cfg] : it->abilities().all_children_view()) {
807 if(it->ability_matches_filter(cfg, key, c.get_parsed_config())) {
808 if (args.u.get_adj_ability_bool(cfg, key, i, args.loc, *it)) {
818 else if (child.first ==
"has_attack") {
820 for(const attack_type& a : args.u.attacks()) {
821 if(a.matches_filter(c.get_parsed_config())) {
829 std::stringstream errmsg;
830 errmsg <<
"encountered a child [" << child.first <<
"] of a standard unit filter, it is being ignored";
Variant for storing WML attributes.
std::string str(const std::string &fallback="") const
auto apply_visitor(const V &visitor) const
Visitor support: Applies a visitor to the underlying variant.
bool blank() const
Tests for an attribute that was never set.
bool empty() const
Tests for an attribute that either was never set or was set to "".
A config object defines a single node in a WML file, with access to child nodes.
auto all_children_view() const
In-order iteration over all children.
const team & get_team(int side) const
This getter takes a 1-based side number, not a 0-based team number.
virtual const gamemap & map() const =0
virtual const unit_map & units() const =0
virtual const display_context & get_disp_context() const =0
virtual game_lua_kernel * get_lua_kernel() const =0
virtual const game_data * get_game_data() const =0
terrain_code get_terrain(const map_location &loc) const
Looks up terrain at a particular location.
bool on_board(const map_location &loc) const
Tell if a location is on the map.
bool is_enemy(int n) const
Visitor helper class to parse the upkeep value from a config.
Visitor helper class to fetch the appropriate upkeep value.
const filter_context * fc_
bool matches(const unit &u, const map_location &loc) const
Determine if *this matches filter at a specified location.
std::vector< const unit * > all_matches_on_map(const map_location *loc=nullptr, const unit *other_unit=nullptr) const
unit_const_ptr first_match_on_map() const
unit_filter_impl::unit_filter_compound impl_
Container associating units to locations.
unit_iterator find(std::size_t id)
const std::string & id() const
const unit_type * find(const std::string &key, unit_type::BUILD_STATUS status=unit_type::FULL) const
Finds a unit_type by its id() and makes sure it is built to the specified level.
A single unit type that the player may recruit.
const std::string & parent_id() const
The id of the original type from which this (variation) descended.
This class represents a single unit of a specific type.
A variable-expanding proxy for the config class.
const config & get_config() const
bool as_bool() const
Returns a boolean state of the variant value.
Definitions for the interface to Wesnoth Markup Language (WML).
std::string deprecated_message(const std::string &elem_name, DEP_LEVEL level, const version_info &version, const std::string &detail)
Interfaces for manipulating version numbers of engine, add-ons, etc.
bool has_ability_type(const std::string &ability) const
Check if the unit has an ability of a specific type.
unit_ability_list get_abilities(const std::string &tag_name, const map_location &loc) const
Gets the unit's active abilities of a particular type if it were on a specified location.
bool get_self_ability_bool(const config &cfg, const std::string &ability, const map_location &loc) const
Checks whether this unit currently possesses a given ability, and that that ability is active.
bool has_ability_by_id(const std::string &ability) const
Check if the unit has an ability of a specific ID.
const config & abilities() const
bool get_adj_ability_bool(const config &cfg, const std::string &ability, int dir, const map_location &loc, const unit &from) const
Checks whether this unit is affected by a given ability, and that that ability is active.
unit_alignments::type alignment() const
The alignment of this unit.
int level() const
The current level of this unit.
std::string usage() const
Gets this unit's usage.
const std::string & get_role() const
Gets this unit's role.
int recall_cost() const
How much gold it costs to recall this unit, or -1 if the side's default recall cost is used.
const std::string & variation() const
The ID of the variation of this unit's type.
bool get_state(const std::string &state) const
Check if the unit is affected by a status effect.
const std::string & type_id() const
The id of this unit's type.
const unit_race * race() const
Gets this unit's race.
const unit_type & type() const
This unit's type, accounting for gender and variation.
bool can_recruit() const
Whether this unit can recruit other units - ie, are they a leader unit.
const std::string & id() const
Gets this unit's id.
int side() const
The side this unit belongs to.
unit_race::GENDER gender() const
The gender of this unit.
const t_string & name() const
Gets this unit's translatable display name.
@ STATE_GUARDIAN
The unit cannot be healed.
int defense_modifier(const t_translation::terrain_code &terrain) const
The unit's defense on a given terrain.
attack_itors attacks()
Gets an iterator over this unit's attacks.
const map_location & get_location() const
The current map location this unit is at.
int movement_cost(const t_translation::terrain_code &terrain) const
Get the unit's movement cost on a particular terrain.
int vision_cost(const t_translation::terrain_code &terrain) const
Get the unit's vision cost on a particular terrain.
int jamming_cost(const t_translation::terrain_code &terrain) const
Get the unit's jamming cost on a particular terrain.
int upkeep() const
Gets the amount of gold this unit costs a side per turn.
utils::variant< upkeep_full, upkeep_loyal, int > upkeep_t
std::vector< std::string > get_traits_list() const
Gets a list of the traits this unit currently has, including hidden traits.
void get_adjacent_tiles(const map_location &a, map_location *res)
Function which, given a location, will place all adjacent locations in res.
Standard logging facilities (interface).
bool in_ranges(const Cmp c, const std::vector< std::pair< Cmp, Cmp >> &ranges)
std::vector< std::pair< int, int > > default_counts
std::stringstream & log_to_chat()
Use this to show WML errors in the ingame chat.
std::function< int(lua_State *)> lua_function
filter_context * filter_con
std::string interpolate_variables_into_string(const std::string &str, const string_map *const symbols)
Function which will interpolate variables, starting with '$' in the string 'str' with the equivalent ...
std::vector< std::pair< int, int > > parse_ranges_unsigned(const std::string &str)
Handles a comma-separated list of inputs to parse_range, in a context that does not expect negative v...
std::vector< std::string > split(const config_attribute_value &val)
std::shared_ptr< const unit > unit_const_ptr
unit_race::GENDER string_gender(const std::string &str, unit_race::GENDER def)
Encapsulates the map of the game.
static std::vector< direction > parse_directions(const std::string &str)
Parse_directions takes a comma-separated list, and filters out any invalid directions.
static std::vector< direction > all_directions()
direction
Valid directions which can be moved in our hexagonal world.
bool matches_range(const std::string &xloc, const std::string &yloc) const
static std::string get_string(enum_type key)
Converts a enum to its string equivalent.
static constexpr utils::optional< enum_type > get_enum(const std::string_view value)
Converts a string into its enum equivalent.
const filter_context & context() const
virtual bool matches(const unit_filter_args &) const =0
bool filter_impl(const unit_filter_args &u) const
void create_child(const vconfig &c, F func)
void create_attribute(const config::attribute_value c, C conv, F func)
std::vector< std::pair< conditional_type::type, unit_filter_compound > > cond_children_
virtual bool matches(const unit_filter_args &u) const override
std::vector< std::shared_ptr< unit_filter_base > > children_
static map_location::direction s
unit_type_data unit_types
static lg::log_domain log_wml("wml")
static lg::log_domain log_config("config")