1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
local util = require 'utility'
local global = require 'vm.node.global'
---@class vm.state
local m = {}
---@type table<string, vm.node.global>
m.globals = util.defaultTable(global)
---@type table<uri, table<string, boolean>>
m.globalSubs = util.defaultTable(function ()
return {}
end)
---@type table<uri, parser.object[]>
m.literals = util.multiTable(2)
---@type table<parser.object, table<parser.object, boolean>>
m.literalSubs = util.multiTable(2, function ()
return setmetatable({}, util.MODE_K)
end)
---@type table<parser.object, boolean>
m.allLiterals = {}
---@param name string
---@param uri uri
---@param source parser.object
---@return vm.node.global
function m.declareGlobal(name, uri, source)
m.globalSubs[uri][name] = true
local node = m.globals[name]
node:addSet(uri, source)
return node
end
---@param name string
---@param uri? uri
---@return vm.node.global
function m.getGlobal(name, uri)
if uri then
m.globalSubs[uri][name] = true
end
return m.globals[name]
end
---@param uri uri
---@param source parser.object
function m.declareLiteral(uri, source)
if m.allLiterals[source] then
return
end
m.allLiterals[source] = true
local literals = m.literals[uri]
literals[#literals+1] = source
end
---@param source parser.object
---@param node vm.node
function m.subscribeLiteral(source, node)
if not node then
return
end
if node.type == 'union'
or node.type == 'cross' then
node:subscribeLiteral(source)
return
end
if not m.allLiterals[source] then
return
end
m.literalSubs[node][source] = true
end
---@param uri uri
function m.dropUri(uri)
local globalSub = m.globalSubs[uri]
m.globalSubs[uri] = nil
for name in pairs(globalSub) do
m.globals[name]:dropUri(uri)
end
local literals = m.literals[uri]
m.literals[uri] = nil
for _, literal in ipairs(literals) do
m.allLiterals[literal] = nil
local literalSubs = m.literalSubs[literal]
m.literalSubs[literal] = nil
for source in pairs(literalSubs) do
source._node = nil
end
end
end
return m
|
