一步步写lua解释器--泛型for循环

写泛型for之前，我发现连这个语法都不是太清晰，因为大都for, in 后面跟着pairs()或ipairs()函数，以为for,in后面只能是这两个函数。查看了官网资料，for, in 的语法规则是：

for namelist in explist do block end 

也就是说for,in后面需要一个表达式列表，但是他们必须要有一定的的规则，例如：

for k,v in ipairsIter, t, i do
  print(k, v)
end

列表第一个必须是一个函数，第二个是要遍历的数据，第三个是初始化控制量。而这个函数ipairsIter要能做如下的事：
以i开始作用在数据t上，并返回下一个控制值和数据。他的实现大致为：

local ipairsIter = function(t, i)
  i = i + 1
  local v = t[i]
  if v then
    return i, v
  end
end

可以想到，pairs的实现为：

pairs = function(t)
    return ipairsIter, t, 0
end

也就是说(i)pairs函数返回的是一个列表。

for循环的原理都能把人搞晕，估计实现起来就没那么容易了。

确实，最后实现出来花了很大功夫，大约一上午的时间，虽然最后代码没有多少行。其实把原理搞懂了，也不复杂，但是得细心，慢慢调试出结果。

泛型for循环有几个要点：

• 迭代函数，迭代数据（主要是table），初始化变量，一般为0
• 显示的k， v

在遇到in 后，要将后面的表达式解析为上面的三个。如果是一个函数也行，例如pairs函数，不过他要返回三个值，其结果仍然是上面的三种类型。

最后是压入显示的k，v属于局部变量，之前在数值型for循环里面说过。然后就是生成一个OpCode_GenericFor的指令。虚拟机遇到这个指令时，先取出k,v保存起来，因为后面要对k赋值的。然后是执行迭代函数。代码如下：

void VM::genericFor(Instruction* ins)
{
    Value* ctrlKey = nullptr;
    std::vector<Value*> vtKeys;
    int num = ins->param.counter.counter1;
    for (int i = 0; i < num; i++)  {
        vtKeys.push_back(_stack->popValue());
    }
    ctrlKey = vtKeys[vtKeys.size() - 1];
    Value* ctrlVal = _stack->popValue();
    Value* data    = _stack->popValue();
    Value* func = _stack->popValue();
    
    do  {
        _stack->Push(data);       //第一个参数
        _stack->Push(ctrlVal);    //第二个参数
        _stack->Push(func);       //迭代函数

        Instruction insCall;
        insCall.param.counter.counter1 = 2;
        insCall.param.counter.counter2 = num;
        call(&insCall);

        for (int i = num - 1; i >=0; i--)  {
            _stack->Push(vtKeys[i]);
        }

        assignVals(num, num, 0);

        ctrlVal = getCurrentClosure()->getTopTable()->GetValue(ctrlKey);
        if (ctrlVal->Type() == Value::TYPE_NIL)  {  //控制值为nil
            break;
        }
        InstructionValue* block = (InstructionValue*)ins->param.value;
        block->setFor(true);
        runBlockCode(ins->param.value);
    } while (true);
}

虽然代码不多，但是却花费了不少功夫。共勉！

 

有问题可以在下面留言，或者加入QQ群 858791125 讨论。

项目地址：
https://github.com/shonm520/mlua