Merge pull request #48 from zmaril/master

Allow for more global attributes in graphviz
aysylu · Jan 10, 2015 · fc3be84 · fc3be84
2 parents 68e89cb + 3145874
commit fc3be84
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/loom/io.clj b/src/loom/io.clj
@@ -56,10 +56,11 @@
                   (if d? "digraph \"" "graph \""))
              (.append (dot-esc graph-name))
              (.append "\" {\n"))]
-    (when (:graph opts)
-      (doto sb
-        (.append "  graph ")
-        (.append (dot-attrs (:graph opts)))))
+    (doseq [k [:graph :node :edge]]
+      (when (k opts)
+        (doto sb
+          (.append (str "  " (name k) " "))
+          (.append (dot-attrs (k opts))))))
     (doseq [[n1 n2] (distinct-edges g)]
       (let [n1l (str (or (node-label n1) n1))
             n2l (str (or (node-label n2) n2))