diff options
Diffstat (limited to 'src/map/magic-expr.cpp')
| -rw-r--r-- | src/map/magic-expr.cpp | 1874 |
1 files changed, 0 insertions, 1874 deletions
diff --git a/src/map/magic-expr.cpp b/src/map/magic-expr.cpp deleted file mode 100644 index 197727e..0000000 --- a/src/map/magic-expr.cpp +++ /dev/null @@ -1,1874 +0,0 @@ -#include "magic-expr.hpp" -// magic-expr.cpp - Pure functions for the old magic backend. -// -// Copyright © 2004-2011 The Mana World Development Team -// Copyright © 2011-2014 Ben Longbons <b.r.longbons@gmail.com> -// -// This file is part of The Mana World (Athena server) -// -// This program is free software: you can redistribute it and/or modify -// it under the terms of the GNU General Public License as published by -// the Free Software Foundation, either version 3 of the License, or -// (at your option) any later version. -// -// This program is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -// GNU General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with this program. If not, see <http://www.gnu.org/licenses/>. - -#include <cassert> - -#include <algorithm> - -#include "../strings/mstring.hpp" -#include "../strings/astring.hpp" -#include "../strings/zstring.hpp" -#include "../strings/vstring.hpp" -#include "../strings/literal.hpp" - -#include "../generic/dumb_ptr.hpp" -#include "../generic/random.hpp" - -#include "../io/cxxstdio.hpp" - -#include "../mmo/cxxstdio_enums.hpp" - -#include "battle.hpp" -#include "itemdb.hpp" -#include "magic-expr-eval.hpp" -#include "magic-interpreter.hpp" -#include "magic-interpreter-base.hpp" -#include "npc.hpp" -#include "pc.hpp" -#include "script-call.hpp" - -#include "../poison.hpp" - - -namespace tmwa -{ -namespace map -{ -namespace magic -{ -static -void free_area(dumb_ptr<area_t> area) -{ - if (!area) - return; - - MATCH_BEGIN (*area) - { - MATCH_CASE (const AreaUnion&, a) - { - free_area(a.a_union[0]); - free_area(a.a_union[1]); - } - } - MATCH_END (); - - area.delete_(); -} - -static -dumb_ptr<area_t> dup_area(dumb_ptr<area_t> area) -{ - MATCH_BEGIN (*area) - { - MATCH_CASE (const location_t&, loc) - { - return dumb_ptr<area_t>::make(loc); - } - MATCH_CASE (const AreaUnion&, a) - { - AreaUnion u; - u.a_union[0] = dup_area(a.a_union[0]); - u.a_union[1] = dup_area(a.a_union[1]); - return dumb_ptr<area_t>::make(u); - } - MATCH_CASE (const AreaRect&, rect) - { - return dumb_ptr<area_t>::make(rect); - } - MATCH_CASE (const AreaBar&, bar) - { - return dumb_ptr<area_t>::make(bar); - } - } - MATCH_END (); - - abort(); -} - -void magic_copy_var(val_t *dest, const val_t *src) -{ - MATCH_BEGIN (*src) - { - MATCH_DEFAULT () - { - abort(); - } - MATCH_CASE (const ValUndef&, s) - { - *dest = s; - } - MATCH_CASE (const ValInt&, s) - { - *dest = s; - } - MATCH_CASE (const ValDir&, s) - { - *dest = s; - } - MATCH_CASE (const ValString&, s) - { - *dest = ValString{s.v_string}; - } - MATCH_CASE (const ValEntityInt&, s) - { - *dest = s; - } - MATCH_CASE (const ValEntityPtr&, s) - { - *dest = s; - } - MATCH_CASE (const ValLocation&, s) - { - *dest = s; - } - MATCH_CASE (const ValArea&, s) - { - *dest = ValArea{dup_area(s.v_area)}; - } - MATCH_CASE (const ValSpell&, s) - { - *dest = s; - } - MATCH_CASE (const ValInvocationInt&, s) - { - *dest = s; - } - MATCH_CASE (const ValInvocationPtr&, s) - { - *dest = s; - } - MATCH_CASE (const ValFail&, s) - { - *dest = s; - } - MATCH_CASE (const ValNegative1&, s) - { - *dest = s; - } - } - MATCH_END (); -} - -void magic_clear_var(val_t *v) -{ - MATCH_BEGIN (*v) - { - MATCH_CASE (ValString&, s) - { - (void)s; - } - MATCH_CASE (const ValArea&, a) - { - free_area(a.v_area); - } - } - MATCH_END (); -} - -static -AString show_entity(dumb_ptr<block_list> entity) -{ - switch (entity->bl_type) - { - case BL::PC: - return entity->is_player()->status_key.name.to__actual(); - case BL::NPC: - return entity->is_npc()->name; - case BL::MOB: - return entity->is_mob()->name; - case BL::ITEM: - assert (0 && "There is no way this code did what it was supposed to do!"_s); - /* Sorry about this one... */ - // WTF? item_data is a Item, not a struct item_data - // return ((struct item_data *) (&entity->is_item()->item_data))->name; - abort(); - case BL::SPELL: - return "%invocation(ERROR:this-should-not-be-an-entity)"_s; - default: - return "%unknown-entity"_s; - } -} - -static -void stringify(val_t *v) -{ - static earray<LString, DIR, DIR::COUNT> dirs //= - {{ - "south"_s, "south-west"_s, - "west"_s, "north-west"_s, - "north"_s, "north-east"_s, - "east"_s, "south-east"_s, - }}; - AString buf; - - MATCH_BEGIN (*v) - { - MATCH_DEFAULT () - { - abort(); - } - MATCH_CASE (const ValUndef&, x) - { - (void)x; - buf = "UNDEF"_s; - } - MATCH_CASE (const ValInt&, x) - { - buf = STRPRINTF("%i"_fmt, x.v_int); - } - MATCH_CASE (const ValString&, x) - { - (void)x; - return; - } - MATCH_CASE (const ValDir&, x) - { - buf = dirs[x.v_dir]; - } - MATCH_CASE (const ValEntityPtr&, x) - { - buf = show_entity(x.v_entity); - } - MATCH_CASE (const ValLocation&, x) - { - buf = STRPRINTF("<\"%s\", %d, %d>"_fmt, - x.v_location.m->name_, - x.v_location.x, - x.v_location.y); - } - MATCH_CASE (const ValArea&, x) - { - buf = "%area"_s; - free_area(x.v_area); - } - MATCH_CASE (const ValSpell&, x) - { - buf = x.v_spell->name; - } - MATCH_CASE (const ValInvocationInt&, x) - { - dumb_ptr<invocation> invocation_ = - map_id2bl(x.v_iid)->is_spell(); - buf = invocation_->spell->name; - } - MATCH_CASE (const ValInvocationPtr&, x) - { - dumb_ptr<invocation> invocation_ = - x.v_invocation; - buf = invocation_->spell->name; - } - } - MATCH_END (); - - *v = ValString{buf}; -} - -static -void intify(val_t *v) -{ - if (v->is<ValInt>()) - return; - - magic_clear_var(v); - *v = ValInt{1}; -} - -static -dumb_ptr<area_t> area_union(dumb_ptr<area_t> area, dumb_ptr<area_t> other_area) -{ - AreaUnion a; - a.a_union[0] = area; - a.a_union[1] = other_area; - return dumb_ptr<area_t>::make(a); -} - -/** - * Turns location into area, leaves other types untouched - */ -static -void make_area(val_t *v) -{ - if (ValLocation *l = v->get_if<ValLocation>()) - { - auto a = dumb_ptr<area_t>::make(l->v_location); - *v = ValArea{a}; - } -} - -static -void make_location(val_t *v) -{ - if (ValArea *a = v->get_if<ValArea>()) - { - MATCH_BEGIN (*a->v_area) - { - MATCH_CASE (const location_t&, location) - { - free_area(a->v_area); - *v = ValLocation{location}; - } - } - MATCH_END (); - } -} - -static -void make_spell(val_t *v) -{ - assert(!v->is<ValInvocationInt>()); - if (ValInvocationPtr *p = v->get_if<ValInvocationPtr>()) - { - dumb_ptr<invocation> invoc = p->v_invocation; - if (!invoc) - { - *v = ValFail{}; - } - else - { - *v = ValSpell{invoc->spell}; - } - } -} - -static -int fun_add(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (args[0].is<ValInt>() && args[1].is<ValInt>()) - { - /* Integer addition */ - *result = ValInt{ARGINT(0) + ARGINT(1)}; - } - else if (ARG_MAY_BE_AREA(0) && ARG_MAY_BE_AREA(1)) - { - /* Area union */ - make_area(&args[0]); - make_area(&args[1]); - *result = ValArea{area_union(ARGAREA(0), ARGAREA(1))}; - ARGAREA(0) = nullptr; args[0] = ValUndef{}; - ARGAREA(1) = nullptr; args[1] = ValUndef{}; - } - else - { - /* Anything else -> string concatenation */ - stringify(&args[0]); - stringify(&args[1]); - /* Yes, we could speed this up. */ - // ugh - MString m; - m += ARGSTR(0); - m += ARGSTR(1); - *result = ValString{AString(m)}; - } - return 0; -} - -static -int fun_sub(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) - ARGINT(1)}; - return 0; -} - -static -int fun_mul(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) * ARGINT(1)}; - return 0; -} - -static -int fun_div(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (!ARGINT(1)) - return 1; /* division by zero */ - *result = ValInt{ARGINT(0) / ARGINT(1)}; - return 0; -} - -static -int fun_mod(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (!ARGINT(1)) - return 1; /* division by zero */ - *result = ValInt{ARGINT(0) % ARGINT(1)}; - return 0; -} - -static -int fun_or(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) || ARGINT(1)}; - return 0; -} - -static -int fun_and(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) && ARGINT(1)}; - return 0; -} - -static -int fun_not(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{!ARGINT(0)}; - return 0; -} - -static -int fun_neg(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{~ARGINT(0)}; - return 0; -} - -static -int fun_gte(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (args[0].is<ValString>() || args[1].is<ValString>()) - { - stringify(&args[0]); - stringify(&args[1]); - *result = ValInt{ARGSTR(0) >= ARGSTR(1)}; - } - else - { - intify(&args[0]); - intify(&args[1]); - *result = ValInt{ARGINT(0) >= ARGINT(1)}; - } - return 0; -} - -static -int fun_lt(dumb_ptr<env_t> env, val_t *result, Slice<val_t> args) -{ - fun_gte(env, result, args); - result->get_if<ValInt>()->v_int ^= 1; - return 0; -} - -static -int fun_gt(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (args[0].is<ValString>() || args[1].is<ValString>()) - { - stringify(&args[0]); - stringify(&args[1]); - *result = ValInt{ARGSTR(0) > ARGSTR(1)}; - } - else - { - intify(&args[0]); - intify(&args[1]); - *result = ValInt{ARGINT(0) > ARGINT(1)}; - } - return 0; -} - -static -int fun_lte(dumb_ptr<env_t> env, val_t *result, Slice<val_t> args) -{ - fun_gt(env, result, args); - result->get_if<ValInt>()->v_int ^= 1; - return 0; -} - -static -int fun_eq(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (args[0].is<ValString>() || args[1].is<ValString>()) - { - stringify(&args[0]); - stringify(&args[1]); - *result = ValInt{ARGSTR(0) == ARGSTR(1)}; - } - else if (args[0].is<ValDir>() && args[1].is<ValDir>()) - *result = ValInt{ARGDIR(0) == ARGDIR(1)}; - else if (args[0].is<ValEntityPtr>() && args[1].is<ValEntityPtr>()) - *result = ValInt{ARGENTITY(0) == ARGENTITY(1)}; - else if (args[0].is<ValLocation>() && args[1].is<ValLocation>()) - *result = ValInt{(ARGLOCATION(0).x == ARGLOCATION(1).x - && ARGLOCATION(0).y == ARGLOCATION(1).y - && ARGLOCATION(0).m == ARGLOCATION(1).m)}; - else if (args[0].is<ValArea>() && args[1].is<ValArea>()) - *result = ValInt{ARGAREA(0) == ARGAREA(1)}; /* Probably not that great an idea... */ - else if (args[0].is<ValSpell>() && args[1].is<ValSpell>()) - *result = ValInt{ARGSPELL(0) == ARGSPELL(1)}; - else if (args[0].is<ValInvocationPtr>() && args[1].is<ValInvocationPtr>()) - *result = ValInt{ARGINVOCATION(0) == ARGINVOCATION(1)}; - else - { - intify(&args[0]); - intify(&args[1]); - *result = ValInt{ARGINT(0) == ARGINT(1)}; - } - return 0; -} - -static -int fun_ne(dumb_ptr<env_t> env, val_t *result, Slice<val_t> args) -{ - fun_eq(env, result, args); - result->get_if<ValInt>()->v_int ^= 1; - return 0; -} - -static -int fun_bitand(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) & ARGINT(1)}; - return 0; -} - -static -int fun_bitor(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) | ARGINT(1)}; - return 0; -} - -static -int fun_bitxor(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) ^ ARGINT(1)}; - return 0; -} - -static -int fun_bitshl(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) << ARGINT(1)}; - return 0; -} - -static -int fun_bitshr(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{ARGINT(0) >> ARGINT(1)}; - return 0; -} - -static -int fun_max(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{std::max(ARGINT(0), ARGINT(1))}; - return 0; -} - -static -int fun_min(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{std::min(ARGINT(0), ARGINT(1))}; - return 0; -} - -static -int fun_if_then_else(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ARGINT(0)) - magic_copy_var(result, &args[1]); - else - magic_copy_var(result, &args[2]); - return 0; -} - -Borrowed<map_local> magic_area_rect(int *x, int *y, int *width, int *height, - area_t& area_) -{ - MATCH_BEGIN (area_) - { - MATCH_CASE (const AreaUnion&, a) - { - (void)a; - abort(); - } - MATCH_CASE (const location_t&, a_loc) - { - P<map_local> m = a_loc.m; - *x = a_loc.x; - *y = a_loc.y; - *width = 1; - *height = 1; - return m; - } - MATCH_CASE (const AreaRect&, a_rect) - { - P<map_local> m = a_rect.loc.m; - *x = a_rect.loc.x; - *y = a_rect.loc.y; - *width = a_rect.width; - *height = a_rect.height; - return m; - } - MATCH_CASE (const AreaBar&, a_bar) - { - int tx = a_bar.loc.x; - int ty = a_bar.loc.y; - int twidth = a_bar.width; - int tdepth = a_bar.width; - P<map_local> m = a_bar.loc.m; - - switch (a_bar.dir) - { - case DIR::S: - *x = tx - twidth; - *y = ty; - *width = twidth * 2 + 1; - *height = tdepth; - break; - - case DIR::W: - *x = tx - tdepth; - *y = ty - twidth; - *width = tdepth; - *height = twidth * 2 + 1; - break; - - case DIR::N: - *x = tx - twidth; - *y = ty - tdepth; - *width = twidth * 2 + 1; - *height = tdepth; - break; - - case DIR::E: - *x = tx; - *y = ty - twidth; - *width = tdepth; - *height = twidth * 2 + 1; - break; - - default: - FPRINTF(stderr, - "Error: Trying to compute area of NE/SE/NW/SW-facing bar"_fmt); - *x = tx; - *y = ty; - *width = *height = 1; - } - return m; - } - } - MATCH_END (); - abort(); -} - -int magic_location_in_area(Borrowed<map_local> m, int x, int y, dumb_ptr<area_t> area) -{ - MATCH_BEGIN (*area) - { - MATCH_CASE (const AreaUnion&, a) - { - return magic_location_in_area(m, x, y, a.a_union[0]) - || magic_location_in_area(m, x, y, a.a_union[1]); - } - MATCH_CASE (const location_t&, a_loc) - { - (void)a_loc; - // TODO this can be simplified - int ax, ay, awidth, aheight; - P<map_local> am = magic_area_rect(&ax, &ay, &awidth, &aheight, *area); - return (am == m - && (x >= ax) && (y >= ay) - && (x < ax + awidth) && (y < ay + aheight)); - } - MATCH_CASE (const AreaRect&, a_rect) - { - (void)a_rect; - // TODO this is too complicated - int ax, ay, awidth, aheight; - P<map_local> am = magic_area_rect(&ax, &ay, &awidth, &aheight, *area); - return (am == m - && (x >= ax) && (y >= ay) - && (x < ax + awidth) && (y < ay + aheight)); - } - MATCH_CASE (const AreaBar&, a_bar) - { - (void)a_bar; - // TODO this is wrong - int ax, ay, awidth, aheight; - P<map_local> am = magic_area_rect(&ax, &ay, &awidth, &aheight, *area); - return (am == m - && (x >= ax) && (y >= ay) - && (x < ax + awidth) && (y < ay + aheight)); - } - } - MATCH_END (); - abort(); -} - -static -int fun_is_in(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{magic_location_in_area(ARGLOCATION(0).m, - ARGLOCATION(0).x, - ARGLOCATION(0).y, ARGAREA(1))}; - return 0; -} - -static -int fun_skill(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ENTITY_TYPE(0) != BL::PC - // don't convert to enum until after the range check - // (actually it would be okay, I checked) - || ARGINT(1) < 0 - || ARGINT(1) >= static_cast<uint16_t>(MAX_SKILL)) - { - *result = ValInt{0}; - } - else - { - SkillID id = static_cast<SkillID>(ARGINT(1)); - *result = ValInt{ARGPC(0)->status.skill[id].lv}; - } - return 0; -} - -static -int fun_his_shroud(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{(ENTITY_TYPE(0) == BL::PC && ARGPC(0)->state.shroud_active)}; - return 0; -} - -#define BATTLE_GETTER(name) \ -static \ -int fun_get_##name(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) \ -{ \ - *result = ValInt{battle_get_##name(ARGENTITY(0))}; \ - return 0; \ -} - -BATTLE_GETTER(str) -BATTLE_GETTER(agi) -BATTLE_GETTER(vit) -BATTLE_GETTER(dex) -BATTLE_GETTER(luk) -BATTLE_GETTER(int) -BATTLE_GETTER(lv) -BATTLE_GETTER(hp) -BATTLE_GETTER(mdef) -BATTLE_GETTER(def) -BATTLE_GETTER(max_hp) -static -int fun_get_dir(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValDir{battle_get_dir(ARGENTITY(0))}; - return 0; -} - -#define MMO_GETTER(name) \ -static \ -int fun_get_##name(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) \ -{ \ - if (ENTITY_TYPE(0) == BL::PC) \ - *result = ValInt{ARGPC(0)->status.name}; \ - else \ - *result = ValInt{0}; \ - return 0; \ -} - -MMO_GETTER(sp) -MMO_GETTER(max_sp) - -static -int fun_name_of(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (args[0].is<ValEntityPtr>()) - { - *result = ValString{show_entity(ARGENTITY(0))}; - return 0; - } - else if (args[0].is<ValSpell>()) - { - *result = ValString{ARGSPELL(0)->name}; - return 0; - } - else if (args[0].is<ValInvocationPtr>()) - { - *result = ValString{ARGINVOCATION(0)->spell->name}; - return 0; - } - return 1; -} - -/* [Freeyorp] I'm putting this one in as name_of seems to have issues with summoned or spawned mobs. */ -static -int fun_mob_id(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ENTITY_TYPE(0) != BL::MOB) - return 1; - *result = ValInt{unwrap<Species>(ARGMOB(0)->mob_class)}; - return 0; -} - -inline -void COPY_LOCATION(block_list& dest, location_t& src) -{ - dest.bl_x = src.x; - dest.bl_y = src.y; - dest.bl_m = src.m; -} - -inline -void COPY_LOCATION(location_t& dest, block_list& src) -{ - dest.x = src.bl_x; - dest.y = src.bl_y; - dest.m = src.bl_m; -} - -static -int fun_location(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - location_t loc; - COPY_LOCATION(loc, *(ARGENTITY(0))); - *result = ValLocation{loc}; - return 0; -} - -static -int fun_random(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - int delta = ARGINT(0); - if (delta < 0) - delta = -delta; - if (delta == 0) - { - *result = ValInt{0}; - return 0; - } - *result = ValInt{random_::to(delta)}; - - if (ARGINT(0) < 0) - result->get_if<ValInt>()->v_int *= -1; - return 0; -} - -static -int fun_random_dir(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ARGINT(0)) - *result = ValDir{random_::choice({DIR::S, DIR::SW, DIR::W, DIR::NW, DIR::N, DIR::NE, DIR::E, DIR::SE})}; - else - *result = ValDir{random_::choice({DIR::S, DIR::W, DIR::N, DIR::E})}; - return 0; -} - -static -int fun_hash_entity(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{static_cast<int32_t>(unwrap<BlockId>(ARGENTITY(0)->bl_id))}; - return 0; -} - -// ret -1: not a string, ret 1: no such item, ret 0: OK -int magic_find_item(Slice<val_t> args, int index, Item *item_, int *stackable) -{ - Option<P<struct item_data>> item_data_ = None; - int must_add_sequentially; - - if (args[index].is<ValInt>()) - item_data_ = itemdb_exists(wrap<ItemNameId>(static_cast<uint16_t>(ARGINT(index)))); - else if (args[index].is<ValString>()) - item_data_ = itemdb_searchname(ARGSTR(index)); - else - return -1; - - P<struct item_data> item_data = TRY_UNWRAP(item_data_, return 1); - - // Very elegant. - must_add_sequentially = ( - item_data->type == ItemType::WEAPON - || item_data->type == ItemType::ARMOR - || item_data->type == ItemType::_7 - || item_data->type == ItemType::_8); - - if (stackable) - *stackable = !must_add_sequentially; - - *item_ = Item(); - item_->nameid = item_data->nameid; - - return 0; -} - -static -int fun_count_item(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - dumb_ptr<map_session_data> chr = (ENTITY_TYPE(0) == BL::PC) ? ARGPC(0) : nullptr; - int stackable; - Item item; - - GET_ARG_ITEM(1, item, stackable); - - if (!chr) - return 1; - - *result = ValInt{pc_count_all_items(chr, item.nameid)}; - return 0; -} - -static -int fun_is_equipped(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - dumb_ptr<map_session_data> chr = (ENTITY_TYPE(0) == BL::PC) ? ARGPC(0) : nullptr; - int stackable; - Item item; - bool retval = false; - - GET_ARG_ITEM(1, item, stackable); - - if (!chr) - return 1; - - for (EQUIP i : EQUIPs) - { - IOff0 idx = chr->equip_index_maybe[i]; - if (idx.ok() && chr->status.inventory[idx].nameid == item.nameid) - { - retval = true; - break; - } - } - - *result = ValInt{retval}; - return 0; -} - -static -int fun_is_married(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{(ENTITY_TYPE(0) == BL::PC && ARGPC(0)->status.partner_id)}; - return 0; -} - -static -int fun_is_dead(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{(ENTITY_TYPE(0) == BL::PC && pc_isdead(ARGPC(0)))}; - return 0; -} - -static -int fun_is_pc(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{(ENTITY_TYPE(0) == BL::PC)}; - return 0; -} - -static -int fun_partner(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ENTITY_TYPE(0) == BL::PC && ARGPC(0)->status.partner_id) - { - *result = - ValEntityPtr{map_nick2sd(map_charid2nick(ARGPC(0)->status.partner_id))}; - return 0; - } - else - return 1; -} - -static -int fun_awayfrom(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - location_t *loc = &ARGLOCATION(0); - int dx = dirx[ARGDIR(1)]; - int dy = diry[ARGDIR(1)]; - int distance = ARGINT(2); - while (distance-- - && !bool(read_gatp(loc->m, loc->x + dx, loc->y + dy) - & MapCell::UNWALKABLE)) - { - loc->x += dx; - loc->y += dy; - } - - *result = ValLocation{*loc}; - return 0; -} - -static -int fun_failed(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{args[0].is<ValFail>()}; - return 0; -} - -static -int fun_npc(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - NpcName name = stringish<NpcName>(ARGSTR(0)); - dumb_ptr<npc_data> npc = npc_name2id(name); - *result = ValEntityPtr{npc}; - return npc == nullptr; -} - -static -int fun_pc(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - CharName name = stringish<CharName>(ARGSTR(0)); - dumb_ptr<map_session_data> chr = map_nick2sd(name); - *result = ValEntityPtr{chr}; - return chr == nullptr; -} - -static -int fun_distance(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ARGLOCATION(0).m != ARGLOCATION(1).m) - *result = ValInt{0x7fffffff}; - else - *result = ValInt{std::max(abs(ARGLOCATION(0).x - ARGLOCATION(1).x), - abs(ARGLOCATION(0).y - ARGLOCATION(1).y))}; - return 0; -} - -static -int fun_rdistance(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ARGLOCATION(0).m != ARGLOCATION(1).m) - *result = ValInt{0x7fffffff}; - else - { - int dx = ARGLOCATION(0).x - ARGLOCATION(1).x; - int dy = ARGLOCATION(0).y - ARGLOCATION(1).y; - *result = ValInt{static_cast<int>(sqrt((dx * dx) + (dy * dy)))}; - } - return 0; -} - -static -int fun_anchor(dumb_ptr<env_t> env, val_t *result, Slice<val_t> args) -{ - dumb_ptr<teleport_anchor_t> anchor = magic_find_anchor(ARGSTR(0)); - - if (!anchor) - return 1; - - magic_eval(env, result, anchor->location); - - make_area(result); - if (!result->is<ValArea>()) - { - magic_clear_var(result); - return 1; - } - - return 0; -} - -static -int fun_line_of_sight(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - block_list e1, e2; - - COPY_LOCATION(e1, ARGLOCATION(0)); - COPY_LOCATION(e2, ARGLOCATION(1)); - - *result = ValInt{battle_check_range(dumb_ptr<block_list>(&e1), dumb_ptr<block_list>(&e2), 0)}; - - return 0; -} - -void magic_random_location(location_t *dest, dumb_ptr<area_t> area) -{ - MATCH_BEGIN (*area) - { - MATCH_CASE (const AreaUnion&, a) - { - if (random_::chance({a.a_union[0]->size, area->size})) - magic_random_location(dest, a.a_union[0]); - else - magic_random_location(dest, a.a_union[1]); - } - MATCH_CASE (const location_t&, a_loc) - { - (void)a_loc; - // TODO this can be simplified - int x, y, w, h; - P<map_local> m = magic_area_rect(&x, &y, &w, &h, *area); - - if (w <= 1) - w = 1; - - if (h <= 1) - h = 1; - - // This is not exactly the same as the old logic, - // but it's better. - auto pair = map_randfreecell(m, x, y, w, h); - - dest->m = m; - dest->x = pair.first; - dest->y = pair.second; - } - MATCH_CASE (const AreaRect&, a_rect) - { - (void)a_rect; - // TODO this can be simplified - int x, y, w, h; - P<map_local> m = magic_area_rect(&x, &y, &w, &h, *area); - - if (w <= 1) - w = 1; - - if (h <= 1) - h = 1; - - // This is not exactly the same as the old logic, - // but it's better. - auto pair = map_randfreecell(m, x, y, w, h); - - dest->m = m; - dest->x = pair.first; - dest->y = pair.second; - } - MATCH_CASE (const AreaBar&, a_bar) - { - (void)a_bar; - // TODO this is wrong - int x, y, w, h; - P<map_local> m = magic_area_rect(&x, &y, &w, &h, *area); - - if (w <= 1) - w = 1; - - if (h <= 1) - h = 1; - - // This is not exactly the same as the old logic, - // but it's better. - auto pair = map_randfreecell(m, x, y, w, h); - - dest->m = m; - dest->x = pair.first; - dest->y = pair.second; - } - } - MATCH_END (); -} - -static -int fun_pick_location(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - location_t loc; - magic_random_location(&loc, ARGAREA(0)); - *result = ValLocation{loc}; - return 0; -} - -static -int fun_read_script_int(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - dumb_ptr<block_list> subject_p = ARGENTITY(0); - VarName var_name = stringish<VarName>(ARGSTR(1)); - int array_index = 0; - - if (subject_p->bl_type != BL::PC) - return 1; - - *result = ValInt{get_script_var_i(subject_p->is_player(), var_name, array_index)}; - return 0; -} - -static -int fun_read_script_str(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - dumb_ptr<block_list> subject_p = ARGENTITY(0); - VarName var_name = stringish<VarName>(ARGSTR(1)); - int array_index = 0; - - if (subject_p->bl_type != BL::PC) - return 1; - - *result = ValString{get_script_var_s(subject_p->is_player(), var_name, array_index)}; - return 0; -} - -static -int fun_rbox(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - location_t loc = ARGLOCATION(0); - int radius = ARGINT(1); - - AreaRect a_rect; - a_rect.loc.m = loc.m; - a_rect.loc.x = loc.x - radius; - a_rect.loc.y = loc.y - radius; - a_rect.width = radius * 2 + 1; - a_rect.height = radius * 2 + 1; - *result = ValArea{dumb_ptr<area_t>::make(a_rect)}; - - return 0; -} - -static -int fun_running_status_update(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - if (ENTITY_TYPE(0) != BL::PC && ENTITY_TYPE(0) != BL::MOB) - return 1; - - StatusChange sc = static_cast<StatusChange>(ARGINT(1)); - *result = ValInt{bool(battle_get_sc_data(ARGENTITY(0))[sc].timer)}; - return 0; -} - -static -int fun_status_option(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{(bool((ARGPC(0))->status.option & static_cast<Opt0>(ARGINT(1))))}; - return 0; -} - -static -int fun_element(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{static_cast<int>(battle_get_element(ARGENTITY(0)).element)}; - return 0; -} - -static -int fun_element_level(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{battle_get_element(ARGENTITY(0)).level}; - return 0; -} - -static -int fun_is_exterior(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ -#warning "Evil assumptions!" - *result = ValInt{ARGLOCATION(0).m->name_[4] == '1'}; - return 0; -} - -static -int fun_contains_string(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{nullptr != strstr(ARGSTR(0).c_str(), ARGSTR(1).c_str())}; - return 0; -} - -static -int fun_strstr(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - const char *offset = strstr(ARGSTR(0).c_str(), ARGSTR(1).c_str()); - *result = ValInt{static_cast<int32_t>(offset - ARGSTR(0).c_str())}; - return offset == nullptr; -} - -static -int fun_strlen(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{static_cast<int32_t>(strlen(ARGSTR(0).c_str()))}; - return 0; -} - -static -int fun_substr(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - RString src = ARGSTR(0); - int offset = ARGINT(1); - int len = ARGINT(2); - - if (len < 0) - len = 0; - if (offset < 0) - offset = 0; - - if (offset > src.size()) - offset = src.size(); - - if (offset + len > src.size()) - len = src.size() - offset; - - auto begin = src.begin() + offset; - auto end = begin + len; - *result = ValString{RString(begin, end)}; - - return 0; -} - -static -int fun_sqrt(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - *result = ValInt{static_cast<int>(sqrt(ARGINT(0)))}; - return 0; -} - -static -int fun_map_level(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ -#warning "Evil assumptions!" - *result = ValInt{ARGLOCATION(0).m->name_[4] - '0'}; - return 0; -} - -static -int fun_map_nr(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ -#warning "Evil assumptions!" - MapName mapname = ARGLOCATION(0).m->name_; - - *result = ValInt{((mapname[0] - '0') * 100) - + ((mapname[1] - '0') * 10) + ((mapname[2] - '0'))}; - return 0; -} - -static -int fun_dir_towards(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - int dx; - int dy; - - if (ARGLOCATION(0).m != ARGLOCATION(1).m) - return 1; - - dx = ARGLOCATION(1).x - ARGLOCATION(0).x; - dy = ARGLOCATION(1).y - ARGLOCATION(0).y; - - if (ARGINT(2)) - { - /* 8-direction mode */ - if (abs(dx) > abs(dy) * 2) - { /* east or west */ - if (dx < 0) - *result = ValDir{DIR::W}; - else - *result = ValDir{DIR::E}; - } - else if (abs(dy) > abs(dx) * 2) - { /* north or south */ - if (dy > 0) - *result = ValDir{DIR::S}; - else - *result = ValDir{DIR::N}; - } - else if (dx < 0) - { /* north-west or south-west */ - if (dy < 0) - *result = ValDir{DIR::NW}; - else - *result = ValDir{DIR::SW}; - } - else - { /* north-east or south-east */ - if (dy < 0) - *result = ValDir{DIR::NE}; - else - *result = ValDir{DIR::SE}; - } - } - else - { - /* 4-direction mode */ - if (abs(dx) > abs(dy)) - { /* east or west */ - if (dx < 0) - *result = ValDir{DIR::W}; - else - *result = ValDir{DIR::E}; - } - else - { /* north or south */ - if (dy > 0) - *result = ValDir{DIR::S}; - else - *result = ValDir{DIR::N}; - } - } - - return 0; -} - -static -int fun_extract_healer_xp(dumb_ptr<env_t>, val_t *result, Slice<val_t> args) -{ - dumb_ptr<map_session_data> sd = (ENTITY_TYPE(0) == BL::PC) ? ARGPC(0) : nullptr; - - if (!sd) - *result = ValInt{0}; - else - *result = ValInt{pc_extract_healer_exp(sd, ARGINT(1))}; - return 0; -} - -#define MAGIC_FUNCTION(name, args, ret, impl) {name, {name, args, ret, impl}} -#define MAGIC_FUNCTION1(name, args, ret) MAGIC_FUNCTION(#name##_s, args, ret, fun_##name) -static // should be LString, but no heterogenous lookup yet -std::map<ZString, fun_t> functions = -{ - MAGIC_FUNCTION("+"_s, ".."_s, '.', fun_add), - MAGIC_FUNCTION("-"_s, "ii"_s, 'i', fun_sub), - MAGIC_FUNCTION("*"_s, "ii"_s, 'i', fun_mul), - MAGIC_FUNCTION("/"_s, "ii"_s, 'i', fun_div), - MAGIC_FUNCTION("%"_s, "ii"_s, 'i', fun_mod), - MAGIC_FUNCTION("||"_s, "ii"_s, 'i', fun_or), - MAGIC_FUNCTION("&&"_s, "ii"_s, 'i', fun_and), - MAGIC_FUNCTION("<"_s, ".."_s, 'i', fun_lt), - MAGIC_FUNCTION(">"_s, ".."_s, 'i', fun_gt), - MAGIC_FUNCTION("<="_s, ".."_s, 'i', fun_lte), - MAGIC_FUNCTION(">="_s, ".."_s, 'i', fun_gte), - MAGIC_FUNCTION("=="_s, ".."_s, 'i', fun_eq), - MAGIC_FUNCTION("!="_s, ".."_s, 'i', fun_ne), - MAGIC_FUNCTION("|"_s, ".."_s, 'i', fun_bitor), - MAGIC_FUNCTION("&"_s, "ii"_s, 'i', fun_bitand), - MAGIC_FUNCTION("^"_s, "ii"_s, 'i', fun_bitxor), - MAGIC_FUNCTION("<<"_s, "ii"_s, 'i', fun_bitshl), - MAGIC_FUNCTION(">>"_s, "ii"_s, 'i', fun_bitshr), - MAGIC_FUNCTION1(not, "i"_s, 'i'), - MAGIC_FUNCTION1(neg, "i"_s, 'i'), - MAGIC_FUNCTION1(max, "ii"_s, 'i'), - MAGIC_FUNCTION1(min, "ii"_s, 'i'), - MAGIC_FUNCTION1(is_in, "la"_s, 'i'), - MAGIC_FUNCTION1(if_then_else, "i__"_s, '_'), - MAGIC_FUNCTION1(skill, "ei"_s, 'i'), - MAGIC_FUNCTION("str"_s, "e"_s, 'i', fun_get_str), - MAGIC_FUNCTION("agi"_s, "e"_s, 'i', fun_get_agi), - MAGIC_FUNCTION("vit"_s, "e"_s, 'i', fun_get_vit), - MAGIC_FUNCTION("dex"_s, "e"_s, 'i', fun_get_dex), - MAGIC_FUNCTION("luk"_s, "e"_s, 'i', fun_get_luk), - MAGIC_FUNCTION("int"_s, "e"_s, 'i', fun_get_int), - MAGIC_FUNCTION("level"_s, "e"_s, 'i', fun_get_lv), - MAGIC_FUNCTION("mdef"_s, "e"_s, 'i', fun_get_mdef), - MAGIC_FUNCTION("def"_s, "e"_s, 'i', fun_get_def), - MAGIC_FUNCTION("hp"_s, "e"_s, 'i', fun_get_hp), - MAGIC_FUNCTION("max_hp"_s, "e"_s, 'i', fun_get_max_hp), - MAGIC_FUNCTION("sp"_s, "e"_s, 'i', fun_get_sp), - MAGIC_FUNCTION("max_sp"_s, "e"_s, 'i', fun_get_max_sp), - MAGIC_FUNCTION("dir"_s, "e"_s, 'd', fun_get_dir), - MAGIC_FUNCTION1(name_of, "."_s, 's'), - MAGIC_FUNCTION1(mob_id, "e"_s, 'i'), - MAGIC_FUNCTION1(location, "e"_s, 'l'), - MAGIC_FUNCTION1(random, "i"_s, 'i'), - MAGIC_FUNCTION1(random_dir, "i"_s, 'd'), - MAGIC_FUNCTION1(hash_entity, "e"_s, 'i'), - MAGIC_FUNCTION1(is_married, "e"_s, 'i'), - MAGIC_FUNCTION1(partner, "e"_s, 'e'), - MAGIC_FUNCTION1(awayfrom, "ldi"_s, 'l'), - MAGIC_FUNCTION1(failed, "_"_s, 'i'), - MAGIC_FUNCTION1(pc, "s"_s, 'e'), - MAGIC_FUNCTION1(npc, "s"_s, 'e'), - MAGIC_FUNCTION1(distance, "ll"_s, 'i'), - MAGIC_FUNCTION1(rdistance, "ll"_s, 'i'), - MAGIC_FUNCTION1(anchor, "s"_s, 'a'), - MAGIC_FUNCTION("random_location"_s, "a"_s, 'l', fun_pick_location), - MAGIC_FUNCTION("script_int"_s, "es"_s, 'i', fun_read_script_int), - MAGIC_FUNCTION("script_str"_s, "es"_s, 's', fun_read_script_str), - MAGIC_FUNCTION1(rbox, "li"_s, 'a'), - MAGIC_FUNCTION1(count_item, "e."_s, 'i'), - MAGIC_FUNCTION1(line_of_sight, "ll"_s, 'i'), - MAGIC_FUNCTION1(running_status_update, "ei"_s, 'i'), - MAGIC_FUNCTION1(status_option, "ei"_s, 'i'), - MAGIC_FUNCTION1(element, "e"_s, 'i'), - MAGIC_FUNCTION1(element_level, "e"_s, 'i'), - MAGIC_FUNCTION1(his_shroud, "e"_s, 'i'), - MAGIC_FUNCTION1(is_equipped, "e."_s, 'i'), - MAGIC_FUNCTION1(is_exterior, "l"_s, 'i'), - MAGIC_FUNCTION1(contains_string, "ss"_s, 'i'), - MAGIC_FUNCTION1(strstr, "ss"_s, 'i'), - MAGIC_FUNCTION1(strlen, "s"_s, 'i'), - MAGIC_FUNCTION1(substr, "sii"_s, 's'), - MAGIC_FUNCTION1(sqrt, "i"_s, 'i'), - MAGIC_FUNCTION1(map_level, "l"_s, 'i'), - MAGIC_FUNCTION1(map_nr, "l"_s, 'i'), - MAGIC_FUNCTION1(dir_towards, "lli"_s, 'd'), - MAGIC_FUNCTION1(is_dead, "e"_s, 'i'), - MAGIC_FUNCTION1(is_pc, "e"_s, 'i'), - MAGIC_FUNCTION("extract_healer_experience"_s, "ei"_s, 'i', fun_extract_healer_xp), -}; - -fun_t *magic_get_fun(ZString name) -{ - auto it = functions.find(name); - if (it == functions.end()) - return nullptr; - return &it->second; -} - -// 1 on failure -static -int eval_location(dumb_ptr<env_t> env, location_t *dest, const e_location_t *expr) -{ - val_t m, x, y; - magic_eval(env, &m, expr->m); - magic_eval(env, &x, expr->x); - magic_eval(env, &y, expr->y); - - if (m.is<ValString>() - && x.is<ValInt>() && y.is<ValInt>()) - { - MapName name = VString<15>(ZString(m.get_if<ValString>()->v_string)); - magic_clear_var(&m); - P<map_local> map_id = TRY_UNWRAP(map_mapname2mapid(name), return 1); - dest->m = map_id; - dest->x = x.get_if<ValInt>()->v_int; - dest->y = y.get_if<ValInt>()->v_int; - return 0; - } - else - { - magic_clear_var(&m); - magic_clear_var(&x); - magic_clear_var(&y); - return 1; - } -} - -static -dumb_ptr<area_t> eval_area(dumb_ptr<env_t> env, const e_area_t& expr_) -{ - MATCH_BEGIN (expr_) - { - MATCH_CASE (const e_location_t&, a_loc) - { - location_t loc; - if (eval_location(env, &loc, &a_loc)) - { - return nullptr; - } - else - { - return dumb_ptr<area_t>::make(loc); - } - } - MATCH_CASE (const ExprAreaUnion&, a) - { - AreaUnion u; - bool fail = false; - for (int i = 0; i < 2; i++) - { - u.a_union[i] = eval_area(env, *a.a_union[i]); - if (!u.a_union[i]) - fail = true; - } - - if (fail) - { - for (int i = 0; i < 2; i++) - { - if (u.a_union[i]) - free_area(u.a_union[i]); - } - return nullptr; - } - return dumb_ptr<area_t>::make(u); - } - MATCH_CASE (const ExprAreaRect&, a_rect) - { - val_t width, height; - magic_eval(env, &width, a_rect.width); - magic_eval(env, &height, a_rect.height); - - AreaRect a_rect_; - if (width.is<ValInt>() - && height.is<ValInt>() - && !eval_location(env, &(a_rect_.loc), - &a_rect.loc)) - { - a_rect_.width = width.get_if<ValInt>()->v_int; - a_rect_.height = height.get_if<ValInt>()->v_int; - - magic_clear_var(&width); - magic_clear_var(&height); - return dumb_ptr<area_t>::make(a_rect_); - } - else - { - magic_clear_var(&width); - magic_clear_var(&height); - return nullptr; - } - } - MATCH_CASE (const ExprAreaBar&, a_bar) - { - val_t width, depth, dir; - magic_eval(env, &width, a_bar.width); - magic_eval(env, &depth, a_bar.depth); - magic_eval(env, &dir, a_bar.dir); - - AreaBar a_bar_; - if (width.is<ValInt>() - && depth.is<ValInt>() - && dir.is<ValDir>() - && !eval_location(env, &a_bar_.loc, - &a_bar.loc)) - { - a_bar_.width = width.get_if<ValInt>()->v_int; - a_bar_.depth = depth.get_if<ValInt>()->v_int; - a_bar_.dir = dir.get_if<ValDir>()->v_dir; - - magic_clear_var(&width); - magic_clear_var(&depth); - magic_clear_var(&dir); - return dumb_ptr<area_t>::make(a_bar_); - } - else - { - magic_clear_var(&width); - magic_clear_var(&depth); - magic_clear_var(&dir); - return nullptr; - } - } - } - MATCH_END (); - abort(); -} - -// This is called on arguments with begin=true, -// and on the return value with begin=false. -// In both cases, the ambiguous types are in pointer mode. -static -bool type_key_matches(char ty_key, val_t *arg, bool begin) -{ - switch (ty_key) - { - case 'i': - if (begin) - intify(arg); - return arg->is<ValInt>(); - case 'd': - return arg->is<ValDir>(); - case 's': - if (begin) - stringify(arg); - return arg->is<ValString>(); - case 'e': - return arg->is<ValEntityPtr>(); - case 'l': - if (begin) - make_location(arg); - return arg->is<ValLocation>(); - case 'a': - if (begin) - make_area(arg); - return arg->is<ValArea>(); - case 'S': - if (begin) - make_spell(arg); - return arg->is<ValSpell>(); - case 'I': - return arg->is<ValInvocationPtr>(); - default: - return true; - } -} - -int magic_signature_check(ZString opname, ZString funname, ZString signature, - Slice<val_t> args, int line, int column) -{ - int i; - for (i = 0; i < args.size(); i++) - { - val_t *arg = &args[i]; - - // whoa, it turns out the second p *does* shadow this one - if (ValEntityInt *p1 = arg->get_if<ValEntityInt>()) - { - /* Dereference entities in preparation for calling function */ - dumb_ptr<block_list> ent = map_id2bl(p1->v_eid); - if (ent) - { - *arg = ValEntityPtr{ent}; - } - else - { - *arg = ValFail{}; - } - } - else if (ValInvocationInt *p2 = arg->get_if<ValInvocationInt>()) - { - dumb_ptr<invocation> invoc = map_id2bl(p2->v_iid)->is_spell(); - if (invoc) - { - *arg = ValInvocationPtr{invoc}; - } - else - { - *arg = ValFail(); - } - } - - char ty_key = signature[i]; - if (!ty_key) - { - FPRINTF(stderr, - "[magic-eval]: L%d:%d: Too many arguments (%zu) to %s `%s'\n"_fmt, - line, column, args.size(), opname, funname); - return 1; - } - - if (arg->is<ValFail>() && ty_key != '_') - return 1; /* Fail `in a sane way': This is a perfectly permissible error */ - - // this also does conversions now - if (type_key_matches(ty_key, arg, true)) - continue; - - if (arg->is<ValUndef>()) - { - FPRINTF(stderr, - "[magic-eval]: L%d:%d: Argument #%d to %s `%s' undefined\n"_fmt, - line, column, i + 1, opname, funname); - return 1; - } - - - { /* Coercion failed? */ - if (!arg->is<ValFail>()) - { - FPRINTF(stderr, - "[magic-eval]: L%d:%d: Argument #%d to %s `%s' of incorrect type (sorry, types aren't integers anymore)\n"_fmt, - line, column, i + 1, opname, funname); - } - return 1; - } - } - - return 0; -} - -void magic_eval(dumb_ptr<env_t> env, val_t *dest, dumb_ptr<expr_t> expr) -{ - MATCH_BEGIN (*expr) - { - MATCH_CASE (const val_t&, e_val) - { - magic_copy_var(dest, &e_val); - } - - MATCH_CASE (const e_location_t&, e_location) - { - location_t loc; - if (eval_location(env, &loc, &e_location)) - *dest = ValFail(); - else - *dest = ValLocation{loc}; - } - MATCH_CASE (const e_area_t&, e_area) - { - if (dumb_ptr<area_t> area = eval_area(env, e_area)) - *dest = ValArea{area}; - else - *dest = ValFail(); - } - MATCH_CASE (const ExprFunApp&, e_funapp) - { - val_t arguments[MAX_ARGS]; - int args_nr = e_funapp.args_nr; - int i; - fun_t *f = e_funapp.funp; - - for (i = 0; i < args_nr; ++i) - magic_eval(env, &arguments[i], e_funapp.args[i]); - if (magic_signature_check("function"_s, f->name, f->signature, Slice<val_t>(arguments, args_nr), - e_funapp.line_nr, e_funapp.column) - || f->fun(env, dest, Slice<val_t>(arguments, args_nr))) - *dest = ValFail(); - else - { - assert (!dest->is<ValInvocationPtr>()); - assert (!dest->is<ValInvocationInt>()); - assert (!dest->is<ValEntityInt>()); - assert (type_key_matches(f->ret_ty, dest, false)); - - /* translate entity back into persistent int */ - if (ValEntityPtr *ent = dest->get_if<ValEntityPtr>()) - { - if (ent->v_entity) - *dest = ValEntityInt{ent->v_entity->bl_id}; - else - *dest = ValFail(); - } - // what about invocation? - } - - for (i = 0; i < args_nr; ++i) - magic_clear_var(&arguments[i]); - } - MATCH_CASE (const ExprId&, e) - { - val_t& v = env->VAR(e.e_id); - magic_copy_var(dest, &v); - } - MATCH_CASE (const ExprField&, e_field) - { - val_t v; - int id = e_field.id; - magic_eval(env, &v, e_field.expr); - - assert(!v.is<ValInvocationPtr>()); - if (ValInvocationInt *ii = v.get_if<ValInvocationInt>()) - { - dumb_ptr<invocation> t = map_id2bl(ii->v_iid)->is_spell(); - - if (!t) - *dest = ValUndef(); - else - { - val_t& val = t->env->VAR(id); - magic_copy_var(dest, &val); - } - } - else - { - FPRINTF(stderr, - "[magic] Attempt to access field %s on non-spell\n"_fmt, - env->base_env->varv[id].name); - *dest = ValFail(); - } - } - } - MATCH_END (); -} - -int magic_eval_int(dumb_ptr<env_t> env, dumb_ptr<expr_t> expr) -{ - val_t result; - magic_eval(env, &result, expr); - - if (result.is<ValFail>() || result.is<ValUndef>()) - return 0; - - intify(&result); - - return result.get_if<ValInt>()->v_int; -} - -AString magic_eval_str(dumb_ptr<env_t> env, dumb_ptr<expr_t> expr) -{ - val_t result; - magic_eval(env, &result, expr); - - if (result.is<ValFail>() || result.is<ValUndef>()) - return "?"_s; - - stringify(&result); - - return result.get_if<ValString>()->v_string; -} -} // namespace magic -} // namespace map -} // namespace tmwa |