diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/mapgen.cpp | 115 | ||||
-rw-r--r-- | src/mapgen.h | 12 | ||||
-rw-r--r-- | src/script/lua_api/luaapi.cpp | 22 | ||||
-rw-r--r-- | src/script/lua_api/luaapi.h | 3 |
4 files changed, 108 insertions, 44 deletions
diff --git a/src/mapgen.cpp b/src/mapgen.cpp index 23af1f398..905e80e89 100644 --- a/src/mapgen.cpp +++ b/src/mapgen.cpp @@ -535,28 +535,10 @@ void DecoSchematic::generate(Mapgen *mg, PseudoRandom *pr, s16 max_y, v3s16 p) { c_place_on != CONTENT_IGNORE) return; - u32 i = 0; - for (s16 z = 0; z != size.Z; z++) - for (s16 y = 0; y != size.Y; y++) { - vi = vm->m_area.index(p.X, p.Y + y, p.Z + z); - for (s16 x = 0; x != size.X; x++, i++, vi++) { - if (!vm->m_area.contains(vi)) - continue; - - if (schematic[i].getContent() == CONTENT_IGNORE) - continue; - - content_t c = vm->m_data[vi].getContent(); - if (c != CONTENT_AIR && c != CONTENT_IGNORE) - continue; - - if (schematic[i].param1 && myrand_range(1, 256) > schematic[i].param1) - continue; - - vm->m_data[vi] = schematic[i]; - vm->m_data[vi].param1 = 0; - } - } + Rotation rot = (rotation == ROTATE_RAND) ? + (Rotation)pr->range(ROTATE_0, ROTATE_270) : rotation; + + blitToVManip(p, vm, rot, false); } @@ -570,32 +552,58 @@ std::string DecoSchematic::getName() { } -void DecoSchematic::placeStructure(Map *map, v3s16 p) { - assert(schematic != NULL); - ManualMapVoxelManipulator *vm = new ManualMapVoxelManipulator(map); +void DecoSchematic::blitToVManip(v3s16 p, ManualMapVoxelManipulator *vm, + int rot, bool force_placement) { + int xstride = 1; + int ystride = size.X; + int zstride = size.X * size.Y; - if (flags & DECO_PLACE_CENTER_X) - p.X -= (size.X + 1) / 2; - if (flags & DECO_PLACE_CENTER_Y) - p.Y -= (size.Y + 1) / 2; - if (flags & DECO_PLACE_CENTER_Z) - p.Z -= (size.Z + 1) / 2; - - v3s16 bp1 = getNodeBlockPos(p); - v3s16 bp2 = getNodeBlockPos(p + size - v3s16(1,1,1)); - vm->initialEmerge(bp1, bp2); + s16 sx = size.X; + s16 sy = size.Y; + s16 sz = size.Z; - u32 i = 0; - for (s16 z = 0; z != size.Z; z++) - for (s16 y = 0; y != size.Y; y++) { - u32 vi = vm->m_area.index(p.X, p.Y + y, p.Z + z); - for (s16 x = 0; x != size.X; x++, i++, vi++) { + int i_start, i_step_x, i_step_z; + switch (rot) { + case ROTATE_90: + i_start = sx - 1; + i_step_x = zstride; + i_step_z = -xstride; + SWAP(s16, sx, sz); + break; + case ROTATE_180: + i_start = zstride * (sz - 1) + sx - 1; + i_step_x = -xstride; + i_step_z = -zstride; + break; + case ROTATE_270: + i_start = zstride * (sz - 1); + i_step_x = -zstride; + i_step_z = xstride; + SWAP(s16, sx, sz); + break; + default: + i_start = 0; + i_step_x = xstride; + i_step_z = zstride; + } + + for (s16 z = 0; z != sz; z++) + for (s16 y = 0; y != sy; y++) { + u32 i = z * i_step_z + y * ystride + i_start; + for (s16 x = 0; x != sx; x++, i += i_step_x) { + u32 vi = vm->m_area.index(p.X + x, p.Y + y, p.Z + z); if (!vm->m_area.contains(vi)) continue; if (schematic[i].getContent() == CONTENT_IGNORE) continue; - + + if (!force_placement) { + content_t c = vm->m_data[vi].getContent(); + if (c != CONTENT_AIR && c != CONTENT_IGNORE) + continue; + } + if (schematic[i].param1 && myrand_range(1, 256) > schematic[i].param1) continue; @@ -603,6 +611,31 @@ void DecoSchematic::placeStructure(Map *map, v3s16 p) { vm->m_data[vi].param1 = 0; } } +} + + +void DecoSchematic::placeStructure(Map *map, v3s16 p) { + assert(schematic != NULL); + ManualMapVoxelManipulator *vm = new ManualMapVoxelManipulator(map); + + Rotation rot = (rotation == ROTATE_RAND) ? + (Rotation)myrand_range(ROTATE_0, ROTATE_270) : rotation; + + v3s16 s = (rot == ROTATE_90 || rot == ROTATE_270) ? + v3s16(size.Z, size.Y, size.X) : size; + + if (flags & DECO_PLACE_CENTER_X) + p.X -= (s.X + 1) / 2; + if (flags & DECO_PLACE_CENTER_Y) + p.Y -= (s.Y + 1) / 2; + if (flags & DECO_PLACE_CENTER_Z) + p.Z -= (s.Z + 1) / 2; + + v3s16 bp1 = getNodeBlockPos(p); + v3s16 bp2 = getNodeBlockPos(p + s - v3s16(1,1,1)); + vm->initialEmerge(bp1, bp2); + + blitToVManip(p, vm, rot, true); std::map<v3s16, MapBlock *> lighting_modified_blocks; std::map<v3s16, MapBlock *> modified_blocks; diff --git a/src/mapgen.h b/src/mapgen.h index 9a8b99c4a..4ed81b99b 100644 --- a/src/mapgen.h +++ b/src/mapgen.h @@ -258,6 +258,14 @@ public: virtual std::string getName(); }; +enum Rotation { + ROTATE_0, + ROTATE_90, + ROTATE_180, + ROTATE_270, + ROTATE_RAND, +}; + class DecoSchematic : public Decoration { public: std::string filename; @@ -266,6 +274,7 @@ public: std::vector<content_t> c_nodes; u32 flags; + Rotation rotation; v3s16 size; MapNode *schematic; @@ -277,6 +286,9 @@ public: virtual int getHeight(); virtual std::string getName(); + void blitToVManip(v3s16 p, ManualMapVoxelManipulator *vm, + int rot, bool force_placement); + bool loadSchematicFile(); void saveSchematicFile(INodeDefManager *ndef); diff --git a/src/script/lua_api/luaapi.cpp b/src/script/lua_api/luaapi.cpp index d457d257e..de0c3a670 100644 --- a/src/script/lua_api/luaapi.cpp +++ b/src/script/lua_api/luaapi.cpp @@ -51,6 +51,16 @@ struct EnumString ModApiBasic::es_DecorationType[] = {0, NULL}, }; +struct EnumString ModApiBasic::es_Rotation[] = +{ + {ROTATE_0, "0"}, + {ROTATE_90, "90"}, + {ROTATE_180, "180"}, + {ROTATE_270, "270"}, + {ROTATE_RAND, "random"}, + {0, NULL}, +}; + ModApiBasic::ModApiBasic() : ModApiBase() { } @@ -767,7 +777,9 @@ int ModApiBasic::l_register_decoration(lua_State *L) break; } case DECO_SCHEMATIC: { DecoSchematic *dschem = (DecoSchematic *)deco; - dschem->flags = getflagsfield(L, index, "flags", flagdesc_deco_schematic); + dschem->flags = getflagsfield(L, index, "flags", flagdesc_deco_schematic); + dschem->rotation = (Rotation)getenumfield(L, index, + "rotation", es_Rotation, ROTATE_0); lua_getfield(L, index, "schematic"); if (!read_schematic(L, -1, dschem, getServer(L))) { @@ -848,7 +860,7 @@ int ModApiBasic::l_create_schematic(lua_State *L) } -// place_schematic(p, schematic) +// place_schematic(p, schematic, rotation) int ModApiBasic::l_place_schematic(lua_State *L) { DecoSchematic dschem; @@ -859,6 +871,12 @@ int ModApiBasic::l_place_schematic(lua_State *L) v3s16 p = read_v3s16(L, 1); if (!read_schematic(L, 2, &dschem, getServer(L))) return 0; + + Rotation rot = ROTATE_0; + if (lua_isstring(L, 3)) + string_to_enum(es_Rotation, (int &)rot, std::string(lua_tostring(L, 3))); + + dschem.rotation = rot; if (!dschem.filename.empty()) { if (!dschem.loadSchematicFile()) { diff --git a/src/script/lua_api/luaapi.h b/src/script/lua_api/luaapi.h index ba76117d4..af73625ba 100644 --- a/src/script/lua_api/luaapi.h +++ b/src/script/lua_api/luaapi.h @@ -136,11 +136,12 @@ private: // create_schematic(p1, p2, filename) static int l_create_schematic(lua_State *L); - // place_schematic(p, filename) + // place_schematic(p, filename, rotation) static int l_place_schematic(lua_State *L); static struct EnumString es_OreType[]; static struct EnumString es_DecorationType[]; + static struct EnumString es_Rotation[]; }; |