1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
/*
* File: l_dgnit.cc
* Summary: Item-related functions in lua library "dgn".
*/
#include "AppHdr.h"
#include "cluautil.h"
#include "l_libs.h"
#include "coord.h"
#include "dungeon.h"
#include "env.h"
#include "items.h"
#include "mapdef.h"
#include "stash.h"
#define ITEMLIST_METATABLE "crawldgn.item_list"
static item_list _lua_get_ilist(lua_State *ls, int ndx)
{
if (lua_isstring(ls, ndx))
{
const char *spec = lua_tostring(ls, ndx);
item_list ilist;
const std::string err = ilist.add_item(spec);
if (!err.empty())
luaL_error(ls, err.c_str());
return (ilist);
}
else
{
item_list **ilist =
clua_get_userdata<item_list*>(ls, ITEMLIST_METATABLE, ndx);
if (ilist)
return (**ilist);
luaL_argerror(ls, ndx, "Expected item list object or string");
return item_list();
}
}
void register_itemlist(lua_State *ls)
{
clua_register_metatable(ls, ITEMLIST_METATABLE, NULL,
lua_object_gc<item_list>);
}
static int dgn_item_from_index(lua_State *ls)
{
const int index = luaL_checkint(ls, 1);
item_def *item = &mitm[index];
if (item->is_valid())
lua_pushlightuserdata(ls, item);
else
lua_pushnil(ls);
return (1);
}
static int dgn_items_at(lua_State *ls)
{
COORDS(c, 1, 2);
lua_push_items(ls, env.igrid(c));
return (1);
}
static int _dgn_item_spec(lua_State *ls)
{
const item_list ilist = _lua_get_ilist(ls, 1);
dlua_push_object_type<item_list>(ls, ITEMLIST_METATABLE, ilist);
return (1);
}
static int dgn_create_item(lua_State *ls)
{
COORDS(c, 1, 2);
item_list ilist = _lua_get_ilist(ls, 3);
const int level =
lua_isnumber(ls, 4) ? lua_tointeger(ls, 4) : you.your_level;
dgn_place_multiple_items(ilist, c, level);
link_items();
return (0);
}
// Returns two arrays: one of floor items, one of shop items.
static int dgn_stash_items(lua_State *ls)
{
unsigned min_value = lua_isnumber(ls, 1) ? luaL_checkint(ls, 1) : 0;
bool skip_stackable = lua_isboolean(ls, 2) ? lua_toboolean(ls, 2)
: false;
std::vector<const item_def*> floor_items;
std::vector<const item_def*> shop_items;
for (ST_ItemIterator stii; stii; ++stii)
{
if (skip_stackable && is_stackable_item(*stii))
continue;
if (min_value > 0)
{
if (stii.shop())
{
if (stii.price() < min_value)
continue;
}
else if (item_value(*stii, true) < min_value)
continue;
}
if (stii.shop())
shop_items.push_back(&(*stii));
else
floor_items.push_back(&(*stii));
}
lua_newtable(ls);
int index = 0;
for (unsigned int i = 0; i < floor_items.size(); i++)
{
lua_pushlightuserdata(ls, const_cast<item_def*>(floor_items[i]));
lua_rawseti(ls, -2, ++index);
}
lua_newtable(ls);
index = 0;
for (unsigned int i = 0; i < shop_items.size(); i++)
{
lua_pushlightuserdata(ls, const_cast<item_def*>(floor_items[i]));
lua_rawseti(ls, -2, ++index);
}
return (2);
}
const struct luaL_reg dgn_item_dlib[] =
{
{ "item_from_index", dgn_item_from_index },
{ "items_at", dgn_items_at },
{ "create_item", dgn_create_item },
{ "item_spec", _dgn_item_spec },
{ "stash_items", dgn_stash_items },
{ NULL, NULL }
};
|
