]> git.lizzy.rs Git - dragonfireclient.git/blob - src/script/lua_api/l_mapgen.cpp
Remove wstrgettext
[dragonfireclient.git] / src / script / lua_api / l_mapgen.cpp
1 /*
2 Minetest
3 Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU Lesser General Public License as published by
7 the Free Software Foundation; either version 2.1 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 GNU Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public License along
16 with this program; if not, write to the Free Software Foundation, Inc.,
17 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "lua_api/l_mapgen.h"
21 #include "lua_api/l_internal.h"
22 #include "lua_api/l_vmanip.h"
23 #include "common/c_converter.h"
24 #include "common/c_content.h"
25 #include "cpp_api/s_security.h"
26 #include "util/serialize.h"
27 #include "server.h"
28 #include "environment.h"
29 #include "emerge.h"
30 #include "mg_biome.h"
31 #include "mg_ore.h"
32 #include "mg_decoration.h"
33 #include "mg_schematic.h"
34 #include "mapgen_v5.h"
35 #include "mapgen_v7.h"
36 #include "filesys.h"
37 #include "settings.h"
38 #include "log.h"
39
40 struct EnumString ModApiMapgen::es_BiomeTerrainType[] =
41 {
42         {BIOME_NORMAL, "normal"},
43         {BIOME_LIQUID, "liquid"},
44         {BIOME_NETHER, "nether"},
45         {BIOME_AETHER, "aether"},
46         {BIOME_FLAT,   "flat"},
47         {0, NULL},
48 };
49
50 struct EnumString ModApiMapgen::es_DecorationType[] =
51 {
52         {DECO_SIMPLE,    "simple"},
53         {DECO_SCHEMATIC, "schematic"},
54         {DECO_LSYSTEM,   "lsystem"},
55         {0, NULL},
56 };
57
58 struct EnumString ModApiMapgen::es_MapgenObject[] =
59 {
60         {MGOBJ_VMANIP,    "voxelmanip"},
61         {MGOBJ_HEIGHTMAP, "heightmap"},
62         {MGOBJ_BIOMEMAP,  "biomemap"},
63         {MGOBJ_HEATMAP,   "heatmap"},
64         {MGOBJ_HUMIDMAP,  "humiditymap"},
65         {MGOBJ_GENNOTIFY, "gennotify"},
66         {0, NULL},
67 };
68
69 struct EnumString ModApiMapgen::es_OreType[] =
70 {
71         {ORE_SCATTER, "scatter"},
72         {ORE_SHEET,   "sheet"},
73         {ORE_PUFF,    "puff"},
74         {ORE_BLOB,    "blob"},
75         {ORE_VEIN,    "vein"},
76         {0, NULL},
77 };
78
79 struct EnumString ModApiMapgen::es_Rotation[] =
80 {
81         {ROTATE_0,    "0"},
82         {ROTATE_90,   "90"},
83         {ROTATE_180,  "180"},
84         {ROTATE_270,  "270"},
85         {ROTATE_RAND, "random"},
86         {0, NULL},
87 };
88
89 struct EnumString ModApiMapgen::es_SchematicFormatType[] =
90 {
91         {SCHEM_FMT_HANDLE, "handle"},
92         {SCHEM_FMT_MTS,    "mts"},
93         {SCHEM_FMT_LUA,    "lua"},
94         {0, NULL},
95 };
96
97 ObjDef *get_objdef(lua_State *L, int index, ObjDefManager *objmgr);
98
99 Biome *get_or_load_biome(lua_State *L, int index,
100         BiomeManager *biomemgr);
101 Biome *read_biome_def(lua_State *L, int index, INodeDefManager *ndef);
102 size_t get_biome_list(lua_State *L, int index,
103         BiomeManager *biomemgr, std::set<u8> *biome_id_list);
104
105 Schematic *get_or_load_schematic(lua_State *L, int index,
106         SchematicManager *schemmgr, StringMap *replace_names);
107 Schematic *load_schematic(lua_State *L, int index, INodeDefManager *ndef,
108         StringMap *replace_names);
109 Schematic *load_schematic_from_def(lua_State *L, int index,
110         INodeDefManager *ndef, StringMap *replace_names);
111 bool read_schematic_def(lua_State *L, int index,
112         Schematic *schem, std::vector<std::string> *names);
113
114 bool read_deco_simple(lua_State *L, DecoSimple *deco);
115 bool read_deco_schematic(lua_State *L, SchematicManager *schemmgr, DecoSchematic *deco);
116
117
118 ///////////////////////////////////////////////////////////////////////////////
119
120 ObjDef *get_objdef(lua_State *L, int index, ObjDefManager *objmgr)
121 {
122         if (index < 0)
123                 index = lua_gettop(L) + 1 + index;
124
125         // If a number, assume this is a handle to an object def
126         if (lua_isnumber(L, index))
127                 return objmgr->get(lua_tointeger(L, index));
128
129         // If a string, assume a name is given instead
130         if (lua_isstring(L, index))
131                 return objmgr->getByName(lua_tostring(L, index));
132
133         return NULL;
134 }
135
136 ///////////////////////////////////////////////////////////////////////////////
137
138 Schematic *get_or_load_schematic(lua_State *L, int index,
139         SchematicManager *schemmgr, StringMap *replace_names)
140 {
141         if (index < 0)
142                 index = lua_gettop(L) + 1 + index;
143
144         Schematic *schem = (Schematic *)get_objdef(L, index, schemmgr);
145         if (schem)
146                 return schem;
147
148         schem = load_schematic(L, index, schemmgr->getNodeDef(),
149                 replace_names);
150         if (!schem)
151                 return NULL;
152
153         if (schemmgr->add(schem) == OBJDEF_INVALID_HANDLE) {
154                 delete schem;
155                 return NULL;
156         }
157
158         return schem;
159 }
160
161
162 Schematic *load_schematic(lua_State *L, int index, INodeDefManager *ndef,
163         StringMap *replace_names)
164 {
165         if (index < 0)
166                 index = lua_gettop(L) + 1 + index;
167
168         Schematic *schem = NULL;
169
170         if (lua_istable(L, index)) {
171                 schem = load_schematic_from_def(L, index, ndef,
172                         replace_names);
173                 if (!schem) {
174                         delete schem;
175                         return NULL;
176                 }
177         } else if (lua_isnumber(L, index)) {
178                 return NULL;
179         } else if (lua_isstring(L, index)) {
180                 schem = SchematicManager::create(SCHEMATIC_NORMAL);
181
182                 std::string filepath = lua_tostring(L, index);
183                 if (!fs::IsPathAbsolute(filepath))
184                         filepath = ModApiBase::getCurrentModPath(L) + DIR_DELIM + filepath;
185
186                 if (!schem->loadSchematicFromFile(filepath, ndef,
187                                 replace_names)) {
188                         delete schem;
189                         return NULL;
190                 }
191         }
192
193         return schem;
194 }
195
196
197 Schematic *load_schematic_from_def(lua_State *L, int index,
198         INodeDefManager *ndef, StringMap *replace_names)
199 {
200         Schematic *schem = SchematicManager::create(SCHEMATIC_NORMAL);
201
202         if (!read_schematic_def(L, index, schem, &schem->m_nodenames)) {
203                 delete schem;
204                 return NULL;
205         }
206
207         size_t num_nodes = schem->m_nodenames.size();
208
209         schem->m_nnlistsizes.push_back(num_nodes);
210
211         if (replace_names) {
212                 for (size_t i = 0; i != num_nodes; i++) {
213                         StringMap::iterator it = replace_names->find(schem->m_nodenames[i]);
214                         if (it != replace_names->end())
215                                 schem->m_nodenames[i] = it->second;
216                 }
217         }
218
219         if (ndef)
220                 ndef->pendNodeResolve(schem);
221
222         return schem;
223 }
224
225
226 bool read_schematic_def(lua_State *L, int index,
227         Schematic *schem, std::vector<std::string> *names)
228 {
229         if (!lua_istable(L, index))
230                 return false;
231
232         //// Get schematic size
233         lua_getfield(L, index, "size");
234         v3s16 size = check_v3s16(L, -1);
235         lua_pop(L, 1);
236
237         schem->size = size;
238
239         //// Get schematic data
240         lua_getfield(L, index, "data");
241         luaL_checktype(L, -1, LUA_TTABLE);
242
243         u32 numnodes = size.X * size.Y * size.Z;
244         schem->schemdata = new MapNode[numnodes];
245
246         size_t names_base = names->size();
247         std::map<std::string, content_t> name_id_map;
248
249         u32 i = 0;
250         for (lua_pushnil(L); lua_next(L, -2); i++, lua_pop(L, 1)) {
251                 if (i >= numnodes)
252                         continue;
253
254                 //// Read name
255                 std::string name;
256                 if (!getstringfield(L, -1, "name", name))
257                         throw LuaError("Schematic data definition with missing name field");
258
259                 //// Read param1/prob
260                 u8 param1;
261                 if (!getintfield(L, -1, "param1", param1) &&
262                         !getintfield(L, -1, "prob", param1))
263                         param1 = MTSCHEM_PROB_ALWAYS_OLD;
264
265                 //// Read param2
266                 u8 param2 = getintfield_default(L, -1, "param2", 0);
267
268                 //// Find or add new nodename-to-ID mapping
269                 std::map<std::string, content_t>::iterator it = name_id_map.find(name);
270                 content_t name_index;
271                 if (it != name_id_map.end()) {
272                         name_index = it->second;
273                 } else {
274                         name_index = names->size() - names_base;
275                         name_id_map[name] = name_index;
276                         names->push_back(name);
277                 }
278
279                 //// Perform probability/force_place fixup on param1
280                 param1 >>= 1;
281                 if (getboolfield_default(L, -1, "force_place", false))
282                         param1 |= MTSCHEM_FORCE_PLACE;
283
284                 //// Actually set the node in the schematic
285                 schem->schemdata[i] = MapNode(name_index, param1, param2);
286         }
287
288         if (i != numnodes) {
289                 errorstream << "read_schematic_def: incorrect number of "
290                         "nodes provided in raw schematic data (got " << i <<
291                         ", expected " << numnodes << ")." << std::endl;
292                 return false;
293         }
294
295         //// Get Y-slice probability values (if present)
296         schem->slice_probs = new u8[size.Y];
297         for (i = 0; i != (u32) size.Y; i++)
298                 schem->slice_probs[i] = MTSCHEM_PROB_ALWAYS;
299
300         lua_getfield(L, index, "yslice_prob");
301         if (lua_istable(L, -1)) {
302                 for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
303                         u16 ypos;
304                         if (!getintfield(L, -1, "ypos", ypos) || (ypos >= size.Y) ||
305                                 !getintfield(L, -1, "prob", schem->slice_probs[ypos]))
306                                 continue;
307
308                         schem->slice_probs[ypos] >>= 1;
309                 }
310         }
311
312         return true;
313 }
314
315
316 void read_schematic_replacements(lua_State *L, int index, StringMap *replace_names)
317 {
318         if (index < 0)
319                 index = lua_gettop(L) + 1 + index;
320
321         lua_pushnil(L);
322         while (lua_next(L, index)) {
323                 std::string replace_from;
324                 std::string replace_to;
325
326                 if (lua_istable(L, -1)) { // Old {{"x", "y"}, ...} format
327                         lua_rawgeti(L, -1, 1);
328                         replace_from = lua_tostring(L, -1);
329                         lua_pop(L, 1);
330
331                         lua_rawgeti(L, -1, 2);
332                         replace_to = lua_tostring(L, -1);
333                         lua_pop(L, 1);
334                 } else { // New {x = "y", ...} format
335                         replace_from = lua_tostring(L, -2);
336                         replace_to = lua_tostring(L, -1);
337                 }
338
339                 replace_names->insert(std::make_pair(replace_from, replace_to));
340                 lua_pop(L, 1);
341         }
342 }
343
344 ///////////////////////////////////////////////////////////////////////////////
345
346 Biome *get_or_load_biome(lua_State *L, int index, BiomeManager *biomemgr)
347 {
348         if (index < 0)
349                 index = lua_gettop(L) + 1 + index;
350
351         Biome *biome = (Biome *)get_objdef(L, index, biomemgr);
352         if (biome)
353                 return biome;
354
355         biome = read_biome_def(L, index, biomemgr->getNodeDef());
356         if (!biome)
357                 return NULL;
358
359         if (biomemgr->add(biome) == OBJDEF_INVALID_HANDLE) {
360                 delete biome;
361                 return NULL;
362         }
363
364         return biome;
365 }
366
367
368 Biome *read_biome_def(lua_State *L, int index, INodeDefManager *ndef)
369 {
370         if (!lua_istable(L, index))
371                 return NULL;
372
373         BiomeType biometype = (BiomeType)getenumfield(L, index, "type",
374                 ModApiMapgen::es_BiomeTerrainType, BIOME_NORMAL);
375         Biome *b = BiomeManager::create(biometype);
376
377         b->name            = getstringfield_default(L, index, "name", "");
378         b->depth_top       = getintfield_default(L,    index, "depth_top",       0);
379         b->depth_filler    = getintfield_default(L,    index, "depth_filler",    -31000);
380         b->depth_water_top = getintfield_default(L,    index, "depth_water_top", 0);
381         b->y_min           = getintfield_default(L,    index, "y_min",           -31000);
382         b->y_max           = getintfield_default(L,    index, "y_max",           31000);
383         b->heat_point      = getfloatfield_default(L,  index, "heat_point",      0.f);
384         b->humidity_point  = getfloatfield_default(L,  index, "humidity_point",  0.f);
385         b->flags           = 0; //reserved
386
387         std::vector<std::string> &nn = b->m_nodenames;
388         nn.push_back(getstringfield_default(L, index, "node_top",         ""));
389         nn.push_back(getstringfield_default(L, index, "node_filler",      ""));
390         nn.push_back(getstringfield_default(L, index, "node_stone",       ""));
391         nn.push_back(getstringfield_default(L, index, "node_water_top",   ""));
392         nn.push_back(getstringfield_default(L, index, "node_water",       ""));
393         nn.push_back(getstringfield_default(L, index, "node_river_water", ""));
394         nn.push_back(getstringfield_default(L, index, "node_dust",        ""));
395         ndef->pendNodeResolve(b);
396
397         return b;
398 }
399
400
401 size_t get_biome_list(lua_State *L, int index,
402         BiomeManager *biomemgr, std::set<u8> *biome_id_list)
403 {
404         if (index < 0)
405                 index = lua_gettop(L) + 1 + index;
406
407         if (lua_isnil(L, index))
408                 return 0;
409
410         bool is_single = true;
411         if (lua_istable(L, index)) {
412                 lua_getfield(L, index, "name");
413                 is_single = !lua_isnil(L, -1);
414                 lua_pop(L, 1);
415         }
416
417         if (is_single) {
418                 Biome *biome = get_or_load_biome(L, index, biomemgr);
419                 if (!biome) {
420                         errorstream << "get_biome_list: failed to get biome '"
421                                 << (lua_isstring(L, index) ? lua_tostring(L, index) : "")
422                                 << "'." << std::endl;
423                         return 1;
424                 }
425
426                 biome_id_list->insert(biome->index);
427                 return 0;
428         }
429
430         // returns number of failed resolutions
431         size_t fail_count = 0;
432         size_t count = 0;
433
434         for (lua_pushnil(L); lua_next(L, index); lua_pop(L, 1)) {
435                 count++;
436                 Biome *biome = get_or_load_biome(L, -1, biomemgr);
437                 if (!biome) {
438                         fail_count++;
439                         errorstream << "get_biome_list: failed to get biome '"
440                                 << (lua_isstring(L, -1) ? lua_tostring(L, -1) : "")
441                                 << "'" << std::endl;
442                         continue;
443                 }
444
445                 biome_id_list->insert(biome->index);
446         }
447
448         return fail_count;
449 }
450
451 ///////////////////////////////////////////////////////////////////////////////
452
453 // get_biome_id(biomename)
454 // returns the biome id used in biomemap
455 int ModApiMapgen::l_get_biome_id(lua_State *L)
456 {
457         const char *biome_str = lua_tostring(L, 1);
458         if (!biome_str)
459                 return 0;
460
461         BiomeManager *bmgr = getServer(L)->getEmergeManager()->biomemgr;
462
463         if (!bmgr)
464                 return 0;
465
466         Biome *biome = (Biome *) bmgr->getByName(biome_str);
467
468         if (!biome || biome->index == OBJDEF_INVALID_INDEX)
469                 return 0;
470
471         lua_pushinteger(L, biome->index);
472
473         return 1;
474 }
475
476
477 // get_mapgen_object(objectname)
478 // returns the requested object used during map generation
479 int ModApiMapgen::l_get_mapgen_object(lua_State *L)
480 {
481         const char *mgobjstr = lua_tostring(L, 1);
482
483         int mgobjint;
484         if (!string_to_enum(es_MapgenObject, mgobjint, mgobjstr ? mgobjstr : ""))
485                 return 0;
486
487         enum MapgenObject mgobj = (MapgenObject)mgobjint;
488
489         EmergeManager *emerge = getServer(L)->getEmergeManager();
490         Mapgen *mg = emerge->getCurrentMapgen();
491         if (!mg)
492                 return 0;
493
494         size_t maplen = mg->csize.X * mg->csize.Z;
495
496         switch (mgobj) {
497         case MGOBJ_VMANIP: {
498                 MMVManip *vm = mg->vm;
499
500                 // VoxelManip object
501                 LuaVoxelManip *o = new LuaVoxelManip(vm, true);
502                 *(void **)(lua_newuserdata(L, sizeof(void *))) = o;
503                 luaL_getmetatable(L, "VoxelManip");
504                 lua_setmetatable(L, -2);
505
506                 // emerged min pos
507                 push_v3s16(L, vm->m_area.MinEdge);
508
509                 // emerged max pos
510                 push_v3s16(L, vm->m_area.MaxEdge);
511
512                 return 3;
513         }
514         case MGOBJ_HEIGHTMAP: {
515                 if (!mg->heightmap)
516                         return 0;
517
518                 lua_newtable(L);
519                 for (size_t i = 0; i != maplen; i++) {
520                         lua_pushinteger(L, mg->heightmap[i]);
521                         lua_rawseti(L, -2, i + 1);
522                 }
523
524                 return 1;
525         }
526         case MGOBJ_BIOMEMAP: {
527                 if (!mg->biomemap)
528                         return 0;
529
530                 lua_newtable(L);
531                 for (size_t i = 0; i != maplen; i++) {
532                         lua_pushinteger(L, mg->biomemap[i]);
533                         lua_rawseti(L, -2, i + 1);
534                 }
535
536                 return 1;
537         }
538         case MGOBJ_HEATMAP: {
539                 if (!mg->heatmap)
540                         return 0;
541
542                 lua_newtable(L);
543                 for (size_t i = 0; i != maplen; i++) {
544                         lua_pushnumber(L, mg->heatmap[i]);
545                         lua_rawseti(L, -2, i + 1);
546                 }
547
548                 return 1;
549         }
550
551         case MGOBJ_HUMIDMAP: {
552                 if (!mg->humidmap)
553                         return 0;
554
555                 lua_newtable(L);
556                 for (size_t i = 0; i != maplen; i++) {
557                         lua_pushnumber(L, mg->humidmap[i]);
558                         lua_rawseti(L, -2, i + 1);
559                 }
560
561                 return 1;
562         }
563         case MGOBJ_GENNOTIFY: {
564                 std::map<std::string, std::vector<v3s16> >event_map;
565                 std::map<std::string, std::vector<v3s16> >::iterator it;
566
567                 mg->gennotify.getEvents(event_map);
568
569                 lua_newtable(L);
570                 for (it = event_map.begin(); it != event_map.end(); ++it) {
571                         lua_newtable(L);
572
573                         for (size_t j = 0; j != it->second.size(); j++) {
574                                 push_v3s16(L, it->second[j]);
575                                 lua_rawseti(L, -2, j + 1);
576                         }
577
578                         lua_setfield(L, -2, it->first.c_str());
579                 }
580
581                 return 1;
582         }
583         }
584
585         return 0;
586 }
587
588
589 int ModApiMapgen::l_get_mapgen_params(lua_State *L)
590 {
591         MapgenParams *params = &getServer(L)->getEmergeManager()->params;
592
593         lua_newtable(L);
594
595         lua_pushstring(L, params->mg_name.c_str());
596         lua_setfield(L, -2, "mgname");
597
598         lua_pushinteger(L, params->seed);
599         lua_setfield(L, -2, "seed");
600
601         lua_pushinteger(L, params->water_level);
602         lua_setfield(L, -2, "water_level");
603
604         lua_pushinteger(L, params->chunksize);
605         lua_setfield(L, -2, "chunksize");
606
607         std::string flagstr = writeFlagString(params->flags, flagdesc_mapgen, U32_MAX);
608         lua_pushstring(L, flagstr.c_str());
609         lua_setfield(L, -2, "flags");
610
611         return 1;
612 }
613
614
615 // set_mapgen_params(params)
616 // set mapgen parameters
617 int ModApiMapgen::l_set_mapgen_params(lua_State *L)
618 {
619         if (!lua_istable(L, 1))
620                 return 0;
621
622         EmergeManager *emerge = getServer(L)->getEmergeManager();
623         if (emerge->isRunning())
624                 throw LuaError("Cannot set parameters while mapgen is running");
625
626         MapgenParams *params = &emerge->params;
627         u32 flags = 0, flagmask = 0;
628
629         lua_getfield(L, 1, "mgname");
630         if (lua_isstring(L, -1)) {
631                 params->mg_name = lua_tostring(L, -1);
632                 delete params->sparams;
633                 params->sparams = NULL;
634         }
635
636         lua_getfield(L, 1, "seed");
637         if (lua_isnumber(L, -1))
638                 params->seed = lua_tointeger(L, -1);
639
640         lua_getfield(L, 1, "water_level");
641         if (lua_isnumber(L, -1))
642                 params->water_level = lua_tointeger(L, -1);
643
644         lua_getfield(L, 1, "chunksize");
645         if (lua_isnumber(L, -1))
646                 params->chunksize = lua_tointeger(L, -1);
647
648         warn_if_field_exists(L, 1, "flagmask",
649                 "Deprecated: flags field now includes unset flags.");
650         lua_getfield(L, 1, "flagmask");
651         if (lua_isstring(L, -1))
652                 params->flags &= ~readFlagString(lua_tostring(L, -1), flagdesc_mapgen, NULL);
653
654         if (getflagsfield(L, 1, "flags", flagdesc_mapgen, &flags, &flagmask)) {
655                 params->flags &= ~flagmask;
656                 params->flags |= flags;
657         }
658
659         return 0;
660 }
661
662
663 // set_noiseparams(name, noiseparams, set_default)
664 // set global config values for noise parameters
665 int ModApiMapgen::l_set_noiseparams(lua_State *L)
666 {
667         const char *name = luaL_checkstring(L, 1);
668
669         NoiseParams np;
670         if (!read_noiseparams(L, 2, &np))
671                 return 0;
672
673         bool set_default = lua_isboolean(L, 3) ? lua_toboolean(L, 3) : true;
674
675         g_settings->setNoiseParams(name, np, set_default);
676
677         return 0;
678 }
679
680
681 // get_noiseparams(name)
682 int ModApiMapgen::l_get_noiseparams(lua_State *L)
683 {
684         std::string name = luaL_checkstring(L, 1);
685
686         NoiseParams np;
687         if (!g_settings->getNoiseParams(name, np))
688                 return 0;
689
690         push_noiseparams(L, &np);
691         return 1;
692 }
693
694
695 // set_gen_notify(flags, {deco_id_table})
696 int ModApiMapgen::l_set_gen_notify(lua_State *L)
697 {
698         u32 flags = 0, flagmask = 0;
699         EmergeManager *emerge = getServer(L)->getEmergeManager();
700
701         if (read_flags(L, 1, flagdesc_gennotify, &flags, &flagmask)) {
702                 emerge->gen_notify_on &= ~flagmask;
703                 emerge->gen_notify_on |= flags;
704         }
705
706         if (lua_istable(L, 2)) {
707                 lua_pushnil(L);
708                 while (lua_next(L, 2)) {
709                         if (lua_isnumber(L, -1))
710                                 emerge->gen_notify_on_deco_ids.insert((u32)lua_tonumber(L, -1));
711                         lua_pop(L, 1);
712                 }
713         }
714
715         return 0;
716 }
717
718
719 // get_gen_notify()
720 int ModApiMapgen::l_get_gen_notify(lua_State *L)
721 {
722         EmergeManager *emerge = getServer(L)->getEmergeManager();
723         push_flags_string(L, flagdesc_gennotify, emerge->gen_notify_on,
724                 emerge->gen_notify_on);
725
726         lua_newtable(L);
727         int i = 1;
728         for (std::set<u32>::iterator it = emerge->gen_notify_on_deco_ids.begin();
729                         it != emerge->gen_notify_on_deco_ids.end(); ++it) {
730                 lua_pushnumber(L, *it);
731                 lua_rawseti(L, -2, i);
732                 i++;
733         }
734         return 2;
735 }
736
737
738 // register_biome({lots of stuff})
739 int ModApiMapgen::l_register_biome(lua_State *L)
740 {
741         int index = 1;
742         luaL_checktype(L, index, LUA_TTABLE);
743
744         INodeDefManager *ndef = getServer(L)->getNodeDefManager();
745         BiomeManager *bmgr    = getServer(L)->getEmergeManager()->biomemgr;
746
747         Biome *biome = read_biome_def(L, index, ndef);
748         if (!biome)
749                 return 0;
750
751         ObjDefHandle handle = bmgr->add(biome);
752         if (handle == OBJDEF_INVALID_HANDLE) {
753                 delete biome;
754                 return 0;
755         }
756
757         lua_pushinteger(L, handle);
758         return 1;
759 }
760
761
762 // register_decoration({lots of stuff})
763 int ModApiMapgen::l_register_decoration(lua_State *L)
764 {
765         int index = 1;
766         luaL_checktype(L, index, LUA_TTABLE);
767
768         INodeDefManager *ndef      = getServer(L)->getNodeDefManager();
769         DecorationManager *decomgr = getServer(L)->getEmergeManager()->decomgr;
770         BiomeManager *biomemgr     = getServer(L)->getEmergeManager()->biomemgr;
771         SchematicManager *schemmgr = getServer(L)->getEmergeManager()->schemmgr;
772
773         enum DecorationType decotype = (DecorationType)getenumfield(L, index,
774                                 "deco_type", es_DecorationType, -1);
775
776         Decoration *deco = decomgr->create(decotype);
777         if (!deco) {
778                 errorstream << "register_decoration: decoration placement type "
779                         << decotype << " not implemented" << std::endl;
780                 return 0;
781         }
782
783         deco->name       = getstringfield_default(L, index, "name", "");
784         deco->fill_ratio = getfloatfield_default(L, index, "fill_ratio", 0.02);
785         deco->y_min      = getintfield_default(L, index, "y_min", -31000);
786         deco->y_max      = getintfield_default(L, index, "y_max", 31000);
787         deco->sidelen    = getintfield_default(L, index, "sidelen", 8);
788         if (deco->sidelen <= 0) {
789                 errorstream << "register_decoration: sidelen must be "
790                         "greater than 0" << std::endl;
791                 delete deco;
792                 return 0;
793         }
794
795         //// Get node name(s) to place decoration on
796         size_t nread = getstringlistfield(L, index, "place_on", &deco->m_nodenames);
797         deco->m_nnlistsizes.push_back(nread);
798
799         //// Get decoration flags
800         getflagsfield(L, index, "flags", flagdesc_deco, &deco->flags, NULL);
801
802         //// Get NoiseParams to define how decoration is placed
803         lua_getfield(L, index, "noise_params");
804         if (read_noiseparams(L, -1, &deco->np))
805                 deco->flags |= DECO_USE_NOISE;
806         lua_pop(L, 1);
807
808         //// Get biomes associated with this decoration (if any)
809         lua_getfield(L, index, "biomes");
810         if (get_biome_list(L, -1, biomemgr, &deco->biomes))
811                 errorstream << "register_decoration: couldn't get all biomes " << std::endl;
812         lua_pop(L, 1);
813
814         //// Handle decoration type-specific parameters
815         bool success = false;
816         switch (decotype) {
817         case DECO_SIMPLE:
818                 success = read_deco_simple(L, (DecoSimple *)deco);
819                 break;
820         case DECO_SCHEMATIC:
821                 success = read_deco_schematic(L, schemmgr, (DecoSchematic *)deco);
822                 break;
823         case DECO_LSYSTEM:
824                 break;
825         }
826
827         if (!success) {
828                 delete deco;
829                 return 0;
830         }
831
832         ndef->pendNodeResolve(deco);
833
834         ObjDefHandle handle = decomgr->add(deco);
835         if (handle == OBJDEF_INVALID_HANDLE) {
836                 delete deco;
837                 return 0;
838         }
839
840         lua_pushinteger(L, handle);
841         return 1;
842 }
843
844
845 bool read_deco_simple(lua_State *L, DecoSimple *deco)
846 {
847         size_t nnames;
848         int index = 1;
849
850         deco->deco_height     = getintfield_default(L, index, "height", 1);
851         deco->deco_height_max = getintfield_default(L, index, "height_max", 0);
852         deco->nspawnby        = getintfield_default(L, index, "num_spawn_by", -1);
853
854         if (deco->deco_height <= 0) {
855                 errorstream << "register_decoration: simple decoration height"
856                         " must be greater than 0" << std::endl;
857                 return false;
858         }
859
860         nnames = getstringlistfield(L, index, "decoration", &deco->m_nodenames);
861         deco->m_nnlistsizes.push_back(nnames);
862         if (nnames == 0) {
863                 errorstream << "register_decoration: no decoration nodes "
864                         "defined" << std::endl;
865                 return false;
866         }
867
868         nnames = getstringlistfield(L, index, "spawn_by", &deco->m_nodenames);
869         deco->m_nnlistsizes.push_back(nnames);
870         if (nnames == 0 && deco->nspawnby != -1) {
871                 errorstream << "register_decoration: no spawn_by nodes defined,"
872                         " but num_spawn_by specified" << std::endl;
873                 return false;
874         }
875
876         return true;
877 }
878
879
880 bool read_deco_schematic(lua_State *L, SchematicManager *schemmgr, DecoSchematic *deco)
881 {
882         int index = 1;
883
884         deco->rotation = (Rotation)getenumfield(L, index, "rotation",
885                 ModApiMapgen::es_Rotation, ROTATE_0);
886
887         StringMap replace_names;
888         lua_getfield(L, index, "replacements");
889         if (lua_istable(L, -1))
890                 read_schematic_replacements(L, -1, &replace_names);
891         lua_pop(L, 1);
892
893         lua_getfield(L, index, "schematic");
894         Schematic *schem = get_or_load_schematic(L, -1, schemmgr, &replace_names);
895         lua_pop(L, 1);
896
897         deco->schematic = schem;
898         return schem != NULL;
899 }
900
901
902 // register_ore({lots of stuff})
903 int ModApiMapgen::l_register_ore(lua_State *L)
904 {
905         int index = 1;
906         luaL_checktype(L, index, LUA_TTABLE);
907
908         INodeDefManager *ndef = getServer(L)->getNodeDefManager();
909         BiomeManager *bmgr    = getServer(L)->getEmergeManager()->biomemgr;
910         OreManager *oremgr    = getServer(L)->getEmergeManager()->oremgr;
911
912         enum OreType oretype = (OreType)getenumfield(L, index,
913                                 "ore_type", es_OreType, ORE_SCATTER);
914         Ore *ore = oremgr->create(oretype);
915         if (!ore) {
916                 errorstream << "register_ore: ore_type " << oretype << " not implemented\n";
917                 return 0;
918         }
919
920         ore->name           = getstringfield_default(L, index, "name", "");
921         ore->ore_param2     = (u8)getintfield_default(L, index, "ore_param2", 0);
922         ore->clust_scarcity = getintfield_default(L, index, "clust_scarcity", 1);
923         ore->clust_num_ores = getintfield_default(L, index, "clust_num_ores", 1);
924         ore->clust_size     = getintfield_default(L, index, "clust_size", 0);
925         ore->nthresh        = getfloatfield_default(L, index, "noise_threshhold", 0);
926         ore->noise          = NULL;
927         ore->flags          = 0;
928
929         //// Get y_min/y_max
930         warn_if_field_exists(L, index, "height_min",
931                 "Deprecated: new name is \"y_min\".");
932         warn_if_field_exists(L, index, "height_max",
933                 "Deprecated: new name is \"y_max\".");
934
935         int ymin, ymax;
936         if (!getintfield(L, index, "y_min", ymin) &&
937                 !getintfield(L, index, "height_min", ymin))
938                 ymin = -31000;
939         if (!getintfield(L, index, "y_max", ymax) &&
940                 !getintfield(L, index, "height_max", ymax))
941                 ymax = 31000;
942         ore->y_min = ymin;
943         ore->y_max = ymax;
944
945         if (ore->clust_scarcity <= 0 || ore->clust_num_ores <= 0) {
946                 errorstream << "register_ore: clust_scarcity and clust_num_ores"
947                         "must be greater than 0" << std::endl;
948                 delete ore;
949                 return 0;
950         }
951
952         //// Get flags
953         getflagsfield(L, index, "flags", flagdesc_ore, &ore->flags, NULL);
954
955         //// Get biomes associated with this decoration (if any)
956         lua_getfield(L, index, "biomes");
957         if (get_biome_list(L, -1, bmgr, &ore->biomes))
958                 errorstream << "register_ore: couldn't get all biomes " << std::endl;
959         lua_pop(L, 1);
960
961         //// Get noise parameters if needed
962         lua_getfield(L, index, "noise_params");
963         if (read_noiseparams(L, -1, &ore->np)) {
964                 ore->flags |= OREFLAG_USE_NOISE;
965         } else if (ore->NEEDS_NOISE) {
966                 errorstream << "register_ore: specified ore type requires valid "
967                         "noise parameters" << std::endl;
968                 delete ore;
969                 return 0;
970         }
971         lua_pop(L, 1);
972
973         //// Get type-specific parameters
974         switch (oretype) {
975                 case ORE_SHEET: {
976                         OreSheet *oresheet = (OreSheet *)ore;
977
978                         oresheet->column_height_min = getintfield_default(L, index,
979                                 "column_height_min", 1);
980                         oresheet->column_height_max = getintfield_default(L, index,
981                                 "column_height_max", ore->clust_size);
982                         oresheet->column_midpoint_factor = getfloatfield_default(L, index,
983                                 "column_midpoint_factor", 0.5f);
984
985                         break;
986                 }
987                 case ORE_PUFF: {
988                         OrePuff *orepuff = (OrePuff *)ore;
989
990                         lua_getfield(L, index, "np_puff_top");
991                         read_noiseparams(L, -1, &orepuff->np_puff_top);
992                         lua_pop(L, 1);
993
994                         lua_getfield(L, index, "np_puff_bottom");
995                         read_noiseparams(L, -1, &orepuff->np_puff_bottom);
996                         lua_pop(L, 1);
997
998                         break;
999                 }
1000                 case ORE_VEIN: {
1001                         OreVein *orevein = (OreVein *)ore;
1002
1003                         orevein->random_factor = getfloatfield_default(L, index,
1004                                 "random_factor", 1.f);
1005
1006                         break;
1007                 }
1008                 default:
1009                         break;
1010         }
1011
1012         ObjDefHandle handle = oremgr->add(ore);
1013         if (handle == OBJDEF_INVALID_HANDLE) {
1014                 delete ore;
1015                 return 0;
1016         }
1017
1018         ore->m_nodenames.push_back(getstringfield_default(L, index, "ore", ""));
1019
1020         size_t nnames = getstringlistfield(L, index, "wherein", &ore->m_nodenames);
1021         ore->m_nnlistsizes.push_back(nnames);
1022
1023         ndef->pendNodeResolve(ore);
1024
1025         lua_pushinteger(L, handle);
1026         return 1;
1027 }
1028
1029
1030 // register_schematic({schematic}, replacements={})
1031 int ModApiMapgen::l_register_schematic(lua_State *L)
1032 {
1033         SchematicManager *schemmgr = getServer(L)->getEmergeManager()->schemmgr;
1034
1035         StringMap replace_names;
1036         if (lua_istable(L, 2))
1037                 read_schematic_replacements(L, 2, &replace_names);
1038
1039         Schematic *schem = load_schematic(L, 1, schemmgr->getNodeDef(),
1040                 &replace_names);
1041         if (!schem)
1042                 return 0;
1043
1044         ObjDefHandle handle = schemmgr->add(schem);
1045         if (handle == OBJDEF_INVALID_HANDLE) {
1046                 delete schem;
1047                 return 0;
1048         }
1049
1050         lua_pushinteger(L, handle);
1051         return 1;
1052 }
1053
1054
1055 // clear_registered_biomes()
1056 int ModApiMapgen::l_clear_registered_biomes(lua_State *L)
1057 {
1058         BiomeManager *bmgr = getServer(L)->getEmergeManager()->biomemgr;
1059         bmgr->clear();
1060         return 0;
1061 }
1062
1063
1064 // clear_registered_decorations()
1065 int ModApiMapgen::l_clear_registered_decorations(lua_State *L)
1066 {
1067         DecorationManager *dmgr = getServer(L)->getEmergeManager()->decomgr;
1068         dmgr->clear();
1069         return 0;
1070 }
1071
1072
1073 // clear_registered_ores()
1074 int ModApiMapgen::l_clear_registered_ores(lua_State *L)
1075 {
1076         OreManager *omgr = getServer(L)->getEmergeManager()->oremgr;
1077         omgr->clear();
1078         return 0;
1079 }
1080
1081
1082 // clear_registered_schematics()
1083 int ModApiMapgen::l_clear_registered_schematics(lua_State *L)
1084 {
1085         SchematicManager *smgr = getServer(L)->getEmergeManager()->schemmgr;
1086         smgr->clear();
1087         return 0;
1088 }
1089
1090
1091 // generate_ores(vm, p1, p2, [ore_id])
1092 int ModApiMapgen::l_generate_ores(lua_State *L)
1093 {
1094         EmergeManager *emerge = getServer(L)->getEmergeManager();
1095
1096         Mapgen mg;
1097         mg.seed = emerge->params.seed;
1098         mg.vm   = LuaVoxelManip::checkobject(L, 1)->vm;
1099         mg.ndef = getServer(L)->getNodeDefManager();
1100
1101         v3s16 pmin = lua_istable(L, 2) ? check_v3s16(L, 2) :
1102                         mg.vm->m_area.MinEdge + v3s16(1,1,1) * MAP_BLOCKSIZE;
1103         v3s16 pmax = lua_istable(L, 3) ? check_v3s16(L, 3) :
1104                         mg.vm->m_area.MaxEdge - v3s16(1,1,1) * MAP_BLOCKSIZE;
1105         sortBoxVerticies(pmin, pmax);
1106
1107         u32 blockseed = Mapgen::getBlockSeed(pmin, mg.seed);
1108
1109         emerge->oremgr->placeAllOres(&mg, blockseed, pmin, pmax);
1110
1111         return 0;
1112 }
1113
1114
1115 // generate_decorations(vm, p1, p2, [deco_id])
1116 int ModApiMapgen::l_generate_decorations(lua_State *L)
1117 {
1118         EmergeManager *emerge = getServer(L)->getEmergeManager();
1119
1120         Mapgen mg;
1121         mg.seed = emerge->params.seed;
1122         mg.vm   = LuaVoxelManip::checkobject(L, 1)->vm;
1123         mg.ndef = getServer(L)->getNodeDefManager();
1124
1125         v3s16 pmin = lua_istable(L, 2) ? check_v3s16(L, 2) :
1126                         mg.vm->m_area.MinEdge + v3s16(1,1,1) * MAP_BLOCKSIZE;
1127         v3s16 pmax = lua_istable(L, 3) ? check_v3s16(L, 3) :
1128                         mg.vm->m_area.MaxEdge - v3s16(1,1,1) * MAP_BLOCKSIZE;
1129         sortBoxVerticies(pmin, pmax);
1130
1131         u32 blockseed = Mapgen::getBlockSeed(pmin, mg.seed);
1132
1133         emerge->decomgr->placeAllDecos(&mg, blockseed, pmin, pmax);
1134
1135         return 0;
1136 }
1137
1138
1139 // create_schematic(p1, p2, probability_list, filename, y_slice_prob_list)
1140 int ModApiMapgen::l_create_schematic(lua_State *L)
1141 {
1142         INodeDefManager *ndef = getServer(L)->getNodeDefManager();
1143
1144         const char *filename = luaL_checkstring(L, 4);
1145         CHECK_SECURE_PATH_OPTIONAL(L, filename);
1146
1147         Map *map = &(getEnv(L)->getMap());
1148         Schematic schem;
1149
1150         v3s16 p1 = check_v3s16(L, 1);
1151         v3s16 p2 = check_v3s16(L, 2);
1152         sortBoxVerticies(p1, p2);
1153
1154         std::vector<std::pair<v3s16, u8> > prob_list;
1155         if (lua_istable(L, 3)) {
1156                 lua_pushnil(L);
1157                 while (lua_next(L, 3)) {
1158                         if (lua_istable(L, -1)) {
1159                                 lua_getfield(L, -1, "pos");
1160                                 v3s16 pos = check_v3s16(L, -1);
1161                                 lua_pop(L, 1);
1162
1163                                 u8 prob = getintfield_default(L, -1, "prob", MTSCHEM_PROB_ALWAYS);
1164                                 prob_list.push_back(std::make_pair(pos, prob));
1165                         }
1166
1167                         lua_pop(L, 1);
1168                 }
1169         }
1170
1171         std::vector<std::pair<s16, u8> > slice_prob_list;
1172         if (lua_istable(L, 5)) {
1173                 lua_pushnil(L);
1174                 while (lua_next(L, 5)) {
1175                         if (lua_istable(L, -1)) {
1176                                 s16 ypos = getintfield_default(L, -1, "ypos", 0);
1177                                 u8 prob  = getintfield_default(L, -1, "prob", MTSCHEM_PROB_ALWAYS);
1178                                 slice_prob_list.push_back(std::make_pair(ypos, prob));
1179                         }
1180
1181                         lua_pop(L, 1);
1182                 }
1183         }
1184
1185         if (!schem.getSchematicFromMap(map, p1, p2)) {
1186                 errorstream << "create_schematic: failed to get schematic "
1187                         "from map" << std::endl;
1188                 return 0;
1189         }
1190
1191         schem.applyProbabilities(p1, &prob_list, &slice_prob_list);
1192
1193         schem.saveSchematicToFile(filename, ndef);
1194         actionstream << "create_schematic: saved schematic file '"
1195                 << filename << "'." << std::endl;
1196
1197         lua_pushboolean(L, true);
1198         return 1;
1199 }
1200
1201
1202 // place_schematic(p, schematic, rotation, replacement)
1203 int ModApiMapgen::l_place_schematic(lua_State *L)
1204 {
1205         Map *map = &(getEnv(L)->getMap());
1206         SchematicManager *schemmgr = getServer(L)->getEmergeManager()->schemmgr;
1207
1208         //// Read position
1209         v3s16 p = check_v3s16(L, 1);
1210
1211         //// Read rotation
1212         int rot = ROTATE_0;
1213         const char *enumstr = lua_tostring(L, 3);
1214         if (enumstr)
1215                 string_to_enum(es_Rotation, rot, std::string(enumstr));
1216
1217         //// Read force placement
1218         bool force_placement = true;
1219         if (lua_isboolean(L, 5))
1220                 force_placement = lua_toboolean(L, 5);
1221
1222         //// Read node replacements
1223         StringMap replace_names;
1224         if (lua_istable(L, 4))
1225                 read_schematic_replacements(L, 4, &replace_names);
1226
1227         //// Read schematic
1228         Schematic *schem = get_or_load_schematic(L, 2, schemmgr, &replace_names);
1229         if (!schem) {
1230                 errorstream << "place_schematic: failed to get schematic" << std::endl;
1231                 return 0;
1232         }
1233
1234         schem->placeStructure(map, p, 0, (Rotation)rot, force_placement);
1235
1236         lua_pushboolean(L, true);
1237         return 1;
1238 }
1239
1240 // serialize_schematic(schematic, format, options={...})
1241 int ModApiMapgen::l_serialize_schematic(lua_State *L)
1242 {
1243         SchematicManager *schemmgr = getServer(L)->getEmergeManager()->schemmgr;
1244
1245         //// Read options
1246         bool use_comments = getboolfield_default(L, 3, "lua_use_comments", false);
1247         u32 indent_spaces = getintfield_default(L, 3, "lua_num_indent_spaces", 0);
1248
1249         //// Get schematic
1250         bool was_loaded = false;
1251         Schematic *schem = (Schematic *)get_objdef(L, 1, schemmgr);
1252         if (!schem) {
1253                 schem = load_schematic(L, 1, NULL, NULL);
1254                 was_loaded = true;
1255         }
1256         if (!schem) {
1257                 errorstream << "serialize_schematic: failed to get schematic" << std::endl;
1258                 return 0;
1259         }
1260
1261         //// Read format of definition to save as
1262         int schem_format = SCHEM_FMT_MTS;
1263         const char *enumstr = lua_tostring(L, 2);
1264         if (enumstr)
1265                 string_to_enum(es_SchematicFormatType, schem_format, std::string(enumstr));
1266
1267         //// Serialize to binary string
1268         std::ostringstream os(std::ios_base::binary);
1269         switch (schem_format) {
1270         case SCHEM_FMT_MTS:
1271                 schem->serializeToMts(&os, schem->m_nodenames);
1272                 break;
1273         case SCHEM_FMT_LUA:
1274                 schem->serializeToLua(&os, schem->m_nodenames,
1275                         use_comments, indent_spaces);
1276                 break;
1277         default:
1278                 return 0;
1279         }
1280
1281         if (was_loaded)
1282                 delete schem;
1283
1284         std::string ser = os.str();
1285         lua_pushlstring(L, ser.c_str(), ser.length());
1286         return 1;
1287 }
1288
1289
1290 void ModApiMapgen::Initialize(lua_State *L, int top)
1291 {
1292         API_FCT(get_biome_id);
1293         API_FCT(get_mapgen_object);
1294
1295         API_FCT(get_mapgen_params);
1296         API_FCT(set_mapgen_params);
1297         API_FCT(set_noiseparams);
1298         API_FCT(get_noiseparams);
1299         API_FCT(set_gen_notify);
1300         API_FCT(get_gen_notify);
1301
1302         API_FCT(register_biome);
1303         API_FCT(register_decoration);
1304         API_FCT(register_ore);
1305         API_FCT(register_schematic);
1306
1307         API_FCT(clear_registered_biomes);
1308         API_FCT(clear_registered_decorations);
1309         API_FCT(clear_registered_ores);
1310         API_FCT(clear_registered_schematics);
1311
1312         API_FCT(generate_ores);
1313         API_FCT(generate_decorations);
1314         API_FCT(create_schematic);
1315         API_FCT(place_schematic);
1316         API_FCT(serialize_schematic);
1317 }