From 6fc0ad3dae4509c1023b4c2dc3a142c0d69604d8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= Date: Fri, 1 Jul 2022 17:49:38 +0800 Subject: stash --- script/vm/runner.lua | 237 +++++++++++++++++++++++++++------------------------ 1 file changed, 127 insertions(+), 110 deletions(-) (limited to 'script') diff --git a/script/vm/runner.lua b/script/vm/runner.lua index 9f9a53a2..f2636717 100644 --- a/script/vm/runner.lua +++ b/script/vm/runner.lua @@ -110,126 +110,64 @@ function mt:_fastWard(pos, node) return node, nil end ----@param action parser.object +---@param exp parser.object ---@param topNode vm.node ---@param outNode? vm.node ---@return vm.node topNode ---@return vm.node outNode -function mt:_lookInto(action, topNode, outNode) - if not action then +function mt:_lookIntoExp(exp, topNode, outNode) + if not exp then return topNode, outNode or topNode end - if self._mark[action] then + if self._mark[exp] then return topNode, outNode or topNode end - self._mark[action] = true + self._mark[exp] = true local top = self._objs[self._index] if not top then return topNode, outNode or topNode end - if not guide.isInRange(action, top.finish) - -- trick for `local tp = type(x);if tp == 'string' then` - and action.type ~= 'binary' then - return topNode, outNode or topNode - end - local set - local value = vm.getObjectValue(action) - if value then - set = action - action = value - end - if action.type == 'function' then - self:_launchBlock(action, topNode:copy()) - elseif action.type == 'loop' - or action.type == 'in' - or action.type == 'repeat' - or action.type == 'for' then - topNode = self:_launchBlock(action, topNode:copy()) - elseif action.type == 'while' then - local blockNode, mainNode = self:_lookInto(action.filter, topNode:copy(), topNode:copy()) - if action.filter then - self:_fastWard(action.filter.finish, blockNode) - end - blockNode = self:_launchBlock(action, blockNode:copy()) - if mainNode then - topNode = mainNode:merge(blockNode) - end - elseif action.type == 'if' then - local hasElse - local mainNode = topNode:copy() - local blockNodes = {} - for _, subBlock in ipairs(action) do - local blockNode = mainNode:copy() - if subBlock.filter then - blockNode, mainNode = self:_lookInto(subBlock.filter, blockNode, mainNode) - self:_fastWard(subBlock.filter.finish, blockNode) - else - hasElse = true - mainNode:clear() - end - blockNode = self:_launchBlock(subBlock, blockNode:copy()) - local neverReturn = subBlock.hasReturn - or subBlock.hasGoTo - or subBlock.hasBreak - or subBlock.hasError - if not neverReturn then - blockNodes[#blockNodes+1] = blockNode - end - end - if not hasElse and not topNode:hasKnownType() then - mainNode:merge(vm.declareGlobal('type', 'unknown')) - end - for _, blockNode in ipairs(blockNodes) do - mainNode:merge(blockNode) - end - topNode = mainNode - elseif action.type == 'getlocal' then - if action.node == self._loc then - topNode = self:_fastWard(action.finish, topNode) - topNode = topNode:copy():setTruthy() - if outNode then - outNode:setFalsy() - end - end - elseif action.type == 'unary' then - if not action[1] then + if exp.type == 'function' then + self:_launchBlock(exp, topNode:copy()) + elseif exp.type == 'unary' then + if not exp[1] then goto RETURN end - if action.op.type == 'not' then + if exp.op.type == 'not' then outNode = outNode or topNode:copy() - outNode, topNode = self:_lookInto(action[1], topNode, outNode) + outNode, topNode = self:_lookIntoExp(exp[1], topNode, outNode) end - elseif action.type == 'binary' then - if not action[1] or not action[2] then + elseif exp.type == 'binary' then + if not exp[1] or not exp[2] then goto RETURN end - if action.op.type == 'and' then - topNode = self:_lookInto(action[1], topNode) - topNode = self:_lookInto(action[2], topNode) - elseif action.op.type == 'or' then + if exp.op.type == 'and' then + topNode = self:_lookIntoExp(exp[1], topNode) + topNode = self:_lookIntoExp(exp[2], topNode) + elseif exp.op.type == 'or' then outNode = outNode or topNode:copy() - local topNode1, outNode1 = self:_lookInto(action[1], topNode, outNode) - local topNode2, outNode2 = self:_lookInto(action[2], outNode1, outNode1:copy()) + local topNode1, outNode1 = self:_lookIntoExp(exp[1], topNode, outNode) + local topNode2, outNode2 = self:_lookIntoExp(exp[2], outNode1, outNode1:copy()) topNode = vm.createNode(topNode1, topNode2) outNode = outNode2:copy() - elseif action.op.type == '==' - or action.op.type == '~=' then - local exp, checker + elseif exp.op.type == '==' + or exp.op.type == '~=' then + local handler, checker for i = 1, 2 do - if guide.isLiteral(action[i]) then - checker = action[i] - exp = action[3-i] -- Copilot tells me use `3-i` instead of `i%2+1` + if guide.isLiteral(exp[i]) then + checker = exp[i] + handler = exp[3-i] -- Copilot tells me use `3-i` instead of `i%2+1` end end - if not exp then + if not handler then goto RETURN end - if exp.type == 'getlocal' - and exp.node == self._loc then + if handler.type == 'getlocal' + and handler.node == self._loc then -- if x == y then self:_fastWard(exp.finish, topNode:copy()) local checkerNode = vm.compileNode(checker) - if action.op.type == '==' then + if exp.op.type == '==' then topNode = checkerNode if outNode then outNode:removeNode(topNode) @@ -249,7 +187,7 @@ function mt:_lookInto(action, topNode, outNode) and exp.args[1].node == self._loc then -- if type(x) == 'string' then self:_fastWard(exp.finish, topNode:copy()) - if action.op.type == '==' then + if exp.op.type == '==' then topNode:narrow(checker[1]) if outNode then outNode:remove(checker[1]) @@ -275,7 +213,7 @@ function mt:_lookInto(action, topNode, outNode) and call.args[1].type == 'getlocal' and call.args[1].node == self._loc then -- `local tp = type(x);if tp == 'string' then` - if action.op.type == '==' then + if exp.op.type == '==' then topNode:narrow(checker[1]) if outNode then outNode:remove(checker[1]) @@ -290,33 +228,112 @@ function mt:_lookInto(action, topNode, outNode) end end end + elseif exp.type == 'getlocal' then + if exp.node == self._loc then + topNode = self:_fastWard(exp.finish, topNode) + topNode = topNode:copy():setTruthy() + if outNode then + outNode:setFalsy() + end + end + elseif exp.type == 'paren' then + topNode, outNode = self:_lookIntoExp(exp.exp, topNode, outNode) + elseif exp.type == 'getindex' then + self:_lookIntoExp(exp.index, topNode) + elseif exp.type == 'table' then + for _, field in ipairs(exp) do + self:_lookIntoExp(field, topNode) + end + end + ::RETURN:: + topNode = self:_fastWard(exp.finish, topNode) + return topNode, outNode or topNode +end + +---@param action parser.object +---@param topNode vm.node +---@return vm.node topNode +function mt:_lookIntoAction(action, topNode) + if not action then + return topNode + end + if self._mark[action] then + return topNode + end + self._mark[action] = true + local top = self._objs[self._index] + if not top then + return topNode + end + if not guide.isInRange(action, top.finish) + -- trick for `local tp = type(x);if tp == 'string' then` + and action.type ~= 'binary' then + return topNode + end + local value = vm.getObjectValue(action) + if value then + topNode = self:_lookIntoExp(value, topNode:copy()) + end + if action.type == 'function' then + self:_launchBlock(action, topNode:copy()) + elseif action.type == 'loop' + or action.type == 'in' + or action.type == 'repeat' + or action.type == 'for' then + topNode = self:_launchBlock(action, topNode:copy()) + elseif action.type == 'while' then + local blockNode, mainNode = self:_lookIntoExp(action.filter, topNode:copy(), topNode:copy()) + if action.filter then + self:_fastWard(action.filter.finish, blockNode) + end + blockNode = self:_launchBlock(action, blockNode:copy()) + if mainNode then + topNode = mainNode:merge(blockNode) + end + elseif action.type == 'if' then + local hasElse + local mainNode = topNode:copy() + local blockNodes = {} + for _, subBlock in ipairs(action) do + local blockNode = mainNode:copy() + if subBlock.filter then + blockNode, mainNode = self:_lookIntoExp(subBlock.filter, blockNode, mainNode) + self:_fastWard(subBlock.filter.finish, blockNode) + else + hasElse = true + mainNode:clear() + end + blockNode = self:_launchBlock(subBlock, blockNode:copy()) + local neverReturn = subBlock.hasReturn + or subBlock.hasGoTo + or subBlock.hasBreak + or subBlock.hasError + if not neverReturn then + blockNodes[#blockNodes+1] = blockNode + end + end + if not hasElse and not topNode:hasKnownType() then + mainNode:merge(vm.declareGlobal('type', 'unknown')) + end + for _, blockNode in ipairs(blockNodes) do + mainNode:merge(blockNode) + end + topNode = mainNode elseif action.type == 'call' then if action.node.special == 'assert' and action.args and action.args[1] then - topNode = self:_lookInto(action.args[1], topNode) + topNode = self:_lookIntoExp(action.args[1], topNode) elseif action.args then for _, arg in ipairs(action.args) do - self:_lookInto(arg, topNode) + self:_lookIntoExp(arg, topNode) end end - elseif action.type == 'paren' then - topNode, outNode = self:_lookInto(action.exp, topNode, outNode) elseif action.type == 'return' then for _, rtn in ipairs(action) do - self:_lookInto(rtn, topNode) - end - elseif action.type == 'getindex' then - self:_lookInto(action.index, topNode) - elseif action.type == 'table' then - for _, field in ipairs(action) do - self:_lookInto(field, topNode) + self:_lookIntoExp(rtn, topNode) end end - ::RETURN:: topNode = self:_fastWard(action.finish, topNode) - if set then - topNode = self:_fastWard(set.range or set.finish, topNode) - end - return topNode, outNode or topNode + return topNode end ---@param block parser.object @@ -331,7 +348,7 @@ function mt:_launchBlock(block, node) if (action.range or action.finish) < (top.range or top.finish) then goto CONTINUE end - topNode = self:_lookInto(action, topNode) + topNode = self:_lookIntoAction(action, topNode) topNode, top = self:_fastWard(action.range or action.finish, topNode) if not top then return topNode -- cgit v1.2.3