From 07bdb57b7358bc62630c6cdaaacd68b151aa6d08 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9C=80=E8=90=8C=E5=B0=8F=E6=B1=90?= <sumneko@hotmail.com>
Date: Tue, 7 Jun 2022 17:23:07 +0800
Subject: #1192 don't search local in simple

use the methods provided by `local-id` to avoid wasting on `gets`
---
 script/vm/def.lua         | 65 -----------------------------------------------
 script/vm/local-id.lua    |  1 +
 script/vm/ref.lua         | 51 -------------------------------------
 test/definition/table.lua | 20 +++++++--------
 4 files changed, 11 insertions(+), 126 deletions(-)

diff --git a/script/vm/def.lua b/script/vm/def.lua
index ad343ae6..accfbb1b 100644
--- a/script/vm/def.lua
+++ b/script/vm/def.lua
@@ -5,72 +5,7 @@ local guide     = require 'parser.guide'
 
 local simpleSwitch
 
-local function searchGetLocal(source, node, pushResult)
-    local key = guide.getKeyName(source)
-    for _, ref in ipairs(node.node.ref) do
-        if  ref.type == 'getlocal'
-        and ref.next
-        and guide.isSet(ref.next)
-        and guide.getKeyName(ref.next) == key then
-            pushResult(ref.next)
-        end
-    end
-end
-
 simpleSwitch = util.switch()
-    : case 'local'
-    : call(function (source, pushResult)
-        pushResult(source)
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'setlocal' then
-                    pushResult(ref)
-                end
-            end
-        end
-    end)
-    : case 'sellf'
-    : call(function (source, pushResult)
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'setlocal' then
-                    pushResult(ref)
-                end
-            end
-        end
-        for _, res in ipairs(vm.getDefs(source.method.node)) do
-            pushResult(res)
-        end
-    end)
-    : case 'getlocal'
-    : case 'setlocal'
-    : call(function (source, pushResult)
-        simpleSwitch('local', source.node, pushResult)
-    end)
-    : case 'field'
-    : call(function (source, pushResult)
-        local parent = source.parent
-        if parent.type ~= 'tablefield' then
-            simpleSwitch(parent.type, parent, pushResult)
-        end
-    end)
-    : case 'setfield'
-    : case 'getfield'
-    : call(function (source, pushResult)
-        local node = source.node
-        if node.type == 'getlocal' then
-            searchGetLocal(source, node, pushResult)
-            return
-        end
-    end)
-    : case 'getindex'
-    : case 'setindex'
-    : call(function (source, pushResult)
-        local node = source.node
-        if node.type == 'getlocal' then
-            searchGetLocal(source, node, pushResult)
-        end
-    end)
     : case 'goto'
     : call(function (source, pushResult)
         if source.node then
diff --git a/script/vm/local-id.lua b/script/vm/local-id.lua
index bb12a927..e0c8770d 100644
--- a/script/vm/local-id.lua
+++ b/script/vm/local-id.lua
@@ -21,6 +21,7 @@ local compileSwitch = util.switch()
             compileLocalID(ref)
         end
     end)
+    : case 'setlocal'
     : case 'getlocal'
     : call(function (source)
         source._localID = ('%d'):format(source.node.start)
diff --git a/script/vm/ref.lua b/script/vm/ref.lua
index 031a2e69..c97ca5e6 100644
--- a/script/vm/ref.lua
+++ b/script/vm/ref.lua
@@ -9,58 +9,7 @@ local lang      = require 'language'
 
 local simpleSwitch
 
-local function searchGetLocal(source, node, pushResult)
-    local key = guide.getKeyName(source)
-    for _, ref in ipairs(node.node.ref) do
-        if  ref.type == 'getlocal'
-        and ref.next
-        and guide.getKeyName(ref.next) == key then
-            pushResult(ref.next)
-        end
-    end
-end
-
 simpleSwitch = util.switch()
-    : case 'local'
-    : call(function (source, pushResult)
-        if source.ref then
-            for _, ref in ipairs(source.ref) do
-                if ref.type == 'setlocal'
-                or ref.type == 'getlocal' then
-                    pushResult(ref)
-                end
-            end
-        end
-    end)
-    : case 'getlocal'
-    : case 'setlocal'
-    : call(function (source, pushResult)
-        simpleSwitch('local', source.node, pushResult)
-    end)
-    : case 'field'
-    : call(function (source, pushResult)
-        local parent = source.parent
-        if parent.type ~= 'tablefield' then
-            simpleSwitch(parent.type, parent, pushResult)
-        end
-    end)
-    : case 'setfield'
-    : case 'getfield'
-    : call(function (source, pushResult)
-        local node = source.node
-        if node.type == 'getlocal' then
-            searchGetLocal(source, node, pushResult)
-            return
-        end
-    end)
-    : case 'getindex'
-    : case 'setindex'
-    : call(function (source, pushResult)
-        local node = source.node
-        if node.type == 'getlocal' then
-            searchGetLocal(source, node, pushResult)
-        end
-    end)
     : case 'goto'
     : call(function (source, pushResult)
         if source.node then
diff --git a/test/definition/table.lua b/test/definition/table.lua
index 66e71b0c..6dbe627c 100644
--- a/test/definition/table.lua
+++ b/test/definition/table.lua
@@ -33,17 +33,17 @@ t.<!x!> = 1
 t.<?x?>()
 ]]
 
-TEST [[
-local t
-t[<!1!>] = 1
-t[<?1?>]()
-]]
+--TEST [[
+--local t
+--t[<!1!>] = 1
+--t[<?1?>]()
+--]]
 
-TEST [[
-local t
-t[<!true!>] = 1
-t[<?true?>]()
-]]
+--TEST [[
+--local t
+--t[<!true!>] = 1
+--t[<?true?>]()
+--]]
 
 TEST [[
 local t
-- 
cgit v1.2.3