• 实现一个四则运算的解释器?

    分析

    按照编译原理相关的知识,我们来设计一下工作,这里我们分成几个步骤。

  • 定义四则运算:产出四则运算的词法定义和语法定义。

  • 词法分析:把输入的字符串流变成 token。
  • 语法分析:把 token 变成抽象语法树 AST。
  • 解释执行:后序遍历 AST,执行得出结果。

    定义四则运算

    四则运算就是加减乘除四种运算,例如:

    1. 1 + 2 * 3

    首先我们来定义词法,四则运算里面只有数字和运算符,所以定义很简单,但是我们还要注意空格和换行符,所以词法定义大概是下面这样的。

  • Token

    • Number: 1 2 3 4 5 6 7 8 9 0 的组合
    • Operator: + 、-、 *、 / 之一
  • Whitespace:
  • LineTerminator:

这里我们对空白和换行符没有任何的处理,所以词法分析阶段会直接丢弃。

接下来我们来定义语法,语法定义多数采用 BNF,但是其实大家写起来都是乱写的,比如 JavaScript 标准里面就是一种跟 BNF 类似的自创语法。

不过语法定义的核心思想不会变,都是几种结构的组合产生一个新的结构,所以语法定义也叫语法产生式。

因为加减乘除有优先级,所以我们可以认为加法是由若干个乘法再由加号或者减号连接成的:

  1. <Expression> ::=
  2. <AdditiveExpression><EOF>
  3. <AdditiveExpression> ::=
  4. <MultiplicativeExpression>
  5. |<AdditiveExpression><+><MultiplicativeExpression>
  6. |<AdditiveExpression><-><MultiplicativeExpression>

这种 BNF 的写法类似递归的原理,你可以理解一下,它表示一个列表。为了方便,我们把普通数字也得当成乘法的一种特例了。

  1. <MultiplicativeExpression> ::=
  2. <Number>
  3. |<MultiplicativeExpression><*><Number>
  4. |<MultiplicativeExpression></><Number>

这就是四则运算的定义了。

词法分析:状态机

词法分析部分,我们把字符流变成 token 流。词法分析有两种方案,一种是状态机,一种是正则表达式,它们是等效的,选择你喜欢的就好,这里我都会你介绍一下状态机。

根据分析,我们可能产生四种输入元素,其中只有两种 token,我们状态机的第一个状态就是根据第一个输入字符来判断进入了哪种状态:

  1. var token = [];
  2. const start = char => {
  3. if(char === '1'
  4. || char === '2'
  5. || char === '3'
  6. || char === '4'
  7. || char === '5'
  8. || char === '6'
  9. || char === '7'
  10. || char === '8'
  11. || char === '9'
  12. || char === '0'
  13. ) {
  14. token.push(char);
  15. return inNumber;
  16. }
  17. if(char === '+'
  18. || char === '-'
  19. || char === '*'
  20. || char === '/'
  21. ) {
  22. emmitToken(char, char);
  23. return start
  24. }
  25. if(char === ' ') {
  26. return start;
  27. }
  28. if(char === '\r'
  29. || char === '\n'
  30. ) {
  31. return start;
  32. }
  33. }
  34. const inNumber = char => {
  35. if(char === '1'
  36. || char === '2'
  37. || char === '3'
  38. || char === '4'
  39. || char === '5'
  40. || char === '6'
  41. || char === '7'
  42. || char === '8'
  43. || char === '9'
  44. || char === '0'
  45. ) {
  46. token.push(char);
  47. return inNumber;
  48. } else {
  49. emmitToken("Number", token.join(""));
  50. token = [];
  51. return start(char); // put back char
  52. }
  53. }

这个状态机非常简单,它只有两个状态,因为我们只有 Number 不是单字符的 token。

这里我的状态机实现是非常经典的方式:用函数表示状态,用 if 表示状态的迁移关系,用 return 值表示下一个状态。

