Antlr4.Tool.DOTGenerator.GetDOT C# (CSharp) Method

        public virtual string GetDOT(DFA dfa, bool isLexer)
        {
            if (dfa.s0.Get() == null)
                return null;

            Template dot = stlib.GetInstanceOf("dfa");
            dot.Add("name", "DFA" + dfa.decision);
            dot.Add("startState", dfa.s0.Get().stateNumber);
            //		dot.add("useBox", Tool.internalOption_ShowATNConfigsInDFA);
            dot.Add("rankdir", rankdir);

            // define stop states first; seems to be a bug in DOT where doublecircle
            foreach (DFAState d in dfa.states.Keys)
            {
                if (!d.IsAcceptState)
                    continue;
                Template st = stlib.GetInstanceOf("stopstate");
                st.Add("name", "s" + d.stateNumber);
                st.Add("label", GetStateLabel(d));
                dot.Add("states", st);
            }

            foreach (DFAState d in dfa.states.Keys)
            {
                if (d.IsAcceptState)
                    continue;
                if (d.stateNumber == int.MaxValue)
                    continue;
                Template st = stlib.GetInstanceOf("state");
                st.Add("name", "s" + d.stateNumber);
                st.Add("label", GetStateLabel(d));
                dot.Add("states", st);
            }

            foreach (DFAState d in dfa.states.Keys)
            {
                IDictionary<int, DFAState> edges = d.EdgeMap;
                foreach (KeyValuePair<int, DFAState> entry in edges)
                {
                    DFAState target = entry.Value;
                    if (target == null)
                        continue;
                    if (target.stateNumber == int.MaxValue)
                        continue;
                    int ttype = entry.Key;
                    string label = ttype.ToString();
                    if (isLexer)
                        label = "'" + GetEdgeLabel(((char)ttype).ToString()) + "'";
                    else if (grammar != null)
                        label = grammar.GetTokenDisplayName(ttype);
                    Template st = stlib.GetInstanceOf("edge");
                    st.Add("label", label);
                    st.Add("src", "s" + d.stateNumber);
                    st.Add("target", "s" + target.stateNumber);
                    st.Add("arrowhead", arrowhead);
                    dot.Add("edges", st);
                }
            }

            string output = dot.Render();
            return Utils.SortLinesInString(output);
        }

        public virtual void GenerateATNs(Grammar g)
        {
            DOTGenerator dotGenerator = new DOTGenerator(g);
            IList<Grammar> grammars = new List<Grammar>();
            grammars.Add(g);
            IList<Grammar> imported = g.GetAllImportedGrammars();
            if (imported != null)
            {
                foreach (Grammar importedGrammar in imported)
                    grammars.Add(importedGrammar);
            }

            foreach (Grammar ig in grammars)
            {
                foreach (Rule r in ig.rules.Values)
                {
                    try
                    {
                        string dot = dotGenerator.GetDOT(g.atn.ruleToStartState[r.index], g.IsLexer());
                        if (dot != null)
                        {
                            WriteDOTFile(g, r, dot);
                        }
                    }
                    catch (IOException ioe)
                    {
                        errMgr.ToolError(ErrorType.CANNOT_WRITE_FILE, ioe);
                    }
                }
            }
        }