]> granicus.if.org Git - python/commitdiff
Utility to help create outline information for PDF.
authorFred Drake <fdrake@acm.org>
Fri, 6 Mar 1998 21:29:00 +0000 (21:29 +0000)
committerFred Drake <fdrake@acm.org>
Fri, 6 Mar 1998 21:29:00 +0000 (21:29 +0000)
Doc/tools/toc2bkm.py [new file with mode: 0755]

diff --git a/Doc/tools/toc2bkm.py b/Doc/tools/toc2bkm.py
new file mode 100755 (executable)
index 0000000..dd74fd4
--- /dev/null
@@ -0,0 +1,115 @@
+#! /usr/bin/env python
+
+"""Convert a LaTeX .toc file to some PDFTeX magic to create that neat outline.
+
+The output file has an extension of '.bkm' instead of '.out', since hyperref
+already uses that extension.  Let's avoid clashing.
+"""
+
+import os
+import re
+import string
+import sys
+
+
+# Ench item in an entry is a tuple of:
+#
+#   Section #,  Title String,  Page #,  List of Sub-entries
+
+cline_re = r"""^
+\\contentsline\ \{([a-z]*)}             # type of section in $1
+\{(?:\\numberline\ \{([0-9.A-Z]+)})?     # section number
+(.*)}                                   # title string
+\{(\d+)}$"""                           # page number
+
+cline_rx = re.compile(cline_re, re.VERBOSE)
+
+OUTER_TO_INNER = -1
+
+_transition_map = {
+    ('chapter', 'section'): OUTER_TO_INNER,
+    ('section', 'subsection'): OUTER_TO_INNER,
+    ('subsection', 'subsubsection'): OUTER_TO_INNER,
+    ('subsubsection', 'subsection'): 1,
+    ('subsection', 'section'): 1,
+    ('section', 'chapter'): 1,
+    ('subsection', 'chapter'): 2,
+    ('subsubsection', 'section'): 2,
+    ('subsubsection', 'chapter'): 3,
+    }
+
+def parse_toc(fp):
+    toc = top = []
+    stack = [toc]
+    level = 'chapter'
+    lineno = 0
+    while 1:
+       line = fp.readline()
+       if not line:
+           break
+       lineno = lineno + 1
+       m = cline_rx.match(line)
+       if m:
+           stype, snum, title, pageno = m.group(1, 2, 3, 4)
+           title = clean_title(title)
+           entry = (stype, snum, title, string.atoi(pageno), [])
+           if stype == level:
+               toc.append(entry)
+           else:
+               direction = _transition_map[(level, stype)]
+               if direction == OUTER_TO_INNER:
+                   toc = toc[-1][-1]
+                   stack.insert(0, toc)
+                   toc.append(entry)
+               else:
+                   for i in range(direction):
+                       del stack[0]
+                       toc = stack[0]
+                   toc.append(entry)
+               level = stype
+       else:
+           sys.stderr.write("l.%s: " + line)
+    return top
+
+
+hackscore_rx = re.compile(r"\\(hackscore|raisebox)\s*{[^}]*}")
+title_rx = re.compile(r"\\[a-zA-Z]+\s*")
+title_trans = string.maketrans("", "")
+
+def clean_title(title):
+    title = hackscore_rx.sub("_", title)
+    while 1:
+       m = title_rx.search(title)
+       if m:
+           title = title[:m.start()] + title[m.end():]
+       else:
+           break
+    return string.translate(title, title_trans, "{}")
+
+
+def write_toc(toc, fp):
+    for entry in toc:
+       write_toc_entry(entry, fp, 0)
+
+def write_toc_entry(entry, fp, layer):
+    stype, snum, title, pageno, toc = entry
+    s = "\\pdfoutline goto name{page.%d}" % pageno
+    if toc:
+       s = "%s count -%d" % (s, len(toc))
+    if snum:
+       title = "%s %s" % (snum, title)
+    s = "%s {%s}\n" % (s, title)
+    fp.write(s)
+    for entry in toc:
+       write_toc_entry(entry, fp, layer + 1)
+
+
+def main():
+    base, ext = os.path.splitext(sys.argv[1])
+    ext = ext or ".toc"
+    toc = parse_toc(open(base + ext))
+    write_toc(toc, open(base + ".bkm", "w"))
+
+
+if __name__ == "__main__":
+    main()