Mercurial > luatraverse / file comparison
comparison: dump.lua
dump.lua
- changeset 13
- 9bd292b35f23
equal
deleted
inserted
replaced
| 12:e3e3cbe544ec | 13:9bd292b35f23 |
|---|---|
| 1 local statewalk = require("luatraverse"); | |
| 2 local format = string.format; | |
| 3 local type = type; | |
| 4 local tostring = tostring; | |
| 5 local io, os = io, os; | |
| 6 | |
| 7 local getsize; | |
| 8 do | |
| 9 local ok, ret = pcall(require, "getsize"); | |
| 10 if ok then | |
| 11 --luacheck: ignore 143/debug | |
| 12 if type(ret) == "function" then | |
| 13 getsize = ret; | |
| 14 elseif debug.getsize then | |
| 15 getsize = debug.getsize; | |
| 16 end | |
| 17 end | |
| 18 end | |
| 19 | |
| 20 local function dump_state(f, config) | |
| 21 if io.type(f) ~= "file" then | |
| 22 local err; | |
| 23 f, err = io.open(f, "w+"); | |
| 24 if not f then return false, err; end | |
| 25 end | |
| 26 | |
| 27 local metadata = config and config.metadata; | |
| 28 local start_time = os.time(); | |
| 29 | |
| 30 f:write("# Lua state graph", "\n"); | |
| 31 f:write("# Generated by dump.lua v2.1", "\n"); | |
| 32 f:write("# at ", os.date("!%F %R UTC", start_time), "\n"); | |
| 33 if metadata then | |
| 34 local keys = {}; | |
| 35 for k in pairs(metadata) do | |
| 36 table.insert(keys, k); | |
| 37 end | |
| 38 table.sort(keys); | |
| 39 if #keys > 0 then | |
| 40 for _, k in ipairs(keys) do | |
| 41 f:write("## ", k, ": ", metadata[k], "\n"); | |
| 42 end | |
| 43 end | |
| 44 end | |
| 45 | |
| 46 local seenobjects, id = {}, 0; | |
| 47 local function edge(from, to, how, value) | |
| 48 if from and (not seenobjects[from] or seenobjects[from] == true) then | |
| 49 id = id + 1; | |
| 50 seenobjects[from] = id; | |
| 51 f:write("V:", id, ":", type(from), ":", (getsize and getsize(from)) or "?", "\n"); | |
| 52 end | |
| 53 if not seenobjects[to] or seenobjects[to] == true then | |
| 54 id = id + 1; | |
| 55 seenobjects[to] = id; | |
| 56 f:write("V:", id, ":", type(to), ":", (getsize and getsize(to)) or "?", "\n"); | |
| 57 end | |
| 58 if seenobjects[from] == true or seenobjects[to] == true then | |
| 59 return; | |
| 60 end | |
| 61 local value_rep = (value and format("%q", tostring(value))) or "?"; | |
| 62 f:write("E:", seenobjects[from] or "?", ":", seenobjects[to], ":", tostring(how or "?"), ":", value_rep, "\n"); | |
| 63 end | |
| 64 local ignore = {f, dump_state, statewalk, seenobjects, edge}; | |
| 65 statewalk.traverse({edge=edge}, ignore, seenobjects); | |
| 66 f:write("# Completed in ", tostring(os.time() - start_time), "s", "\n"); | |
| 67 f:close(); | |
| 68 end | |
| 69 | |
| 70 return { | |
| 71 dump_state = dump_state; | |
| 72 }; |
