Tweak parsing of /placeschem

[minetest_schemedit.git] / init.lua

local S = minetest.get_translator("schemedit")
local F = minetest.formspec_escape

local schemedit = {}

local DIR_DELIM = "/"

-- Set to true to enable `make_schemedit_readme` command
local MAKE_README = false

local export_path_full = table.concat({minetest.get_worldpath(), "schems"}, DIR_DELIM)

-- truncated export path so the server directory structure is not exposed publicly
local export_path_trunc = table.concat({S("<world path>"), "schems"}, DIR_DELIM)

local text_color = "#D79E9E"
local text_color_number = 0xD79E9E

local can_import = minetest.read_schematic ~= nil

schemedit.markers = {}

-- [local function] Renumber table
local function renumber(t)
        local res = {}
        for _, i in pairs(t) do
                res[#res + 1] = i
        end
        return res
end

local NEEDED_PRIV = "server"
local function check_priv(player_name, quit)
        local privs = minetest.get_player_privs(player_name)
        if privs[NEEDED_PRIV] then
                return true
        else
                if not quit then
                        minetest.chat_send_player(player_name, minetest.colorize("red",
                                        S("Insufficient privileges! You need the “@1” privilege to use this.", NEEDED_PRIV)))
                end
                return false
        end
end

-- Lua export
local export_schematic_to_lua
if can_import then
        export_schematic_to_lua = function(schematic, filepath, options)
                if not options then options = {} end
                local str = minetest.serialize_schematic(schematic, "lua", options)
                local file = io.open(filepath, "w")
                if file and str then
                        file:write(str)
                        file:flush()
                        file:close()
                        return true
                else
                        return false
                end
        end
end

---
--- Formspec API
---

local contexts = {}
local form_data = {}
local tabs = {}
local forms = {}
local displayed_waypoints = {}

-- Sadly, the probabilities presented in Lua (0-255) are not identical to the REAL probabilities in the
-- schematic file (0-127). There are two converter functions to convert from one probability type to another.
-- This mod tries to retain the “Lua probability” as long as possible and only switches to “schematic probability”
-- on an actual export to a schematic.

function schemedit.lua_prob_to_schematic_prob(lua_prob)
        return math.floor(lua_prob / 2)
end

function schemedit.schematic_prob_to_lua_prob(schematic_prob)
        return schematic_prob * 2

end

-- [function] Add form
function schemedit.add_form(name, def)
        def.name = name
        forms[name] = def

        if def.tab then
                tabs[#tabs + 1] = name
        end
end

-- [function] Generate tabs
function schemedit.generate_tabs(current)
        local retval = "tabheader[0,0;tabs;"
        for _, t in pairs(tabs) do
                local f = forms[t]
                if f.tab ~= false and f.caption then
                        retval = retval..f.caption..","

                        if type(current) ~= "number" and current == f.name then
                                current = _
                        end
                end
        end
        retval = retval:sub(1, -2) -- Strip last comma
        retval = retval..";"..current.."]" -- Close tabheader
        return retval
end

-- [function] Handle tabs
function schemedit.handle_tabs(pos, name, fields)
        local tab = tonumber(fields.tabs)
        if tab and tabs[tab] and forms[tabs[tab]] then
                schemedit.show_formspec(pos, name, forms[tabs[tab]].name)
                return true
        end
end

-- [function] Show formspec
function schemedit.show_formspec(pos, player, tab, show, ...)
        if forms[tab] then
                if type(player) == "string" then
                        player = minetest.get_player_by_name(player)
                end
                local name = player:get_player_name()

                if show ~= false then
                        if not form_data[name] then
                                form_data[name] = {}
                        end

                        local form = forms[tab].get(form_data[name], pos, name, ...)
                        if forms[tab].tab then
                                form = form..schemedit.generate_tabs(tab)
                        end

                        minetest.show_formspec(name, "schemedit:"..tab, form)
                        contexts[name] = pos

                        -- Update player attribute
                        if forms[tab].cache_name ~= false then
                                local pmeta = player:get_meta()
                                pmeta:set_string("schemedit:tab", tab)
                        end
                else
                        minetest.close_formspec(pname, "schemedit:"..tab)
                end
        end
end

-- [event] On receive fields
minetest.register_on_player_receive_fields(function(player, formname, fields)
        local formname = formname:split(":")

        if formname[1] == "schemedit" and forms[formname[2]] then
                local handle = forms[formname[2]].handle
                local name = player:get_player_name()
                if contexts[name] then
                        if not form_data[name] then
                                form_data[name] = {}
                        end

                        if not schemedit.handle_tabs(contexts[name], name, fields) and handle then
                                handle(form_data[name], contexts[name], name, fields)
                        end
                end
        end
end)

-- Helper function. Scans probabilities of all nodes in the given area and returns a prob_list
schemedit.scan_metadata = function(pos1, pos2)
        local prob_list = {}

        for x=pos1.x, pos2.x do
        for y=pos1.y, pos2.y do
        for z=pos1.z, pos2.z do
                local scanpos = {x=x, y=y, z=z}
                local node = minetest.get_node_or_nil(scanpos)

                local prob, force_place
                if node == nil or node.name == "schemedit:void" then
                        prob = 0
                        force_place = false
                else
                        local meta = minetest.get_meta(scanpos)

                        prob = tonumber(meta:get_string("schemedit_prob")) or 255
                        local fp = meta:get_string("schemedit_force_place")
                        if fp == "true" then
                                force_place = true
                        else
                                force_place = false
                        end
                end

                local hashpos = minetest.hash_node_position(scanpos)
                prob_list[hashpos] = {
                        pos = scanpos,
                        prob = prob,
                        force_place = force_place,
                }
        end
        end
        end

        return prob_list
end

-- Sets probability and force_place metadata of an item.
-- Also updates item description.
-- The itemstack is updated in-place.
local function set_item_metadata(itemstack, prob, force_place)
        local smeta = itemstack:get_meta()
        local prob_desc = "\n"..S("Probability: @1", prob or
                        smeta:get_string("schemedit_prob") or S("Not Set"))
        -- Update probability
        if prob and prob >= 0 and prob < 255 then
                smeta:set_string("schemedit_prob", tostring(prob))
        elseif prob and prob == 255 then
                -- Clear prob metadata for default probability
                prob_desc = ""
                smeta:set_string("schemedit_prob", nil)
        else
                prob_desc = "\n"..S("Probability: @1", smeta:get_string("schemedit_prob") or
                                S("Not Set"))
        end

        -- Update force place
        if force_place == true then
                smeta:set_string("schemedit_force_place", "true")
        elseif force_place == false then
                smeta:set_string("schemedit_force_place", nil)
        end

        -- Update description
        local desc = minetest.registered_items[itemstack:get_name()].description
        local meta_desc = smeta:get_string("description")
        if meta_desc and meta_desc ~= "" then
                desc = meta_desc
        end

        local original_desc = smeta:get_string("original_description")
        if original_desc and original_desc ~= "" then
                desc = original_desc
        else
                smeta:set_string("original_description", desc)
        end

        local force_desc = ""
        if smeta:get_string("schemedit_force_place") == "true" then
                force_desc = "\n"..S("Force placement")
        end

        desc = desc..minetest.colorize(text_color, prob_desc..force_desc)

        smeta:set_string("description", desc)

        return itemstack
end

---
--- Formspec Tabs
---
local import_btn = ""
if can_import then
        import_btn = "button[0.5,2.5;6,1;import;"..F(S("Import schematic")).."]"
end
schemedit.add_form("main", {
        tab = true,
        caption = S("Main"),
        get = function(self, pos, name)
                local meta = minetest.get_meta(pos):to_table().fields
                local strpos = minetest.pos_to_string(pos)
                local hashpos = minetest.hash_node_position(pos)

                local border_button
                if meta.schem_border == "true" and schemedit.markers[hashpos] then
                        border_button = "button[3.5,7.5;3,1;border;"..F(S("Hide border")).."]"
                else
                        border_button = "button[3.5,7.5;3,1;border;"..F(S("Show border")).."]"
                end

                local xs, ys, zs = meta.x_size or 1, meta.y_size or 1, meta.z_size or 1
                local size = {x=xs, y=ys, z=zs}
                local schem_name = meta.schem_name or ""

                local form = [[
                        size[7,8]
                        label[0.5,-0.1;]]..F(S("Position: @1", strpos))..[[]
                        label[3,-0.1;]]..F(S("Owner: @1", name))..[[]
                        label[0.5,0.4;]]..F(S("Schematic name: @1", F(schem_name)))..[[]
                        label[0.5,0.9;]]..F(S("Size: @1", minetest.pos_to_string(size)))..[[]

                        field[0.8,2;5,1;name;]]..F(S("Schematic name:"))..[[;]]..F(schem_name or "")..[[]
                        button[5.3,1.69;1.2,1;save_name;]]..F(S("OK"))..[[]
                        tooltip[save_name;]]..F(S("Save schematic name"))..[[]
                        field_close_on_enter[name;false]

                        button[0.5,3.5;6,1;export;]]..F(S("Export schematic")).."]"..
                        import_btn..[[
                        textarea[0.8,4.5;6.2,1;;]]..F(S("Export/import path:\n@1",
                        export_path_trunc .. DIR_DELIM .. F(S("<name>"))..".mts"))..[[;]
                        button[0.5,5.5;3,1;air2void;]]..F(S("Air to voids"))..[[]
                        button[3.5,5.5;3,1;void2air;]]..F(S("Voids to air"))..[[]
                        tooltip[air2void;]]..F(S("Turn all air nodes into schematic void nodes"))..[[]
                        tooltip[void2air;]]..F(S("Turn all schematic void nodes into air nodes"))..[[]
                        field[0.8,7;2,1;x;]]..F(S("X size:"))..[[;]]..xs..[[]
                        field[2.8,7;2,1;y;]]..F(S("Y size:"))..[[;]]..ys..[[]
                        field[4.8,7;2,1;z;]]..F(S("Z size:"))..[[;]]..zs..[[]
                        field_close_on_enter[x;false]
                        field_close_on_enter[y;false]
                        field_close_on_enter[z;false]
                        button[0.5,7.5;3,1;save;]]..F(S("Save size"))..[[]
                ]]..
                border_button
                if minetest.get_modpath("doc") then
                        form = form .. "image_button[6.4,-0.2;0.8,0.8;doc_button_icon_lores.png;doc;]" ..
                        "tooltip[doc;"..F(S("Help")).."]"
                end
                return form
        end,
        handle = function(self, pos, name, fields)
                if fields.doc then
                        doc.show_entry(name, "nodes", "schemedit:creator", true)
                        return
                end

                if not check_priv(name, fields.quit) then
                        return
                end

                local realmeta = minetest.get_meta(pos)
                local meta = realmeta:to_table().fields
                local hashpos = minetest.hash_node_position(pos)

                -- Save size vector values
                if (fields.x and fields.x ~= "") then
                        local x = tonumber(fields.x)
                        if x then
                                meta.x_size = math.max(x, 1)
                        end
                end
                if (fields.y and fields.y ~= "") then
                        local y = tonumber(fields.y)
                        if y then
                                meta.y_size = math.max(y, 1)
                        end
                end
                if (fields.z and fields.z ~= "") then
                        local z = tonumber(fields.z)
                        if z then
                                meta.z_size = math.max(z, 1)
                        end
                end

                -- Save schematic name
                if fields.name then
                        meta.schem_name = fields.name
                end

                -- Node conversion
                if (fields.air2void) then
                        local pos1, pos2 = schemedit.size(pos)
                        pos1, pos2 = schemedit.sort_pos(pos1, pos2)
                        local nodes = minetest.find_nodes_in_area(pos1, pos2, {"air"})
                        minetest.bulk_set_node(nodes, {name="schemedit:void"})
                        return
                elseif (fields.void2air) then
                        local pos1, pos2 = schemedit.size(pos)
                        pos1, pos2 = schemedit.sort_pos(pos1, pos2)
                        local nodes = minetest.find_nodes_in_area(pos1, pos2, {"schemedit:void"})
                        minetest.bulk_set_node(nodes, {name="air"})
                        return
                end

                -- Toggle border
                if fields.border then
                        if meta.schem_border == "true" and schemedit.markers[hashpos] then
                                schemedit.unmark(pos)
                                meta.schem_border = "false"
                        else
                                schemedit.mark(pos)
                                meta.schem_border = "true"
                        end
                end

                -- Export schematic
                if fields.export and meta.schem_name and meta.schem_name ~= "" then
                        local pos1, pos2 = schemedit.size(pos)
                        pos1, pos2 = schemedit.sort_pos(pos1, pos2)
                        local path = export_path_full .. DIR_DELIM
                        minetest.mkdir(path)

                        local plist = schemedit.scan_metadata(pos1, pos2)
                        local probability_list = {}
                        for hash, i in pairs(plist) do
                                local prob = schemedit.lua_prob_to_schematic_prob(i.prob)
                                if i.force_place == true then
                                        prob = prob + 128
                                end

                                table.insert(probability_list, {
                                        pos = minetest.get_position_from_hash(hash),
                                        prob = prob,
                                })
                        end

                        local slist = minetest.deserialize(meta.slices)
                        local slice_list = {}
                        for _, i in pairs(slist) do
                                slice_list[#slice_list + 1] = {
                                        ypos = pos.y + i.ypos,
                                        prob = schemedit.lua_prob_to_schematic_prob(i.prob),
                                }
                        end

                        local filepath = path..meta.schem_name..".mts"
                        local res = minetest.create_schematic(pos1, pos2, probability_list, filepath, slice_list)

                        if res then
                                minetest.chat_send_player(name, minetest.colorize("#00ff00",
                                                S("Exported schematic to @1", filepath)))
                                -- Additional export to Lua file if MTS export was successful
                                local schematic = minetest.read_schematic(filepath, {})
                                if schematic and minetest.settings:get_bool("schemedit_export_lua") then
                                        local filepath_lua = path..meta.schem_name..".lua"
                                        res = export_schematic_to_lua(schematic, filepath_lua)
                                        if res then
                                                minetest.chat_send_player(name, minetest.colorize("#00ff00",
                                                                S("Exported schematic to @1", filepath_lua)))
                                        end
                                end
                        else
                                minetest.chat_send_player(name, minetest.colorize("red",
                                                S("Failed to export schematic to @1", filepath)))
                        end
                end

                -- Import schematic
                if fields.import and meta.schem_name and meta.schem_name ~= "" then
                        if not can_import then
                                return
                        end
                        local pos1
                        local node = minetest.get_node(pos)
                        local path = export_path_full .. DIR_DELIM

                        local filepath = path..meta.schem_name..".mts"
                        local schematic = minetest.read_schematic(filepath, {write_yslice_prob="low"})
                        local success = false

                        if schematic then
                                meta.x_size = schematic.size.x
                                meta.y_size = schematic.size.y
                                meta.z_size = schematic.size.z
                                meta.slices = minetest.serialize(renumber(schematic.yslice_prob))
                                local special_x_size = meta.x_size
                                local special_y_size = meta.y_size
                                local special_z_size = meta.z_size

                                if node.param2 == 1 then
                                        pos1 = vector.add(pos, {x=1,y=0,z=-meta.z_size+1})
                                        meta.x_size, meta.z_size = meta.z_size, meta.x_size
                                elseif node.param2 == 2 then
                                        pos1 = vector.add(pos, {x=-meta.x_size+1,y=0,z=-meta.z_size})
                                elseif node.param2 == 3 then
                                        pos1 = vector.add(pos, {x=-meta.x_size,y=0,z=0})
                                        meta.x_size, meta.z_size = meta.z_size, meta.x_size
                                else
                                        pos1 = vector.add(pos, {x=0,y=0,z=1})
                                end

                                local schematic_for_meta = table.copy(schematic)
                                -- Strip probability data for placement
                                schematic.yslice_prob = {}
                                for d=1, #schematic.data do
                                        schematic.data[d].prob = nil
                                end

                                -- Place schematic
                                success = minetest.place_schematic(pos1, schematic, "0", nil, true)

                                -- Add special schematic data to nodes
                                if success then
                                        local d = 1
                                        for z=0, special_z_size-1 do
                                        for y=0, special_y_size-1 do
                                        for x=0, special_x_size-1 do
                                                local data = schematic_for_meta.data[d]
                                                local pp = {x=pos1.x+x, y=pos1.y+y, z=pos1.z+z}
                                                if data.prob == 0 then
                                                        minetest.set_node(pp, {name="schemedit:void"})
                                                else
                                                        local meta = minetest.get_meta(pp)
                                                        if data.prob and data.prob ~= 255 and data.prob ~= 254 then
                                                                meta:set_string("schemedit_prob", tostring(data.prob))
                                                        else
                                                                meta:set_string("schemedit_prob", "")
                                                        end
                                                        if data.force_place then
                                                                meta:set_string("schemedit_force_place", "true")
                                                        else
                                                                meta:set_string("schemedit_force_place", "")
                                                        end
                                                end
                                                d = d + 1
                                        end
                                        end
                                        end
                                end
                        end
                        if success then
                                minetest.chat_send_player(name, minetest.colorize("#00ff00",
                                                S("Imported schematic from @1", filepath)))
                        else
                                minetest.chat_send_player(name, minetest.colorize("red",
                                                S("Failed to import schematic from @1", filepath)))
                        end
                end



                -- Save meta before updating visuals
                local inv = realmeta:get_inventory():get_lists()
                realmeta:from_table({fields = meta, inventory = inv})

                -- Update border
                if not fields.border and meta.schem_border == "true" then
                        schemedit.mark(pos)
                end

                -- Update formspec
                if not fields.quit then
                        schemedit.show_formspec(pos, minetest.get_player_by_name(name), "main")
                end
        end,
})

schemedit.add_form("slice", {
        caption = S("Y Slices"),
        tab = true,
        get = function(self, pos, name, visible_panel)
                local meta = minetest.get_meta(pos):to_table().fields

                self.selected = self.selected or 1
                local selected = tostring(self.selected)
                local slice_list = minetest.deserialize(meta.slices)
                local slices = ""
                for _, i in pairs(slice_list) do
                        local insert = F(S("Y = @1; Probability = @2", tostring(i.ypos), tostring(i.prob)))
                        slices = slices..insert..","
                end
                slices = slices:sub(1, -2) -- Remove final comma

                local form = [[
                        size[7,8]
                        table[0,0;6.8,6;slices;]]..slices..[[;]]..selected..[[]
                ]]

                if self.panel_add or self.panel_edit then
                        local ypos_default, prob_default = "", ""
                        local done_button = "button[5,7.18;2,1;done_add;"..F(S("Add")).."]"
                        if self.panel_edit then
                                done_button = "button[5,7.18;2,1;done_edit;"..F(S("Apply")).."]"
                                if slice_list[self.selected] then
                                        ypos_default = slice_list[self.selected].ypos
                                        prob_default = slice_list[self.selected].prob
                                end
                        end

                        local field_ypos = ""
                        if self.panel_add then
                                field_ypos = "field[0.3,7.5;2.5,1;ypos;"..F(S("Y position (max. @1):", (meta.y_size - 1)))..";"..ypos_default.."]"
                        end

                        form = form..[[
                                ]]..field_ypos..[[
                                field[2.8,7.5;2.5,1;prob;]]..F(S("Probability (0-255):"))..[[;]]..prob_default..[[]
                                field_close_on_enter[ypos;false]
                                field_close_on_enter[prob;false]
                        ]]..done_button
                end

                if not self.panel_edit then
                        if self.panel_add then
                                form = form.."button[0,6;2.4,1;add;"..F(S("Cancel")).."]"
                        else
                                form = form.."button[0,6;2.4,1;add;"..F(S("Add slice")).."]"
                        end
                end

                if slices ~= "" and self.selected and not self.panel_add then
                        if not self.panel_edit then
                                form = form..[[
                                        button[2.4,6;2.4,1;remove;]]..F(S("Remove slice"))..[[]
                                        button[4.8,6;2.4,1;edit;]]..F(S("Edit slice"))..[[]
                                ]]
                        else
                                form = form..[[
                                        button[4.8,6;2.4,1;edit;]]..F(S("Back"))..[[]
                                ]]
                        end
                end

                return form
        end,
        handle = function(self, pos, name, fields)
                if not check_priv(name, fields.quit) then
                        return
                end

                local meta = minetest.get_meta(pos)
                local player = minetest.get_player_by_name(name)

                if fields.slices then
                        local slices = fields.slices:split(":")
                        self.selected = tonumber(slices[2])
                end

                if fields.add then
                        if not self.panel_add then
                                self.panel_add = true
                                schemedit.show_formspec(pos, player, "slice")
                        else
                                self.panel_add = nil
                                schemedit.show_formspec(pos, player, "slice")
                        end
                end

                local ypos, prob = tonumber(fields.ypos), tonumber(fields.prob)
                if fields.done_edit then
                        ypos = 0
                end
                if (fields.done_add or fields.done_edit) and ypos and prob and
                                 ypos <= (meta:get_int("y_size") - 1) and prob >= 0 and prob <= 255 then
                        local slice_list = minetest.deserialize(meta:get_string("slices"))
                        local index = #slice_list + 1
                        if fields.done_edit then
                                index = self.selected
                        end

                        local dupe = false
                        if fields.done_add then
                                for k,v in pairs(slice_list) do
                                        if v.ypos == ypos then
                                                v.prob = prob
                                                dupe = true
                                        end
                                end
                        end
                        if fields.done_edit and slice_list[index] then
                                ypos = slice_list[index].ypos
                        end
                        if not dupe then
                                slice_list[index] = {ypos = ypos, prob = prob}
                        end

                        meta:set_string("slices", minetest.serialize(slice_list))

                        -- Update and show formspec
                        self.panel_add = nil
                        schemedit.show_formspec(pos, player, "slice")
                end

                if fields.remove and self.selected then
                        local slice_list = minetest.deserialize(meta:get_string("slices"))
                        slice_list[self.selected] = nil
                        meta:set_string("slices", minetest.serialize(renumber(slice_list)))

                        -- Update formspec
                        self.selected = math.max(1, self.selected-1)
                        self.panel_edit = nil
                        schemedit.show_formspec(pos, player, "slice")
                end

                if fields.edit then
                        if not self.panel_edit then
                                self.panel_edit = true
                                schemedit.show_formspec(pos, player, "slice")
                        else
                                self.panel_edit = nil
                                schemedit.show_formspec(pos, player, "slice")
                        end
                end
        end,
})

schemedit.add_form("probtool", {
        cache_name = false,
        caption = S("Schematic Node Probability Tool"),
        get = function(self, pos, name)
                local player = minetest.get_player_by_name(name)
                if not player then
                        return
                end
                local probtool = player:get_wielded_item()
                if probtool:get_name() ~= "schemedit:probtool" then
                        return
                end

                local meta = probtool:get_meta()
                local prob = tonumber(meta:get_string("schemedit_prob"))
                local force_place = meta:get_string("schemedit_force_place")

                if not prob then
                        prob = 255
                end
                if force_place == nil or force_place == "" then
                        force_place = "false"
                end
                local form = "size[5,4]"..
                        "label[0,0;"..F(S("Schematic Node Probability Tool")).."]"..
                        "field[0.75,1;4,1;prob;"..F(S("Probability (0-255)"))..";"..prob.."]"..
                        "checkbox[0.60,1.5;force_place;"..F(S("Force placement"))..";" .. force_place .. "]" ..
                        "button_exit[0.25,3;2,1;cancel;"..F(S("Cancel")).."]"..
                        "button_exit[2.75,3;2,1;submit;"..F(S("Apply")).."]"..
                        "tooltip[prob;"..F(S("Probability that the node will be placed")).."]"..
                        "tooltip[force_place;"..F(S("If enabled, the node will replace nodes other than air and ignore")).."]"..
                        "field_close_on_enter[prob;false]"
                return form
        end,
        handle = function(self, pos, name, fields)
                if not check_priv(name, fields.quit) then
                        return
                end

                if fields.submit then
                        local prob = tonumber(fields.prob)
                        if prob then
                                local player = minetest.get_player_by_name(name)
                                if not player then
                                        return
                                end
                                local probtool = player:get_wielded_item()
                                if probtool:get_name() ~= "schemedit:probtool" then
                                        return
                                end

                                local force_place = self.force_place == true

                                set_item_metadata(probtool, prob, force_place)

                                -- Repurpose the tool's wear bar to display the set probability
                                probtool:set_wear(math.floor(((255-prob)/255)*65535))

                                player:set_wielded_item(probtool)
                        end
                end
                if fields.force_place == "true" then
                        self.force_place = true
                elseif fields.force_place == "false" then
                        self.force_place = false
                end
        end,
})

---
--- API
---

--- Copies and modifies positions `pos1` and `pos2` so that each component of
-- `pos1` is less than or equal to the corresponding component of `pos2`.
-- Returns the new positions.
function schemedit.sort_pos(pos1, pos2)
        if not pos1 or not pos2 then
                return
        end

        pos1, pos2 = table.copy(pos1), table.copy(pos2)
        if pos1.x > pos2.x then
                pos2.x, pos1.x = pos1.x, pos2.x
        end
        if pos1.y > pos2.y then
                pos2.y, pos1.y = pos1.y, pos2.y
        end
        if pos1.z > pos2.z then
                pos2.z, pos1.z = pos1.z, pos2.z
        end
        return pos1, pos2
end

-- [function] Prepare size
function schemedit.size(pos)
        local pos1   = vector.new(pos)
        local meta   = minetest.get_meta(pos)
        local node   = minetest.get_node(pos)
        local param2 = node.param2
        local size   = {
                x = meta:get_int("x_size"),
                y = math.max(meta:get_int("y_size") - 1, 0),
                z = meta:get_int("z_size"),
        }

        if param2 == 1 then
                local new_pos = vector.add({x = size.z, y = size.y, z = -size.x}, pos)
                pos1.x = pos1.x + 1
                new_pos.z = new_pos.z + 1
                return pos1, new_pos
        elseif param2 == 2 then
                local new_pos = vector.add({x = -size.x, y = size.y, z = -size.z}, pos)
                pos1.z = pos1.z - 1
                new_pos.x = new_pos.x + 1
                return pos1, new_pos
        elseif param2 == 3 then
                local new_pos = vector.add({x = -size.z, y = size.y, z = size.x}, pos)
                pos1.x = pos1.x - 1
                new_pos.z = new_pos.z - 1
                return pos1, new_pos
        else
                local new_pos = vector.add(size, pos)
                pos1.z = pos1.z + 1
                new_pos.x = new_pos.x - 1
                return pos1, new_pos
        end
end

-- [function] Mark region
function schemedit.mark(pos)
        schemedit.unmark(pos)

        local id = minetest.hash_node_position(pos)
        local owner = minetest.get_meta(pos):get_string("owner")
        local pos1, pos2 = schemedit.size(pos)
        pos1, pos2 = schemedit.sort_pos(pos1, pos2)

        local thickness = 0.2
        local sizex, sizey, sizez = (1 + pos2.x - pos1.x) / 2, (1 + pos2.y - pos1.y) / 2, (1 + pos2.z - pos1.z) / 2
        local m = {}
        local low = true
        local offset

        -- XY plane markers
        for _, z in ipairs({pos1.z - 0.5, pos2.z + 0.5}) do
                if low then
                        offset = -0.01
                else
                        offset = 0.01
                end
                local marker = minetest.add_entity({x = pos1.x + sizex - 0.5, y = pos1.y + sizey - 0.5, z = z + offset}, "schemedit:display")
                if marker ~= nil then
                        marker:set_properties({
                                visual_size={x=(sizex+0.01) * 2, y=(sizey+0.01) * 2},
                        })
                        marker:get_luaentity().id = id
                        marker:get_luaentity().owner = owner
                        table.insert(m, marker)
                end
                low = false
        end

        low = true
        -- YZ plane markers
        for _, x in ipairs({pos1.x - 0.5, pos2.x + 0.5}) do
                if low then
                        offset = -0.01
                else
                        offset = 0.01
                end

                local marker = minetest.add_entity({x = x + offset, y = pos1.y + sizey - 0.5, z = pos1.z + sizez - 0.5}, "schemedit:display")
                if marker ~= nil then
                        marker:set_properties({
                                visual_size={x=(sizez+0.01) * 2, y=(sizey+0.01) * 2},
                        })
                        marker:set_rotation({x=0, y=math.pi / 2, z=0})
                        marker:get_luaentity().id = id
                        marker:get_luaentity().owner = owner
                        table.insert(m, marker)
                end
                low = false
        end

        low = true
        -- XZ plane markers
        for _, y in ipairs({pos1.y - 0.5, pos2.y + 0.5}) do
                if low then
                        offset = -0.01
                else
                        offset = 0.01
                end

                local marker = minetest.add_entity({x = pos1.x + sizex - 0.5, y = y + offset, z = pos1.z + sizez - 0.5}, "schemedit:display")
                if marker ~= nil then
                        marker:set_properties({
                                visual_size={x=(sizex+0.01) * 2, y=(sizez+0.01) * 2},
                        })
                        marker:set_rotation({x=math.pi/2, y=0, z=0})
                        marker:get_luaentity().id = id
                        marker:get_luaentity().owner = owner
                        table.insert(m, marker)
                end
                low = false
        end



        schemedit.markers[id] = m
        return true
end

-- [function] Unmark region
function schemedit.unmark(pos)
        local id = minetest.hash_node_position(pos)
        if schemedit.markers[id] then
                local retval
                for _, entity in ipairs(schemedit.markers[id]) do
                        entity:remove()
                        retval = true
                end
                return retval
        end
end

---
--- Mark node probability values near player
---

-- Show probability and force_place status of a particular position for player in HUD.
-- Probability is shown as a number followed by “[F]” if the node is force-placed.
function schemedit.display_node_prob(player, pos, prob, force_place)
        local wpstring
        if prob and force_place == true then
                wpstring = string.format("%s [F]", prob)
        elseif prob and type(tonumber(prob)) == "number" then
                wpstring = prob
        elseif force_place == true then
                wpstring = "[F]"
        end
        if wpstring then
                return player:hud_add({
                        hud_elem_type = "waypoint",
                        name = wpstring,
                        precision = 0,
                        text = "m", -- For the distance artifact [legacy]
                        number = text_color_number,
                        world_pos = pos,
                })
        end
end

-- Display the node probabilities and force_place status of the nodes in a region.
-- By default, this is done for nodes near the player (distance: 5).
-- But the boundaries can optionally be set explicitly with pos1 and pos2.
function schemedit.display_node_probs_region(player, pos1, pos2)
        local playername = player:get_player_name()
        local pos = vector.round(player:get_pos())

        local dist = 5
        -- Default: 5 nodes away from player in any direction
        if not pos1 then
                pos1 = vector.subtract(pos, dist)
        end
        if not pos2 then
                pos2 = vector.add(pos, dist)
        end
        for x=pos1.x, pos2.x do
                for y=pos1.y, pos2.y do
                        for z=pos1.z, pos2.z do
                                local checkpos = {x=x, y=y, z=z}
                                local nodehash = minetest.hash_node_position(checkpos)

                                -- If node is already displayed, remove it so it can re replaced later
                                if displayed_waypoints[playername][nodehash] then
                                        player:hud_remove(displayed_waypoints[playername][nodehash])
                                        displayed_waypoints[playername][nodehash] = nil
                                end

                                local prob, force_place
                                local meta = minetest.get_meta(checkpos)
                                prob = meta:get_string("schemedit_prob")
                                force_place = meta:get_string("schemedit_force_place") == "true"
                                local hud_id = schemedit.display_node_prob(player, checkpos, prob, force_place)
                                if hud_id then
                                        displayed_waypoints[playername][nodehash] = hud_id
                                        displayed_waypoints[playername].display_active = true
                                end
                        end
                end
        end
end

-- Remove all active displayed node statuses.
function schemedit.clear_displayed_node_probs(player)
        local playername = player:get_player_name()
        for nodehash, hud_id in pairs(displayed_waypoints[playername]) do
                if nodehash ~= "display_active" then
                        player:hud_remove(hud_id)
                        displayed_waypoints[playername][nodehash] = nil
                        displayed_waypoints[playername].display_active = false
                end
        end
end

minetest.register_on_joinplayer(function(player)
        displayed_waypoints[player:get_player_name()] = {
                display_active = false  -- If true, there *might* be at least one active node prob HUD display
                                        -- If false, no node probabilities are displayed for sure.
        }
end)

minetest.register_on_leaveplayer(function(player)
        displayed_waypoints[player:get_player_name()] = nil
end)

-- Regularily clear the displayed node probabilities and force_place
-- for all players who do not wield the probtool.
-- This makes sure the screen is not spammed with information when it
-- isn't needed.
local cleartimer = 0
minetest.register_globalstep(function(dtime)
        cleartimer = cleartimer + dtime
        if cleartimer > 2 then
                local players = minetest.get_connected_players()
                for p = 1, #players do
                        local player = players[p]
                        local pname = player:get_player_name()
                        if displayed_waypoints[pname].display_active then
                                local item = player:get_wielded_item()
                                if item:get_name() ~= "schemedit:probtool" then
                                        schemedit.clear_displayed_node_probs(player)
                                end
                        end
                end
                cleartimer = 0
        end
end)

---
--- Registrations
---

-- [priv] schematic_override
minetest.register_privilege("schematic_override", {
        description = S("Allows you to access schemedit nodes not owned by you"),
        give_to_singleplayer = false,
})

local help_import = ""
if can_import then
        help_import = S("Importing a schematic will load a schematic from the world directory, place it in front of the schematic creator and sets probability and force-place data accordingly.").."\n"
end

-- [node] Schematic creator
minetest.register_node("schemedit:creator", {
        description = S("Schematic Creator"),
        _doc_items_longdesc = S("The schematic creator is used to save a region of the world into a schematic file (.mts)."),
        _doc_items_usagehelp = S("To get started, place the block facing directly in front of any bottom left corner of the structure you want to save. This block can only be accessed by the placer or by anyone with the “schematic_override” privilege.").."\n"..
S("To save a region, use the block, enter the size and a schematic name and hit “Export schematic”. The file will always be saved in the world directory. Note you can use this name in the /placeschem command to place the schematic again.").."\n\n"..
help_import..
S("The other features of the schematic creator are optional and are used to allow to add randomness and fine-tuning.").."\n\n"..
S("Y slices are used to remove entire slices based on chance. For each slice of the schematic region along the Y axis, you can specify that it occurs only with a certain chance. In the Y slice tab, you have to specify the Y slice height (0 = bottom) and a probability from 0 to 255 (255 is for 100%). By default, all Y slices occur always.").."\n\n"..
S("With a schematic node probability tool, you can set a probability for each node and enable them to overwrite all nodes when placed as schematic. This tool must be used prior to the file export."),
        tiles = {"schemedit_creator_top.png", "schemedit_creator_bottom.png",
                        "schemedit_creator_sides.png"},
        groups = { dig_immediate = 2},
        paramtype2 = "facedir",
        is_ground_content = false,

        after_place_node = function(pos, player)
                local name = player:get_player_name()
                local meta = minetest.get_meta(pos)

                meta:set_string("owner", name)
                meta:set_string("infotext", S("Schematic Creator").."\n"..S("(owned by @1)", name))
                meta:set_string("prob_list", minetest.serialize({}))
                meta:set_string("slices", minetest.serialize({}))

                local node = minetest.get_node(pos)
                local dir  = minetest.facedir_to_dir(node.param2)

                meta:set_int("x_size", 1)
                meta:set_int("y_size", 1)
                meta:set_int("z_size", 1)

                -- Don't take item from itemstack
                return true
        end,
        can_dig = function(pos, player)
                local name = player:get_player_name()
                local meta = minetest.get_meta(pos)
                if meta:get_string("owner") == name or
                                minetest.check_player_privs(player, "schematic_override") == true then
                        return true
                end

                return false
        end,
        on_rightclick = function(pos, node, player)
                local meta = minetest.get_meta(pos)
                local name = player:get_player_name()
                if meta:get_string("owner") == name or
                                minetest.check_player_privs(player, "schematic_override") == true then
                        -- Get player attribute
                        local pmeta = player:get_meta()
                        local tab = pmeta:get_string("schemedit:tab")
                        if not forms[tab] or not tab then
                                tab = "main"
                        end

                        schemedit.show_formspec(pos, player, tab, true)
                end
        end,
        after_destruct = function(pos)
                schemedit.unmark(pos)
        end,

        -- No support for Minetest Game's screwdriver
        on_rotate = false,
})

minetest.register_tool("schemedit:probtool", {
        description = S("Schematic Node Probability Tool"),
        _doc_items_longdesc =
S("This is an advanced tool which only makes sense when used together with a schematic creator. It is used to finetune the way how nodes from a schematic are placed.").."\n"..
S("It allows you to set two things:").."\n"..
S("1) Set probability: Chance for any particular node to be actually placed (default: always placed)").."\n"..
S("2) Enable force placement: These nodes replace node other than air and ignore when placed in a schematic (default: off)"),
        _doc_items_usagehelp = "\n"..
S("BASIC USAGE:").."\n"..
S("Punch to configure the tool. Select a probability (0-255; 255 is for 100%) and enable or disable force placement. Now place the tool on any node to apply these values to the node. This information is preserved in the node until it is destroyed or changed by the tool again. This tool has no effect on schematic voids.").."\n"..
S("Now you can use a schematic creator to save a region as usual, the nodes will now be saved with the special node settings applied.").."\n\n"..
S("NODE HUD:").."\n"..
S("To help you remember the node values, the nodes with special values are labelled in the HUD. The first line shows probability and force placement (with “[F]”). The second line is the current distance to the node. Nodes with default settings and schematic voids are not labelled.").."\n"..
S("To disable the node HUD, unselect the tool or hit “place” while not pointing anything.").."\n\n"..
S("UPDATING THE NODE HUD:").."\n"..
S("The node HUD is not updated automatically and may be outdated. The node HUD only updates the HUD for nodes close to you whenever you place the tool or press the punch and sneak keys simultaneously. If you sneak-punch a schematic creator, then the node HUD is updated for all nodes within the schematic creator's region, even if this region is very big."),
        wield_image = "schemedit_probtool.png",
        inventory_image = "schemedit_probtool.png",
        liquids_pointable = true,
        groups = { disable_repair = 1 },
        on_use = function(itemstack, user, pointed_thing)
                local uname = user:get_player_name()
                if uname and not check_priv(uname) then
                        return
                end

                local ctrl = user:get_player_control()
                -- Simple use
                if not ctrl.sneak then
                        -- Open dialog to change the probability to apply to nodes
                        schemedit.show_formspec(user:get_pos(), user, "probtool", true)

                -- Use + sneak
                else
                        -- Display the probability and force_place values for nodes.

                        -- If a schematic creator was punched, only enable display for all nodes
                        -- within the creator's region.
                        local use_creator_region = false
                        if pointed_thing and pointed_thing.type == "node" and pointed_thing.under then
                                local punchpos = pointed_thing.under
                                local node = minetest.get_node(punchpos)
                                if node.name == "schemedit:creator" then
                                        local pos1, pos2 = schemedit.size(punchpos)
                                        pos1, pos2 = schemedit.sort_pos(pos1, pos2)
                                        schemedit.display_node_probs_region(user, pos1, pos2)
                                        return
                                end
                        end

                        -- Otherwise, just display the region close to the player
                        schemedit.display_node_probs_region(user)
                end
        end,
        on_secondary_use = function(itemstack, user, pointed_thing)
                local uname = user:get_player_name()
                if uname and not check_priv(uname) then
                        return
                end

                schemedit.clear_displayed_node_probs(user)
        end,
        -- Set note probability and force_place and enable node probability display
        on_place = function(itemstack, placer, pointed_thing)
                local pname = placer:get_player_name()
                if pname and not check_priv(pname) then
                        return
                end

                -- Use pointed node's on_rightclick function first, if present
                local node = minetest.get_node(pointed_thing.under)
                if placer and not placer:get_player_control().sneak then
                        if minetest.registered_nodes[node.name] and minetest.registered_nodes[node.name].on_rightclick then
                                return minetest.registered_nodes[node.name].on_rightclick(pointed_thing.under, node, placer, itemstack) or itemstack
                        end
                end

                -- This sets the node probability of pointed node to the
                -- currently used probability stored in the tool.
                local pos = pointed_thing.under
                local node = minetest.get_node(pos)
                -- Schematic void are ignored, they always have probability 0
                if node.name == "schemedit:void" then
                        return itemstack
                end
                local nmeta = minetest.get_meta(pos)
                local imeta = itemstack:get_meta()
                local prob = tonumber(imeta:get_string("schemedit_prob"))
                local force_place = imeta:get_string("schemedit_force_place")

                if not prob or prob == 255 then
                        nmeta:set_string("schemedit_prob", nil)
                else
                        nmeta:set_string("schemedit_prob", prob)
                end
                if force_place == "true" then
                        nmeta:set_string("schemedit_force_place", "true")
                else
                        nmeta:set_string("schemedit_force_place", nil)
                end

                -- Enable node probablity display
                schemedit.display_node_probs_region(placer)

                return itemstack
        end,
})

local use_texture_alpha_void
if minetest.features.use_texture_alpha_string_modes then
        use_texture_alpha_void = "clip"
else
        use_texture_alpha_void = true
end

minetest.register_node("schemedit:void", {
        description = S("Schematic Void"),
        _doc_items_longdesc = S("This is an utility block used in the creation of schematic files. It should be used together with a schematic creator. When saving a schematic, all nodes with a schematic void will be left unchanged when the schematic is placed again. Technically, this is equivalent to a block with the node probability set to 0."),
        _doc_items_usagehelp = S("Just place the schematic void like any other block and use the schematic creator to save a portion of the world."),
        tiles = { "schemedit_void.png" },
        use_texture_alpha = use_texture_alpha_void,
        drawtype = "nodebox",
        is_ground_content = false,
        paramtype = "light",
        walkable = false,
        sunlight_propagates = true,
        node_box = {
                type = "fixed",
                fixed = {
                        { -4/16, -4/16, -4/16, 4/16, 4/16, 4/16 },
                },
        },
        groups = { dig_immediate = 3},
})

-- [entity] Visible schematic border
minetest.register_entity("schemedit:display", {
        visual = "upright_sprite",
        textures = {"schemedit_border.png"},
        visual_size = {x=10, y=10},
        pointable = false,
        physical = false,
        static_save = false,
        glow = minetest.LIGHT_MAX,

        on_step = function(self, dtime)
                if not self.id then
                        self.object:remove()
                elseif not schemedit.markers[self.id] then
                        self.object:remove()
                end
        end,
        on_activate = function(self)
                self.object:set_armor_groups({immortal = 1})
        end,
})

minetest.register_lbm({
        label = "Reset schematic creator border entities",
        name = "schemedit:reset_border",
        nodenames = "schemedit:creator",
        run_at_every_load = true,
        action = function(pos, node)
                local meta = minetest.get_meta(pos)
                meta:set_string("schem_border", "false")
        end,
})

local function add_suffix(schem)
        -- Automatically add file name suffix if omitted
        local schem_full, schem_lua
        if string.sub(schem, string.len(schem)-3, string.len(schem)) == ".mts" then
                schem_full = schem
                schem_lua = string.sub(schem, 1, -5) .. ".lua"
        else
                schem_full = schem .. ".mts"
                schem_lua = schem .. ".lua"
        end
        return schem_full, schem_lua
end

-- [chatcommand] Place schematic
minetest.register_chatcommand("placeschem", {
        description = S("Place schematic at the position specified or the current player position (loaded from @1). “-c” will clear the area first", export_path_trunc),
        privs = {server = true},
        params = S("<schematic name>[.mts] [-c] [<x> <y> <z>]"),
        func = function(name, param)
                local schem, clear, p = string.match(param, "^([^ ]+) +(%-c) +(.*)$")
                if not schem then
                        schem, p = string.match(param, "^([^ ]+) +(.*)$")
                end
                clear = clear == "-c"

                local pos = minetest.string_to_pos(p)

                if not schem then
                        return false, S("No schematic file specified.")
                end

                if not pos then
                        pos = minetest.get_player_by_name(name):get_pos()
                end

                local schem_full, schem_lua = add_suffix(schem)
                local success = false
                local schem_path = export_path_full .. DIR_DELIM .. schem_full
                if minetest.read_schematic then
                        -- We don't call minetest.place_schematic with the path name directly because
                        -- this would trigger the caching and we wouldn't get any updates to the schematic
                        -- files when we reload. minetest.read_schematic circumvents that.
                        local schematic = minetest.read_schematic(schem_path, {})
                        if schematic then
                                if clear then
                                        -- Clear same size for X and Z because
                                        -- because schematic is randomly rotated
                                        local max_xz = math.max(schematic.size.x, schematic.size.z)
                                        local posses = {}
                                        for z=pos.z, pos.z+max_xz-1 do
                                        for y=pos.y, pos.y+schematic.size.y-1 do
                                        for x=pos.x, pos.x+max_xz-1 do
                                                table.insert(posses, {x=x,y=y,z=z})
                                        end
                                        end
                                        end
                                        minetest.bulk_set_node(posses, {name="air"})
                                end
                                success = minetest.place_schematic(pos, schematic, "random", nil, false)
                        end
                else
                        -- Legacy support for Minetest versions that do not have minetest.read_schematic.
                        -- Note: "-c" is ignored here.
                        success = minetest.place_schematic(schem_path, schematic, "random", nil, false)
                end

                if success == nil then
                        return false, S("Schematic file could not be loaded!")
                else
                        return true
                end
        end,
})

if can_import then
-- [chatcommand] Convert MTS schematic file to .lua file
minetest.register_chatcommand("mts2lua", {
        description = S("Convert .mts schematic file to .lua file (loaded from @1)", export_path_trunc),
        privs = {server = true},
        params = S("<schematic name>[.mts] [comments]"),
        func = function(name, param)
                local schem, comments_str = string.match(param, "^([^ ]+) *(.*)$")

                if not schem then
                        return false, S("No schematic file specified.")
                end

                local comments = comments_str == "comments"

                -- Automatically add file name suffix if omitted
                local schem_full, schem_lua = add_suffix(schem)
                local schem_path = export_path_full .. DIR_DELIM .. schem_full
                local schematic = minetest.read_schematic(schem_path, {})

                if schematic then
                        local str = minetest.serialize_schematic(schematic, "lua", {lua_use_comments=comments})
                        local lua_path = export_path_full .. DIR_DELIM .. schem_lua
                        local file = io.open(lua_path, "w")
                        if file and str then
                                file:write(str)
                                file:flush()
                                file:close()
                                return true, S("Exported schematic to @1", lua_path)
                        else
                                return false, S("Failed!")
                        end
                end
        end,
})
end

if MAKE_README then
        dofile(minetest.get_modpath("schemedit")..DIR_DELIM.."make_readme.lua")
end