412 lines
11 KiB
Lua
412 lines
11 KiB
Lua
|
local util = require 'utility'
|
|||
|
|
local guide = require 'parser.guide'
|
|||
|
|
---@class vm
|
|||
|
|
local vm = require 'vm.vm'
|
|||
|
|
|
|||
|
|
---@class vm.variable
|
|||
|
|
---@field uri uri
|
|||
|
|
---@field root parser.object
|
|||
|
|
---@field id string
|
|||
|
|
---@field base parser.object
|
|||
|
|
---@field sets parser.object[]
|
|||
|
|
---@field gets parser.object[]
|
|||
|
|
local mt = {}
|
|||
|
|
mt.__index = mt
|
|||
|
|
mt.type = 'variable'
|
|||
|
|
|
|||
|
|
---@param id string
|
|||
|
|
---@return vm.variable
|
|||
|
|
local function createVariable(root, id)
|
|||
|
|
local variable = setmetatable({
|
|||
|
|
root = root,
|
|||
|
|
uri = root.uri,
|
|||
|
|
id = id,
|
|||
|
|
sets = {},
|
|||
|
|
gets = {},
|
|||
|
|
}, mt)
|
|||
|
|
return variable
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@class parser.object
|
|||
|
|
---@field package _variableNode vm.variable|false
|
|||
|
|
---@field package _variableNodes table<string, vm.variable>
|
|||
|
|
|
|||
|
|
local compileVariables, getLoc
|
|||
|
|
|
|||
|
|
---@param id string
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param base parser.object
|
|||
|
|
---@return vm.variable
|
|||
|
|
local function insertVariableID(id, source, base)
|
|||
|
|
local root = guide.getRoot(source)
|
|||
|
|
if not root._variableNodes then
|
|||
|
|
root._variableNodes = util.multiTable(2, function (lid)
|
|||
|
|
local variable = createVariable(root, lid)
|
|||
|
|
return variable
|
|||
|
|
end)
|
|||
|
|
end
|
|||
|
|
local variable = root._variableNodes[id]
|
|||
|
|
variable.base = base
|
|||
|
|
if guide.isAssign(source) then
|
|||
|
|
variable.sets[#variable.sets+1] = source
|
|||
|
|
else
|
|||
|
|
variable.gets[#variable.gets+1] = source
|
|||
|
|
end
|
|||
|
|
return variable
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
local compileSwitch = util.switch()
|
|||
|
|
: case 'local'
|
|||
|
|
: case 'self'
|
|||
|
|
: call(function (source, base)
|
|||
|
|
local id = ('%d'):format(source.start)
|
|||
|
|
local variable = insertVariableID(id, source, base)
|
|||
|
|
source._variableNode = variable
|
|||
|
|
if not source.ref then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
for _, ref in ipairs(source.ref) do
|
|||
|
|
compileVariables(ref, base)
|
|||
|
|
end
|
|||
|
|
end)
|
|||
|
|
: case 'setlocal'
|
|||
|
|
: case 'getlocal'
|
|||
|
|
: call(function (source, base)
|
|||
|
|
local id = ('%d'):format(source.node.start)
|
|||
|
|
local variable = insertVariableID(id, source, base)
|
|||
|
|
source._variableNode = variable
|
|||
|
|
compileVariables(source.next, base)
|
|||
|
|
end)
|
|||
|
|
: case 'getfield'
|
|||
|
|
: case 'setfield'
|
|||
|
|
: call(function (source, base)
|
|||
|
|
local parentNode = source.node._variableNode
|
|||
|
|
if not parentNode then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local key = guide.getKeyName(source)
|
|||
|
|
if type(key) ~= 'string' then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local id = parentNode.id .. vm.ID_SPLITE .. key
|
|||
|
|
local variable = insertVariableID(id, source, base)
|
|||
|
|
source._variableNode = variable
|
|||
|
|
source.field._variableNode = variable
|
|||
|
|
if source.type == 'getfield' then
|
|||
|
|
compileVariables(source.next, base)
|
|||
|
|
end
|
|||
|
|
end)
|
|||
|
|
: case 'getmethod'
|
|||
|
|
: case 'setmethod'
|
|||
|
|
: call(function (source, base)
|
|||
|
|
local parentNode = source.node._variableNode
|
|||
|
|
if not parentNode then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local key = guide.getKeyName(source)
|
|||
|
|
if type(key) ~= 'string' then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local id = parentNode.id .. vm.ID_SPLITE .. key
|
|||
|
|
local variable = insertVariableID(id, source, base)
|
|||
|
|
source._variableNode = variable
|
|||
|
|
source.method._variableNode = variable
|
|||
|
|
if source.type == 'getmethod' then
|
|||
|
|
compileVariables(source.next, base)
|
|||
|
|
end
|
|||
|
|
end)
|
|||
|
|
: case 'getindex'
|
|||
|
|
: case 'setindex'
|
|||
|
|
: call(function (source, base)
|
|||
|
|
local parentNode = source.node._variableNode
|
|||
|
|
if not parentNode then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local key = guide.getKeyName(source)
|
|||
|
|
if type(key) ~= 'string' then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local id = parentNode.id .. vm.ID_SPLITE .. key
|
|||
|
|
local variable = insertVariableID(id, source, base)
|
|||
|
|
source._variableNode = variable
|
|||
|
|
source.index._variableNode = variable
|
|||
|
|
if source.type == 'setindex' then
|
|||
|
|
compileVariables(source.next, base)
|
|||
|
|
end
|
|||
|
|
end)
|
|||
|
|
|
|||
|
|
local leftSwitch = util.switch()
|
|||
|
|
: case 'field'
|
|||
|
|
: case 'method'
|
|||
|
|
: call(function (source)
|
|||
|
|
return getLoc(source.parent)
|
|||
|
|
end)
|
|||
|
|
: case 'getfield'
|
|||
|
|
: case 'setfield'
|
|||
|
|
: case 'getmethod'
|
|||
|
|
: case 'setmethod'
|
|||
|
|
: case 'getindex'
|
|||
|
|
: case 'setindex'
|
|||
|
|
: call(function (source)
|
|||
|
|
return getLoc(source.node)
|
|||
|
|
end)
|
|||
|
|
: case 'getlocal'
|
|||
|
|
: call(function (source)
|
|||
|
|
return source.node
|
|||
|
|
end)
|
|||
|
|
: case 'local'
|
|||
|
|
: case 'self'
|
|||
|
|
: call(function (source)
|
|||
|
|
return source
|
|||
|
|
end)
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@return parser.object?
|
|||
|
|
function getLoc(source)
|
|||
|
|
return leftSwitch(source.type, source)
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@return parser.object
|
|||
|
|
function mt:getBase()
|
|||
|
|
return self.base
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@return string
|
|||
|
|
function mt:getCodeName()
|
|||
|
|
local name = self.id:gsub(vm.ID_SPLITE, '.'):gsub('^%d+', self.base[1])
|
|||
|
|
return name
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@return vm.variable?
|
|||
|
|
function mt:getParent()
|
|||
|
|
local parentID = self.id:match('^(.+)' .. vm.ID_SPLITE)
|
|||
|
|
if not parentID then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
return self.root._variableNodes[parentID]
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@return string?
|
|||
|
|
function mt:getFieldName()
|
|||
|
|
return self.id:match(vm.ID_SPLITE .. '(.-)$')
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param key? string
|
|||
|
|
function mt:getSets(key)
|
|||
|
|
if not key then
|
|||
|
|
return self.sets
|
|||
|
|
end
|
|||
|
|
local id = self.id .. vm.ID_SPLITE .. key
|
|||
|
|
local variable = self.root._variableNodes[id]
|
|||
|
|
return variable.sets
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param includeGets boolean?
|
|||
|
|
function mt:getFields(includeGets)
|
|||
|
|
local id = self.id
|
|||
|
|
local root = self.root
|
|||
|
|
-- TODO：optimize
|
|||
|
|
local clock = os.clock()
|
|||
|
|
local fields = {}
|
|||
|
|
for lid, variable in pairs(root._variableNodes) do
|
|||
|
|
if lid ~= id
|
|||
|
|
and util.stringStartWith(lid, id)
|
|||
|
|
and lid:sub(#id + 1, #id + 1) == vm.ID_SPLITE
|
|||
|
|
-- only one field
|
|||
|
|
and not lid:find(vm.ID_SPLITE, #id + 2) then
|
|||
|
|
for _, src in ipairs(variable.sets) do
|
|||
|
|
fields[#fields+1] = src
|
|||
|
|
end
|
|||
|
|
if includeGets then
|
|||
|
|
for _, src in ipairs(variable.gets) do
|
|||
|
|
fields[#fields+1] = src
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
local cost = os.clock() - clock
|
|||
|
|
if cost > 1.0 then
|
|||
|
|
log.warn('variable-id getFields takes %.3f seconds', cost)
|
|||
|
|
end
|
|||
|
|
return fields
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param base parser.object
|
|||
|
|
function compileVariables(source, base)
|
|||
|
|
if not source then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
source._variableNode = false
|
|||
|
|
if not compileSwitch:has(source.type) then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
compileSwitch(source.type, source, base)
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@return string?
|
|||
|
|
function vm.getVariableID(source)
|
|||
|
|
local variable = vm.getVariableNode(source)
|
|||
|
|
if not variable then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
return variable.id
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param key? string
|
|||
|
|
---@return vm.variable?
|
|||
|
|
function vm.getVariable(source, key)
|
|||
|
|
local variable = vm.getVariableNode(source)
|
|||
|
|
if not variable then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
if not key then
|
|||
|
|
return variable
|
|||
|
|
end
|
|||
|
|
local root = guide.getRoot(source)
|
|||
|
|
if not root._variableNodes then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
local id = variable.id .. vm.ID_SPLITE .. key
|
|||
|
|
return root._variableNodes[id]
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@return vm.variable?
|
|||
|
|
function vm.getVariableNode(source)
|
|||
|
|
local variable = source._variableNode
|
|||
|
|
if variable ~= nil then
|
|||
|
|
return variable or nil
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
source._variableNode = false
|
|||
|
|
local loc = getLoc(source)
|
|||
|
|
if not loc then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
compileVariables(loc, loc)
|
|||
|
|
return source._variableNode or nil
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param name string
|
|||
|
|
---@return vm.variable?
|
|||
|
|
function vm.getVariableInfoByCodeName(source, name)
|
|||
|
|
local id = vm.getVariableID(source)
|
|||
|
|
if not id then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
local root = guide.getRoot(source)
|
|||
|
|
if not root._variableNodes then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
local headPos = name:find('.', 1, true)
|
|||
|
|
if not headPos then
|
|||
|
|
return root._variableNodes[id]
|
|||
|
|
end
|
|||
|
|
local vid = id .. name:sub(headPos):gsub('%.', vm.ID_SPLITE)
|
|||
|
|
return root._variableNodes[vid]
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param key? string
|
|||
|
|
---@return parser.object[]?
|
|||
|
|
function vm.getVariableSets(source, key)
|
|||
|
|
local variable = vm.getVariable(source, key)
|
|||
|
|
if not variable then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
return variable.sets
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param key? string
|
|||
|
|
---@return parser.object[]?
|
|||
|
|
function vm.getVariableGets(source, key)
|
|||
|
|
local variable = vm.getVariable(source, key)
|
|||
|
|
if not variable then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
return variable.gets
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@param includeGets boolean
|
|||
|
|
---@return parser.object[]?
|
|||
|
|
function vm.getVariableFields(source, includeGets)
|
|||
|
|
local variable = vm.getVariable(source)
|
|||
|
|
if not variable then
|
|||
|
|
return nil
|
|||
|
|
end
|
|||
|
|
return variable:getFields(includeGets)
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
---@return boolean
|
|||
|
|
function vm.compileByVariable(source)
|
|||
|
|
local variable = vm.getVariableNode(source)
|
|||
|
|
if not variable then
|
|||
|
|
return false
|
|||
|
|
end
|
|||
|
|
vm.setNode(source, variable)
|
|||
|
|
return true
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
local function compileSelf(source)
|
|||
|
|
if source.parent.type ~= 'funcargs' then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
---@type parser.object
|
|||
|
|
local node = source.parent.parent and source.parent.parent.parent and source.parent.parent.parent.node
|
|||
|
|
if not node then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local fields = vm.getVariableFields(source, false)
|
|||
|
|
if not fields then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
local variableNode = vm.getVariableNode(node)
|
|||
|
|
local globalNode = vm.getGlobalNode(node)
|
|||
|
|
if not variableNode and not globalNode then
|
|||
|
|
return
|
|||
|
|
end
|
|||
|
|
for _, field in ipairs(fields) do
|
|||
|
|
if field.type == 'setfield' then
|
|||
|
|
local key = guide.getKeyName(field)
|
|||
|
|
if key then
|
|||
|
|
if variableNode then
|
|||
|
|
local myID = variableNode.id .. vm.ID_SPLITE .. key
|
|||
|
|
insertVariableID(myID, field, variableNode.base)
|
|||
|
|
end
|
|||
|
|
if globalNode then
|
|||
|
|
local myID = globalNode:getName() .. vm.ID_SPLITE .. key
|
|||
|
|
local myGlobal = vm.declareGlobal('variable', myID, guide.getUri(node))
|
|||
|
|
myGlobal:addSet(guide.getUri(node), field)
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
---@param source parser.object
|
|||
|
|
local function compileAst(source)
|
|||
|
|
--[[
|
|||
|
|
local mt
|
|||
|
|
function mt:xxx()
|
|||
|
|
self.a = 1
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
mt.a --> find this definition
|
|||
|
|
]]
|
|||
|
|
guide.eachSourceType(source, 'self', function (src)
|
|||
|
|
compileSelf(src)
|
|||
|
|
end)
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
compileAst = compileAst,
|
|||
|
|
}
|