1 files changed, 141 insertions, 14 deletions
diff --git a/content/resources/rst_graphviz/rst2latex.py b/content/resources/rst_graphviz/rst2latex.py
index 9185405..615d2e5 100755
--- a/content/resources/rst_graphviz/rst2latex.py
+++ b/content/resources/rst_graphviz/rst2latex.py
@@ -8,6 +8,10 @@
 A minimal front end to the Docutils Publisher, producing LaTeX.
 """
 
+import sys
+reload(sys)
+sys.setdefaultencoding('utf-8')
+
 try:
     import locale
     locale.setlocale(locale.LC_ALL, '')
@@ -19,7 +23,7 @@ import os.path
 
 from docutils.core import publish_cmdline
 from docutils.parsers.rst import directives
-from docutils.parsers.rst.directives.images import Figure
+from docutils.parsers.rst.directives.images import Figure, Image
 
 description = ('Generates LaTeX documents from standalone reStructuredText '
                'sources. '
@@ -28,11 +32,103 @@ description = ('Generates LaTeX documents from standalone reStructuredText '
                '<http://docutils.sourceforge.net/docs/user/latex.html> for '
                'the full reference.')
 
+DOT2TEX = "dot2tex"
+
+def processContent(node):
+
+    if node.content.count("..") == 0:
+        sep = -1 
+        text = '\n'.join(node.content)
+    else:
+        sep = node.content.index("..")
+        text = '\n'.join(node.content[:sep])
+    extension = 'pdf'
+    pdfFile = "tmp/%s.pdf" % (hash(text))
+
+
+    if not os.path.exists(pdfFile):
+        conversion = subprocess.Popen(['dot',
+            '-T', extension,
+            "-Gcharset=utf8",
+            ],
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            )
+
+        ps2pdf = subprocess.Popen(['ps2pdf',
+            "-",
+            pdfFile
+            ],
+            stdin = conversion.stdout
+            )
+        conversion.stdin.write("%s G {\n" %
+                node.arguments[0].encode("utf-8"))
+        conversion.stdin.write(text.encode("utf-8"))
+        conversion.stdin.write("\n}")
+        conversion.stdin.close()
+        conversion.stdout.close()
+
+        ps2pdf.wait()
+        conversion.wait()
+        if conversion.returncode != 0:
+            raise node.error(
+                'Error in "%s" directive:\n' % node.name)
+
+    return (sep, pdfFile)
+
+class GraphvizImage(Image):
+    """ Generate a graphviz image
+    """
+    required_arguments = 1
+    optional_arguments = 0
+    final_argument_whitespace = True
+
+    has_content = True
+
+    option_spec = Image.option_spec.copy()
+
+    def run(self):
+
+
+        sep, arguments = processContent(self)
+        self.arguments = [arguments]
+        self.options['scale'] = 66
+        if sep == -1:
+            self.content = None
+        else:
+            self.content = self.content[sep+1:]
+
+        return Image.run(self)
+
 class Graphviz(Figure):
     """ Generate a graphviz image
     """
-    required_arguments = 0
-    optional_arguments = 1
+    required_arguments = 1
+    optional_arguments = 0
+    final_argument_whitespace = True
+
+    has_content = True
+
+    option_spec = Figure.option_spec.copy()
+
+    def run(self):
+
+        sep, arguments = processContent(self)
+        self.arguments = [arguments]
+        self.options['scale'] = 66
+        if sep == -1:
+            self.content = None
+        else:
+            self.content = self.content[sep+1:]
+
+        return Figure.run(self)
+
+
+class Dot2Tex(Figure):
+    """ Generate a graphviz image using dot2tex
+    """
+    required_arguments = 1
+    optional_arguments = 0
     final_argument_whitespace = True
 
     has_content = True
@@ -41,24 +137,55 @@ class Graphviz(Figure):
 
     def run(self):
 
-        text = '\n'.join(self.content)
-        extension = 'png'
-        imageFile = "tmp/%s.%s" % (hash(text), extension)
-        if not os.path.exists(imageFile):
-            conversion = subprocess.Popen(['/usr/bin/dot',
-                '-T', 'png',
-                '-o', imageFile,
+        if self.content.count("..") == 0:
+            sep = -1 
+            text = '\n'.join(self.content)
+        else:
+            sep = self.content.index("..")
+            text = '\n'.join(self.content[:sep])
+        pdfFile = "tmp/%s.pdf" % (hash(text))
+        texFile = "tmp/%s.tex" % (hash(text))
+
+
+        if not os.path.exists(pdfFile):
+            conversion = subprocess.Popen([DOT2TEX,
+                '-c',
+                '-o', texFile,
                 ],
                 stdin=subprocess.PIPE
                 )
-            conversion.communicate(text)
+            conversion.communicate("%s G { \n %s \n}" 
+                    % (self.arguments[0], text))
             conversion.wait()
 
-        self.arguments = [imageFile]
-        self.options['scale'] = 50
-        self.content = None
+            ps2pdf = subprocess.Popen(['pdflatex',
+                '-output-directory', 'tmp',
+                texFile,
+                ],
+                )
+            ps2pdf.wait()
+            if conversion.returncode != 0:
+                raise self.error(
+                    'Error in "%s" directive:\n' % self.name)
+
+        self.arguments = [pdfFile]
+        self.options['scale'] = 66
+        if sep == -1:
+            self.content = None
+        else:
+            self.content = self.content[sep+1:]
+
         return Figure.run(self)
 
+#try:
+#    # Check if dot2tex exists and register the conversion with dot2tex and
+#    # pdftex
+#    subprocess.check_call([DOT2TEX, "-V"])
+#    directives.register_directive('graphviz', Dot2Tex)
+#    directives.register_directive('graphimg', GraphvizImage)
+#except:
+#    # Otherwise use the old graphviz program
 directives.register_directive('graphviz', Graphviz)
+directives.register_directive('graphimg', GraphvizImage)
 
 publish_cmdline(writer_name='latex', description=description)