[T] 添加一些测试样例

Author: Starrah

parser/simai/ErrorStrategy.cs

@@ -20,14 +20,14 @@ public override void SyntaxError(TextWriter output, IRecognizer recognizer, ITok
             {
                 simaiParser.alerts.Add(new Alert(Warning, 
                     string.Format(Locale.RecoverInlineExtraneousToken, GetTokenErrorDisplay(offendingSymbol)), 
-                    line: line, relevantNote: RelevantNote(parser.Context)));
+                    line: line, relevantNote: RelevantNote(parser.Context, recognizer, offendingSymbol)));
                 return;
             }
             else if (msg.StartsWith("missing"))
             {
                 simaiParser.alerts.Add(new Alert(Warning, 
                     string.Format(Locale.RecoverInlineMissingToken, GetTokenErrorDisplay(offendingSymbol), parser.GetExpectedTokens().ToString(parser.Vocabulary)), 
-                    line: line, relevantNote: RelevantNote(parser.Context)));
+                    line: line, relevantNote: RelevantNote(parser.Context, recognizer, offendingSymbol)));
                 return;
             }
         }
@@ -50,7 +50,7 @@ public override void SyntaxError(TextWriter output, IRecognizer recognizer, ITok
                 message = string.Format(Locale.AntlrUnknownError, msg);
                 break;
         }
-        simaiParser.alerts.Add(new Alert(level, message, line: line, relevantNote: RelevantNote(parser.Context)));
+        simaiParser.alerts.Add(new Alert(level, message, line: line, relevantNote: RelevantNote(parser.Context, recognizer, offendingSymbol)));
     }
 
     // 词法分析的错误报告函数
@@ -75,14 +75,26 @@ public void SyntaxError(TextWriter output, IRecognizer recognizer, int offending
     }
 
     // 从context获得为适合放进relevantNote里的形式
-    private static string? RelevantNote(RuleContext? context)
+    private static string? RelevantNote(RuleContext? context, IRecognizer? recognizer, IToken? offendingSymbol)
     {
-        while (true)
+        string? result = null;
+        while (context != null)
         {
-            if (context == null) return null;
-            if (context.GetText().Length >= 5) return context.GetText();
+            if (context.GetText().Length >= 5)
+            {
+                result = context.GetText();
+                break;
+            }
             context = context.Parent;
         }
+
+        if (recognizer != null && offendingSymbol is { TokenIndex: >= 0 } && result is { Length: > 35 })
+        { // 根据offendingToken的位置，前截取5个、后截取三个token
+            var i = offendingSymbol.TokenIndex;
+            result = ((ITokenStream)recognizer.InputStream).GetText(
+                new Interval(Math.Max(i - 5, 0), Math.Min(i + 3, recognizer.InputStream.Size - 1)));
+        }
+        return result;
     }
 
     # region 用于暴露DefaultErrorStrategy内部的GetTokenErrorDisplay函数

tests/MA2转Simai测试.cs

@@ -32,8 +32,10 @@ private void TestChart(TestInput c)
         var inote = maidata.Levels[c.LevelId].Inote;
         var ma2Text = File.ReadAllText(c.MA2, Encoding.UTF8);
 
-        var (chart, _) = new MA2Parser().Parse(ma2Text);
-        var (simai, _) = new SimaiGenerator().Generate(chart);
+        var (chart, alerts) = new MA2Parser().Parse(ma2Text);
+        var (simai, alerts2) = new SimaiGenerator().Generate(chart);
+        _output.WriteLine(string.Join('\n', alerts));
+        _output.WriteLine(string.Join('\n', alerts2));
 
         Assert.Equal(TestUtils.TryParseMa2ClkDef(ma2Text), chart.ClockCount * 96);
         var expectedTimeline = SimaiCommaTimeline.Flatten(inote);

tests/Simai转MA2测试.cs

@@ -28,8 +28,10 @@ private void TestChart(TestInput input)
         var chartInfo = maidata.Levels[input.LevelId];
         var expectedMa2 = File.ReadAllText(input.MA2, Encoding.UTF8);
 
-        var (chart, _) = new SimaiParser(bigTouch: false, clockCount: maidata.ClockCount).Parse(chartInfo.Inote);
-        var (ma2, _) = new MA2Generator(isUtage: false).Generate(chart);
+        var (chart, alerts) = new SimaiParser(bigTouch: false, clockCount: maidata.ClockCount).Parse(chartInfo.Inote);
+        var (ma2, alerts2) = new MA2Generator(isUtage: false).Generate(chart);
+        _output.WriteLine(string.Join('\n', alerts));
+        _output.WriteLine(string.Join('\n', alerts2));
 
         Assert.Equal(maidata.ClockCount * 96, TestUtils.TryParseMa2ClkDef(ma2));
         ma2 = KeepNotesOnly(ma2);