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         --make area stay loaded\r
  37         local manip = minetest.get_voxel_manip()\r
  38         manip:read_from_map(pos1, pos2)\r
  39 \r
  40         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  41         local pos = {x=pos1.x, y=0, z=0}\r
  42         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  43         while pos.x <= pos2.x do\r
  44                 pos.y = pos1.y\r
  45                 while pos.y <= pos2.y do\r
  46                         pos.z = pos1.z\r
  47                         while pos.z <= pos2.z do\r
  48                                 local node = get_node(pos)\r
  49                                 if node.name ~= "worldedit:placeholder" then\r
  50                                         local data = get_meta(pos):to_table() --obtain metadata of original node\r
  51                                         data.fields.worldedit_placeholder = node.name --add the node's name\r
  52                                         node.name = "worldedit:placeholder" --set node name\r
  53                                         add_node(pos, node) --add placeholder node\r
  54                                         get_meta(pos):from_table(data) --set placeholder metadata to the original node's metadata\r
  55                                 end\r
  56                                 pos.z = pos.z + 1\r
  57                         end\r
  58                         pos.y = pos.y + 1\r
  59                 end\r
  60                 pos.x = pos.x + 1\r
  61         end\r
  62         return worldedit.volume(pos1, pos2)\r
  63 end\r
  64 \r
  65 --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
  66 worldedit.suppress = function(pos1, pos2, nodename)\r
  67         --ignore placeholder supression\r
  68         if nodename == "worldedit:placeholder" then\r
  69                 return 0\r
  70         end\r
  71 \r
  72         --make area stay loaded\r
  73         local manip = minetest.get_voxel_manip()\r
  74         manip:read_from_map(pos1, pos2)\r
  75 \r
  76         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  77         local nodes = minetest.find_nodes_in_area(pos1, pos2, nodename)\r
  78         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  79         for _, pos in ipairs(nodes) do\r
  80                 local node = get_node(pos)\r
  81                 local data = get_meta(pos):to_table() --obtain metadata of original node\r
  82                 data.fields.worldedit_placeholder = node.name --add the node's name\r
  83                 node.name = "worldedit:placeholder" --set node name\r
  84                 add_node(pos, node) --add placeholder node\r
  85                 get_meta(pos):from_table(data) --set placeholder metadata to the original node's metadata\r
  86         end\r
  87         return #nodes\r
  88 end\r
  89 \r
  90 --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
  91 worldedit.highlight = function(pos1, pos2, nodename)\r
  92         --make area stay loaded\r
  93         local manip = minetest.get_voxel_manip()\r
  94         manip:read_from_map(pos1, pos2)\r
  95 \r
  96         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
  97         local pos = {x=pos1.x, y=0, z=0}\r
  98         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
  99         local count = 0\r
 100         while pos.x <= pos2.x do\r
 101                 pos.y = pos1.y\r
 102                 while pos.y <= pos2.y do\r
 103                         pos.z = pos1.z\r
 104                         while pos.z <= pos2.z do\r
 105                                 local node = get_node(pos)\r
 106                                 if node.name == nodename then --node found\r
 107                                         count = count + 1\r
 108                                 elseif node.name ~= "worldedit:placeholder" then --hide other nodes\r
 109                                         local data = get_meta(pos):to_table() --obtain metadata of original node\r
 110                                         data.fields.worldedit_placeholder = node.name --add the node's name\r
 111                                         node.name = "worldedit:placeholder" --set node name\r
 112                                         add_node(pos, node) --add placeholder node\r
 113                                         get_meta(pos):from_table(data) --set placeholder metadata to the original node's metadata\r
 114                                 end\r
 115                                 pos.z = pos.z + 1\r
 116                         end\r
 117                         pos.y = pos.y + 1\r
 118                 end\r
 119                 pos.x = pos.x + 1\r
 120         end\r
 121         return count\r
 122 end\r
 123 \r
 124 --restores all nodes hidden with WorldEdit functions in a region defined by positions `pos1` and `pos2`, returning the number of nodes restored\r
 125 worldedit.restore = function(pos1, pos2)\r
 126         --make area stay loaded\r
 127         local manip = minetest.get_voxel_manip()\r
 128         manip:read_from_map(pos1, pos2)\r
 129 \r
 130         local pos1, pos2 = worldedit.sort_pos(pos1, pos2)\r
 131         local nodes = minetest.find_nodes_in_area(pos1, pos2, "worldedit:placeholder")\r
 132         local get_node, get_meta, add_node = minetest.get_node, minetest.get_meta, minetest.add_node\r
 133         for _, pos in ipairs(nodes) do\r
 134                 local node = get_node(pos)\r
 135                 local data = get_meta(pos):to_table() --obtain node metadata\r
 136                 node.name = data.fields.worldedit_placeholder --set node name\r
 137                 data.fields.worldedit_placeholder = nil --delete old nodename\r
 138                 add_node(pos, node) --add original node\r
 139                 get_meta(pos):from_table(data) --set original node metadata\r
 140         end\r
 141         return #nodes\r
 142 end\r