Merge pull request #44 from abap34/json

Dot言語への変換を実装

src/almo.cpp

@@ -12,6 +12,7 @@ int main(int argc, char* argv[]) {
     bool debug = false;
     std::string editor_theme = "ace/theme/xcode";
     std::string syntax_theme = "github.min";
+    bool plot_graph = false;
 
     // 出力ファイルのデフォルト値を設定。 .md -> .html に置き換える
     std::string out_path = argv[1];
@@ -48,6 +49,9 @@ int main(int argc, char* argv[]) {
             else if (argv[i][1] == 's') {
                 syntax_theme = argv[i + 1];
             }
+            else if (argv[i][1] == 'g') {
+                plot_graph = true;
+            }
             else if (argv[i][1] == 'h') {
                 if (argc > 3) {
                     throw InvalidCommandLineArgumentsError("不正なコマンドライン引数です。 -h オプションと他のオプションは同時に指定できません。");
@@ -60,6 +64,7 @@ int main(int argc, char* argv[]) {
                 std::cout << "  -c <CSS>      CSSファイルを指定します。デフォルトは テーマに付属するものが使用されます。" << std::endl;
                 std::cout << "  -e <テーマ>   エディタのテーマを指定します。デフォルトは  です。" << std::endl;
                 std::cout << "  -d            デバッグモードで実行します。" << std::endl;
+                std::cout << "  -g            構文木をdot言語として出力します。" << std::endl;
                 std::cout << "  -h            ヘルプを表示します。" << std::endl;
             }
             else {
@@ -108,6 +113,18 @@ int main(int argc, char* argv[]) {
         std::cout << output << std::endl;
     }
 
+    if (plot_graph) {
+        std::string graph = ast.to_dot();
+        std::cout << "digraph g {" << std::endl;
+        std::cout << "    graph [" << std::endl;
+        std::cout << "        labelloc=\"t\";" << std::endl;
+        std::cout << "        ];" << std::endl;
+        std::cout << graph << std::endl;
+        std::cout << "}" << std::endl;
+    }
+
+
+
     almo::meta_data = meta_data;
 
     almo::render(ast, meta_data, out_path);

src/ast.hpp

@@ -58,6 +58,27 @@ namespace almo {
             return json + "}";
         }
 
+        // 部分木を dot 言語に変換する.
+        std::string to_dot() const {
+            std::map<std::string, std::string> properties = get_properties();
+
+            std::string node = properties["uuid"];
+            std::string label = "";
+
+            std::string label_header = properties["class"] + " | ";
+
+            int i = 1;
+            for (auto property : properties) {
+                if (property.first == "class" || property.first == "uuid") {
+                    continue;
+                }
+                label += "<f" + std::to_string(i) + "> " + property.first + ": " + escape(property.second) + " | ";
+                i++;
+            }
+
+            return node + "[label=\"" + label_header + label + "\", shape=\"record\"]\n";
+        }
+
     };
 
 
@@ -106,6 +127,44 @@ namespace almo {
             return json;
         }
 
+        // 部分木を dot 言語に変換する.
+        virtual std::string to_dot() const {
+            std::map<std::string, std::string> properties = get_properties();
+
+            std::string node = properties["uuid"];
+            std::string label = "";
+
+            std::string label_header = "<f0> " + properties["class"] + " | ";
+
+            int i = 1;
+            for (auto property : properties) {
+                if (property.first == "class" || property.first == "uuid") {
+                    continue;
+                }
+                label += "<f" + std::to_string(i) + "> " + property.first + ": " + escape(property.second) + " | ";
+                i++;
+            }
+
+            // 子ノードを追加する
+            std::string childs_dot = "";
+            for (auto child : childs) {
+                if (child->is_leaf()) {
+                    childs_dot += std::dynamic_pointer_cast<LeafNode>(child)->to_dot();
+                }
+                else {
+                    childs_dot += std::dynamic_pointer_cast<NonLeafNode>(child)->to_dot();
+                }
+            }
+
+            // 子ノードと繋ぐ
+            std::string edges = "";
+            for (auto child : childs) {
+                edges += node + ":f" + std::to_string(edges.length()) + " -> " + child->get_properties()["uuid"] + "\n";
+            }
+
+            return node + "[label=\"" + label_header + label + "\", shape=\"record\"]\n" + childs_dot + edges;
+        }
+
     };
 
     // H1~6 に対応する構文木のノード

src/utils.hpp

@@ -279,6 +279,15 @@ std::string escape(std::string s) {
         else if (s[i] == '\\') {
             result += "\\\\";
         }
+        else if (s[i] == '\r') {
+            result += "\\r";
+        }
+        else if (s[i] == '{') {
+            result += "\\{";
+        }
+        else if (s[i] == '}') {
+            result += "\\}";
+        }
         else {
             result += s[i];
         }