print_DOT() can translate numbers to ASCII chars

VeriFIT · Mar 1, 2024 · 485b91e · 485b91e
1 parent fcb8507
commit 485b91e
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 12 deletions.
diff --git a/include/mata/nfa/nfa.hh b/include/mata/nfa/nfa.hh
@@ -256,13 +256,16 @@ public:
     /**
      * @brief Prints the automaton in DOT format
      *
+     * @param[in] ascii Whether to use ASCII characters for the output.
      * @return automaton in DOT format
      */
-    std::string print_to_DOT() const;
+    std::string print_to_DOT(const bool ascii = false) const;
     /**
      * @brief Prints the automaton to the output stream in DOT format
+     *
+     * @param[in] ascii Whether to use ASCII characters for the output.
      */
-    void print_to_DOT(std::ostream &output) const;
+    void print_to_DOT(std::ostream &output, const bool ascii = false) const;
     /**
      * @brief Prints the automaton in mata format
      *

diff --git a/include/mata/nft/nft.hh b/include/mata/nft/nft.hh
@@ -240,13 +240,16 @@ public:
     /**
      * @brief Prints the automaton in DOT format
      *
+     * @param[in] ascii Whether to use ASCII characters for the output.
      * @return automaton in DOT format
      */
-    std::string print_to_DOT() const;
+    std::string print_to_DOT(const bool ascii = false) const;
     /**
      * @brief Prints the automaton to the output stream in DOT format
+     *
+     * @param[in] ascii Whether to use ASCII characters for the output.
      */
-    void print_to_DOT(std::ostream &output) const;
+    void print_to_DOT(std::ostream &output, const bool ascii = false) const;
     /**
      * @brief Prints the automaton in mata format
      *

diff --git a/src/nfa/nfa.cc b/src/nfa/nfa.cc
@@ -379,13 +379,21 @@ bool Nfa::is_acyclic() const {
     return acyclic;
 }
 
-std::string Nfa::print_to_DOT() const {
+std::string Nfa::print_to_DOT(const bool ascii) const {
     std::stringstream output;
-    print_to_DOT(output);
+    print_to_DOT(output, ascii);
     return output.str();
 }
 
-void Nfa::print_to_DOT(std::ostream &output) const {
+void Nfa::print_to_DOT(std::ostream &output, const bool ascii) const {
+    auto to_ascii = [&](const Symbol symbol) {
+        // Translate only printable ASCII characters.
+        if (symbol < 33) {
+            return std::to_string(symbol);
+        }
+        return "\\'" + std::string(1, static_cast<char>(symbol)) + "\\'";
+    };
+
     output << "digraph finiteAutomaton {" << std::endl
                  << "node [shape=circle];" << std::endl;
 
@@ -400,7 +408,11 @@ void Nfa::print_to_DOT(std::ostream &output) const {
             for (State target: move.targets) {
                 output << target << " ";
             }
-            output << "} [label=" << move.symbol << "];" << std::endl;
+            if (ascii) {
+                output << "} [label=\"" << to_ascii(move.symbol) << "\"];" << std::endl;
+            } else {
+                output << "} [label=\"" << move.symbol << "\"];" << std::endl;
+            }
         }
     }
 

diff --git a/src/nft/nft.cc b/src/nft/nft.cc
@@ -76,13 +76,30 @@ Nft& Nft::trim(StateRenaming* state_renaming) {
     return *this;
 }
 
-std::string Nft::print_to_DOT() const {
+std::string Nft::print_to_DOT(const bool ascii) const {
     std::stringstream output;
-    print_to_DOT(output);
+    print_to_DOT(output, ascii);
     return output.str();
 }
 
-void Nft::print_to_DOT(std::ostream &output) const {
+void Nft::print_to_DOT(std::ostream &output, const bool ascii) const {
+    auto translate_special_symbols = [&](const Symbol symbol) -> std::string {
+        if (symbol == EPSILON) {
+            return "<eps>";
+        }
+        if (symbol == DONT_CARE) {
+            return "<dcare>";
+        }
+        return std::to_string(symbol);
+    };
+
+    auto to_ascii = [&](const Symbol symbol) {
+        // Translate only printable ASCII characters.
+        if (symbol < 33) {
+            return std::to_string(symbol);
+        }
+        return "\\'" + std::string(1, static_cast<char>(symbol)) + "\\'";
+    };
     output << "digraph finiteAutomaton {" << std::endl
                  << "node [shape=circle];" << std::endl;
 
@@ -97,7 +114,12 @@ void Nft::print_to_DOT(std::ostream &output) const {
             for (State target: move.targets) {
                 output << target << " ";
             }
-            output << "} [label=" << move.symbol << "];" << std::endl;
+
+            if (ascii && move.symbol < 128) {
+                output << "} [label=\"" << to_ascii(move.symbol) << "\"];" << std::endl;
+            } else {
+                output << "} [label=\"" << translate_special_symbols(move.symbol) << "\"];" << std::endl;
+            }
         }
     }