前回まででVisitorの使い方がわかったので、計算機を完成させる。

実装

文法ファイル

Calculator.g4

grammar Calculator;

PLUS : '+';
MINUS: '-';
MULTI: '*';
DIV  : '/';
NUMBER : [0-9]+;
WHITESPACE : [ \r\n\t]+ -> skip;

expression 
	: NUMBER						# Number
	| '(' inside=expression')'			# Parentheses
	| left=expression MULTI right=expression	# Multiplication
	| left=expression DIV right=expression		# Division
	| left=expression PLUS right=expression	# Addition
	| left=expression MINUS right=expression	# Subtraction
;

演算(+ - / *)の左辺右辺を区別するために、Rule Element Labelsという仕組みを使う。
そのため、grammarファイルで演算の左のexpressionをleft,右のexpressionをrightとラベルを付ける。
そのために、grammarファイルに下記のようにleft=,right=を追記する。
またカッコについても大本のexpressionと区別するためinsideのラベルを付ける。

	| left=expression MULTI right=expression	# Multiplication

Visitorクラス作成

CalculatorBaseVisitorを継承して作成。

CalculatorVisitor.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Antlr4.Runtime.Misc;

namespace Calculator02_2
{
    class CalculatorVisitor: CalculatorBaseVisitor<float>
    {
        public override float VisitAddition([NotNull] CalculatorParser.AdditionContext context)
        {
            return base.Visit(context.left) + base.Visit(context.right);
        }
        public override float VisitDivision([NotNull] CalculatorParser.DivisionContext context)
        {
            return base.Visit(context.left) / base.Visit(context.right);
        }
        public override float VisitMultiplication([NotNull] CalculatorParser.MultiplicationContext context)
        {
            return base.Visit(context.left) * base.Visit(context.right);
        }
        public override float VisitSubtraction([NotNull] CalculatorParser.SubtractionContext context)
        {
            return base.Visit(context.left) - base.Visit(context.right);
        }
        public override float VisitParentheses([NotNull] CalculatorParser.ParenthesesContext context)
        {
            return base.Visit(context.inside);
        }
        public override float VisitNumber([NotNull] CalculatorParser.NumberContext context)
        {
            return float.Parse(context.NUMBER().GetText());
        }
    }
}

各演算から、左辺/右辺をvisit。数値は数値をそのまま返す。

Program.cs

visitがfloatを返すようになったので、それを表示するように変更。

Program.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Antlr4.Runtime;

namespace Calculator02_2
{
    class Program
    {
        static void Main(string[] args)
        {
            string parsedString;
            if (args.Length == 0)
            {
                Console.WriteLine("引数に数式を指定してください");
                return;
            }
            else
            {
                parsedString = args[0];
            }
            var inputStream = new AntlrInputStream(parsedString);
            var lexer = new CalculatorLexer(inputStream);
            var commonTokenStream = new CommonTokenStream(lexer);
            var parser = new CalculatorParser(commonTokenStream);
            var graphContext = parser.expression();
            CalculatorVisitor visitor = new CalculatorVisitor();
            Console.WriteLine(visitor.Visit(graphContext));
        }
    }
}

実行結果

>Calculator02_2.exe 3-1
2

>Calculator02_2.exe 3-1*5
-2

>Calculator02_2.exe 3-1*5/2
0.5

>Calculator02_2.exe (3-1)*5/2
5

>Calculator02_2.exe 1*2+3/4
2.75

メモ

Visitorから別のVisitorをvisitする際に、base.を付ける必要あるか。
付ける場合と付けない場合の違いはなにか。
後日調べる？