src/lib/metalua/base.lua

   1 ----------------------------------------------------------------------
   2 ----------------------------------------------------------------------
   3 --
   4 -- Base library extension
   5 --
   6 ----------------------------------------------------------------------
   7 ----------------------------------------------------------------------
   8
   9 if not metalua then rawset(getfenv(), 'metalua', { }) end
  10 metalua.version             = "v-0.5"
  11
  12 if not rawpairs then
  13    rawpairs, rawipairs, rawtype = pairs, ipairs, type
  14 end
  15
  16 function pairs(x)
  17    assert(type(x)=='table', 'pairs() expects a table')
  18    local mt = getmetatable(x)
  19    if mt then
  20       local mtp = mt.__pairs
  21       if mtp then return mtp(x) end
  22    end
  23    return rawpairs(x)
  24 end
  25
  26 function ipairs(x)
  27    assert(type(x)=='table', 'ipairs() expects a table')
  28    local mt = getmetatable(x)
  29    if mt then
  30       local mti = mt.__ipairs
  31       if mti then return mti(x) end
  32    end
  33    return rawipairs(x)
  34 end
  35
  36 --[[
  37 function type(x)
  38    local mt = getmetatable(x)
  39    if mt then
  40       local mtt = mt.__type
  41       if mtt then return mtt end
  42    end
  43    return rawtype(x)
  44 end
  45 ]]
  46
  47 function min (a, ...)
  48    for n in values{...} do if n<a then a=n end end
  49    return a
  50 end
  51
  52 function max (a, ...)
  53    for n in values{...} do if n>a then a=n end end
  54    return a
  55 end
  56
  57 function o (...)
  58    local args = {...}
  59    local function g (...)
  60       local result = {...}
  61       for i=#args, 1, -1 do result = {args[i](unpack(result))} end
  62       return unpack (result)
  63    end
  64    return g
  65 end
  66
  67 function id (...) return ... end
  68 function const (k) return function () return k end end
  69
  70 function printf(...) return print(string.format(...)) end
  71 function eprintf(...)
  72    io.stderr:write(string.format(...).."\n")
  73 end
  74
  75 function ivalues (x)
  76    assert(type(x)=='table', 'ivalues() expects a table')
  77    local i = 1
  78    local function iterator ()
  79       local r = x[i]; i=i+1; return r
  80    end
  81    return iterator
  82 end
  83
  84
  85 function values (x)
  86    assert(type(x)=='table', 'values() expects a table')
  87    local function iterator (state)
  88       local it
  89       state.content, it = next(state.list, state.content)
  90       return it
  91    end
  92    return iterator, { list = x }
  93 end
  94
  95 function keys (x)
  96    assert(type(x)=='table', 'keys() expects a table')
  97    local function iterator (state)
  98       local it = next(state.list, state.content)
  99       state.content = it
 100       return it
 101    end
 102    return iterator, { list = x }
 103 end
 104