builtin/common/misc_helpers.lua

   1 -- Minetest: builtin/misc_helpers.lua
   2
   3 --------------------------------------------------------------------------------
   4 -- Localize functions to avoid table lookups (better performance).
   5 local string_sub, string_find = string.sub, string.find
   6
   7 --------------------------------------------------------------------------------
   8 local function basic_dump(o)
   9         local tp = type(o)
  10         if tp == "number" then
  11                 return tostring(o)
  12         elseif tp == "string" then
  13                 return string.format("%q", o)
  14         elseif tp == "boolean" then
  15                 return tostring(o)
  16         elseif tp == "nil" then
  17                 return "nil"
  18         -- Uncomment for full function dumping support.
  19         -- Not currently enabled because bytecode isn't very human-readable and
  20         -- dump's output is intended for humans.
  21         --elseif tp == "function" then
  22         --      return string.format("loadstring(%q)", string.dump(o))
  23         elseif tp == "userdata" then
  24                 return tostring(o)
  25         else
  26                 return string.format("<%s>", tp)
  27         end
  28 end
  29
  30 local keywords = {
  31         ["and"] = true,
  32         ["break"] = true,
  33         ["do"] = true,
  34         ["else"] = true,
  35         ["elseif"] = true,
  36         ["end"] = true,
  37         ["false"] = true,
  38         ["for"] = true,
  39         ["function"] = true,
  40         ["goto"] = true,  -- Lua 5.2
  41         ["if"] = true,
  42         ["in"] = true,
  43         ["local"] = true,
  44         ["nil"] = true,
  45         ["not"] = true,
  46         ["or"] = true,
  47         ["repeat"] = true,
  48         ["return"] = true,
  49         ["then"] = true,
  50         ["true"] = true,
  51         ["until"] = true,
  52         ["while"] = true,
  53 }
  54 local function is_valid_identifier(str)
  55         if not str:find("^[a-zA-Z_][a-zA-Z0-9_]*$") or keywords[str] then
  56                 return false
  57         end
  58         return true
  59 end
  60
  61 --------------------------------------------------------------------------------
  62 -- Dumps values in a line-per-value format.
  63 -- For example, {test = {"Testing..."}} becomes:
  64 --   _["test"] = {}
  65 --   _["test"][1] = "Testing..."
  66 -- This handles tables as keys and circular references properly.
  67 -- It also handles multiple references well, writing the table only once.
  68 -- The dumped argument is internal-only.
  69
  70 function dump2(o, name, dumped)
  71         name = name or "_"
  72         -- "dumped" is used to keep track of serialized tables to handle
  73         -- multiple references and circular tables properly.
  74         -- It only contains tables as keys.  The value is the name that
  75         -- the table has in the dump, eg:
  76         -- {x = {"y"}} -> dumped[{"y"}] = '_["x"]'
  77         dumped = dumped or {}
  78         if type(o) ~= "table" then
  79                 return string.format("%s = %s\n", name, basic_dump(o))
  80         end
  81         if dumped[o] then
  82                 return string.format("%s = %s\n", name, dumped[o])
  83         end
  84         dumped[o] = name
  85         -- This contains a list of strings to be concatenated later (because
  86         -- Lua is slow at individual concatenation).
  87         local t = {}
  88         for k, v in pairs(o) do
  89                 local keyStr
  90                 if type(k) == "table" then
  91                         if dumped[k] then
  92                                 keyStr = dumped[k]
  93                         else
  94                                 -- Key tables don't have a name, so use one of
  95                                 -- the form _G["table: 0xFFFFFFF"]
  96                                 keyStr = string.format("_G[%q]", tostring(k))
  97                                 -- Dump key table
  98                                 t[#t + 1] = dump2(k, keyStr, dumped)
  99                         end
 100                 else
 101                         keyStr = basic_dump(k)
 102                 end
 103                 local vname = string.format("%s[%s]", name, keyStr)
 104                 t[#t + 1] = dump2(v, vname, dumped)
 105         end
 106         return string.format("%s = {}\n%s", name, table.concat(t))
 107 end
 108
 109 --------------------------------------------------------------------------------
 110 -- This dumps values in a one-statement format.
 111 -- For example, {test = {"Testing..."}} becomes:
 112 -- [[{
 113 --      test = {
 114 --              "Testing..."
 115 --      }
 116 -- }]]
 117 -- This supports tables as keys, but not circular references.
 118 -- It performs poorly with multiple references as it writes out the full
 119 -- table each time.
 120 -- The indent field specifies a indentation string, it defaults to a tab.
 121 -- Use the empty string to disable indentation.
 122 -- The dumped and level arguments are internal-only.
 123
 124 function dump(o, indent, nested, level)
 125         local t = type(o)
 126         if not level and t == "userdata" then
 127                 -- when userdata (e.g. player) is passed directly, print its metatable:
 128                 return "userdata metatable: " .. dump(getmetatable(o))
 129         end
 130         if t ~= "table" then
 131                 return basic_dump(o)
 132         end
 133
 134         -- Contains table -> true/nil of currently nested tables
 135         nested = nested or {}
 136         if nested[o] then
 137                 return "<circular reference>"
 138         end
 139         nested[o] = true
 140         indent = indent or "\t"
 141         level = level or 1
 142
 143         local ret = {}
 144         local dumped_indexes = {}
 145         for i, v in ipairs(o) do
 146                 ret[#ret + 1] = dump(v, indent, nested, level + 1)
 147                 dumped_indexes[i] = true
 148         end
 149         for k, v in pairs(o) do
 150                 if not dumped_indexes[k] then
 151                         if type(k) ~= "string" or not is_valid_identifier(k) then
 152                                 k = "["..dump(k, indent, nested, level + 1).."]"
 153                         end
 154                         v = dump(v, indent, nested, level + 1)
 155                         ret[#ret + 1] = k.." = "..v
 156                 end
 157         end
 158         nested[o] = nil
 159         if indent ~= "" then
 160                 local indent_str = "\n"..string.rep(indent, level)
 161                 local end_indent_str = "\n"..string.rep(indent, level - 1)
 162                 return string.format("{%s%s%s}",
 163                                 indent_str,
 164                                 table.concat(ret, ","..indent_str),
 165                                 end_indent_str)
 166         end
 167         return "{"..table.concat(ret, ", ").."}"
 168 end
 169
 170 --------------------------------------------------------------------------------
 171 function string.split(str, delim, include_empty, max_splits, sep_is_pattern)
 172         delim = delim or ","
 173         max_splits = max_splits or -2
 174         local items = {}
 175         local pos, len = 1, #str
 176         local plain = not sep_is_pattern
 177         max_splits = max_splits + 1
 178         repeat
 179                 local np, npe = string_find(str, delim, pos, plain)
 180                 np, npe = (np or (len+1)), (npe or (len+1))
 181                 if (not np) or (max_splits == 1) then
 182                         np = len + 1
 183                         npe = np
 184                 end
 185                 local s = string_sub(str, pos, np - 1)
 186                 if include_empty or (s ~= "") then
 187                         max_splits = max_splits - 1
 188                         items[#items + 1] = s
 189                 end
 190                 pos = npe + 1
 191         until (max_splits == 0) or (pos > (len + 1))
 192         return items
 193 end
 194
 195 --------------------------------------------------------------------------------
 196 function table.indexof(list, val)
 197         for i, v in ipairs(list) do
 198                 if v == val then
 199                         return i
 200                 end
 201         end
 202         return -1
 203 end
 204
 205 --------------------------------------------------------------------------------
 206 function string:trim()
 207         return (self:gsub("^%s*(.-)%s*$", "%1"))
 208 end
 209
 210 --------------------------------------------------------------------------------
 211 function math.hypot(x, y)
 212         local t
 213         x = math.abs(x)
 214         y = math.abs(y)
 215         t = math.min(x, y)
 216         x = math.max(x, y)
 217         if x == 0 then return 0 end
 218         t = t / x
 219         return x * math.sqrt(1 + t * t)
 220 end
 221
 222 --------------------------------------------------------------------------------
 223 function math.sign(x, tolerance)
 224         tolerance = tolerance or 0
 225         if x > tolerance then
 226                 return 1
 227         elseif x < -tolerance then
 228                 return -1
 229         end
 230         return 0
 231 end
 232
 233 --------------------------------------------------------------------------------
 234 function math.factorial(x)
 235         assert(x % 1 == 0 and x >= 0, "factorial expects a non-negative integer")
 236         if x >= 171 then
 237                 -- 171! is greater than the biggest double, no need to calculate
 238                 return math.huge
 239         end
 240         local v = 1
 241         for k = 2, x do
 242                 v = v * k
 243         end
 244         return v
 245 end
 246
 247
 248 function math.round(x)
 249         if x >= 0 then
 250                 return math.floor(x + 0.5)
 251         end
 252         return math.ceil(x - 0.5)
 253 end
 254
 255
 256 function core.formspec_escape(text)
 257         if text ~= nil then
 258                 text = string.gsub(text,"\\","\\\\")
 259                 text = string.gsub(text,"%]","\\]")
 260                 text = string.gsub(text,"%[","\\[")
 261                 text = string.gsub(text,";","\\;")
 262                 text = string.gsub(text,",","\\,")
 263         end
 264         return text
 265 end
 266
 267
 268 function core.wrap_text(text, max_length, as_table)
 269         local result = {}
 270         local line = {}
 271         if #text <= max_length then
 272                 return as_table and {text} or text
 273         end
 274
 275         for word in text:gmatch('%S+') do
 276                 local cur_length = #table.concat(line, ' ')
 277                 if cur_length > 0 and cur_length + #word + 1 >= max_length then
 278                         -- word wouldn't fit on current line, move to next line
 279                         table.insert(result, table.concat(line, ' '))
 280                         line = {}
 281                 end
 282                 table.insert(line, word)
 283         end
 284
 285         table.insert(result, table.concat(line, ' '))
 286         return as_table and result or table.concat(result, '\n')
 287 end
 288
 289 --------------------------------------------------------------------------------
 290
 291 if INIT == "game" then
 292         local dirs1 = {9, 18, 7, 12}
 293         local dirs2 = {20, 23, 22, 21}
 294
 295         function core.rotate_and_place(itemstack, placer, pointed_thing,
 296                         infinitestacks, orient_flags, prevent_after_place)
 297                 orient_flags = orient_flags or {}
 298
 299                 local unode = core.get_node_or_nil(pointed_thing.under)
 300                 if not unode then
 301                         return
 302                 end
 303                 local undef = core.registered_nodes[unode.name]
 304                 local sneaking = placer and placer:get_player_control().sneak
 305                 if undef and undef.on_rightclick and not sneaking then
 306                         return undef.on_rightclick(pointed_thing.under, unode, placer,
 307                                         itemstack, pointed_thing)
 308                 end
 309                 local fdir = placer and core.dir_to_facedir(placer:get_look_dir()) or 0
 310
 311                 local above = pointed_thing.above
 312                 local under = pointed_thing.under
 313                 local iswall = (above.y == under.y)
 314                 local isceiling = not iswall and (above.y < under.y)
 315
 316                 if undef and undef.buildable_to then
 317                         iswall = false
 318                 end
 319
 320                 if orient_flags.force_floor then
 321                         iswall = false
 322                         isceiling = false
 323                 elseif orient_flags.force_ceiling then
 324                         iswall = false
 325                         isceiling = true
 326                 elseif orient_flags.force_wall then
 327                         iswall = true
 328                         isceiling = false
 329                 elseif orient_flags.invert_wall then
 330                         iswall = not iswall
 331                 end
 332
 333                 local param2 = fdir
 334                 if iswall then
 335                         param2 = dirs1[fdir + 1]
 336                 elseif isceiling then
 337                         if orient_flags.force_facedir then
 338                                 param2 = 20
 339                         else
 340                                 param2 = dirs2[fdir + 1]
 341                         end
 342                 else -- place right side up
 343                         if orient_flags.force_facedir then
 344                                 param2 = 0
 345                         end
 346                 end
 347
 348                 local old_itemstack = ItemStack(itemstack)
 349                 local new_itemstack = core.item_place_node(itemstack, placer,
 350                                 pointed_thing, param2, prevent_after_place)
 351                 return infinitestacks and old_itemstack or new_itemstack
 352         end
 353
 354
 355 --------------------------------------------------------------------------------
 356 --Wrapper for rotate_and_place() to check for sneak and assume Creative mode
 357 --implies infinite stacks when performing a 6d rotation.
 358 --------------------------------------------------------------------------------
 359         core.rotate_node = function(itemstack, placer, pointed_thing)
 360                 local name = placer and placer:get_player_name() or ""
 361                 local invert_wall = placer and placer:get_player_control().sneak or false
 362                 return core.rotate_and_place(itemstack, placer, pointed_thing,
 363                         core.is_creative_enabled(name),
 364                         {invert_wall = invert_wall}, true)
 365         end
 366 end
 367
 368 --------------------------------------------------------------------------------
 369 function core.explode_table_event(evt)
 370         if evt ~= nil then
 371                 local parts = evt:split(":")
 372                 if #parts == 3 then
 373                         local t = parts[1]:trim()
 374                         local r = tonumber(parts[2]:trim())
 375                         local c = tonumber(parts[3]:trim())
 376                         if type(r) == "number" and type(c) == "number"
 377                                         and t ~= "INV" then
 378                                 return {type=t, row=r, column=c}
 379                         end
 380                 end
 381         end
 382         return {type="INV", row=0, column=0}
 383 end
 384
 385 --------------------------------------------------------------------------------
 386 function core.explode_textlist_event(evt)
 387         if evt ~= nil then
 388                 local parts = evt:split(":")
 389                 if #parts == 2 then
 390                         local t = parts[1]:trim()
 391                         local r = tonumber(parts[2]:trim())
 392                         if type(r) == "number" and t ~= "INV" then
 393                                 return {type=t, index=r}
 394                         end
 395                 end
 396         end
 397         return {type="INV", index=0}
 398 end
 399
 400 --------------------------------------------------------------------------------
 401 function core.explode_scrollbar_event(evt)
 402         local retval = core.explode_textlist_event(evt)
 403
 404         retval.value = retval.index
 405         retval.index = nil
 406
 407         return retval
 408 end
 409
 410 --------------------------------------------------------------------------------
 411 function core.rgba(r, g, b, a)
 412         return a and string.format("#%02X%02X%02X%02X", r, g, b, a) or
 413                         string.format("#%02X%02X%02X", r, g, b)
 414 end
 415
 416 --------------------------------------------------------------------------------
 417 function core.pos_to_string(pos, decimal_places)
 418         local x = pos.x
 419         local y = pos.y
 420         local z = pos.z
 421         if decimal_places ~= nil then
 422                 x = string.format("%." .. decimal_places .. "f", x)
 423                 y = string.format("%." .. decimal_places .. "f", y)
 424                 z = string.format("%." .. decimal_places .. "f", z)
 425         end
 426         return "(" .. x .. "," .. y .. "," .. z .. ")"
 427 end
 428
 429 --------------------------------------------------------------------------------
 430 function core.string_to_pos(value)
 431         if value == nil then
 432                 return nil
 433         end
 434
 435         local x, y, z = string.match(value, "^([%d.-]+)[, ] *([%d.-]+)[, ] *([%d.-]+)$")
 436         if x and y and z then
 437                 x = tonumber(x)
 438                 y = tonumber(y)
 439                 z = tonumber(z)
 440                 return vector.new(x, y, z)
 441         end
 442         x, y, z = string.match(value, "^%( *([%d.-]+)[, ] *([%d.-]+)[, ] *([%d.-]+) *%)$")
 443         if x and y and z then
 444                 x = tonumber(x)
 445                 y = tonumber(y)
 446                 z = tonumber(z)
 447                 return vector.new(x, y, z)
 448         end
 449         return nil
 450 end
 451
 452
 453 --------------------------------------------------------------------------------
 454 function core.string_to_area(value)
 455         local p1, p2 = unpack(value:split(") ("))
 456         if p1 == nil or p2 == nil then
 457                 return nil
 458         end
 459
 460         p1 = core.string_to_pos(p1 .. ")")
 461         p2 = core.string_to_pos("(" .. p2)
 462         if p1 == nil or p2 == nil then
 463                 return nil
 464         end
 465
 466         return p1, p2
 467 end
 468
 469 local function test_string_to_area()
 470         local p1, p2 = core.string_to_area("(10.0, 5, -2) (  30.2,   4, -12.53)")
 471         assert(p1.x == 10.0 and p1.y == 5 and p1.z == -2)
 472         assert(p2.x == 30.2 and p2.y == 4 and p2.z == -12.53)
 473
 474         p1, p2 = core.string_to_area("(10.0, 5, -2  30.2,   4, -12.53")
 475         assert(p1 == nil and p2 == nil)
 476
 477         p1, p2 = core.string_to_area("(10.0, 5,) -2  fgdf2,   4, -12.53")
 478         assert(p1 == nil and p2 == nil)
 479 end
 480
 481 test_string_to_area()
 482
 483 --------------------------------------------------------------------------------
 484 function table.copy(t, seen)
 485         local n = {}
 486         seen = seen or {}
 487         seen[t] = n
 488         for k, v in pairs(t) do
 489                 n[(type(k) == "table" and (seen[k] or table.copy(k, seen))) or k] =
 490                         (type(v) == "table" and (seen[v] or table.copy(v, seen))) or v
 491         end
 492         return n
 493 end
 494
 495
 496 function table.insert_all(t, other)
 497         for i=1, #other do
 498                 t[#t + 1] = other[i]
 499         end
 500         return t
 501 end
 502
 503
 504 function table.key_value_swap(t)
 505         local ti = {}
 506         for k,v in pairs(t) do
 507                 ti[v] = k
 508         end
 509         return ti
 510 end
 511
 512
 513 function table.shuffle(t, from, to, random)
 514         from = from or 1
 515         to = to or #t
 516         random = random or math.random
 517         local n = to - from + 1
 518         while n > 1 do
 519                 local r = from + n-1
 520                 local l = from + random(0, n-1)
 521                 t[l], t[r] = t[r], t[l]
 522                 n = n-1
 523         end
 524 end
 525
 526
 527 --------------------------------------------------------------------------------
 528 -- mainmenu only functions
 529 --------------------------------------------------------------------------------
 530 if INIT == "mainmenu" then
 531         function core.get_game(index)
 532                 local games = core.get_games()
 533
 534                 if index > 0 and index <= #games then
 535                         return games[index]
 536                 end
 537
 538                 return nil
 539         end
 540 end
 541
 542 if INIT == "client" or INIT == "mainmenu" then
 543         function fgettext_ne(text, ...)
 544                 text = core.gettext(text)
 545                 local arg = {n=select('#', ...), ...}
 546                 if arg.n >= 1 then
 547                         -- Insert positional parameters ($1, $2, ...)
 548                         local result = ''
 549                         local pos = 1
 550                         while pos <= text:len() do
 551                                 local newpos = text:find('[$]', pos)
 552                                 if newpos == nil then
 553                                         result = result .. text:sub(pos)
 554                                         pos = text:len() + 1
 555                                 else
 556                                         local paramindex =
 557                                                 tonumber(text:sub(newpos+1, newpos+1))
 558                                         result = result .. text:sub(pos, newpos-1)
 559                                                 .. tostring(arg[paramindex])
 560                                         pos = newpos + 2
 561                                 end
 562                         end
 563                         text = result
 564                 end
 565                 return text
 566         end
 567
 568         function fgettext(text, ...)
 569                 return core.formspec_escape(fgettext_ne(text, ...))
 570         end
 571 end
 572
 573 local ESCAPE_CHAR = string.char(0x1b)
 574
 575 function core.get_color_escape_sequence(color)
 576         return ESCAPE_CHAR .. "(c@" .. color .. ")"
 577 end
 578
 579 function core.get_background_escape_sequence(color)
 580         return ESCAPE_CHAR .. "(b@" .. color .. ")"
 581 end
 582
 583 function core.colorize(color, message)
 584         local lines = tostring(message):split("\n", true)
 585         local color_code = core.get_color_escape_sequence(color)
 586
 587         for i, line in ipairs(lines) do
 588                 lines[i] = color_code .. line
 589         end
 590
 591         return table.concat(lines, "\n") .. core.get_color_escape_sequence("#ffffff")
 592 end
 593
 594
 595 function core.strip_foreground_colors(str)
 596         return (str:gsub(ESCAPE_CHAR .. "%(c@[^)]+%)", ""))
 597 end
 598
 599 function core.strip_background_colors(str)
 600         return (str:gsub(ESCAPE_CHAR .. "%(b@[^)]+%)", ""))
 601 end
 602
 603 function core.strip_colors(str)
 604         return (str:gsub(ESCAPE_CHAR .. "%([bc]@[^)]+%)", ""))
 605 end
 606
 607 function core.translate(textdomain, str, ...)
 608         local start_seq
 609         if textdomain == "" then
 610                 start_seq = ESCAPE_CHAR .. "T"
 611         else
 612                 start_seq = ESCAPE_CHAR .. "(T@" .. textdomain .. ")"
 613         end
 614         local arg = {n=select('#', ...), ...}
 615         local end_seq = ESCAPE_CHAR .. "E"
 616         local arg_index = 1
 617         local translated = str:gsub("@(.)", function(matched)
 618                 local c = string.byte(matched)
 619                 if string.byte("1") <= c and c <= string.byte("9") then
 620                         local a = c - string.byte("0")
 621                         if a ~= arg_index then
 622                                 error("Escape sequences in string given to core.translate " ..
 623                                         "are not in the correct order: got @" .. matched ..
 624                                         "but expected @" .. tostring(arg_index))
 625                         end
 626                         if a > arg.n then
 627                                 error("Not enough arguments provided to core.translate")
 628                         end
 629                         arg_index = arg_index + 1
 630                         return ESCAPE_CHAR .. "F" .. arg[a] .. ESCAPE_CHAR .. "E"
 631                 elseif matched == "n" then
 632                         return "\n"
 633                 else
 634                         return matched
 635                 end
 636         end)
 637         if arg_index < arg.n + 1 then
 638                 error("Too many arguments provided to core.translate")
 639         end
 640         return start_seq .. translated .. end_seq
 641 end
 642
 643 function core.get_translator(textdomain)
 644         return function(str, ...) return core.translate(textdomain or "", str, ...) end
 645 end
 646
 647 --------------------------------------------------------------------------------
 648 -- Returns the exact coordinate of a pointed surface
 649 --------------------------------------------------------------------------------
 650 function core.pointed_thing_to_face_pos(placer, pointed_thing)
 651         -- Avoid crash in some situations when player is inside a node, causing
 652         -- 'above' to equal 'under'.
 653         if vector.equals(pointed_thing.above, pointed_thing.under) then
 654                 return pointed_thing.under
 655         end
 656
 657         local eye_height = placer:get_properties().eye_height
 658         local eye_offset_first = placer:get_eye_offset()
 659         local node_pos = pointed_thing.under
 660         local camera_pos = placer:get_pos()
 661         local pos_off = vector.multiply(
 662                         vector.subtract(pointed_thing.above, node_pos), 0.5)
 663         local look_dir = placer:get_look_dir()
 664         local offset, nc
 665         local oc = {}
 666
 667         for c, v in pairs(pos_off) do
 668                 if nc or v == 0 then
 669                         oc[#oc + 1] = c
 670                 else
 671                         offset = v
 672                         nc = c
 673                 end
 674         end
 675
 676         local fine_pos = {[nc] = node_pos[nc] + offset}
 677         camera_pos.y = camera_pos.y + eye_height + eye_offset_first.y / 10
 678         local f = (node_pos[nc] + offset - camera_pos[nc]) / look_dir[nc]
 679
 680         for i = 1, #oc do
 681                 fine_pos[oc[i]] = camera_pos[oc[i]] + look_dir[oc[i]] * f
 682         end
 683         return fine_pos
 684 end
 685
 686 function core.string_to_privs(str, delim)
 687         assert(type(str) == "string")
 688         delim = delim or ','
 689         local privs = {}
 690         for _, priv in pairs(string.split(str, delim)) do
 691                 privs[priv:trim()] = true
 692         end
 693         return privs
 694 end
 695
 696 function core.privs_to_string(privs, delim)
 697         assert(type(privs) == "table")
 698         delim = delim or ','
 699         local list = {}
 700         for priv, bool in pairs(privs) do
 701                 if bool then
 702                         list[#list + 1] = priv
 703                 end
 704         end
 705         return table.concat(list, delim)
 706 end
 707
 708 function core.is_nan(number)
 709         return number ~= number
 710 end