1 /* Copyright (c) 2012-2017 The ANTLR Project. All rights reserved.
2  * Use of this file is governed by the BSD 3-clause license that
3  * can be found in the LICENSE.txt file in the project root.
4  */
5 
6 #include "atn/PredictionContext.h"
7 #include "atn/ATNConfig.h"
8 #include "atn/ATNConfigSet.h"
9 #include "Parser.h"
10 #include "misc/Interval.h"
11 #include "dfa/DFA.h"
12 
13 #include "DiagnosticErrorListener.h"
14 
15 using namespace antlr4;
16 
DiagnosticErrorListener()17 DiagnosticErrorListener::DiagnosticErrorListener() : DiagnosticErrorListener(true) {
18 }
19 
DiagnosticErrorListener(bool exactOnly_)20 DiagnosticErrorListener::DiagnosticErrorListener(bool exactOnly_) : exactOnly(exactOnly_) {
21 }
22 
reportAmbiguity(Parser * recognizer,const dfa::DFA & dfa,size_t startIndex,size_t stopIndex,bool exact,const antlrcpp::BitSet & ambigAlts,atn::ATNConfigSet * configs)23 void DiagnosticErrorListener::reportAmbiguity(Parser *recognizer, const dfa::DFA &dfa, size_t startIndex, size_t stopIndex,
24    bool exact, const antlrcpp::BitSet &ambigAlts, atn::ATNConfigSet *configs) {
25   if (exactOnly && !exact) {
26     return;
27   }
28 
29   std::string decision = getDecisionDescription(recognizer, dfa);
30   antlrcpp::BitSet conflictingAlts = getConflictingAlts(ambigAlts, configs);
31   std::string text = recognizer->getTokenStream()->getText(misc::Interval(startIndex, stopIndex));
32   std::string message = "reportAmbiguity d=" + decision + ": ambigAlts=" + conflictingAlts.toString() +
33     ", input='" + text + "'";
34 
35   recognizer->notifyErrorListeners(message);
36 }
37 
reportAttemptingFullContext(Parser * recognizer,const dfa::DFA & dfa,size_t startIndex,size_t stopIndex,const antlrcpp::BitSet &,atn::ATNConfigSet *)38 void DiagnosticErrorListener::reportAttemptingFullContext(Parser *recognizer, const dfa::DFA &dfa, size_t startIndex,
39   size_t stopIndex, const antlrcpp::BitSet &/*conflictingAlts*/, atn::ATNConfigSet * /*configs*/) {
40   std::string decision = getDecisionDescription(recognizer, dfa);
41   std::string text = recognizer->getTokenStream()->getText(misc::Interval(startIndex, stopIndex));
42   std::string message = "reportAttemptingFullContext d=" + decision + ", input='" + text + "'";
43   recognizer->notifyErrorListeners(message);
44 }
45 
reportContextSensitivity(Parser * recognizer,const dfa::DFA & dfa,size_t startIndex,size_t stopIndex,size_t,atn::ATNConfigSet *)46 void DiagnosticErrorListener::reportContextSensitivity(Parser *recognizer, const dfa::DFA &dfa, size_t startIndex,
47   size_t stopIndex, size_t /*prediction*/, atn::ATNConfigSet * /*configs*/) {
48   std::string decision = getDecisionDescription(recognizer, dfa);
49   std::string text = recognizer->getTokenStream()->getText(misc::Interval(startIndex, stopIndex));
50   std::string message = "reportContextSensitivity d=" + decision + ", input='" + text + "'";
51   recognizer->notifyErrorListeners(message);
52 }
53 
getDecisionDescription(Parser * recognizer,const dfa::DFA & dfa)54 std::string DiagnosticErrorListener::getDecisionDescription(Parser *recognizer, const dfa::DFA &dfa) {
55   size_t decision = dfa.decision;
56   size_t ruleIndex = (reinterpret_cast<atn::ATNState*>(dfa.atnStartState))->ruleIndex;
57 
58   const std::vector<std::string>& ruleNames = recognizer->getRuleNames();
59   if (ruleIndex == INVALID_INDEX || ruleIndex >= ruleNames.size()) {
60     return std::to_string(decision);
61   }
62 
63   std::string ruleName = ruleNames[ruleIndex];
64   if (ruleName == "" || ruleName.empty())  {
65     return std::to_string(decision);
66   }
67 
68   return std::to_string(decision) + " (" + ruleName + ")";
69 }
70 
getConflictingAlts(const antlrcpp::BitSet & reportedAlts,atn::ATNConfigSet * configs)71 antlrcpp::BitSet DiagnosticErrorListener::getConflictingAlts(const antlrcpp::BitSet &reportedAlts,
72                                                              atn::ATNConfigSet *configs) {
73   if (reportedAlts.count() > 0) { // Not exactly like the original Java code, but this listener is only used
74                                   // in the TestRig (where it never provides a good alt set), so it's probably ok so.
75     return reportedAlts;
76   }
77 
78   antlrcpp::BitSet result;
79   for (auto &config : configs->configs) {
80     result.set(config->alt);
81   }
82 
83   return result;
84 }
85