Use information about source and sinks to create better charts

jeroenvandijk · Jul 19, 2013 · 3255948 · 3255948
1 parent ed9ab2a
commit 3255948
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/adgoji/cascalog/graph.clj b/src/adgoji/cascalog/graph.clj
@@ -163,10 +163,13 @@
   [graph & [options]]
   (let [{:keys [graph-name]
          :or {graph-name "graph"}} options
-         graph (graph/->graph graph)]
+         graph (graph/->graph graph)
+         {sink-options :sink-options} (tap-options graph)]
     (println "digraph" (pr-str (name graph-name)) "{")
     (when (map? graph)
       (doseq [sym (keys graph)
               dep (when-let [node (graph sym)] (fnk-input-keys node))]
-        (println "  " (pr-str (name dep)) "->" (pr-str (name sym)) ";")))
+        (if (sink-options dep)
+          (println "  " (pr-str (name sym)) "->" (pr-str (name dep)) ";")
+          (println "  " (pr-str (name dep)) "->" (pr-str (name sym)) ";"))))
     (println "}")))