下面我们来运行一下这个状态机试试看:

  1. function emmitToken(type, value) {
  2. console.log(value);
  3. }
  4. var input = "1024 + 2 * 256"
  5. var state = start;
  6. for(var c of input.split(''))
  7. state = state(c);
  8. state(Symbol('EOF'))

运行后我们发现输出如下:

  1. 1024
  2. +
  3. 2
  4. *
  5. 256

语法分析:LL

做完了词法分析,我们开始进行语法分析,LL 语法分析根据每一个产生式来写一个函数,首先我们来写好函数名:

  1. function AdditiveExpression( ){
  2. }
  3. function MultiplicativeExpression(){
  4. }

为了便于理解,我们就不做流式处理了,实际上一般编译代码都应该支持流式处理。

所以我们假设 token 已经都拿到了:

  1. var tokens = [{
  2. type:"Number",
  3. value: "1024"
  4. }, {
  5. type:"+"
  6. value: "+"
  7. }, {
  8. type:"Number",
  9. value: "2"
  10. }, {
  11. type:"*"
  12. value: "*"
  13. }, {
  14. type:"Number",
  15. value: "256"
  16. }, {
  17. type:"EOF"
  18. }];

每个产生式对应着一个函数,例如:根据产生式,我们的 AdditiveExpression 需要处理三种情况:

  1. <AdditiveExpression> ::=
  2. <MultiplicativeExpression>
  3. |<AdditiveExpression><+><MultiplicativeExpression>
  4. |<AdditiveExpression><-><MultiplicativeExpression>

那么 AddititveExpression 中就要写三个 if 分支,来处理三种情况。

AdditiveExpression 的写法是根传入的节点,利用产生式合成新的节点

  1. function AdditiveExpression(source){
  2. if(source[0].type === "MultiplicativeExpression") {
  3. let node = {
  4. type:"AdditiveExpression",
  5. children:[source[0]]
  6. }
  7. source[0] = node;
  8. return node;
  9. }
  10. if(source[0].type === "AdditiveExpression" && source[1].type === "+") {
  11. let node = {
  12. type:"AdditiveExpression",
  13. operator:"+",
  14. children:[source.shift(), source.shift(), MultiplicativeExpression(source)]
  15. }
  16. source.unshift(node);
  17. }
  18. if(source[0].type === "AdditiveExpression" && source[1].type === "-") {
  19. let node = {
  20. type:"AdditiveExpression",
  21. operator:"-",
  22. children:[source.shift(), source.shift(), MultiplicativeExpression(source)]
  23. }
  24. source.unshift(node);
  25. }
  26. }

那么下一步我们就把解析好的 token 传给我们的顶层处理函数 Expression。

  1. Expression(tokens);

接下来,我们看 Expression 该怎么处理它。

我们 Expression 收到第一个 token,是个 Number,这个时候,Expression 就傻了,这是因为产生式只告诉我们,收到了 AdditiveExpression 怎么办。

