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)
133 std::size_t radius = cfg_[
"radius"].to_int(1);
136 std::vector<map_location::direction> dirs;
137 for(
const unit& u : units) {
140 if(from_loc == args.
loc || distance > radius || !child_.matches(
unit_filter_args{u, from_loc, &args.u, args.fc, args.use_flat_tod} )) {
144 for(
unsigned j = 0; j < adjacent.size(); ++j) {
145 bool adj_or_dist = distance != 1 ?
distance_between(adjacent[j], from_loc) == (distance - 1) : adjacent[j] == from_loc;
151 assert(dir >= 0 && dir <= 5);
153 if(!i_adjacent.
empty()) {
158 auto is_enemy = cfg_[
"is_enemy"];
166 if(i_count.
empty() && match_count == 0) {
183 unit_filter_child_literal(
const vconfig& v,
const F&
f) : v_(v) , f_(
f) {}
192 template<
typename T,
typename F>
195 unit_filter_attribute_parsed(T&& v, F&&
f) : v_(std::move(v)), f_(std::move(
f)) {}
204 template<
typename C,
typename F>
207 unit_filter_attribute_literal(std::string&& v, C&&
c, F&&
f) : v_(std::move(v)), c_(std::move(
c)), f_(std::move(
f)) {}
212 return f_(c_(v), args);
219 class contains_dollar_visitor
220 #ifdef USING_BOOST_VARIANT
221 :
public boost::static_visitor<bool>
225 contains_dollar_visitor() {}
229 bool operator()(
const T&)
const {
return false; }
231 bool operator()(
const t_string&)
const {
return true; }
233 bool operator()(
const std::string&
s)
const
235 return s.find(
'$') != std::string::npos;
242 unit_filter_compound::unit_filter_compound(
const vconfig& cfg)
270 case conditional_type::type::filter_and:
271 res = res &&
filter.second.matches(args);
273 case conditional_type::type::filter_or:
274 res = res ||
filter.second.matches(args);
276 case conditional_type::type::filter_not:
277 res = res && !
filter.second.matches(args);
287 if (!
filter->matches(args)) {
297 children_.emplace_back(
new unit_filter_child_literal<F>(
c, func));
300 template<
typename C,
typename F>
306 children_.emplace_back(
new unit_filter_attribute_literal<C, F>(std::move(v.
str()), std::move(conv), std::move(func)));
309 children_.emplace_back(
new unit_filter_attribute_parsed<decltype(conv(v)), F>(std::move(conv(v)), std::move(func)));
318 if(literal.
empty()) {
return; }
329 return std::find(id_list.begin(), id_list.end(), args.
u.
id()) != id_list.end();
345 std::set<std::string> types_expanded;
346 for(
const std::string&
type : types) {
347 if(types_expanded.count(
type)) {
351 const auto& tree = ut->advancement_tree();
352 types_expanded.insert(tree.begin(), tree.end());
353 types_expanded.insert(
type);
356 return types_expanded.find(args.
u.
type_id()) != types_expanded.end();
376 for(
const std::string& variation_id : types) {
377 if (
type->has_variation(variation_id)) {
389 for(
const std::string& ability_id : abilities) {
402 for(
const std::string& ability : abilities) {
431 utils::optional<int> dir;
433 for(std::size_t j = 0; j < adjacent.size(); ++j) {
434 bool adj_or_dist = distance != 1 ?
distance_between(adjacent[j], args.
loc) == (distance - 1) : adjacent[j] == args.
loc;
454 for(
const std::string& ability : abilities) {
467 std::sort(res.begin(), res.end());
471 [](
const std::vector<std::string>& check_traits,
const unit_filter_args& args)
475 std::vector<std::string> isect;
476 std::sort(have_traits.begin(), have_traits.end());
477 std::set_intersection(check_traits.begin(), check_traits.end(), have_traits.begin(), have_traits.end(), std::back_inserter(isect));
478 return !isect.empty();
486 return std::find(races.begin(), races.end(), args.
u.
race()->
id()) != races.end();
494 return gender == args.
u.
gender();
503 }
catch(std::invalid_argument&) {
516 std::vector<int> res;
519 res.push_back(std::stoi(s));
520 }
catch(std::invalid_argument&) {
521 WRN_CF <<
"ignored invalid side='" << s <<
"' in filter";
528 return std::find(sides.begin(), sides.end(), args.
u.
side()) != sides.end();
532 create_attribute(literal[
"status"],
536 for(
const std::string& status : statuses) {
545 create_attribute(literal[
"has_weapon"],
551 if(a.id() == weapon) {
559 create_attribute(literal[
"role"],
567 create_attribute(literal[
"alignment"],
575 create_attribute(literal[
"ai_special"],
583 create_attribute(literal[
"usage"],
587 for(
const std::string& usage : usages) {
588 if(args.
u.
usage() == usage) {
596 create_attribute(literal[
"canrecruit"],
604 create_attribute(literal[
"recall_cost"],
606 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
612 create_attribute(literal[
"level"],
614 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
620 create_attribute(literal[
"defense"],
622 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
625 return in_ranges(actual_defense, ranges);
629 create_attribute(literal[
"movement_cost"],
631 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
638 create_attribute(literal[
"vision_cost"],
640 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
647 create_attribute(literal[
"jamming_cost"],
649 [](
const std::vector<std::pair<int,int>>& ranges,
const unit_filter_args& args)
656 create_attribute(literal[
"lua_function"],
667 create_attribute(literal[
"formula"],
673 lg::log_to_chat() <<
"Formula error while evaluating formula in unit filter: " <<
e.type <<
" at "
674 <<
e.filename <<
':' <<
e.line <<
")\n";
675 ERR_WML <<
"Formula error while evaluating formula in unit filter: " <<
e.type <<
" at "
676 <<
e.filename <<
':' <<
e.line <<
")";
686 auto secondary = std::make_shared<wfl::unit_callable>(*args.
u2);
695 lg::log_to_chat() <<
"Formula error in unit filter: " <<
e.type <<
" at " <<
e.filename <<
':' <<
e.line <<
")\n";
696 ERR_WML <<
"Formula error in unit filter: " <<
e.type <<
" at " <<
e.filename <<
':' <<
e.line <<
")";
703 create_attribute(literal[
"find_in"],
711 for (
const config&
c : gd->get_variable_access_read(find_in).as_array())
713 if(
c[
"id"] == args.
u.
id()) {
728 if (!literal[
"x"].
blank() || !literal[
"y"].blank()) {
729 children_.emplace_back(
new unit_filter_xy(literal[
"x"], literal[
"y"]));
732 for(
auto child : cfg.all_ordered()) {
735 cond_children_.emplace_back(std::piecewise_construct_t(), std::tuple(*cond), std::tuple(child.second));
737 else if (child.first ==
"filter_wml") {
739 config fwml = c.get_parsed_config();
743 config::all_children_itors ci = fwml.all_children_range();
744 if (fwml.all_children_count() == 1 && fwml.attribute_count() == 1 && ci.front().key ==
"variables") {
745 return args.u.variables().matches(ci.front().cfg);
749 return ucfg.matches(fwml);
753 else if (child.first ==
"filter_vision") {
755 std::set<int> viewers;
756 side_filter ssf(c, args.fc);
757 std::vector<int> sides = ssf.get_teams();
758 viewers.insert(sides.begin(), sides.end());
760 for (const int viewer : viewers) {
761 bool fogged = args.context().get_disp_context().get_team(viewer).fogged(args.loc);
763 bool hiding = args.context().get_disp_context().get_team(viewer).is_enemy(args.u.side()) && args.u.invisible(args.loc);
764 bool unit_hidden = fogged || hiding;
765 if (c[
"visible"].to_bool(true) != unit_hidden) {
772 else if (child.first ==
"filter_adjacent") {
773 children_.emplace_back(
new unit_filter_adjacent(child.second));
775 else if (child.first ==
"filter_location") {
777 return terrain_filter(c, args.fc, args.use_flat_tod).match(args.loc);
780 else if (child.first ==
"filter_side") {
782 return side_filter(c, args.fc).match(args.u.side());
785 else if ((child.first ==
"filter_ability") || (child.first ==
"experimental_filter_ability")) {
786 if(child.first ==
"experimental_filter_ability"){
790 if(!(c.get_parsed_config())[
"active"].to_bool()){
791 for(const auto [key, cfg] : args.u.abilities().all_children_view()) {
792 if(args.u.ability_matches_filter(cfg, key, c.get_parsed_config())) {
797 const unit_map& units = args.context().get_disp_context().units();
798 for(const auto [key, cfg] : args.u.abilities().all_children_view()) {
799 if(args.u.ability_matches_filter(cfg, key, c.get_parsed_config())) {
800 if (args.u.get_self_ability_bool(cfg, key, args.loc)) {
806 if(
c.get_parsed_config()[
"affect_adjacent"].to_bool(
true)) {
807 for(const unit& unit : units) {
808 if(!unit.has_ability_distant() || unit.incapacitated() || &unit == args.u.shared_from_this().get()) {
811 const map_location& from_loc = unit.get_location();
812 std::size_t distance = distance_between(from_loc, args.loc);
813 if(distance > *unit.has_ability_distant()) {
816 utils::optional<int> dir;
817 const auto adjacent = get_adjacent_tiles(from_loc);
818 for(std::size_t j = 0; j < adjacent.size(); ++j) {
819 bool adj_or_dist = distance != 1 ? distance_between(adjacent[j], args.loc) == (distance - 1) : adjacent[j] == args.loc;
825 for(const auto [key, cfg] : unit.abilities().all_children_view()) {
826 if(args.u.get_adj_ability_bool(cfg, key, distance, *dir, args.loc, unit, from_loc)) {
836 else if (child.first ==
"has_attack") {
838 for(const attack_type& a : args.u.attacks()) {
839 if(a.matches_filter(c.get_parsed_config())) {
847 std::stringstream errmsg;
848 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.
unit_filter(const vconfig &cfg)
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.
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 get_adj_ability_bool(const config &cfg, const std::string &ability, std::size_t dist, int dir, const map_location &loc, const unit &from, const map_location &from_loc) const
Checks whether this unit is affected by a given ability, and that that ability is active.
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
unit_alignments::type alignment() const
The alignment of this unit.
bool incapacitated() const
Check if the unit has been petrified.
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::optional< std::size_t > has_ability_distant() const
Gets if this unit own ability with [affect_adjacent] subtags.
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.
std::size_t distance_between(const map_location &a, const map_location &b)
Function which gives the number of hexes between two tiles (i.e.
Standard logging facilities (interface).
bool in_ranges(const Cmp c, const std::vector< std::pair< Cmp, Cmp >> &ranges)
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::set< std::string > split_set(std::string_view s, char sep, const int flags)
bool contains(const Container &container, const Value &value)
Returns true iff value is found in container.
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)
auto * find(Container &container, const Value &value)
Convenience wrapper for using find on a container without needing to comare to end()
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.
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 fill(const vconfig &cfg)
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")