vitchyr · March 23, 2025 16:24 · Jul 1, 2015 · Jul 1, 2015 · Jul 1, 2015 · Jul 1, 2015
diff --git a/pdf2svgs.py b/pdf2svgs.py
@@ -63,7 +63,7 @@ def file_name(path):
 def pdf_to_svg(out_dir, fpath):
     """
     Convert a pdf to an svg.
-    
+
     Parameters
     ----------
     out_dir : string

diff --git a/pdf2svg.py → pdf2svgs.py b/pdf2svg.py → pdf2svgs.py
@@ -11,7 +11,7 @@
         Use 'python pdf2svg.py -h' for more information
 
 Example Usage:
-    $ python pdf_splitter.py path/to/foo.pdf path/to/svgs new_svg_name
+    $ python pdf2svgs.py path/to/foo.pdf path/to/svgs new_svg_name
 
 will result in the new files:
     path/to/svgs/new_svg_name_all_p0.svg

diff --git a/pdf2svg.py b/pdf2svg.py
@@ -27,8 +27,8 @@ def split_pdf(fpath, name, tmp_dir):
     """
     Split a pdf into multiple PDFs, one per page.
 
-    Parameter
-    ---------
+    Parameters
+    ----------
     fpath : string
         Path to a PDF.
     name : string
@@ -63,6 +63,13 @@ def file_name(path):
 def pdf_to_svg(out_dir, fpath):
     """
     Convert a pdf to an svg.
+    
+    Parameters
+    ----------
+    out_dir : string
+        Directory where to save the SVG.
+    fpath : string
+        Path to a PDF. The SVG will have the same name as this but with a .svg extension.
     """
     out_name = file_name(fpath) + ".svg"
     out_fpath = os.path.join(out_dir, out_name)

diff --git a/pdf2svg.py b/pdf2svg.py
@@ -0,0 +1,92 @@
+"""
+Author: Vitchyr Pong
+
+Convert a PDF to svg files. Note that this is pretty slow since it makes
+subprocess calls to inkscape's PDF-to-SVG command line convert.
+
+Requirements:
+    - Inkscape (https://inkscape.org/)
+    - pyPdf (http://pybrary.net/pyPdf/)
+    - A '/tmp' directory. If not, you must pass in another directory.
+        Use 'python pdf2svg.py -h' for more information
+
+Example Usage:
+    $ python pdf_splitter.py path/to/foo.pdf path/to/svgs new_svg_name
+
+will result in the new files:
+    path/to/svgs/new_svg_name_all_p0.svg
+    path/to/svgs/new_svg_name_all_p1.svg
+    etc.
+"""
+import argparse
+import os
+import subprocess
+from pyPdf import PdfFileWriter, PdfFileReader
+
+def split_pdf(fpath, name, tmp_dir):
+    """
+    Split a pdf into multiple PDFs, one per page.
+
+    Parameter
+    ---------
+    fpath : string
+        Path to a PDF.
+    name : string
+        Base name for the SVGs.
+    tmp_dir: string
+        A directory where temporary PDFs are saved
+
+    Return
+    ------
+    pdfs: list of strings
+        A list of path directories to the temporary PDFs, in order.
+    """
+    input_pdf = PdfFileReader(open(fpath, "rb"))
+
+    directory = os.path.dirname(fpath)
+
+    pdfs = []
+    for i in xrange(input_pdf.numPages):
+        output = PdfFileWriter()
+        output.addPage(input_pdf.getPage(i))
+        new_path = os.path.join(tmp_dir, name + "_p{0}.pdf".format(i))
+        with open(new_path, "wb") as outputStream:
+            output.write(outputStream)
+        pdfs.append(new_path)
+
+    return pdfs
+
+def file_name(path):
+    """ Get the name of a file without the extension. """
+    return os.path.split(path)[-1].split(".")[0]
+
+def pdf_to_svg(out_dir, fpath):
+    """
+    Convert a pdf to an svg.
+    """
+    out_name = file_name(fpath) + ".svg"
+    out_fpath = os.path.join(out_dir, out_name)
+    subprocess.call(["inkscape", "-l", out_fpath, fpath])
+
+def main():
+    parser = argparse.ArgumentParser(description="Convert a PDF to svg files.")
+
+    parser.add_argument("pdf_path", help="Path to the pdf.")
+    parser.add_argument("svg_dir", help="Directory to save the svg.")
+    parser.add_argument("svg_name", help="Base name of the svgs.")
+    parser.add_argument("-t",
+                        "--tmp_dir",
+                        help="Where to save temporary PDFs.",
+                        default="/tmp")
+
+    args = parser.parse_args()
+    pdfs = split_pdf(args.pdf_path, args.svg_name, args.tmp_dir)
+
+    if not os.path.exists(args.svg_dir):
+        os.makedirs(args.svg_dir)
+
+    for p in pdfs:
+        pdf_to_svg(args.svg_dir, p)
+
+if __name__ == '__main__':
+    main()
No results found