这个时候,我们就需要对产生式的首项层层展开,根据所有可能性调用相应的处理函数,这个过程在编译原理中称为求“closure”。

  1. function Expression(source){
  2. if(source[0].type === "AdditiveExpression" && source[1] && source[1].type === "EOF" ) {
  3. let node = {
  4. type:"Expression",
  5. children:[source.shift(), source.shift()]
  6. }
  7. source.unshift(node);
  8. return node;
  9. }
  10. AdditiveExpression(source);
  11. return Expression(source);
  12. }
  13. function AdditiveExpression(source){
  14. if(source[0].type === "MultiplicativeExpression") {
  15. let node = {
  16. type:"AdditiveExpression",
  17. children:[source[0]]
  18. }
  19. source[0] = node;
  20. return AdditiveExpression(source);
  21. }
  22. if(source[0].type === "AdditiveExpression" && source[1] && source[1].type === "+") {
  23. let node = {
  24. type:"AdditiveExpression",
  25. operator:"+",
  26. children:[]
  27. }
  28. node.children.push(source.shift());
  29. node.children.push(source.shift());
  30. MultiplicativeExpression(source);
  31. node.children.push(source.shift());
  32. source.unshift(node);
  33. return AdditiveExpression(source);
  34. }
  35. if(source[0].type === "AdditiveExpression" && source[1] && source[1].type === "-") {
  36. let node = {
  37. type:"AdditiveExpression",
  38. operator:"-",
  39. children:[]
  40. }
  41. node.children.push(source.shift());
  42. node.children.push(source.shift());
  43. MultiplicativeExpression(source);
  44. node.children.push(source.shift());
  45. source.unshift(node);
  46. return AdditiveExpression(source);
  47. }
  48. if(source[0].type === "AdditiveExpression")
  49. return source[0];
  50. MultiplicativeExpression(source);
  51. return AdditiveExpression(source);
  52. }
  53. function MultiplicativeExpression(source){
  54. if(source[0].type === "Number") {
  55. let node = {
  56. type:"MultiplicativeExpression",
  57. children:[source[0]]
  58. }
  59. source[0] = node;
  60. return MultiplicativeExpression(source);
  61. }
  62. if(source[0].type === "MultiplicativeExpression" && source[1] && source[1].type === "*") {
  63. let node = {
  64. type:"MultiplicativeExpression",
  65. operator:"*",
  66. children:[]
  67. }
  68. node.children.push(source.shift());
  69. node.children.push(source.shift());
  70. node.children.push(source.shift());
  71. source.unshift(node);
  72. return MultiplicativeExpression(source);
  73. }
  74. if(source[0].type === "MultiplicativeExpression"&& source[1] && source[1].type === "/") {
  75. let node = {
  76. type:"MultiplicativeExpression",
  77. operator:"/",
  78. children:[]
  79. }
  80. node.children.push(source.shift());
  81. node.children.push(source.shift());
  82. node.children.push(source.shift());
  83. source.unshift(node);
  84. return MultiplicativeExpression(source);
  85. }
  86. if(source[0].type === "MultiplicativeExpression")
  87. return source[0];
  88. return MultiplicativeExpression(source);
  89. };
  90. var source = [{
  91. type:"Number",
  92. value: "3"
  93. }, {
  94. type:"*",
  95. value: "*"
  96. }, {
  97. type:"Number",
  98. value: "300"
  99. }, {
  100. type:"+",
  101. value: "+"
  102. }, {
  103. type:"Number",
  104. value: "2"
  105. }, {
  106. type:"*",
  107. value: "*"
  108. }, {
  109. type:"Number",
  110. value: "256"
  111. }, {
  112. type:"EOF"
  113. }];
  114. var ast = Expression(source);
  115. console.log(ast);

解释执行

得到了 AST 之后,最困难的一步我们已经解决了。这里我们就不对这颗树做任何的优化和精简了,那么接下来,直接进入执行阶段。我们只需要对这个树做遍历操作执行即可。

我们根据不同的节点类型和其它信息,写 if 分别处理即可:

  1. function evaluate(node) {
  2. if(node.type === "Expression") {
  3. return evaluate(node.children[0])
  4. }
  5. if(node.type === "AdditiveExpression") {
  6. if(node.operator === '-') {
  7. return evaluate(node.children[0]) - evaluate(node.children[2]);
  8. }
  9. if(node.operator === '+') {
  10. return evaluate(node.children[0]) + evaluate(node.children[2]);
  11. }
  12. return evaluate(node.children[0])
  13. }
  14. if(node.type === "MultiplicativeExpression") {
  15. if(node.operator === '*') {
  16. return evaluate(node.children[0]) * evaluate(node.children[2]);
  17. }
  18. if(node.operator === '/') {
  19. return evaluate(node.children[0]) / evaluate(node.children[2]);
  20. }
  21. return evaluate(node.children[0])
  22. }
  23. if(node.type === "Number") {
  24. return Number(node.value);
  25. }
  26. }

作业:

  • 补全 emmitToken,使得我们的代码能完整工作起来。
  • 为四则运算加入小数。
  • 引入负数。
  • 添加括号功能。

🔚