worldedit/visualization.lua

   1 worldedit = worldedit or {}\r
   2 local minetest = minetest --local copy of global\r
   3 \r
   4 --modifies positions `pos1` and `pos2` so that each component of `pos1` is less than or equal to its corresponding conent of `pos2`, returning two new positions\r
   5 worldedit.sort_pos = function(pos1, pos2)\r
   6         pos1 = {x=pos1.x, y=pos1.y, z=pos1.z}\r
   7         pos2 = {x=pos2.x, y=pos2.y, z=pos2.z}\r
   8         if pos1.x > pos2.x then\r
   9                 pos2.x, pos1.x = pos1.x, pos2.x\r
  10         end\r
  11         if pos1.y > pos2.y then\r
  12                 pos2.y, pos1.y = pos1.y, pos2.y\r
  13         end\r
  14         if pos1.z > pos2.z then\r
  15                 pos2.z, pos1.z = pos1.z, pos2.z\r
  16         end\r
  17         return pos1, pos2\r
  18 end\r
  19 \r
  20 --determines the volume of the region defined by positions `pos1` and `pos2`, returning the volume\r
  21 worldedit.volume = function(pos1, pos2)\r
  22         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  23         return (pos2.x - pos1.x + 1) * (pos2.y - pos1.y + 1) * (pos2.z - pos1.z + 1)\r
  24 end\r
  25 \r
  26 minetest.register_node("worldedit:placeholder", {\r
  27         drawtype = "airlike",\r
  28         paramtype = "light",\r
  29         sunlight_propagates = true,\r
  30         diggable = false,\r
  31         groups = {not_in_creative_inventory=1},\r
  32 })\r
  33 \r
  34 --hides all nodes in a region defined by positions `pos1` and `pos2` by non-destructively replacing them with invisible nodes, returning the number of nodes hidden\r
  35 worldedit.hide = function(pos1, pos2)\r
  36         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  37         local pos = {x=pos1.x, y=0, z=0}\r
  38         local placeholder = {name="worldedit:placeholder", param1=0, param2=0}\r
  39         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  40         while pos.x <= pos2.x do\r
  41                 pos.y = pos1.y\r
  42                 while pos.y <= pos2.y do\r
  43                         pos.z = pos1.z\r
  44                         while pos.z <= pos2.z do\r
  45                                 local node = get_node(pos)\r
  46                                 placeholder.param1, placeholder.param2 = node.param1, node.param2 --copy node's param1 and param2\r
  47                                 local data = get_meta(pos):to_table() --obtain metadata of original node\r
  48                                 add_node(pos, placeholder) --add placeholder node\r
  49                                 local meta = get_meta(pos) --obtain placeholder meta\r
  50                                 meta:from_table(data) --set placeholder metadata to the original node's metadata\r
  51                                 meta:set_string("worldedit_placeholder", node.name)  --add the node's name\r
  52                                 pos.z = pos.z + 1\r
  53                         end\r
  54                         pos.y = pos.y + 1\r
  55                 end\r
  56                 pos.x = pos.x + 1\r
  57         end\r
  58         return worldedit.volume(pos1, pos2)\r
  59 end\r
  60 \r
  61 --suppresses all instances of `nodename` in a region defined by positions `pos1` and `pos2` by non-destructively replacing them with invisible nodes, returning the number of nodes suppressed\r
  62 worldedit.suppress = function(pos1, pos2, nodename)\r
  63         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  64         local placeholder = {name="worldedit:placeholder", param1=0, param2=0}\r
  65         local nodes = minetest.find_nodes_in_area(pos1, pos2, nodename)\r
  66         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  67         for _, pos in ipairs(nodes) do\r
  68                 local node = get_node(pos)\r
  69                 placeholder.param1, placeholder.param2 = node.param1, node.param2 --copy node's param1 and param2\r
  70                 local data = get_meta(pos):to_table() --obtain metadata of original node\r
  71                 add_node(pos, placeholder) --add placeholder node\r
  72                 local meta = get_meta(pos) --obtain placeholder meta\r
  73                 meta:from_table(data) --set placeholder metadata to the original node's metadata\r
  74                 meta:set_string("worldedit_placeholder", nodename)  --add the node's name\r
  75         end\r
  76         return #nodes\r
  77 end\r
  78 \r
  79 --highlights all instances of `nodename` in a region defined by positions `pos1` and `pos2` by non-destructively hiding all other nodes, returning the number of nodes found\r
  80 worldedit.highlight = function(pos1, pos2, nodename) --wip: speed this up with voxmanip get_data\r
  81         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  82         local pos = {x=pos1.x, y=0, z=0}\r
  83         local placeholder = {name="worldedit:placeholder", param1=0, param2=0}\r
  84         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  85         local count = 0\r
  86         while pos.x <= pos2.x do\r
  87                 pos.y = pos1.y\r
  88                 while pos.y <= pos2.y do\r
  89                         pos.z = pos1.z\r
  90                         while pos.z <= pos2.z do\r
  91                                 local node = get_node(pos)\r
  92                                 if node.name == nodename then --node found\r
  93                                         count = count + 1\r
  94                                 else --hide other nodes\r
  95                                         placeholder.param1, placeholder.param2 = node.param1, node.param2 --copy node's param1 and param2\r
  96                                         local data = get_meta(pos):to_table() --obtain metadata of original node\r
  97                                         add_node(pos, placeholder) --add placeholder node\r
  98                                         local meta = get_meta(pos) --obtain placeholder meta\r
  99                                         meta:from_table(data) --set placeholder metadata to the original node's metadata\r
 100                                         meta:set_string("worldedit_placeholder", node.name)  --add the node's name\r
 101                                 end\r
 102                                 pos.z = pos.z + 1\r
 103                         end\r
 104                         pos.y = pos.y + 1\r
 105                 end\r
 106                 pos.x = pos.x + 1\r
 107         end\r
 108         return count\r
 109 end\r
 110 \r
 111 --restores all nodes hidden with WorldEdit functions in a region defined by positions `pos1` and `pos2`, returning the number of nodes restored\r
 112 worldedit.restore = function(pos1, pos2)\r
 113         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
 114         local node = {name="", param1=0, param2=0}\r
 115         local nodes = minetest.find_nodes_in_area(pos1, pos2, "worldedit:placeholder")\r
 116         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
 117         for _, pos in ipairs(nodes) do\r
 118                 local currentnode = get_node(pos)\r
 119                 node.param1, node.param2 = currentnode.param1, currentnode.param2 --copy node's param1 and param2\r
 120                 local data = get_meta(pos):to_table() --obtain node metadata\r
 121                 node.name = data.fields.worldedit_placeholder --set node name\r
 122                 data.fields.worldedit_placeholder = nil --delete old nodename\r
 123                 add_node(pos, placeholder) --add original node\r
 124                 get_meta(pos):from_table(data) --set original node metadata\r
 125         end\r
 126         return #nodes\r
 127 end\r