Added inputs and outputs to \factor command

mkk20 · Sep 10, 2012 · dc5a733 · dc5a733
1 parent fb0eb65
commit dc5a733
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/README.rst b/README.rst
@@ -25,7 +25,7 @@ Provides the following node styles:
 
 Provides the following commands:
 
-* :code:`\factor [options] {name} {caption}`
+* :code:`\factor [options] {name} {caption} {inputs} {outputs}`
 
 * :code:`\plate [options] {name} {fitlist} {caption}`
 

diff --git a/tikzlibrarybayesnet.code.tex b/tikzlibrarybayesnet.code.tex
@@ -86,9 +86,13 @@
   } ;
 }
 
-% \factor [options] {name} {caption}
-\newcommand{\factor}[3][]{ %
-  \node[factor, label={[name=#2-caption]#3}, #1] (#2) {} ;  %
+% \factor [options] {name} {caption} {inputs} {outputs}
+\newcommand{\factor}[5][]{ %
+  % Draw the factor node. Use alias to allow empty names.
+  \node[factor, label={[name=#2-caption]#3}, name=#2, alias=#2-alias,
+  #1] {} ; %
+  % Connect all inputs to outputs via this factor
+  \factoredge {#4} {#2-alias} {#5} ; %
 }
 
 % \plate [options] {name} {fitlist} {caption}